fix test

Signed-off-by: Shivansh Arora <[email protected]>
opensearch-project · Mar 18, 2024 · b252e71 · b252e71
1 parent 5639a69
commit b252e71
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/server/src/main/java/org/opensearch/gateway/ReplicaShardBatchAllocator.java b/server/src/main/java/org/opensearch/gateway/ReplicaShardBatchAllocator.java
@@ -181,9 +181,8 @@ private Map<DiscoveryNode, StoreFilesMetadata> convertToNodeStoreFilesMetadataMa
         ShardRouting unassignedShard,
         FetchResult<NodeStoreFilesMetadataBatch> data
     ) {
-
         if (!data.hasData()) {
-            return new HashMap<>();
+            return null;
         }
 
         Map<DiscoveryNode, StoreFilesMetadata> map = new HashMap<>();

diff --git a/server/src/test/java/org/opensearch/gateway/ReplicaShardBatchAllocatorTests.java b/server/src/test/java/org/opensearch/gateway/ReplicaShardBatchAllocatorTests.java
@@ -834,7 +834,11 @@ protected AsyncShardFetch.FetchResult<NodeStoreFilesMetadataBatch> fetchData(
                     );
                 }
             }
-            return new AsyncShardFetch.FetchResult<>(tData, Collections.<ShardId, Set<String>>emptyMap());
+            return new AsyncShardFetch.FetchResult<>(tData, new HashMap<>() {
+                {
+                    put(shardId, Collections.emptySet());
+                }
+            });
         }
 
         @Override