Fix test coverage

Signed-off-by: Anshu Agarwal <[email protected]>
opensearch-project · May 27, 2023 · 628a7fe · 628a7fe
1 parent 29c12ce
commit 628a7fe
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/server/src/test/java/org/opensearch/cluster/routing/FailAwareWeightedRoutingTests.java b/server/src/test/java/org/opensearch/cluster/routing/FailAwareWeightedRoutingTests.java
@@ -434,6 +434,12 @@ public void testUpdateFailOpenStatsForOneHealthyCopy() {
         boolean update = FailAwareWeightedRouting.getInstance()
             .updateFailOpenStatsForOneHealthyCopy(searchShardIterator, clusterState, shardRoutingA.currentNodeId());
         assertTrue(update);
+
+        ShardsIterator shardIterator = new PlainShardIterator(shardId, shardRoutings);
+
+        update = FailAwareWeightedRouting.getInstance()
+            .updateFailOpenStatsForOneHealthyCopy(shardIterator, clusterState, shardRoutingA.currentNodeId());
+        assertTrue(update);
     }
 
     public void testUpdateFailOpenStatsForWithTwoHealthyCopies() {
@@ -483,6 +489,12 @@ public void testUpdateFailOpenStatsForWithTwoHealthyCopies() {
         boolean update = FailAwareWeightedRouting.getInstance()
             .updateFailOpenStatsForOneHealthyCopy(searchShardIterator, clusterState, shardRoutingA.currentNodeId());
         assertFalse(update);
+
+        ShardsIterator shardIterator = new PlainShardIterator(shardId, shardRoutings);
+
+        update = FailAwareWeightedRouting.getInstance()
+            .updateFailOpenStatsForOneHealthyCopy(shardIterator, clusterState, shardRoutingA.currentNodeId());
+        assertFalse(update);
     }
 
     public void testUpdateFailOpenStatsForOneHealthyCopyWithIgnoreWeightedRouting() {
@@ -532,6 +544,12 @@ public void testUpdateFailOpenStatsForOneHealthyCopyWithIgnoreWeightedRouting()
         boolean update = FailAwareWeightedRouting.getInstance()
             .updateFailOpenStatsForOneHealthyCopy(searchShardIterator, clusterState, shardRoutingA.currentNodeId());
         assertFalse(update);
+
+        ShardsIterator shardIterator = new PlainShardIterator(shardId, shardRoutings);
+
+        update = FailAwareWeightedRouting.getInstance()
+            .updateFailOpenStatsForOneHealthyCopy(shardIterator, clusterState, shardRoutingA.currentNodeId());
+        assertFalse(update);
     }
 
 }