pytests: fix flaky checkpoint sync test

abhizer · abhizer · commit 6e8fbd8077c2 · 2026-01-30T10:19:18.000Z
In sync tests, wait until the most recently checkpoint, the one with all
the data is synced.
This didn't show up previously, as all the data was inserted and
executed within the checkpoint interval of 5s, and this checkpoint was
synced within the interval of 10s.

Signed-off-by: Abhinav Gyawali &lt;22275402+abhizer@users.noreply.github.com&gt;
diff --git a/python/tests/platform/test_checkpoint_sync.py b/python/tests/platform/test_checkpoint_sync.py
@@ -3,7 +3,7 @@
 import sys
 import time
 from typing import Optional
-from uuid import uuid4
+from uuid import uuid4, UUID
 
 from feldera.enums import FaultToleranceModel, PipelineStatus
 from feldera.runtime_config import RuntimeConfig, Storage
@@ -144,13 +144,22 @@ def test_checkpoint_sync(
                 time.sleep(0.5)
 
         print("Checkpoint UUID:", chk_uuid, file=sys.stderr)
+        time.sleep(1)
 
         if automated_sync_interval is not None:
             timeout = time.monotonic() + 30
             success = None
             while time.monotonic() < timeout and success is None:
                 try:
-                    success = self.pipeline.last_successful_checkpoint_sync()
+                    synced = self.pipeline.last_successful_checkpoint_sync()
+                    print(
+                        "Automatically synced checkpoint UUID:", synced, file=sys.stderr
+                    )
+                    if synced is not None and chk_uuid is not None:
+                        if synced >= UUID(chk_uuid):
+                            success = synced
+                    else:
+                        success = synced
                 except RuntimeError:
                     time.sleep(0.5)
                     continue
@@ -160,6 +169,9 @@ def test_checkpoint_sync(
                 )
         else:
             uuid = self.pipeline.sync_checkpoint(wait=True)
+            print("Synced Checkpoint UUID:", uuid, file=sys.stderr)
+            if chk_uuid is not None:
+                assert UUID(uuid) >= UUID(chk_uuid)
 
         self.pipeline.stop(force=True)
 
