1 package de.juplo.kafka.chat.backend;
3 import de.juplo.kafka.chat.backend.api.ChatRoomInfoTo;
4 import de.juplo.kafka.chat.backend.api.MessageTo;
5 import lombok.extern.slf4j.Slf4j;
6 import org.junit.jupiter.api.BeforeEach;
7 import org.junit.jupiter.api.Disabled;
8 import org.junit.jupiter.api.Test;
9 import org.springframework.http.HttpStatus;
10 import org.springframework.http.MediaType;
11 import org.springframework.web.reactive.function.client.WebClient;
12 import org.testcontainers.junit.jupiter.Testcontainers;
13 import pl.rzrz.assertj.reactor.Assertions;
14 import reactor.core.publisher.Flux;
15 import reactor.core.publisher.Mono;
17 import java.util.List;
18 import java.util.concurrent.CompletableFuture;
23 public abstract class AbstractHandoverIT
25 static final int NUM_CHATROOMS = 23;
26 static final int NUM_CLIENTS = 17;
29 private final AbstractHandoverITContainers containers;
32 AbstractHandoverIT(AbstractHandoverITContainers containers)
34 this.containers = containers;
40 void test() throws InterruptedException
42 ChatRoomInfoTo[] chatRooms = Flux
43 .range(0, NUM_CHATROOMS)
44 .flatMap(i -> createChatRoom("room-" + i))
46 .toArray(size -> new ChatRoomInfoTo[size]);
48 int port = containers.haproxy.getMappedPort(8400);
50 CompletableFuture<Void>[] testWriterFutures = new CompletableFuture[NUM_CLIENTS];
51 TestWriter[] testWriters = new TestWriter[NUM_CLIENTS];
52 for (int i = 0; i < NUM_CLIENTS; i++)
54 TestWriter testWriter = new TestWriter(
56 chatRooms[i % NUM_CHATROOMS],
58 testWriters[i] = testWriter;
59 testWriterFutures[i] = testWriter
64 TestListener testListener = new TestListener(port, chatRooms);
67 .subscribe(message -> log.info(
68 "Received message: {}",
71 log.info("Sleeping for 3 seconds...");
74 log.info("Starting Backend 2...");
75 containers.backend2.start();
76 log.info("Backend 2 started");
78 log.info("Sleeping for 3 seconds...");
81 log.info("Starting Backend 3...");
82 containers.backend3.start();
83 log.info("Backend 3 started");
85 log.info("Sleeping for 3 seconds...");
88 log.info("Shutting down writers...");
89 for (int i = 0; i < NUM_CLIENTS; i++)
91 log.info("Shutting down writer {}", i);
92 testWriters[i].running = false;
93 testWriterFutures[i].join();
94 log.info("Joined TestWriter {}", testWriters[i].user);
97 // Yield the work, so that the last messages can be received
100 for (int i = 0; i < NUM_CLIENTS; i++)
102 TestWriter testWriter = testWriters[i];
103 ChatRoomInfoTo chatRoom = testWriter.chatRoom;
104 List<MessageTo> receivedMessages = testListener.receivedMessages.get(chatRoom.getId());
106 Assertions.assertThat(receivedMessages
108 .filter(message -> message.getUser().equals(testWriter.user.getName()))
109 ).containsExactlyElementsOf(testWriter.sentMessages);
113 Mono<ChatRoomInfoTo> createChatRoom(String name)
118 .contentType(MediaType.TEXT_PLAIN)
120 .accept(MediaType.APPLICATION_JSON)
121 .exchangeToMono(response ->
123 if (response.statusCode().equals(HttpStatus.OK))
125 return response.bodyToMono(ChatRoomInfoTo.class);
129 return response.createError();
138 void setUp() throws Exception
142 Integer port = containers.haproxy.getMappedPort(8400);
143 webClient = WebClient.create("http://localhost:" + port);