X-Git-Url: http://juplo.de/gitweb/?a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fde%2Fjuplo%2Fkafka%2Fchat%2Fbackend%2Fpersistence%2Finmemory%2FShardedChatHomeTest.java;h=9b4bde7ea328618d94b216b130582728b3b1c5a8;hb=586d8662e763c75eb689735bd6d99e240fee8e4b;hp=ff82986a9403556d29aba0a7ac55a98ff2c31f51;hpb=bb1f9a4b980e35790cef4892063ab6186f2b7084;p=demos%2Fkafka%2Fchat

diff --git a/src/test/java/de/juplo/kafka/chat/backend/persistence/inmemory/ShardedChatHomeTest.java b/src/test/java/de/juplo/kafka/chat/backend/persistence/inmemory/ShardedChatHomeTest.java
index ff82986a..9b4bde7e 100644
--- a/src/test/java/de/juplo/kafka/chat/backend/persistence/inmemory/ShardedChatHomeTest.java
+++ b/src/test/java/de/juplo/kafka/chat/backend/persistence/inmemory/ShardedChatHomeTest.java
@@ -11,7 +11,6 @@ import java.nio.file.Paths;
 import java.time.Clock;
 import java.util.stream.IntStream;
 
-
 public class ShardedChatHomeTest extends ChatHomeWithShardsTestBase
 {
   @TestConfiguration
@@ -19,55 +18,44 @@ public class ShardedChatHomeTest extends ChatHomeWithShardsTestBase
   {
     @Bean
     ShardedChatHome chatHome(
-        Integer numShards,
-        int[] ownedShards,
         InMemoryChatHomeService chatHomeService)
     {
-      SimpleChatHome[] chatHomes = new SimpleChatHome[numShards];
+      SimpleChatHome[] chatHomes = new SimpleChatHome[NUM_SHARDS];
 
       IntStream
-          .of(ownedShards)
+          .of(ownedShards())
           .forEach(shard -> chatHomes[shard] = new SimpleChatHome(chatHomeService, shard));
 
-      ShardingStrategy strategy = new KafkaLikeShardingStrategy(numShards);
+      ShardingStrategy strategy = new KafkaLikeShardingStrategy(NUM_SHARDS);
 
       return new ShardedChatHome(chatHomes, strategy);
     }
 
     @Bean
     InMemoryChatHomeService chatHomeService(
-        Integer numShards,
-        int[] ownedShards,
         StorageStrategy storageStrategy)
     {
       return new InMemoryChatHomeService(
-          numShards,
-          ownedShards,
+          NUM_SHARDS,
+          ownedShards(),
           storageStrategy.read());
     }
 
     @Bean
-    public FilesStorageStrategy storageStrategy(Integer numShards)
+    public FilesStorageStrategy storageStrategy()
     {
       return new FilesStorageStrategy(
           Paths.get("target", "test-classes", "data", "files"),
           Clock.systemDefaultZone(),
           8,
-          new KafkaLikeShardingStrategy(numShards),
+          new KafkaLikeShardingStrategy(NUM_SHARDS),
           messageFlux -> new InMemoryChatRoomService(messageFlux),
           new ObjectMapper());
     }
 
-    @Bean
-    Integer numShards()
-    {
-      return 10;
-    }
-
-    @Bean
     int[] ownedShards()
     {
-      return new int[] { 2 };
+      return new int[] { OWNED_SHARD };
     }
   }
 }