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=0ec0bc199fb5320fa81b7ba904f2aa5c9eb4331c;hpb=47d4a54aaa1ce13180ab6fdad0569cf2d0851563;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 0ec0bc19..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,15 +1,11 @@
 package de.juplo.kafka.chat.backend;
 
-import de.juplo.kafka.chat.backend.domain.ChatHomeService;
-import de.juplo.kafka.chat.backend.implementation.StorageStrategy;
-import de.juplo.kafka.chat.backend.implementation.inmemory.InMemoryServicesConfiguration;
 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;
-
 
 @ContextConfiguration(classes = InMemoryTestUtils.class)
 @Slf4j
@@ -17,10 +13,12 @@ public abstract class AbstractInMemoryStorageIT extends AbstractStorageStrategyI
 {
   @Autowired
   InMemoryTestUtils testUtils;
+  @Autowired
+  SimpleChatHomeService simpleChatHomeService;
 
   @Override
-  ChatHomeService getChatHome()
+  void restore()
   {
-    return testUtils.createNoneShardingChatHomeService();
+    testUtils.restore(simpleChatHomeService).block();
   }
 }