Fixed GatewayAllocatorTests

Signed-off-by: Gaurav Chandani <[email protected]>
Gaurav614 · Apr 12, 2024 · e8d5600 · e8d5600
1 parent baca309
commit e8d5600
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/server/src/test/java/org/opensearch/gateway/GatewayAllocatorTests.java b/server/src/test/java/org/opensearch/gateway/GatewayAllocatorTests.java
@@ -122,15 +122,15 @@ public void testCorrectnessOfBatch() {
             .values()
             .stream()
             .map(ShardsBatchGatewayAllocator.ShardsBatch::getBatchedShardRoutings)
-            .flatMap(Set::stream)
+            .flatMap(List::stream)
             .collect(Collectors.toSet());
         primariesInAllBatches.forEach(shardRouting -> assertTrue(shardRouting.unassigned() && shardRouting.primary() == true));
 
         Set<ShardRouting> replicasInAllBatches = testShardsBatchGatewayAllocator.getBatchIdToStoreShardBatch()
             .values()
             .stream()
             .map(ShardsBatchGatewayAllocator.ShardsBatch::getBatchedShardRoutings)
-            .flatMap(Set::stream)
+            .flatMap(List::stream)
             .collect(Collectors.toSet());
 
         replicasInAllBatches.forEach(shardRouting -> assertTrue(shardRouting.unassigned() && shardRouting.primary() == false));