fix flaky tests

Signed-off-by: Nikolay Ulmasov <[email protected]>

crates/deltalake-core/tests/command_restore.rs

@@ -116,12 +116,25 @@ async fn test_restore_by_version() -> Result<(), Box<dyn Error>> {
 
 #[tokio::test]
 async fn test_restore_by_datetime() -> Result<(), Box<dyn Error>> {
+    // use chrono::{Duration};
+    use std::time::{SystemTime, UNIX_EPOCH};
+
     let context = setup_test().await?;
     let mut table = context.table;
     let history = table.history(Some(10)).await?;
     let timestamp = history.get(1).unwrap().timestamp.unwrap();
     let naive = NaiveDateTime::from_timestamp_millis(timestamp).unwrap();
-    let datetime: DateTime<Utc> = Utc.from_utc_datetime(&naive);
+    let datetime: DateTime<Utc> = Utc.from_utc_datetime(&naive); // - Duration::hours(1);
+
+    let now = SystemTime::now()
+        .duration_since(UNIX_EPOCH)
+        .unwrap()
+        .as_millis();
+    if now as i64 <= datetime.timestamp_millis() {
+        println!("--> Now: {} < history[1].timestamp: {}", now, datetime.timestamp_millis());
+    }
+    assert!(now as i64 > datetime.timestamp_millis());
+
 
     let result = DeltaOps(table)
         .restore()

python/tests/test_writer.py

@@ -3,6 +3,7 @@
 import os
 import pathlib
 import random
+import shutil
 import threading
 from datetime import date, datetime
 from math import inf
@@ -871,6 +872,11 @@ def test_replace_where_overwrite(
     filter_string: str,
 ):
     table_path = tmp_path / sub_path
+
+    # Sometimes the path is already there resulting in failed test
+    if table_path.exists():
+        shutil.rmtree(table_path)
+
     sample_data = pa.table(
         {
             "p1": pa.array(["1", "1", "2", "2"], pa.string()),