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.awaitility.Awaitility;
7 import org.junit.jupiter.api.BeforeEach;
8 import org.junit.jupiter.api.Disabled;
9 import org.junit.jupiter.api.Test;
10 import org.springframework.http.HttpStatus;
11 import org.springframework.http.MediaType;
12 import org.springframework.web.reactive.function.client.WebClient;
13 import org.testcontainers.junit.jupiter.Testcontainers;
14 import pl.rzrz.assertj.reactor.Assertions;
15 import reactor.core.publisher.Flux;
16 import reactor.core.publisher.Mono;
18 import java.time.Duration;
19 import java.util.List;
20 import java.util.concurrent.CompletableFuture;
25 public abstract class AbstractHandoverIT
27 static final int NUM_CHATROOMS = 23;
28 static final int NUM_CLIENTS = 17;
31 private final AbstractHandoverITContainers containers;
34 AbstractHandoverIT(AbstractHandoverITContainers containers)
36 this.containers = containers;
42 void test() throws InterruptedException
44 log.info("Starting backend-1...");
45 containers.startBackend(containers.backend1, new TestWriter[0]);
46 log.info("backend-1 started!");
48 ChatRoomInfoTo[] chatRooms = Flux
49 .range(0, NUM_CHATROOMS)
50 .flatMap(i -> createChatRoom("room-" + i))
52 .toArray(size -> new ChatRoomInfoTo[size]);
54 int port = containers.haproxy.getMappedPort(8400);
56 CompletableFuture<Void>[] testWriterFutures = new CompletableFuture[NUM_CLIENTS];
57 TestWriter[] testWriters = new TestWriter[NUM_CLIENTS];
58 for (int i = 0; i < NUM_CLIENTS; i++)
60 TestWriter testWriter = new TestWriter(
62 chatRooms[i % NUM_CHATROOMS],
64 testWriters[i] = testWriter;
65 testWriterFutures[i] = testWriter
70 TestListener testListener = new TestListener(port, chatRooms);
73 .subscribe(message -> log.info(
74 "Received message: {}",
77 log.info("Starting backend-2...");
78 containers.startBackend(containers.backend2, testWriters);
79 log.info("backend-2 started!");
81 for (int i = 0; i < NUM_CLIENTS; i++)
83 testWriters[i].running = false;
84 testWriterFutures[i].join();
85 log.info("Joined TestWriter {}", testWriters[i].user);
90 .atMost(Duration.ofSeconds(30))
91 .untilAsserted(() -> assertAllSentMessagesReceived(testWriters, testListener));
94 private void assertAllSentMessagesReceived(
95 TestWriter[] testWriters,
96 TestListener testListener)
98 for (int i = 0; i < NUM_CLIENTS; i++)
100 TestWriter testWriter = testWriters[i];
101 ChatRoomInfoTo chatRoom = testWriter.chatRoom;
102 List<MessageTo> receivedMessages = testListener.receivedMessages.get(chatRoom.getId());
104 Assertions.assertThat(receivedMessages
106 .filter(message -> message.getUser().equals(testWriter.user.getName()))
107 ).containsExactlyElementsOf(testWriter.sentMessages);
111 Mono<ChatRoomInfoTo> createChatRoom(String name)
116 .contentType(MediaType.TEXT_PLAIN)
118 .accept(MediaType.APPLICATION_JSON)
119 .exchangeToMono(response ->
121 if (response.statusCode().equals(HttpStatus.OK))
123 return response.bodyToMono(ChatRoomInfoTo.class);
127 return response.createError();
136 void setUp() throws Exception
140 Integer port = containers.haproxy.getMappedPort(8400);
141 webClient = WebClient.create("http://localhost:" + port);