From 38f24fd2ed2c8a7581bd43d3131242236208ef7a Mon Sep 17 00:00:00 2001 From: Corneil du Plessis Date: Thu, 17 Nov 2022 14:16:18 +0200 Subject: [PATCH] [SB3] Fix tests for changes in SB3. --- .../websocket/WebsocketConsumerTests.java | 19 +++++++++++++++---- 1 file changed, 15 insertions(+), 4 deletions(-) diff --git a/functions/consumer/websocket-consumer/src/test/java/org/springframework/cloud/fn/consumer/websocket/WebsocketConsumerTests.java b/functions/consumer/websocket-consumer/src/test/java/org/springframework/cloud/fn/consumer/websocket/WebsocketConsumerTests.java index d3ae375a..a19f179f 100644 --- a/functions/consumer/websocket-consumer/src/test/java/org/springframework/cloud/fn/consumer/websocket/WebsocketConsumerTests.java +++ b/functions/consumer/websocket-consumer/src/test/java/org/springframework/cloud/fn/consumer/websocket/WebsocketConsumerTests.java @@ -107,13 +107,24 @@ public class WebsocketConsumerTests { // submit a single message String payload = UUID.randomUUID().toString(); websocketConsumer.accept(MessageBuilder.withPayload(payload).build()); - + List responses = new ArrayList<>(); // await completion on each handler for (WebsocketConsumerClientHandler handler : handlers) { - handler.await(); - assertThat(handler.getReceivedMessages().size()).isEqualTo(1); - assertThat(handler.getReceivedMessages().get(0)).isEqualTo(payload); + Awaitility.await() + .atMost(Duration.ofSeconds(2)) + .until(() -> { + handler.await(); + if (!handler.getReceivedMessages().isEmpty()) { + responses.add(handler.getReceivedMessages().get(0)); + } + return !handler.getReceivedMessages().isEmpty(); + }); + } + assertThat(responses.size()).isEqualTo(handlers.size()); + responses.forEach(s -> { + assertThat(s).isEqualTo(payload); + }); } @Test