import reactor.core.publisher.Flux;
import reactor.core.publisher.Mono;
+import java.util.Arrays;
+import java.util.concurrent.ExecutorService;
+import java.util.concurrent.Executors;
+import java.util.stream.IntStream;
+
@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);
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)
+ ChatRoomInfoTo[] chatRooms = Flux
+ .range(0, NUM_CHATROOMS)
+ .flatMap(i -> createChatRoom("room-" + i))
+ .toStream()
+ .toArray(size -> new ChatRoomInfoTo[size]);
+
+ TestClient[] testClients = Flux
+ .fromStream(IntStream.range(0, NUM_CLIENTS).mapToObj(i -> "user-" + Integer.toString(i)))
+ .map(i -> new TestClient(
+ containers.haproxy.getMappedPort(8400),
+ chatRooms,
+ i))
+ .doOnNext(testClient -> executorService.execute(testClient))
+ .toStream()
+ .toArray(size -> new TestClient[size]);
+
+ Thread.sleep(2000);
+
+ Arrays
+ .stream(testClients)
+ .forEach(testClient -> testClient.running = false);
+
+ Flux
+ .fromArray(chatRooms)
+ .flatMap(chatRoom ->receiveMessages(chatRoom).take(50))
.doOnNext(message -> log.info("message: {}", message))
.then()
.block();