Update tests to use PushFramesResult

Author: avinassh

libsql/src/sync.rs

@@ -98,6 +98,7 @@ struct InfoResult {
     current_generation: u32,
 }
 
+#[derive(Debug)]
 struct PushFramesResult {
     max_frame_no: u32,
     baton: Option<String>,

libsql/src/sync/test.rs

@@ -28,9 +28,9 @@ async fn test_sync_context_push_frame() {
     let mut sync_ctx = sync_ctx;
 
     // Push a frame and verify the response
-    let durable_frame = sync_ctx.push_frames(frame, 1, 0, 1).await.unwrap();
+    let durable_frame = sync_ctx.push_frames(frame, 1, 0, 1, None).await.unwrap();
     sync_ctx.write_metadata().await.unwrap();
-    assert_eq!(durable_frame, 0); // First frame should return max_frame_no = 0
+    assert_eq!(durable_frame.max_frame_no, 0); // First frame should return max_frame_no = 0
 
     // Verify internal state was updated
     assert_eq!(sync_ctx.durable_frame_num(), 0);
@@ -56,9 +56,9 @@ async fn test_sync_context_with_auth() {
     let frame = Bytes::from("test frame with auth");
     let mut sync_ctx = sync_ctx;
 
-    let durable_frame = sync_ctx.push_frames(frame, 1, 0, 1).await.unwrap();
+    let durable_frame = sync_ctx.push_frames(frame, 1, 0, 1, None).await.unwrap();
     sync_ctx.write_metadata().await.unwrap();
-    assert_eq!(durable_frame, 0);
+    assert_eq!(durable_frame.max_frame_no, 0);
     assert_eq!(server.frame_count(), 1);
 }
 
@@ -82,9 +82,9 @@ async fn test_sync_context_multiple_frames() {
     // Push multiple frames and verify incrementing frame numbers
     for i in 0..3 {
         let frame = Bytes::from(format!("frame data {}", i));
-        let durable_frame = sync_ctx.push_frames(frame, 1, i, 1).await.unwrap();
+        let durable_frame = sync_ctx.push_frames(frame, 1, i, 1, None).await.unwrap();
         sync_ctx.write_metadata().await.unwrap();
-        assert_eq!(durable_frame, i);
+        assert_eq!(durable_frame.max_frame_no, i);
         assert_eq!(sync_ctx.durable_frame_num(), i);
         assert_eq!(server.frame_count(), i + 1);
     }
@@ -108,9 +108,9 @@ async fn test_sync_context_corrupted_metadata() {
 
     let mut sync_ctx = sync_ctx;
     let frame = Bytes::from("test frame data");
-    let durable_frame = sync_ctx.push_frames(frame, 1, 0, 1).await.unwrap();
+    let durable_frame = sync_ctx.push_frames(frame, 1, 0, 1, None).await.unwrap();
     sync_ctx.write_metadata().await.unwrap();
-    assert_eq!(durable_frame, 0);
+    assert_eq!(durable_frame.max_frame_no, 0);
     assert_eq!(server.frame_count(), 1);
 
     // Update metadata path to use -info instead of .meta
@@ -152,9 +152,12 @@ async fn test_sync_restarts_with_lower_max_frame_no() {
 
     let mut sync_ctx = sync_ctx;
     let frame = Bytes::from("test frame data");
-    let durable_frame = sync_ctx.push_frames(frame.clone(), 1, 0, 1).await.unwrap();
+    let durable_frame = sync_ctx
+        .push_frames(frame.clone(), 1, 0, 1, None)
+        .await
+        .unwrap();
     sync_ctx.write_metadata().await.unwrap();
-    assert_eq!(durable_frame, 0);
+    assert_eq!(durable_frame.max_frame_no, 0);
     assert_eq!(server.frame_count(), 1);
 
     // Bump the durable frame num so that the next time we call the
@@ -180,14 +183,17 @@ async fn test_sync_restarts_with_lower_max_frame_no() {
     // This push should fail because we are ahead of the server and thus should get an invalid
     // frame no error.
     sync_ctx
-        .push_frames(frame.clone(), 1, frame_no, 1)
+        .push_frames(frame.clone(), 1, frame_no, 1, None)
         .await
         .unwrap_err();
 
     let frame_no = sync_ctx.durable_frame_num() + 1;
     // This then should work because when the last one failed it updated our state of the server
     // durable_frame_num and we should then start writing from there.
-    sync_ctx.push_frames(frame, 1, frame_no, 1).await.unwrap();
+    sync_ctx
+        .push_frames(frame, 1, frame_no, 1, None)
+        .await
+        .unwrap();
 }
 
 #[tokio::test]
@@ -215,7 +221,7 @@ async fn test_sync_context_retry_on_error() {
     server.return_error.store(true, Ordering::SeqCst);
 
     // First attempt should fail but retry
-    let result = sync_ctx.push_frames(frame.clone(), 1, 0, 1).await;
+    let result = sync_ctx.push_frames(frame.clone(), 1, 0, 1, None).await;
     assert!(result.is_err());
 
     // Advance time to trigger retries faster
@@ -228,9 +234,9 @@ async fn test_sync_context_retry_on_error() {
     server.return_error.store(false, Ordering::SeqCst);
 
     // Next attempt should succeed
-    let durable_frame = sync_ctx.push_frames(frame, 1, 0, 1).await.unwrap();
+    let durable_frame = sync_ctx.push_frames(frame, 1, 0, 1, None).await.unwrap();
     sync_ctx.write_metadata().await.unwrap();
-    assert_eq!(durable_frame, 0);
+    assert_eq!(durable_frame.max_frame_no, 0);
     assert_eq!(server.frame_count(), 1);
 }