Fix flaky test. (#8441) (#8454)

(cherry picked from commit d8cc450)

Signed-off-by: Rishikesh1159 <rishireddy1159@gmail.com>
Signed-off-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>

Author: opensearch-trigger-bot[bot]

server/src/internalClusterTest/java/org/opensearch/indices/replication/SegmentReplicationIT.java

@@ -431,6 +431,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