@powersync/service-core-tests 0.0.0-dev-20250611110033 → 0.0.0-dev-20250618131818

package/src/tests/register-data-storage-tests.ts

@@ -1769,14 +1769,12 @@ bucket_definitions:
       .watchCheckpointChanges({ user_id: 'user1', signal: abortController.signal })
       [Symbol.asyncIterator]();
 
-    await bucketStorage.batchCreateCustomWriteCheckpoints([
-      {
+    await bucketStorage.startBatch(test_utils.BATCH_OPTIONS, async (batch) => {
+      await batch.addCustomWriteCheckpoint({
         checkpoint: 5n,
         user_id: 'user1'
-      }
-    ]);
-
-    await bucketStorage.startBatch(test_utils.BATCH_OPTIONS, async (batch) => {
+      });
+      await batch.flush();
       await batch.keepalive('5/0');
     });
 
@@ -1785,7 +1783,6 @@ bucket_definitions:
       done: false,
       value: {
         base: {
-          checkpoint: 0n,
           lsn: '5/0'
         },
         writeCheckpoint: 5n
@@ -1822,23 +1819,18 @@ bucket_definitions:
       done: false,
       value: {
         base: {
-          checkpoint: 0n,
           lsn: '5/0'
         },
         writeCheckpoint: null
       }
     });
 
-    await bucketStorage.batchCreateCustomWriteCheckpoints([
-      {
+    await bucketStorage.startBatch(test_utils.BATCH_OPTIONS, async (batch) => {
+      batch.addCustomWriteCheckpoint({
         checkpoint: 6n,
         user_id: 'user1'
-      }
-    ]);
-    // We have to trigger a new keepalive after the checkpoint, at least to cover postgres storage.
-    // This is what is effetively triggered with RouteAPI.createReplicationHead().
-    // MongoDB storage doesn't explicitly need this anymore.
-    await bucketStorage.startBatch(test_utils.BATCH_OPTIONS, async (batch) => {
+      });
+      await batch.flush();
       await batch.keepalive('6/0');
     });
 
@@ -1847,7 +1839,6 @@ bucket_definitions:
       done: false,
       value: {
         base: {
-          checkpoint: 0n
           // can be 5/0 or 6/0 - actual value not relevant for custom write checkpoints
           // lsn: '6/0'
         },
@@ -1855,13 +1846,12 @@ bucket_definitions:
       }
     });
 
-    await bucketStorage.batchCreateCustomWriteCheckpoints([
-      {
+    await bucketStorage.startBatch(test_utils.BATCH_OPTIONS, async (batch) => {
+      batch.addCustomWriteCheckpoint({
         checkpoint: 7n,
         user_id: 'user1'
-      }
-    ]);
-    await bucketStorage.startBatch(test_utils.BATCH_OPTIONS, async (batch) => {
+      });
+      await batch.flush();
       await batch.keepalive('7/0');
     });
 
@@ -1870,7 +1860,6 @@ bucket_definitions:
       done: false,
       value: {
         base: {
-          checkpoint: 0n
           // can be 5/0, 6/0 or 7/0 - actual value not relevant for custom write checkpoints
           // lsn: '7/0'
         },

package/src/tests/register-sync-tests.ts

@@ -519,12 +519,12 @@ bucket_definitions:
 
               await batch.commit('0/2');
             });
+          }
 
-            if (sentRows >= 1000 && sentRows <= 2001) {
-              // pause for a bit to give the stream time to process interruptions.
-              // This covers the data batch above and the next one.
-              await timers.setTimeout(50);
-            }
+          if (sentRows >= 1000 && sentRows <= 2001) {
+            // pause for a bit to give the stream time to process interruptions.
+            // This covers the data batch above and the next one.
+            await timers.setTimeout(50);
           }
         }
         if ('checkpoint_complete' in next) {