package de.juplo.kafka.chat.backend;
import de.juplo.kafka.chat.backend.api.ChatRoomInfoTo;
+import de.juplo.kafka.chat.backend.api.MessageTo;
import lombok.extern.slf4j.Slf4j;
+import org.awaitility.Awaitility;
import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Disabled;
import org.junit.jupiter.api.Test;
import org.springframework.http.HttpStatus;
import org.springframework.http.MediaType;
import org.springframework.web.reactive.function.client.WebClient;
import org.testcontainers.junit.jupiter.Testcontainers;
+import pl.rzrz.assertj.reactor.Assertions;
import reactor.core.publisher.Flux;
import reactor.core.publisher.Mono;
+import java.time.Duration;
+import java.util.List;
import java.util.concurrent.CompletableFuture;
}
+ @Disabled
@Test
void test() throws InterruptedException
{
+ log.info("Starting backend-1...");
+ containers.startBackend(containers.backend1, new TestWriter[0]);
+ log.info("backend-1 started!");
+
ChatRoomInfoTo[] chatRooms = Flux
.range(0, NUM_CHATROOMS)
.flatMap(i -> createChatRoom("room-" + i))
}
TestListener testListener = new TestListener(port, chatRooms);
- CompletableFuture<Void> testListenerFuture = testListener
+ testListener
.run()
- .toFuture();
+ .subscribe(message -> log.info(
+ "Received message: {}",
+ message));
- log.info("Sleeping for 3 seconds...");
- Thread.sleep(3000);
+ log.info("Starting backend-2...");
+ containers.startBackend(containers.backend2, testWriters);
+ log.info("backend-2 started!");
for (int i = 0; i < NUM_CLIENTS; i++)
{
log.info("Joined TestWriter {}", testWriters[i].user);
}
+ Awaitility
+ .await()
+ .atMost(Duration.ofSeconds(30))
+ .untilAsserted(() -> assertAllSentMessagesReceived(testWriters, testListener));
+ }
- log.info("Sleeping for 3 seconds...");
- Thread.sleep(3000);
- log.info("Joining TestListener...");
- testListener.running = false;
- testListenerFuture.join();
- log.info("Joined TestListener");
+ private void assertAllSentMessagesReceived(
+ TestWriter[] testWriters,
+ TestListener testListener)
+ {
+ for (int i = 0; i < NUM_CLIENTS; i++)
+ {
+ TestWriter testWriter = testWriters[i];
+ ChatRoomInfoTo chatRoom = testWriter.chatRoom;
+ List<MessageTo> receivedMessages = testListener.receivedMessages.get(chatRoom.getId());
+
+ Assertions.assertThat(receivedMessages
+ .stream()
+ .filter(message -> message.getUser().equals(testWriter.user.getName()))
+ ).containsExactlyElementsOf(testWriter.sentMessages);
+ }
}
Mono<ChatRoomInfoTo> createChatRoom(String name)