SpringCloud/spring-functions-catalog
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[SB3] Fix tests for changes in SB3.

Browse Source
This commit is contained in:
Corneil du Plessis
2022-11-17 13:11:50 +02:00
parent 1318cbde93
commit b168cb9828
1 changed files with 13 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
consumer/websocket-consumer/src/test/java/org/springframework/cloud/fn/consumer/websocket/WebsocketConsumerTests.java
View File

@@ -16,12 +16,14 @@
package org.springframework.cloud.fn.consumer.websocket;
import java.time.Duration;
import java.util.ArrayList;
import java.util.List;
import java.util.UUID;
import java.util.concurrent.ExecutionException;
import java.util.function.Consumer;
import org.awaitility.Awaitility;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.Timeout;
@@ -80,12 +82,19 @@ public class WebsocketConsumerTests {
public void testMultipleMessageSingleSubscriber() throws Exception {
WebsocketConsumerClientHandler handler = new WebsocketConsumerClientHandler("handler_0", MESSAGE_COUNT, TIMEOUT);
doHandshake(handler);
List<String> messagesToSend = submitMultipleMessages(MESSAGE_COUNT);
handler.await();
List<String> received = new ArrayList<>();
Awaitility.await()
.atMost(Duration.ofSeconds(10))
.until(() -> {
handler.await();
received.addAll(handler.getReceivedMessages());
return received.size() == MESSAGE_COUNT;
}
);
assertThat(handler.getReceivedMessages().size()).isEqualTo(MESSAGE_COUNT);
messagesToSend.forEach(s -> assertThat(handler.getReceivedMessages().contains(s)).isTrue());
assertThat(received.size()).isEqualTo(MESSAGE_COUNT);
messagesToSend.forEach(s -> assertThat(received.contains(s)).isTrue());
}
@Test