@powersync/service-core-tests 0.10.3 → 0.11.0

package/src/tests/register-parameter-compacting-tests.ts

@@ -0,0 +1,172 @@
+import { storage } from '@powersync/service-core';
+import { ParameterLookup } from '@powersync/service-sync-rules';
+import { expect, test } from 'vitest';
+import * as test_utils from '../test-utils/test-utils-index.js';
+
+const TEST_TABLE = test_utils.makeTestTable('test', ['id']);
+
+export function registerParameterCompactTests(generateStorageFactory: storage.TestStorageFactory) {
+  test('compacting parameters', async () => {
+    await using factory = await generateStorageFactory();
+    const syncRules = await factory.updateSyncRules({
+      content: `
+bucket_definitions:
+  test:
+    parameters: select id from test where id = request.user_id()
+    data: []
+    `
+    });
+    const bucketStorage = factory.getInstance(syncRules);
+
+    await bucketStorage.startBatch(test_utils.BATCH_OPTIONS, async (batch) => {
+      await batch.save({
+        sourceTable: TEST_TABLE,
+        tag: storage.SaveOperationTag.INSERT,
+        after: {
+          id: 't1'
+        },
+        afterReplicaId: 't1'
+      });
+
+      await batch.save({
+        sourceTable: TEST_TABLE,
+        tag: storage.SaveOperationTag.INSERT,
+        after: {
+          id: 't2'
+        },
+        afterReplicaId: 't2'
+      });
+
+      await batch.commit('1/1');
+    });
+
+    const lookup = ParameterLookup.normalized('test', '1', ['t1']);
+
+    const checkpoint1 = await bucketStorage.getCheckpoint();
+    const parameters1 = await checkpoint1.getParameterSets([lookup]);
+    expect(parameters1).toEqual([{ id: 't1' }]);
+
+    await bucketStorage.startBatch(test_utils.BATCH_OPTIONS, async (batch) => {
+      await batch.save({
+        sourceTable: TEST_TABLE,
+        tag: storage.SaveOperationTag.UPDATE,
+        before: {
+          id: 't1'
+        },
+        beforeReplicaId: 't1',
+        after: {
+          id: 't1'
+        },
+        afterReplicaId: 't1'
+      });
+
+      await batch.save({
+        sourceTable: TEST_TABLE,
+        tag: storage.SaveOperationTag.DELETE,
+        before: {
+          id: 't1'
+        },
+        beforeReplicaId: 't1'
+      });
+      await batch.commit('1/2');
+    });
+    const checkpoint2 = await bucketStorage.getCheckpoint();
+    const parameters2 = await checkpoint2.getParameterSets([lookup]);
+    expect(parameters2).toEqual([]);
+
+    const statsBefore = await bucketStorage.factory.getStorageMetrics();
+    await bucketStorage.compact({ compactParameterData: true });
+
+    // Check consistency
+    const parameters1b = await checkpoint1.getParameterSets([lookup]);
+    const parameters2b = await checkpoint2.getParameterSets([lookup]);
+    expect(parameters1b).toEqual([{ id: 't1' }]);
+    expect(parameters2b).toEqual([]);
+
+    // Check storage size
+    const statsAfter = await bucketStorage.factory.getStorageMetrics();
+    expect(statsAfter.parameters_size_bytes).toBeLessThan(statsBefore.parameters_size_bytes);
+  });
+
+  for (let cacheLimit of [1, 10]) {
+    test(`compacting deleted parameters with cache size ${cacheLimit}`, async () => {
+      await using factory = await generateStorageFactory();
+      const syncRules = await factory.updateSyncRules({
+        content: `
+bucket_definitions:
+  test:
+    parameters: select id from test where uid = request.user_id()
+    data: []
+    `
+      });
+      const bucketStorage = factory.getInstance(syncRules);
+
+      await bucketStorage.startBatch(test_utils.BATCH_OPTIONS, async (batch) => {
+        await batch.save({
+          sourceTable: TEST_TABLE,
+          tag: storage.SaveOperationTag.INSERT,
+          after: {
+            id: 't1',
+            uid: 'u1'
+          },
+          afterReplicaId: 't1'
+        });
+        // Interleave with another operation, to evict the other cache entry when compacting.
+        await batch.save({
+          sourceTable: TEST_TABLE,
+          tag: storage.SaveOperationTag.INSERT,
+          after: {
+            id: 't2',
+            uid: 'u1'
+          },
+          afterReplicaId: 't2'
+        });
+
+        await batch.commit('1/1');
+      });
+
+      await bucketStorage.startBatch(test_utils.BATCH_OPTIONS, async (batch) => {
+        await batch.save({
+          sourceTable: TEST_TABLE,
+          tag: storage.SaveOperationTag.DELETE,
+          before: {
+            id: 't1',
+            uid: 'u1'
+          },
+          beforeReplicaId: 't1'
+        });
+        await batch.commit('2/1');
+      });
+
+      await bucketStorage.startBatch(test_utils.BATCH_OPTIONS, async (batch) => {
+        await batch.save({
+          sourceTable: TEST_TABLE,
+          tag: storage.SaveOperationTag.UPDATE,
+          after: {
+            id: 't2',
+            uid: 'u2'
+          },
+          afterReplicaId: 't2'
+        });
+        await batch.commit('3/1');
+      });
+
+      const lookup = ParameterLookup.normalized('test', '1', ['u1']);
+
+      const checkpoint1 = await bucketStorage.getCheckpoint();
+      const parameters1 = await checkpoint1.getParameterSets([lookup]);
+      expect(parameters1).toEqual([]);
+
+      const statsBefore = await bucketStorage.factory.getStorageMetrics();
+      await bucketStorage.compact({ compactParameterData: true, compactParameterCacheLimit: cacheLimit });
+
+      // Check consistency
+      const parameters1b = await checkpoint1.getParameterSets([lookup]);
+      expect(parameters1b).toEqual([]);
+
+      // Check storage size
+      const statsAfter = await bucketStorage.factory.getStorageMetrics();
+      expect(statsAfter.parameters_size_bytes).toBeLessThan(statsBefore.parameters_size_bytes);
+    });
+  }
+}

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

@@ -54,7 +54,6 @@ export function registerSyncTests(factory: storage.TestStorageFactory) {
     });
 
     const bucketStorage = f.getInstance(syncRules);
-    await bucketStorage.autoActivate();
 
     const result = await bucketStorage.startBatch(test_utils.BATCH_OPTIONS, async (batch) => {
       await batch.save({
@@ -116,7 +115,6 @@ bucket_definitions:
     });
 
     const bucketStorage = f.getInstance(syncRules);
-    await bucketStorage.autoActivate();
 
     const result = await bucketStorage.startBatch(test_utils.BATCH_OPTIONS, async (batch) => {
       await batch.save({
@@ -178,7 +176,6 @@ bucket_definitions:
     });
 
     const bucketStorage = f.getInstance(syncRules);
-    await bucketStorage.autoActivate();
 
     await bucketStorage.startBatch(test_utils.BATCH_OPTIONS, async (batch) => {
       // Initial data: Add one priority row and 10k low-priority rows.
@@ -289,7 +286,6 @@ bucket_definitions:
     });
 
     const bucketStorage = f.getInstance(syncRules);
-    await bucketStorage.autoActivate();
 
     await bucketStorage.startBatch(test_utils.BATCH_OPTIONS, async (batch) => {
       // Initial data: Add one priority row and 10k low-priority rows.
@@ -431,7 +427,6 @@ bucket_definitions:
     });
 
     const bucketStorage = f.getInstance(syncRules);
-    await bucketStorage.autoActivate();
 
     await bucketStorage.startBatch(test_utils.BATCH_OPTIONS, async (batch) => {
       // Initial data: Add one priority row and 10k low-priority rows.
@@ -561,7 +556,6 @@ bucket_definitions:
       content: BASIC_SYNC_RULES
     });
     const bucketStorage = f.getInstance(syncRules);
-    await bucketStorage.autoActivate();
 
     await bucketStorage.startBatch(test_utils.BATCH_OPTIONS, async (batch) => {
       await batch.save({
@@ -626,7 +620,6 @@ bucket_definitions:
     });
 
     const bucketStorage = await f.getInstance(syncRules);
-    await bucketStorage.autoActivate();
 
     const result = await bucketStorage.startBatch(test_utils.BATCH_OPTIONS, async (batch) => {
       await batch.save({
@@ -671,7 +664,6 @@ bucket_definitions:
     });
 
     const bucketStorage = await f.getInstance(syncRules);
-    await bucketStorage.autoActivate();
 
     const stream = sync.streamResponse({
       syncContext,
@@ -699,7 +691,10 @@ bucket_definitions:
     });
 
     const bucketStorage = await f.getInstance(syncRules);
-    await bucketStorage.autoActivate();
+    // Activate
+    await bucketStorage.startBatch(test_utils.BATCH_OPTIONS, async (batch) => {
+      await batch.keepalive('0/0');
+    });
 
     const stream = sync.streamResponse({
       syncContext,
@@ -770,7 +765,10 @@ bucket_definitions:
     const listsTable = test_utils.makeTestTable('lists', ['id']);
 
     const bucketStorage = await f.getInstance(syncRules);
-    await bucketStorage.autoActivate();
+    // Activate
+    await bucketStorage.startBatch(test_utils.BATCH_OPTIONS, async (batch) => {
+      await batch.keepalive('0/0');
+    });
 
     const stream = sync.streamResponse({
       syncContext,
@@ -833,7 +831,6 @@ bucket_definitions:
     const listsTable = test_utils.makeTestTable('lists', ['id']);
 
     const bucketStorage = await f.getInstance(syncRules);
-    await bucketStorage.autoActivate();
 
     await bucketStorage.startBatch(test_utils.BATCH_OPTIONS, async (batch) => {
       await batch.save({
@@ -911,7 +908,10 @@ bucket_definitions:
     const listsTable = test_utils.makeTestTable('lists', ['id']);
 
     const bucketStorage = await f.getInstance(syncRules);
-    await bucketStorage.autoActivate();
+    // Activate
+    await bucketStorage.startBatch(test_utils.BATCH_OPTIONS, async (batch) => {
+      await batch.keepalive('0/0');
+    });
 
     const stream = sync.streamResponse({
       syncContext,
@@ -974,7 +974,10 @@ bucket_definitions:
     });
 
     const bucketStorage = await f.getInstance(syncRules);
-    await bucketStorage.autoActivate();
+    // Activate
+    await bucketStorage.startBatch(test_utils.BATCH_OPTIONS, async (batch) => {
+      await batch.keepalive('0/0');
+    });
 
     const exp = Date.now() / 1000 + 0.1;
 
@@ -1016,7 +1019,6 @@ bucket_definitions:
     });
 
     const bucketStorage = await f.getInstance(syncRules);
-    await bucketStorage.autoActivate();
 
     await bucketStorage.startBatch(test_utils.BATCH_OPTIONS, async (batch) => {
       await batch.save({
@@ -1157,7 +1159,6 @@ bucket_definitions:
     });
 
     const bucketStorage = f.getInstance(syncRules);
-    await bucketStorage.autoActivate();
 
     await bucketStorage.startBatch(test_utils.BATCH_OPTIONS, async (batch) => {
       // <= the managed write checkpoint LSN below

package/src/tests/tests-index.ts

@@ -1,5 +1,6 @@
 export * from './register-bucket-validation-tests.js';
 export * from './register-compacting-tests.js';
+export * from './register-parameter-compacting-tests.js';
 export * from './register-data-storage-tests.js';
 export * from './register-migration-tests.js';
 export * from './register-sync-tests.js';