X-Git-Url: https://juplo.de/gitweb/?a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fde%2Fjuplo%2Fkafka%2Fchat%2Fbackend%2FAbstractInMemoryStorageIT.java;h=843b9490c7c3a9c03100c786f6a6fe60aa73d7ea;hb=64ede95835a496e84857c38213dbf8ea451878e0;hp=703911de72f97957c2a221f7dd38731522edd0d7;hpb=9ac32a888a2a03e7d40679226213c3b3b67d586e;p=demos%2Fkafka%2Fchat

diff --git a/src/test/java/de/juplo/kafka/chat/backend/AbstractInMemoryStorageIT.java b/src/test/java/de/juplo/kafka/chat/backend/AbstractInMemoryStorageIT.java
index 703911de..843b9490 100644
--- a/src/test/java/de/juplo/kafka/chat/backend/AbstractInMemoryStorageIT.java
+++ b/src/test/java/de/juplo/kafka/chat/backend/AbstractInMemoryStorageIT.java
@@ -1,36 +1,24 @@
 package de.juplo.kafka.chat.backend;
 
-import de.juplo.kafka.chat.backend.domain.ChatHomeService;
-import de.juplo.kafka.chat.backend.persistence.inmemory.SimpleChatHomeService;
-import lombok.RequiredArgsConstructor;
+import de.juplo.kafka.chat.backend.implementation.inmemory.InMemoryTestUtils;
+import de.juplo.kafka.chat.backend.implementation.inmemory.SimpleChatHomeService;
 import lombok.extern.slf4j.Slf4j;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.test.context.ContextConfiguration;
 
-import java.time.Clock;
 
-
-@RequiredArgsConstructor
+@ContextConfiguration(classes = InMemoryTestUtils.class)
 @Slf4j
 public abstract class AbstractInMemoryStorageIT extends AbstractStorageStrategyIT
 {
-  final Clock clock;
+  @Autowired
+  InMemoryTestUtils testUtils;
+  @Autowired
+  SimpleChatHomeService simpleChatHomeService;
 
   @Override
-  protected StorageStrategyITConfig getConfig()
+  void restore()
   {
-    return new StorageStrategyITConfig()
-    {
-      int bufferSize = 8;
-
-      SimpleChatHomeService simpleChatHome = new SimpleChatHomeService(
-          getStorageStrategy(),
-          clock,
-          bufferSize);
-
-      @Override
-      public ChatHomeService getChatHome()
-      {
-        return simpleChatHome;
-      }
-    };
+    testUtils.restore(simpleChatHomeService).block();
   }
 }