Modifying IT to restore snapshot

Signed-off-by: Shubh Sahu <[email protected]>
opensearch-project · Jan 8, 2025 · 4bd2acb · 4bd2acb
1 parent 3947531
commit 4bd2acb
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/server/src/internalClusterTest/java/org/opensearch/remotestore/RemoteStoreIT.java b/server/src/internalClusterTest/java/org/opensearch/remotestore/RemoteStoreIT.java
@@ -1084,7 +1084,7 @@ public void testCloseIndexWithNoOpSyncAndFlushForAsyncTranslog() throws Interrup
 
     public void testSuccessfulShallowV1SnapshotPostIndexClose() throws Exception {
         internalCluster().startClusterManagerOnlyNode();
-        internalCluster().startDataOnlyNodes(1);
+        String dataNode = internalCluster().startDataOnlyNodes(1).get(0);
         createIndex(INDEX_NAME, remoteStoreIndexSettings(0, 10000L, -1));
         ensureGreen(INDEX_NAME);
 
@@ -1143,5 +1143,17 @@ public void testSuccessfulShallowV1SnapshotPostIndexClose() throws Exception {
         assertEquals(SnapshotState.SUCCESS, snapshotInfo2.state());
         assertTrue(snapshotInfo2.successfulShards() > 0);
         assertEquals(0, snapshotInfo2.failedShards());
+
+        // delete the index
+        cluster().wipeIndices(INDEX_NAME);
+        // try restoring the snapshot
+        RestoreSnapshotResponse restoreSnapshotResponse = clusterAdmin().prepareRestoreSnapshot(shallowSnapshotRepoName, snapshot2)
+            .setWaitForCompletion(true)
+            .execute()
+            .actionGet();
+        assertThat(restoreSnapshotResponse.getRestoreInfo().totalShards(), greaterThan(0));
+        ensureGreen(INDEX_NAME);
+        flushAndRefresh(INDEX_NAME);
+        assertBusy(() -> { assertHitCount(client(dataNode).prepareSearch(INDEX_NAME).setSize(0).get(), 20); });
     }
 }