import reactor.core.publisher.Flux;
import reactor.core.publisher.Mono;
+import java.util.Arrays;
+import java.util.concurrent.ExecutorService;
+import java.util.concurrent.Executors;
+
@Testcontainers
@Slf4j
public abstract class AbstractHandoverIT
{
static final ParameterizedTypeReference<ServerSentEvent<String>> SSE_TYPE = new ParameterizedTypeReference<>() {};
+ static final int NUM_CHATROOMS = 23;
+ static final int NUM_CLIENTS = 17;
private final AbstractHandoverITContainers containers;
+ private final ExecutorService executorService = Executors.newFixedThreadPool(NUM_CLIENTS + 1);
AbstractHandoverIT(AbstractHandoverITContainers containers)
@Test
void test() throws InterruptedException
{
- ChatRoomInfoTo chatRoom = createChatRoom("bar").block();
- TestClient testClient = new TestClient(
- containers.haproxy.getMappedPort(8400),
- chatRoom,
- "nerd");
- testClient.run();
-
- receiveMessages(chatRoom)
- .take(100)
- .doOnNext(message -> log.info("message: {}", message))
- .then()
- .block();
+ ChatRoomInfoTo[] chatRooms = Flux
+ .range(0, NUM_CHATROOMS)
+ .flatMap(i -> createChatRoom("room-" + i))
+ .toStream()
+ .toArray(size -> new ChatRoomInfoTo[size]);
+
+ int port = containers.haproxy.getMappedPort(8400);
+
+ TestWriter[] testWriters = Flux
+ .range(0, NUM_CLIENTS)
+ .map(i -> new TestWriter(
+ port,
+ chatRooms[i % NUM_CHATROOMS],
+ "user-" + Integer.toString(i)))
+ .doOnNext(testClient -> executorService.execute(testClient))
+ .toStream()
+ .toArray(size -> new TestWriter[size]);
+
+ TestListener testListener = new TestListener(port, chatRooms);
+ executorService.execute(testListener);
+
+ Thread.sleep(10000);
+
+ Arrays
+ .stream(testWriters)
+ .forEach(testClient -> testClient.running = false);
+ testListener.running = false;
}
Mono<ChatRoomInfoTo> createChatRoom(String name)
});
}
- Flux<ServerSentEvent<String>> receiveMessages(ChatRoomInfoTo chatRoom)
- {
- return webClient
- .get()
- .uri(
- "/{chatRoomId}/listen",
- chatRoom.getId())
- .accept(MediaType.TEXT_EVENT_STREAM)
- .retrieve()
- .bodyToFlux(SSE_TYPE);
- }
-
WebClient webClient;