X-Git-Url: http://juplo.de/gitweb/?a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fde%2Fjuplo%2Fkafka%2Fchat%2Fbackend%2Fpersistence%2FAbstractInMemoryStorageIT.java;h=142f70997e34b52f2da4ff2603722d28d6e5de60;hb=df207aa9a8cd349fd43785270d250a7f55593801;hp=a4defeb14606b89b5630c221c0d021d5daff8711;hpb=d7689f059964f8c6105b65bb3ee5a097fd106cb7;p=demos%2Fkafka%2Fchat

diff --git a/src/test/java/de/juplo/kafka/chat/backend/persistence/AbstractInMemoryStorageIT.java b/src/test/java/de/juplo/kafka/chat/backend/persistence/AbstractInMemoryStorageIT.java
index a4defeb1..142f7099 100644
--- a/src/test/java/de/juplo/kafka/chat/backend/persistence/AbstractInMemoryStorageIT.java
+++ b/src/test/java/de/juplo/kafka/chat/backend/persistence/AbstractInMemoryStorageIT.java
@@ -1,9 +1,7 @@
 package de.juplo.kafka.chat.backend.persistence;
 
-import de.juplo.kafka.chat.backend.persistence.inmemory.ChatHomeService;
-import de.juplo.kafka.chat.backend.domain.ChatRoomFactory;
-import de.juplo.kafka.chat.backend.persistence.inmemory.InMemoryChatHomeService;
-import de.juplo.kafka.chat.backend.persistence.inmemory.InMemoryChatRoomFactory;
+import de.juplo.kafka.chat.backend.domain.ChatHome;
+import de.juplo.kafka.chat.backend.persistence.inmemory.SimpleChatHome;
 import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
 
@@ -21,27 +19,17 @@ public abstract class AbstractInMemoryStorageIT extends AbstractStorageStrategyI
   {
     return new StorageStrategyITConfig()
     {
-      InMemoryChatHomeService chatHomeService = new InMemoryChatHomeService(
-          1,
-          new int[] { 0 },
-          getStorageStrategy().read());
+      int bufferSize = 8;
 
-      InMemoryChatRoomFactory chatRoomFactory = new InMemoryChatRoomFactory(
-          chatHomeService,
-          chatRoomId -> 0,
+      SimpleChatHome simpleChatHome = new SimpleChatHome(
+          getStorageStrategy(),
           clock,
-          8);
+          bufferSize);
 
       @Override
-      public ChatHomeService getChatHomeService()
+      public ChatHome getChatHome()
       {
-        return chatHomeService;
-      }
-
-      @Override
-      public ChatRoomFactory getChatRoomFactory()
-      {
-        return chatRoomFactory;
+        return simpleChatHome;
       }
     };
   }