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

ZeroMQ test: Add sleep between SUB & PUB

Browse Source
This commit is contained in:
Artem Bilan
2021-07-07 15:57:44 -04:00
parent a5776524e2
commit 920b8ae744
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
spring-integration-zeromq/src/test/java/org/springframework/integration/zeromq/outbound/ZeroMqMessageHandlerTests.java
View File

@@ -86,7 +86,7 @@ public class ZeroMqMessageHandlerTests {
}
@Test
void testMessageHandlerForPubSub() {
void testMessageHandlerForPubSub() throws InterruptedException {
ZMQ.Socket subSocket = CONTEXT.createSocket(SocketType.SUB);
subSocket.setReceiveTimeOut(20_000);
int port = subSocket.bindToRandomPort("tcp://*");
@@ -98,12 +98,15 @@ public class ZeroMqMessageHandlerTests {
new FunctionExpression<Message<?>>((message) -> message.getHeaders().get("topic")));
messageHandler.setMessageMapper(new EmbeddedJsonHeadersMessageMapper());
messageHandler.afterPropertiesSet();
subSocket.subscribe("test");
// Give it some time to connect and subscribe
Thread.sleep(2000);
Message<?> testMessage = MessageBuilder.withPayload("test").setHeader("topic", "testTopic").build();
await().atMost(Duration.ofSeconds(20)).pollDelay(Duration.ofMillis(100))
.untilAsserted(() -> {
subSocket.subscribe("test");
messageHandler.handleMessage(testMessage).subscribe();
ZMsg msg = ZMsg.recvMsg(subSocket);
assertThat(msg).isNotNull();