- 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;