[Segment Replication] Fix flaky testReplicaHasDiffFilesThanPrimary test

Signed-off-by: Suraj Singh <[email protected]>
opensearch-project · Apr 4, 2023 · 5ab7285 · 5ab7285
1 parent 4859e15
commit 5ab7285
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
@@ -737,6 +737,7 @@ public void testReplicaHasDiffFilesThanPrimary() throws Exception {
 
         final SegmentInfos segmentInfos = SegmentInfos.readLatestCommit(replicaShard.store().directory());
         replicaShard.finalizeReplication(segmentInfos);
+        ensureYellow(INDEX_NAME);
 
         final int docCount = scaledRandomIntBetween(10, 200);
         for (int i = 0; i < docCount; i++) {