package de.juplo.kafka.chat.backend;
import de.juplo.kafka.chat.backend.api.ChatRoomInfoTo;
-import de.juplo.kafka.chat.backend.api.MessageTo;
-import lombok.EqualsAndHashCode;
-import lombok.Getter;
-import lombok.Setter;
-import lombok.ToString;
import lombok.extern.slf4j.Slf4j;
+import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.springframework.core.ParameterizedTypeReference;
import org.springframework.http.HttpStatus;
import org.springframework.http.MediaType;
import org.springframework.http.codec.ServerSentEvent;
import org.springframework.web.reactive.function.client.WebClient;
-import org.testcontainers.images.ImagePullPolicy;
import org.testcontainers.junit.jupiter.Testcontainers;
import reactor.core.publisher.Flux;
import reactor.core.publisher.Mono;
-import reactor.util.retry.Retry;
-import java.time.Duration;
+import java.util.Arrays;
+import java.util.concurrent.ExecutorService;
+import java.util.concurrent.Executors;
import java.util.stream.IntStream;
@Slf4j
public abstract class AbstractHandoverIT
{
- static final ImagePullPolicy NEVER_PULL = imageName -> false;
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)
+ {
+ this.containers = containers;
+ }
@Test
void test() throws InterruptedException
{
- ChatRoomInfoTo chatRoom = createChatRoom("bar").block();
- User user = new User("nerd");
- IntStream
- .rangeClosed(1,100)
- .mapToObj(i ->sendMessage(chatRoom, user, "Message #" + i))
- .map(result -> result
- .map(MessageTo::toString)
- .retryWhen(Retry.fixedDelay(10, Duration.ofSeconds(1)))
- .block())
- .forEach(result -> log.info("{}", result));
-
- 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();
});
}
- Mono<MessageTo> sendMessage(
- ChatRoomInfoTo chatRoom,
- User user,
- String message)
- {
- return webClient
- .put()
- .uri(
- "/{chatRoomId}/{username}/{serial}",
- chatRoom.getId(),
- user.getName(),
- user.nextSerial())
- .contentType(MediaType.TEXT_PLAIN)
- .accept(MediaType.APPLICATION_JSON)
- .bodyValue(message)
- .exchangeToMono(response ->
- {
- if (response.statusCode().equals(HttpStatus.OK))
- {
- return response.bodyToMono(MessageTo.class);
- }
- else
- {
- return response.createError();
- }
- });
- }
-
Flux<ServerSentEvent<String>> receiveMessages(ChatRoomInfoTo chatRoom)
{
return webClient
WebClient webClient;
-
- @EqualsAndHashCode
- @ToString
- class User
+ @BeforeEach
+ void setUp() throws Exception
{
- @Getter
- private final String name;
- private int serial = 0;
-
-
- User (String name)
- {
- this.name = name;
- }
-
+ containers.setUp();
- int nextSerial()
- {
- return ++serial;
- }
- }
-
- @Getter
- @Setter
- static class StatusTo
- {
- String status;
+ Integer port = containers.haproxy.getMappedPort(8400);
+ webClient = WebClient.create("http://localhost:" + port);
}
}