1 package de.juplo.kafka.chat.backend.persistence;
3 import de.juplo.kafka.chat.backend.api.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.InMemoryWithMongoDbStorageStrategyIT.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 InMemoryWithMongoDbStorageStrategyIT extends AbstractStorageStrategyIT
43 MongoDbStorageStrategy storageStrategy;
49 protected StorageStrategy getStorageStrategy()
51 return storageStrategy;
55 protected Supplier<ChatHomeService> getChatHomeServiceSupplier()
57 return () -> new InMemoryChatHomeService(1, getStorageStrategy().read());
61 protected ChatRoomFactory getChatRoomFactory()
63 ShardingStrategy strategy = chatRoomId -> 0;
64 return new InMemoryChatRoomFactory(strategy, clock, 8);
68 static class InMemoryWithMongoDbStorageStrategyITConfig
71 MongoDbStorageStrategy storageStrategy(
72 ChatRoomRepository chatRoomRepository,
75 return new MongoDbStorageStrategy(
79 messageFlux -> new InMemoryChatRoomService(messageFlux));
85 return Clock.systemDefaultZone();
89 private static final int MONGODB_PORT = 27017;
92 private static final GenericContainer CONTAINER =
93 new GenericContainer("mongo:6")
94 .withExposedPorts(MONGODB_PORT);
96 public static class DataSourceInitializer
97 implements ApplicationContextInitializer<ConfigurableApplicationContext>
100 public void initialize(ConfigurableApplicationContext applicationContext)
102 TestPropertySourceUtils.addInlinedPropertiesToEnvironment(
104 "spring.data.mongodb.host=localhost",
105 "spring.data.mongodb.port=" + CONTAINER.getMappedPort(MONGODB_PORT),
106 "spring.data.mongodb.database=test");
113 Slf4jLogConsumer logConsumer = new Slf4jLogConsumer(log);
114 CONTAINER.followOutput(logConsumer);