Fix testIndexDeletionDuringSnapshotCreationInQueue flaky test (opense…

…arch-project#11726) * Fix testIndexDeletionDuringSnapshotCreationInQueue flaky test Signed-off-by: Suraj Singh <[email protected]> * Update comment Signed-off-by: Suraj Singh <[email protected]> --------- Signed-off-by: Suraj Singh <[email protected]> Signed-off-by: Shivansh Arora <[email protected]>
Gaurav614 · Apr 25, 2024 · dbd2cc6 · dbd2cc6
1 parent 57fed55
commit dbd2cc6
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/.../internalClusterTest/java/org/opensearch/snapshots/DedicatedClusterSnapshotRestoreIT.java b/.../internalClusterTest/java/org/opensearch/snapshots/DedicatedClusterSnapshotRestoreIT.java
@@ -1457,6 +1457,13 @@ public void testIndexDeletionDuringSnapshotCreationInQueue() throws Exception {
 
         clusterAdmin().prepareRestoreSnapshot("test-repo", "test-snap").get();
         ensureGreen("test-idx");
+
+        // Wait for snapshot process to complete to prevent conflict with repository clean up
+        assertBusy(() -> {
+            SnapshotInfo snapshotInfo = getSnapshot("test-repo", "test-snap-2");
+            assertTrue(snapshotInfo.state().completed());
+            assertEquals(SnapshotState.PARTIAL, snapshotInfo.state());
+        }, 1, TimeUnit.MINUTES);
     }
 
     private long calculateTotalFilesSize(List<Path> files) {