Fix flaky test_cleanup_intermediate_files (apache#5645)

tustvold · Apr 15, 2024 · 0124307 · 0124307
1 parent 8731760
commit 0124307
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/object_store/src/local.rs b/object_store/src/local.rs
@@ -1450,10 +1450,14 @@ mod not_wasm_tests {
         assert_eq!(file_count, 1);
         drop(upload);
 
-        tokio::time::sleep(Duration::from_millis(1)).await;
-
-        let file_count = std::fs::read_dir(root.path()).unwrap().count();
-        assert_eq!(file_count, 0);
+        for _ in 0..100 {
+            tokio::time::sleep(Duration::from_millis(1)).await;
+            let file_count = std::fs::read_dir(root.path()).unwrap().count();
+            if file_count == 0 {
+                return;
+            }
+        }
+        panic!("Failed to cleanup file in 100ms")
     }
 }