Minor fix

Signed-off-by: Rishab Nahata <rnnahata@amazon.com>
imRishN · Jul 9, 2024 · 50d7235 · 50d7235
1 parent 4cc0676
commit 50d7235
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/server/src/test/java/org/opensearch/gateway/ReplicaShardBatchAllocatorTests.java b/server/src/test/java/org/opensearch/gateway/ReplicaShardBatchAllocatorTests.java
@@ -53,7 +53,6 @@
 import org.opensearch.snapshots.SnapshotShardSizeInfo;
 import org.junit.Before;
 
-import java.sql.Array;
 import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.Collection;
@@ -277,8 +276,8 @@ public void testCancelRecoveryIfFoundCopyWithNoopRetentionLease() {
         );
         Collection<ShardRouting> replicaShards = allocation.routingNodes().shardsWithState(ShardRoutingState.UNASSIGNED);
         List<ShardRouting> shardRoutingBatch = new ArrayList<>(replicaShards);
-        List<Set<ShardRouting>> shardBatchList = Collections.singletonList(
-            new HashSet<>(allocation.routingNodes().shardsWithState(ShardRoutingState.INITIALIZING))
+        List<List<ShardRouting>> shardBatchList = Collections.singletonList(
+            new ArrayList<>(allocation.routingNodes().shardsWithState(ShardRoutingState.INITIALIZING))
         );
 
         testBatchAllocator.processExistingRecoveries(allocation, shardBatchList);
@@ -530,7 +529,7 @@ public void testCancelRecoveryBetterSyncId() {
             .addData(node3, "MATCH", null, new StoreFileMetadata("file1", 10, "MATCH_CHECKSUM", MIN_SUPPORTED_LUCENE_VERSION));
         testBatchAllocator.processExistingRecoveries(
             allocation,
-            Collections.singletonList(new HashSet<>(allocation.routingNodes().shardsWithState(ShardRoutingState.INITIALIZING)))
+            Collections.singletonList(new ArrayList<>(allocation.routingNodes().shardsWithState(ShardRoutingState.INITIALIZING)))
         );
         assertThat(allocation.routingNodesChanged(), equalTo(true));
         assertThat(allocation.routingNodes().shardsWithState(ShardRoutingState.UNASSIGNED).size(), equalTo(1));