1 package de.juplo.kafka.chat.backend.persistence;
3 import de.juplo.kafka.chat.backend.domain.ShardingStrategy;
4 import de.juplo.kafka.chat.backend.domain.ChatHomeService;
5 import de.juplo.kafka.chat.backend.domain.ChatRoomFactory;
6 import de.juplo.kafka.chat.backend.persistence.inmemory.InMemoryChatHomeService;
7 import de.juplo.kafka.chat.backend.persistence.InMemoryWithMongoDbStorageIT.DataSourceInitializer;
8 import de.juplo.kafka.chat.backend.persistence.inmemory.InMemoryChatRoomFactory;
9 import de.juplo.kafka.chat.backend.persistence.inmemory.InMemoryChatRoomService;
10 import de.juplo.kafka.chat.backend.persistence.storage.mongodb.ChatRoomRepository;
11 import de.juplo.kafka.chat.backend.persistence.storage.mongodb.MongoDbStorageStrategy;
12 import lombok.extern.slf4j.Slf4j;
13 import org.junit.jupiter.api.BeforeEach;
14 import org.junit.jupiter.api.extension.ExtendWith;
15 import org.springframework.beans.factory.annotation.Autowired;
16 import org.springframework.boot.autoconfigure.EnableAutoConfiguration;
17 import org.springframework.boot.test.autoconfigure.data.mongo.AutoConfigureDataMongo;
18 import org.springframework.boot.test.context.TestConfiguration;
19 import org.springframework.context.ApplicationContextInitializer;
20 import org.springframework.context.ConfigurableApplicationContext;
21 import org.springframework.context.annotation.Bean;
22 import org.springframework.test.context.ContextConfiguration;
23 import org.springframework.test.context.junit.jupiter.SpringExtension;
24 import org.springframework.test.context.support.TestPropertySourceUtils;
25 import org.testcontainers.containers.GenericContainer;
26 import org.testcontainers.containers.output.Slf4jLogConsumer;
27 import org.testcontainers.junit.jupiter.Container;
28 import org.testcontainers.junit.jupiter.Testcontainers;
30 import java.time.Clock;
31 import java.util.function.Supplier;
35 @ExtendWith({SpringExtension.class})
36 @EnableAutoConfiguration
37 @AutoConfigureDataMongo
38 @ContextConfiguration(initializers = DataSourceInitializer.class)
40 public class InMemoryWithMongoDbStorageIT extends AbstractStorageStrategyIT
43 MongoDbStorageStrategy storageStrategy;
45 ChatRoomRepository repository;
51 protected StorageStrategy getStorageStrategy()
53 return storageStrategy;
57 protected Supplier<ChatHomeService> getChatHomeServiceSupplier()
59 return () -> new InMemoryChatHomeService(
62 getStorageStrategy().read());
66 protected ChatRoomFactory getChatRoomFactory()
68 ShardingStrategy strategy = chatRoomId -> 0;
69 return new InMemoryChatRoomFactory(strategy, clock, 8);
73 static class InMemoryWithMongoDbStorageStrategyITConfig
76 MongoDbStorageStrategy storageStrategy(
77 ChatRoomRepository chatRoomRepository,
80 return new MongoDbStorageStrategy(
85 messageFlux -> new InMemoryChatRoomService(messageFlux));
91 return Clock.systemDefaultZone();
95 private static final int MONGODB_PORT = 27017;
98 private static final GenericContainer CONTAINER =
99 new GenericContainer("mongo:6")
100 .withExposedPorts(MONGODB_PORT);
102 public static class DataSourceInitializer
103 implements ApplicationContextInitializer<ConfigurableApplicationContext>
106 public void initialize(ConfigurableApplicationContext applicationContext)
108 TestPropertySourceUtils.addInlinedPropertiesToEnvironment(
110 "spring.data.mongodb.host=localhost",
111 "spring.data.mongodb.port=" + CONTAINER.getMappedPort(MONGODB_PORT),
112 "spring.data.mongodb.database=test");
119 Slf4jLogConsumer logConsumer = new Slf4jLogConsumer(log);
120 CONTAINER.followOutput(logConsumer);
121 repository.deleteAll();