[Segment Replication] Fix flaky testReplicaHasDiffFilesThanPrimary te…

…st (#6979) * [Segment Replication] Fix flaky testReplicaHasDiffFilesThanPrimary test Signed-off-by: Suraj Singh <[email protected]> * Use existing doc assertion methods Signed-off-by: Suraj Singh <[email protected]> --------- Signed-off-by: Suraj Singh <[email protected]>
opensearch-project · Apr 4, 2023 · 4511354 · 4511354
1 parent 5094541
commit 4511354
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/...src/internalClusterTest/java/org/opensearch/indices/replication/SegmentReplicationIT.java b/...src/internalClusterTest/java/org/opensearch/indices/replication/SegmentReplicationIT.java
@@ -737,14 +737,16 @@ 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++) {
             client().prepareIndex(INDEX_NAME).setId(Integer.toString(i)).setSource("field", "value" + i).execute().get();
             refresh(INDEX_NAME);
         }
         // Refresh, this should trigger round of segment replication
-        assertBusy(() -> { assertDocCounts(docCount, replicaNode); });
+        waitForSearchableDocs(docCount, primaryNode, replicaNode);
+        verifyStoreContent();
         final IndexShard replicaAfterFailure = getIndexShard(replicaNode, INDEX_NAME);
         assertNotEquals(replicaAfterFailure.routingEntry().allocationId().getId(), replicaShard.routingEntry().allocationId().getId());
     }