@powersync/service-module-postgres-storage 0.0.0-dev-20250507154604 → 0.0.0-dev-20250617143654

package/test/src/__snapshots__/storage_sync.test.ts.snap

@@ -309,6 +309,153 @@ exports[`sync - postgres > sync global data 1`] = `
 ]
 `;
 
+exports[`sync - postgres > sync interrupts low-priority buckets on new checkpoints (2) 1`] = `
+[
+  {
+    "checkpoint": {
+      "buckets": [
+        {
+          "bucket": "b0a[]",
+          "checksum": -659831575,
+          "count": 2000,
+          "priority": 2,
+        },
+        {
+          "bucket": "b0b[]",
+          "checksum": -659831575,
+          "count": 2000,
+          "priority": 2,
+        },
+        {
+          "bucket": "b1[]",
+          "checksum": -1096116670,
+          "count": 1,
+          "priority": 1,
+        },
+      ],
+      "last_op_id": "4001",
+      "write_checkpoint": undefined,
+    },
+  },
+  {
+    "data": {
+      "after": "0",
+      "bucket": "b1[]",
+      "data": undefined,
+      "has_more": false,
+      "next_after": "1",
+    },
+  },
+  {
+    "partial_checkpoint_complete": {
+      "last_op_id": "4001",
+      "priority": 1,
+    },
+  },
+  {
+    "data": {
+      "after": "0",
+      "bucket": "b0a[]",
+      "data": undefined,
+      "has_more": true,
+      "next_after": "2000",
+    },
+  },
+  {
+    "data": {
+      "after": "2000",
+      "bucket": "b0a[]",
+      "data": undefined,
+      "has_more": true,
+      "next_after": "4000",
+    },
+  },
+  {
+    "checkpoint_diff": {
+      "last_op_id": "4004",
+      "removed_buckets": [],
+      "updated_buckets": [
+        {
+          "bucket": "b0a[]",
+          "checksum": 883076828,
+          "count": 2001,
+          "priority": 2,
+        },
+        {
+          "bucket": "b0b[]",
+          "checksum": 883076828,
+          "count": 2001,
+          "priority": 2,
+        },
+        {
+          "bucket": "b1[]",
+          "checksum": 1841937527,
+          "count": 2,
+          "priority": 1,
+        },
+      ],
+      "write_checkpoint": undefined,
+    },
+  },
+  {
+    "data": {
+      "after": "1",
+      "bucket": "b1[]",
+      "data": undefined,
+      "has_more": false,
+      "next_after": "4002",
+    },
+  },
+  {
+    "partial_checkpoint_complete": {
+      "last_op_id": "4004",
+      "priority": 1,
+    },
+  },
+  {
+    "data": {
+      "after": "4000",
+      "bucket": "b0a[]",
+      "data": undefined,
+      "has_more": false,
+      "next_after": "4003",
+    },
+  },
+  {
+    "data": {
+      "after": "0",
+      "bucket": "b0b[]",
+      "data": undefined,
+      "has_more": true,
+      "next_after": "1999",
+    },
+  },
+  {
+    "data": {
+      "after": "1999",
+      "bucket": "b0b[]",
+      "data": undefined,
+      "has_more": true,
+      "next_after": "3999",
+    },
+  },
+  {
+    "data": {
+      "after": "3999",
+      "bucket": "b0b[]",
+      "data": undefined,
+      "has_more": false,
+      "next_after": "4004",
+    },
+  },
+  {
+    "checkpoint_complete": {
+      "last_op_id": "4004",
+    },
+  },
+]
+`;
+
 exports[`sync - postgres > sync legacy non-raw data 1`] = `
 [
   {

package/test/src/migrations.test.ts

@@ -1,15 +1,23 @@
-import { describe, expect, it } from 'vitest';
+import { beforeEach, describe, expect, it } from 'vitest';
 
+import { Direction } from '@powersync/lib-services-framework';
 import { register } from '@powersync/service-core-tests';
 import { PostgresMigrationAgent } from '../../src/migrations/PostgresMigrationAgent.js';
 import { env } from './env.js';
-import { POSTGRES_STORAGE_FACTORY } from './util.js';
+import { POSTGRES_STORAGE_FACTORY, POSTGRES_STORAGE_SETUP } from './util.js';
 
 const MIGRATION_AGENT_FACTORY = () => {
   return new PostgresMigrationAgent({ type: 'postgresql', uri: env.PG_STORAGE_TEST_URL, sslmode: 'disable' });
 };
 
 describe('Migrations', () => {
+  beforeEach(async () => {
+    // The migration tests clear the migration store, without running the down migrations.
+    // This ensures all the down migrations have been run before.
+    const setup = POSTGRES_STORAGE_SETUP;
+    await setup.migrate(Direction.Down);
+  });
+
   register.registerMigrationTests(MIGRATION_AGENT_FACTORY);
 
   it('Should have tables declared', async () => {

package/test/src/util.ts

@@ -2,7 +2,10 @@ import path from 'path';
 import { fileURLToPath } from 'url';
 import { normalizePostgresStorageConfig } from '../../src//types/types.js';
 import { PostgresMigrationAgent } from '../../src/migrations/PostgresMigrationAgent.js';
-import { PostgresTestStorageFactoryGenerator } from '../../src/storage/PostgresTestStorageFactoryGenerator.js';
+import {
+  postgresTestSetup,
+  PostgresTestStorageFactoryGenerator
+} from '../../src/storage/PostgresTestStorageFactoryGenerator.js';
 import { env } from './env.js';
 
 const __filename = fileURLToPath(import.meta.url);
@@ -28,7 +31,9 @@ class TestPostgresMigrationAgent extends PostgresMigrationAgent {
   }
 }
 
-export const POSTGRES_STORAGE_FACTORY = PostgresTestStorageFactoryGenerator({
+export const POSTGRES_STORAGE_SETUP = postgresTestSetup({
   url: env.PG_STORAGE_TEST_URL,
   migrationAgent: (config) => new TestPostgresMigrationAgent(config)
 });
+
+export const POSTGRES_STORAGE_FACTORY = POSTGRES_STORAGE_SETUP.factory;