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

   1 package de.juplo.kafka.chat.backend.persistence;
   2
   3 import de.juplo.kafka.chat.backend.persistence.InMemoryWithMongoDbStorageIT.DataSourceInitializer;
   4 import de.juplo.kafka.chat.backend.persistence.inmemory.InMemoryChatRoomService;
   5 import de.juplo.kafka.chat.backend.persistence.storage.mongodb.ChatRoomRepository;
   6 import de.juplo.kafka.chat.backend.persistence.storage.mongodb.MongoDbStorageStrategy;
   7 import lombok.extern.slf4j.Slf4j;
   8 import org.junit.jupiter.api.BeforeEach;
   9 import org.junit.jupiter.api.extension.ExtendWith;
  10 import org.springframework.beans.factory.annotation.Autowired;
  11 import org.springframework.boot.autoconfigure.EnableAutoConfiguration;
  12 import org.springframework.boot.test.autoconfigure.data.mongo.AutoConfigureDataMongo;
  13 import org.springframework.boot.test.context.TestConfiguration;
  14 import org.springframework.context.ApplicationContextInitializer;
  15 import org.springframework.context.ConfigurableApplicationContext;
  16 import org.springframework.context.annotation.Bean;
  17 import org.springframework.test.context.ContextConfiguration;
  18 import org.springframework.test.context.junit.jupiter.SpringExtension;
  19 import org.springframework.test.context.support.TestPropertySourceUtils;
  20 import org.testcontainers.containers.GenericContainer;
  21 import org.testcontainers.containers.output.Slf4jLogConsumer;
  22 import org.testcontainers.junit.jupiter.Container;
  23 import org.testcontainers.junit.jupiter.Testcontainers;
  24
  25 import java.time.Clock;
  26
  27
  28 @Testcontainers
  29 @ExtendWith({SpringExtension.class})
  30 @EnableAutoConfiguration
  31 @AutoConfigureDataMongo
  32 @ContextConfiguration(initializers = DataSourceInitializer.class)
  33 @Slf4j
  34 public class InMemoryWithMongoDbStorageIT extends AbstractInMemoryStorageIT
  35 {
  36   @Autowired
  37   MongoDbStorageStrategy storageStrategy;
  38   @Autowired
  39   ChatRoomRepository repository;
  40   @Autowired
  41   Clock clock;
  42
  43
  44   public InMemoryWithMongoDbStorageIT()
  45   {
  46     super(Clock.systemDefaultZone());
  47   }
  48
  49
  50   @Override
  51   protected StorageStrategy getStorageStrategy()
  52   {
  53     return storageStrategy;
  54   }
  55
  56   @TestConfiguration
  57   static class InMemoryWithMongoDbStorageStrategyITConfig
  58   {
  59     @Bean
  60     MongoDbStorageStrategy storageStrategy(
  61         ChatRoomRepository chatRoomRepository,
  62         Clock clock)
  63     {
  64       return new MongoDbStorageStrategy(
  65           chatRoomRepository,
  66           clock,
  67           8,
  68           chatRoomId -> 0,
  69           messageFlux -> new InMemoryChatRoomService(messageFlux));
  70     }
  71
  72     @Bean
  73     Clock clock()
  74     {
  75       return Clock.systemDefaultZone();
  76     }
  77   }
  78
  79   private static final int MONGODB_PORT = 27017;
  80
  81   @Container
  82   private static final GenericContainer CONTAINER =
  83       new GenericContainer("mongo:6")
  84           .withExposedPorts(MONGODB_PORT);
  85
  86   public static class DataSourceInitializer
  87       implements ApplicationContextInitializer<ConfigurableApplicationContext>
  88   {
  89     @Override
  90     public void initialize(ConfigurableApplicationContext applicationContext)
  91     {
  92       TestPropertySourceUtils.addInlinedPropertiesToEnvironment(
  93           applicationContext,
  94           "spring.data.mongodb.host=localhost",
  95           "spring.data.mongodb.port=" + CONTAINER.getMappedPort(MONGODB_PORT),
  96           "spring.data.mongodb.database=test");
  97     }
  98   }
  99
 100   @BeforeEach
 101   void setUpLogging()
 102   {
 103     Slf4jLogConsumer logConsumer = new Slf4jLogConsumer(log);
 104     CONTAINER.followOutput(logConsumer);
 105     repository.deleteAll();
 106   }
 107 }