do not delete snapshot before the consistency check

Author: agourlay

src/process.rs

@@ -115,9 +115,9 @@ impl ProcessManager {
         let source_storage_dir = PathBuf::from(&self.working_dir).join("storage");
 
         if backup_exists {
-            fs::remove_dir_all(backup_dir_path)
-                .await
-                .with_context(|| format!("failed to remove backup storage dir {backup_dir_path:?}"))?;
+            fs::remove_dir_all(backup_dir_path).await.with_context(|| {
+                format!("failed to remove backup storage dir {backup_dir_path:?}")
+            })?;
         }
 
         util::copy_dir(&source_storage_dir, backup_dir_path).await?;

src/workload.rs

@@ -332,7 +332,6 @@ impl Workload {
                 restore_collection_snapshot(&self.collection_name, snapshot_name, http_client)
                     .await?;
 
-                delete_collection_snapshot(client, &self.collection_name, snapshot_name).await?;
                 let restored_count = get_exact_points_count(client, &self.collection_name).await?;
 
                 if restored_count == 0 {
@@ -343,9 +342,14 @@ impl Workload {
                 log::info!(
                     "Run: data consistency check over {restored_count} points restored for snapshot '{snapshot_name}'"
                 );
-                self.data_consistency_check(client, restored_count, "snapshot-restore-check")
-                    .await?;
-
+                self.data_consistency_check(
+                    client,
+                    restored_count,
+                    &format!("snapshot-restore-check-{snapshot_name}"),
+                )
+                .await?;
+                // cleanup
+                delete_collection_snapshot(client, &self.collection_name, snapshot_name).await?;
                 delete_points(client, &self.collection_name).await?;
                 self.reset_max_confirmed_point_id();
             }