juplo.de Git - demos/kafka/chat/blob - src/test/java/de/juplo/kafka/chat/backend/persistence/InMemoryWithMongoDbStorageStrategyIT.java

   1 package de.juplo.kafka.chat.backend.persistence;
   2
   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;
  29
  30 import java.time.Clock;
  31 import java.util.function.Supplier;
  32
  33
  34 @Testcontainers
  35 @ExtendWith({SpringExtension.class})
  36 @EnableAutoConfiguration
  37 @AutoConfigureDataMongo
  38 @ContextConfiguration(initializers = DataSourceInitializer.class)
  39 @Slf4j
  40 public class InMemoryWithMongoDbStorageStrategyIT extends AbstractStorageStrategyIT
  41 {
  42   @Autowired
  43   MongoDbStorageStrategy storageStrategy;
  44   @Autowired
  45   Clock clock;
  46
  47
  48   @Override
  49   protected StorageStrategy getStorageStrategy()
  50   {
  51     return storageStrategy;
  52   }
  53
  54   @Override
  55   protected Supplier<ChatHomeService> getChatHomeServiceSupplier()
  56   {
  57     return () -> new InMemoryChatHomeService(1, getStorageStrategy().read());
  58   }
  59
  60   @Override
  61   protected ChatRoomFactory getChatRoomFactory()
  62   {
  63     ShardingStrategy strategy = chatRoomId -> 0;
  64     return new InMemoryChatRoomFactory(strategy, clock, 8);
  65   }
  66
  67   @TestConfiguration
  68   static class InMemoryWithMongoDbStorageStrategyITConfig
  69   {
  70     @Bean
  71     MongoDbStorageStrategy storageStrategy(
  72         ChatRoomRepository chatRoomRepository,
  73         Clock clock)
  74     {
  75       return new MongoDbStorageStrategy(
  76           chatRoomRepository,
  77           clock,
  78           8,
  79           messageFlux -> new InMemoryChatRoomService(messageFlux));
  80     }
  81
  82     @Bean
  83     Clock clock()
  84     {
  85       return Clock.systemDefaultZone();
  86     }
  87   }
  88
  89   private static final int MONGODB_PORT = 27017;
  90
  91   @Container
  92   private static final GenericContainer CONTAINER =
  93       new GenericContainer("mongo:6")
  94           .withExposedPorts(MONGODB_PORT);
  95
  96   public static class DataSourceInitializer
  97       implements ApplicationContextInitializer<ConfigurableApplicationContext>
  98   {
  99     @Override
 100     public void initialize(ConfigurableApplicationContext applicationContext)
 101     {
 102       TestPropertySourceUtils.addInlinedPropertiesToEnvironment(
 103           applicationContext,
 104           "spring.data.mongodb.host=localhost",
 105           "spring.data.mongodb.port=" + CONTAINER.getMappedPort(MONGODB_PORT),
 106           "spring.data.mongodb.database=test");
 107     }
 108   }
 109
 110   @BeforeEach
 111   void setUpLogging()
 112   {
 113     Slf4jLogConsumer logConsumer = new Slf4jLogConsumer(log);
 114     CONTAINER.followOutput(logConsumer);
 115   }
 116 }