From eb32ea49e67c4413736d1926b40f4683b12ee6a8 Mon Sep 17 00:00:00 2001
From: Nikita Sivukhin <sivukhin@turso.tech>
Date: Mon, 28 Jul 2025 17:06:48 +0400
Subject: [PATCH] fix tests

---
 tests/integration/functions/test_wal_api.rs | 24 +++++++--------------
 1 file changed, 8 insertions(+), 16 deletions(-)

diff --git a/tests/integration/functions/test_wal_api.rs b/tests/integration/functions/test_wal_api.rs
index f945183a2..e096e6729 100644
--- a/tests/integration/functions/test_wal_api.rs
+++ b/tests/integration/functions/test_wal_api.rs
@@ -42,8 +42,7 @@ fn test_wal_frame_transfer_no_schema_changes() {
     let mut frame = [0u8; 24 + 4096];
     conn2.wal_insert_begin().unwrap();
     for frame_id in 1..=conn1.wal_frame_count().unwrap() as u32 {
-        let c = conn1.wal_get_frame(frame_id, &mut frame).unwrap();
-        db1.io.wait_for_completion(c).unwrap();
+        conn1.wal_get_frame(frame_id, &mut frame).unwrap();
         conn2.wal_insert_frame(frame_id, &frame).unwrap();
     }
     conn2.wal_insert_end().unwrap();
@@ -137,8 +136,7 @@ fn test_wal_frame_transfer_schema_changes() {
     let mut frame = [0u8; 24 + 4096];
     conn2.wal_insert_begin().unwrap();
     for frame_id in 1..=conn1.wal_frame_count().unwrap() as u32 {
-        let c = conn1.wal_get_frame(frame_id, &mut frame).unwrap();
-        db1.io.wait_for_completion(c).unwrap();
+        conn1.wal_get_frame(frame_id, &mut frame).unwrap();
         conn2.wal_insert_frame(frame_id, &frame).unwrap();
     }
     conn2.wal_insert_end().unwrap();
@@ -172,8 +170,7 @@ fn test_wal_frame_transfer_no_schema_changes_rollback() {
     let mut frame = [0u8; 24 + 4096];
     conn2.wal_insert_begin().unwrap();
     for frame_id in 1..=(conn1.wal_frame_count().unwrap() as u32 - 1) {
-        let c = conn1.wal_get_frame(frame_id, &mut frame).unwrap();
-        db1.io.wait_for_completion(c).unwrap();
+        conn1.wal_get_frame(frame_id, &mut frame).unwrap();
         conn2.wal_insert_frame(frame_id, &frame).unwrap();
     }
     conn2.wal_insert_end().unwrap();
@@ -208,8 +205,7 @@ fn test_wal_frame_transfer_schema_changes_rollback() {
     let mut frame = [0u8; 24 + 4096];
     conn2.wal_insert_begin().unwrap();
     for frame_id in 1..=(conn1.wal_frame_count().unwrap() as u32 - 1) {
-        let c = conn1.wal_get_frame(frame_id, &mut frame).unwrap();
-        db1.io.wait_for_completion(c).unwrap();
+        conn1.wal_get_frame(frame_id, &mut frame).unwrap();
         conn2.wal_insert_frame(frame_id, &frame).unwrap();
     }
     conn2.wal_insert_end().unwrap();
@@ -243,8 +239,7 @@ fn test_wal_frame_conflict() {
     assert_eq!(conn1.wal_frame_count().unwrap(), 2);
     let mut frame = [0u8; 24 + 4096];
     conn2.wal_insert_begin().unwrap();
-    let c = conn1.wal_get_frame(1, &mut frame).unwrap();
-    db1.io.wait_for_completion(c).unwrap();
+    conn1.wal_get_frame(1, &mut frame).unwrap();
     assert!(conn2.wal_insert_frame(1, &frame).is_err());
 }
 
@@ -267,12 +262,10 @@ fn test_wal_frame_far_away_write() {
     let mut frame = [0u8; 24 + 4096];
     conn2.wal_insert_begin().unwrap();
 
-    let c = conn1.wal_get_frame(3, &mut frame).unwrap();
-    db1.io.wait_for_completion(c).unwrap();
+    conn1.wal_get_frame(3, &mut frame).unwrap();
     conn2.wal_insert_frame(3, &frame).unwrap();
 
-    let c = conn1.wal_get_frame(5, &mut frame).unwrap();
-    db1.io.wait_for_completion(c).unwrap();
+    conn1.wal_get_frame(5, &mut frame).unwrap();
     assert!(conn2.wal_insert_frame(5, &frame).is_err());
 }
 
@@ -311,8 +304,7 @@ fn test_wal_frame_api_no_schema_changes_fuzz() {
                 let mut frame = [0u8; 24 + 4096];
                 conn2.wal_insert_begin().unwrap();
                 for frame_no in (synced_frame + 1)..=next_frame {
-                    let c = conn1.wal_get_frame(frame_no as u32, &mut frame).unwrap();
-                    db1.io.wait_for_completion(c).unwrap();
+                    conn1.wal_get_frame(frame_no as u32, &mut frame).unwrap();
                     conn2.wal_insert_frame(frame_no as u32, &frame[..]).unwrap();
                 }
                 conn2.wal_insert_end().unwrap();