From 47cf8e2c90a7bbb2611d104950a36f90f6812482 Mon Sep 17 00:00:00 2001
From: Otto Westerlund <westerlundotto@gmail.com>
Date: Tue, 20 Jan 2026 18:10:31 +0200
Subject: [PATCH] flaky test wait for lock cleanup

Signed-off-by: Otto Westerlund <westerlundotto@gmail.com>
---
 tests/failpoint_tests.rs | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tests/failpoint_tests.rs b/tests/failpoint_tests.rs
index 20379bcc..b3d23707 100644
--- a/tests/failpoint_tests.rs
+++ b/tests/failpoint_tests.rs
@@ -337,7 +337,8 @@ async fn txn_cleanup_2pc_locks() -> Result<()> {
         )
         .await?;
         let keys = write_data(&client, false, false).await?;
-        assert_eq!(count_locks(&client).await?, 0);
+        let remaining = wait_for_locks_count(&client, 0).await?;
+        assert_eq!(remaining, 0);
 
         let safepoint = client.current_timestamp().await?;
         let options = ResolveLocksOptions {