Spring/spring-amqp
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Use controlled atomic for the isOpen() stub

Browse Source 
https://github.com/spring-projects/spring-amqp/pull/584/files#r105809037
This commit is contained in:
Artem Bilan
2017-03-14 12:59:28 -04:00
parent b238d6a601
commit d7e702a765
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
spring-rabbit/src/test/java/org/springframework/amqp/rabbit/listener/DirectMessageListenerContainerMockTests.java
View File

@@ -30,6 +30,7 @@ import static org.mockito.Mockito.mock;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicBoolean;
import java.util.concurrent.atomic.AtomicInteger;
import org.junit.Test;
@@ -61,7 +62,8 @@ public class DirectMessageListenerContainerMockTests {
given(connectionFactory.createConnection()).willReturn(connection);
given(connection.createChannel(anyBoolean())).willReturn(channel);
given(channel.isOpen()).willReturn(true, false);
final AtomicBoolean isOpen = new AtomicBoolean(true);
willAnswer(i -> isOpen.get()).given(channel).isOpen();
given(channel.queueDeclarePassive(Mockito.anyString()))
.willAnswer(invocation -> mock(AMQP.Queue.DeclareOk.class));
given(channel.basicConsume(anyString(), anyBoolean(), anyString(), anyBoolean(), anyBoolean(),
@@ -89,6 +91,7 @@ public class DirectMessageListenerContainerMockTests {
assertTrue(latch1.await(10, TimeUnit.SECONDS));
assertThat(qos.get(), equalTo(2));
isOpen.set(false);
assertTrue(latch2.await(10, TimeUnit.SECONDS));
container.stop();
}