@powersync/service-module-postgres-storage 0.7.4 → 0.8.0

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;