Fix flaky test. (opensearch-project#8441)

Signed-off-by: Rishikesh1159 <rishireddy1159@gmail.com>
raghuvanshraj · Jul 12, 2023 · 516b1d4 · 516b1d4
1 parent f5da98d
commit 516b1d4
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/...src/internalClusterTest/java/org/opensearch/indices/replication/SegmentReplicationIT.java b/...src/internalClusterTest/java/org/opensearch/indices/replication/SegmentReplicationIT.java
@@ -435,6 +435,7 @@ public void testNodeDropWithOngoingReplication() throws Exception {
         refresh(INDEX_NAME);
         blockFileCopy.countDown();
         internalCluster().stopRandomNode(InternalTestCluster.nameFilter(primaryNode));
+        ensureYellow(INDEX_NAME);
         assertBusy(() -> { assertDocCounts(docCount, replicaNode); });
         state = client().admin().cluster().prepareState().execute().actionGet().getState();
         // replica now promoted as primary should have same allocation id