@powersync/service-module-postgres-storage 0.0.0-dev-20260203155513 → 0.0.0-dev-20260223080959

package/src/utils/db.ts

@@ -9,6 +9,9 @@ export const NOTIFICATION_CHANNEL = 'powersynccheckpoints';
  */
 export const sql = lib_postgres.sql;
 
+/**
+ * Drop all Postgres storage tables used by the service, including migrations.
+ */
 export const dropTables = async (client: lib_postgres.DatabaseClient) => {
   // Lock a connection for automatic schema search paths
   await client.lockConnection(async (db) => {
@@ -23,5 +26,39 @@ export const dropTables = async (client: lib_postgres.DatabaseClient) => {
     await db.sql`DROP TABLE IF EXISTS custom_write_checkpoints`.execute();
     await db.sql`DROP SEQUENCE IF EXISTS op_id_sequence`.execute();
     await db.sql`DROP SEQUENCE IF EXISTS sync_rules_id_sequence`.execute();
+    await db.sql`DROP TABLE IF EXISTS migrations`.execute();
   });
 };
+
+/**
+ * Clear all Postgres storage tables and reset sequences.
+ *
+ * Does not clear migration state.
+ */
+export const truncateTables = async (db: lib_postgres.DatabaseClient) => {
+  // Lock a connection for automatic schema search paths
+  await db.query(
+    {
+      statement: `TRUNCATE TABLE bucket_data,
+        bucket_parameters,
+        sync_rules,
+        instance,
+        current_data,
+        source_tables,
+        write_checkpoints,
+        custom_write_checkpoints,
+        connection_report_events RESTART IDENTITY CASCADE
+    `
+    },
+    {
+      statement: `ALTER SEQUENCE IF EXISTS op_id_sequence RESTART
+      WITH
+        1`
+    },
+    {
+      statement: `ALTER SEQUENCE IF EXISTS sync_rules_id_sequence RESTART
+      WITH
+        1`
+    }
+  );
+};

package/src/utils/test-utils.ts

@@ -3,6 +3,7 @@ import { PostgresMigrationAgent } from '../migrations/PostgresMigrationAgent.js'
 import { normalizePostgresStorageConfig, PostgresStorageConfigDecoded } from '../types/types.js';
 import { PostgresReportStorage } from '../storage/PostgresReportStorage.js';
 import { PostgresBucketStorageFactory } from '../storage/PostgresBucketStorageFactory.js';
+import { truncateTables } from './db.js';
 
 export type PostgresTestStorageOptions = {
   url: string;
@@ -22,7 +23,7 @@ export function postgresTestSetup(factoryOptions: PostgresTestStorageOptions) {
 
   const TEST_CONNECTION_OPTIONS = normalizePostgresStorageConfig(BASE_CONFIG);
 
-  const migrate = async (direction: framework.migrations.Direction) => {
+  const runMigrations = async (options: { down: boolean; up: boolean }) => {
     await using migrationManager: PowerSyncMigrationManager = new framework.MigrationManager();
     await using migrationAgent = factoryOptions.migrationAgent
       ? factoryOptions.migrationAgent(BASE_CONFIG)
@@ -31,14 +32,16 @@ export function postgresTestSetup(factoryOptions: PostgresTestStorageOptions) {
 
     const mockServiceContext = { configuration: { storage: BASE_CONFIG } } as unknown as ServiceContext;
 
-    await migrationManager.migrate({
-      direction: framework.migrations.Direction.Down,
-      migrationContext: {
-        service_context: mockServiceContext
-      }
-    });
+    if (options.down) {
+      await migrationManager.migrate({
+        direction: framework.migrations.Direction.Down,
+        migrationContext: {
+          service_context: mockServiceContext
+        }
+      });
+    }
 
-    if (direction == framework.migrations.Direction.Up) {
+    if (options.up) {
       await migrationManager.migrate({
         direction: framework.migrations.Direction.Up,
         migrationContext: {
@@ -48,11 +51,28 @@ export function postgresTestSetup(factoryOptions: PostgresTestStorageOptions) {
     }
   };
 
+  const migrate = async (direction: framework.migrations.Direction) => {
+    await runMigrations({
+      down: true,
+      up: direction == framework.migrations.Direction.Up
+    });
+  };
+
+  const clearStorage = async () => {
+    await runMigrations({ down: false, up: true });
+
+    await using storageFactory = new PostgresBucketStorageFactory({
+      config: TEST_CONNECTION_OPTIONS,
+      slot_name_prefix: 'test_'
+    });
+    await truncateTables(storageFactory.db);
+  };
+
   return {
     reportFactory: async (options?: TestStorageOptions) => {
       try {
         if (!options?.doNotClear) {
-          await migrate(framework.migrations.Direction.Up);
+          await clearStorage();
         }
 
         return new PostgresReportStorage({
@@ -67,7 +87,7 @@ export function postgresTestSetup(factoryOptions: PostgresTestStorageOptions) {
     factory: async (options?: TestStorageOptions) => {
       try {
         if (!options?.doNotClear) {
-          await migrate(framework.migrations.Direction.Up);
+          await clearStorage();
         }
 
         return new PostgresBucketStorageFactory({