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

   1 package de.juplo.kafka.chat.backend;
   2
   3 import de.juplo.kafka.chat.backend.InMemoryWithMongoDbStorageIT.DataSourceInitializer;
   4 import de.juplo.kafka.chat.backend.implementation.StorageStrategy;
   5 import de.juplo.kafka.chat.backend.storage.mongodb.ChatRoomRepository;
   6 import de.juplo.kafka.chat.backend.storage.mongodb.MessageRepository;
   7 import de.juplo.kafka.chat.backend.storage.mongodb.MongoDbStorageStrategy;
   8 import lombok.extern.slf4j.Slf4j;
   9 import org.junit.jupiter.api.BeforeEach;
  10 import org.junit.jupiter.api.extension.ExtendWith;
  11 import org.springframework.beans.factory.annotation.Autowired;
  12 import org.springframework.boot.autoconfigure.EnableAutoConfiguration;
  13 import org.springframework.boot.test.autoconfigure.data.mongo.AutoConfigureDataMongo;
  14 import org.springframework.boot.test.context.TestConfiguration;
  15 import org.springframework.context.ApplicationContextInitializer;
  16 import org.springframework.context.ConfigurableApplicationContext;
  17 import org.springframework.context.annotation.Bean;
  18 import org.springframework.test.context.ContextConfiguration;
  19 import org.springframework.test.context.junit.jupiter.SpringExtension;
  20 import org.springframework.test.context.support.TestPropertySourceUtils;
  21 import org.testcontainers.containers.GenericContainer;
  22 import org.testcontainers.containers.output.Slf4jLogConsumer;
  23 import org.testcontainers.junit.jupiter.Container;
  24 import org.testcontainers.junit.jupiter.Testcontainers;
  25
  26 import java.time.Clock;
  27
  28
  29 @Testcontainers
  30 @ExtendWith({SpringExtension.class})
  31 @EnableAutoConfiguration
  32 @AutoConfigureDataMongo
  33 @ContextConfiguration(initializers = DataSourceInitializer.class)
  34 @Slf4j
  35 public class InMemoryWithMongoDbStorageIT extends AbstractInMemoryStorageIT
  36 {
  37   @Autowired
  38   MongoDbStorageStrategy storageStrategy;
  39   @Autowired
  40   ChatRoomRepository chatRoomRepository;
  41   @Autowired
  42   MessageRepository messageRepository;
  43
  44
  45   public InMemoryWithMongoDbStorageIT()
  46   {
  47     super(Clock.systemDefaultZone());
  48   }
  49
  50
  51   @Override
  52   protected StorageStrategy getStorageStrategy()
  53   {
  54     return storageStrategy;
  55   }
  56
  57   @TestConfiguration
  58   static class InMemoryWithMongoDbStorageStrategyITConfig
  59   {
  60     @Bean
  61     MongoDbStorageStrategy storageStrategy(
  62         ChatRoomRepository chatRoomRepository,
  63         MessageRepository messageRepository,
  64         Clock clock)
  65     {
  66       return new MongoDbStorageStrategy(
  67           chatRoomRepository,
  68           messageRepository,
  69           chatRoomId -> 0);
  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     chatRoomRepository.deleteAll();
 106     messageRepository.deleteAll();
 107   }
 108 }