Add more cases in integ test

Signed-off-by: Anshu Agarwal <anshukag@amazon.com>
opensearch-project · Feb 7, 2023 · 9dd3632 · 9dd3632
1 parent 3e3053a
commit 9dd3632
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/server/src/internalClusterTest/java/org/opensearch/cluster/routing/WeightedRoutingIT.java b/server/src/internalClusterTest/java/org/opensearch/cluster/routing/WeightedRoutingIT.java
@@ -662,5 +662,20 @@ public void testClusterHealthResponseWithEnsureNodeWeighedInParam() throws Excep
         assertFalse(nodeLocalHealth.hasDiscoveredClusterManager());
 
         networkDisruption.stopDisrupting();
+        Thread.sleep(1000);
+
+        // delete weights
+        ClusterDeleteWeightedRoutingResponse deleteResponse = client().admin().cluster().prepareDeleteWeightedRouting().setVersion(0).get();
+        assertTrue(deleteResponse.isAcknowledged());
+
+        // Check local cluster health
+        nodeLocalHealth = client(nodes_in_zone_c.get(0)).admin()
+            .cluster()
+            .prepareHealth()
+            .setLocal(true)
+            .setEnsureNodeWeighedIn(true)
+            .get();
+        assertFalse(nodeLocalHealth.isTimedOut());
+        assertTrue(nodeLocalHealth.hasDiscoveredClusterManager());
     }
 }