@proteinjs/db-driver-spanner 1.23.5 → 1.25.0

package/test/util/columnEncryptionTestHarness.ts

@@ -0,0 +1,54 @@
+import { DataEncryptionKeyTable, EncryptedColumns, Table, TableManager } from '@proteinjs/db';
+import { cascadeDeleteTestTables } from '@proteinjs/db/test';
+import { SpannerDriver } from '@proteinjs/db-driver-spanner';
+
+/**
+ * Schema loading for the column-encryption suites. `Db.delete` consults the reverse-cascade
+ * edge index over the WHOLE static table registry, and dynamic-reference edges are
+ * target-agnostic — so a delete from ANY table queries every registered table holding a
+ * `reverseCascadeDelete` dynamic-reference column (the cascade-delete test tables). A
+ * single-file run must therefore create those alongside the suite's own tables, plus the
+ * framework's data-key table. Serial `loadTable` (never batch `loadTables`): the registry
+ * deliberately holds same-name table pairs for schema-change tests, which a batch create
+ * would collide on.
+ *
+ * @internal This module is intended to be used only in tests.
+ */
+export const loadColumnEncryptionTestSchema = async (
+  tableManager: TableManager,
+  suiteTables: Table<any>[]
+): Promise<void> => {
+  const tables: Table<any>[] = [
+    ...Object.values(cascadeDeleteTestTables),
+    new DataEncryptionKeyTable(),
+    ...suiteTables,
+  ];
+  for (const table of tables) {
+    await tableManager.loadTable(table);
+  }
+};
+
+/**
+ * Clean-slate purge of a suite's tables (rows + their derived token tables) in beforeAll:
+ * a prior run killed mid-flight (or a teardown starved out by shared-emulator contention)
+ * leaves seeded rows behind, and exact-match search assertions must never depend on a
+ * previous process having exited cleanly.
+ */
+export const purgeColumnEncryptionTestRows = async (
+  spannerDriver: SpannerDriver,
+  suiteTables: Table<any>[]
+): Promise<void> => {
+  const encryptedColumns = new EncryptedColumns();
+  const tables: Table<any>[] = [];
+  for (const table of suiteTables) {
+    tables.push(table);
+    const tokenTable = encryptedColumns.tokenTableFor(table);
+    if (tokenTable) {
+      tables.push(tokenTable);
+    }
+  }
+
+  for (const table of tables) {
+    await spannerDriver.runDml(() => ({ sql: `DELETE FROM \`${table.name}\` WHERE TRUE` }));
+  }
+};

package/test/util/columnEncryptionTestTables.ts

@@ -0,0 +1,115 @@
+import { Record, StringColumn, Table, withRecordColumns } from '@proteinjs/db';
+
+/**
+ * Tables for the column-encryption suites (`ColumnEncryption.test.ts`,
+ * `EncryptedSearch.test.ts`, `EncryptionSortAndUnique.test.ts`,
+ * `EncryptionLifecycleWalker.test.ts`, `EncryptionPerf.test.ts`).
+ *
+ * Defined here rather than in the test files so the reflection build registers them as
+ * `Table` loadables — the driver's statement generation and the framework's derived
+ * token-table subqueries resolve tables through `tableByName`.
+ */
+
+/** Core seam suite: contains + equality + opaque encrypted + plaintext metadata. */
+export interface EncNote extends Record {
+  scope: string;
+  title?: string | null; // encrypted + contains search
+  label?: string | null; // encrypted + equality lookup
+  body?: string | null; // encrypted, never queried by value
+  status?: string | null; // plaintext metadata
+}
+
+export class EncNoteTable extends Table<EncNote> {
+  name = 'db_test_enc_note';
+  columns: Table<EncNote>['columns'] = withRecordColumns<EncNote>({
+    scope: new StringColumn('scope', {}, 36),
+    title: new StringColumn('title', { encrypted: { searchable: 'contains' } }),
+    label: new StringColumn('label', { encrypted: { searchable: 'equality' } }),
+    body: new StringColumn('body', { encrypted: {} }, 'MAX'),
+    status: new StringColumn('status', { encrypted: false }),
+  });
+}
+
+/** Search-exactness suite: one contains column beside plaintext metadata. */
+export interface EncSearchDoc extends Record {
+  scope: string;
+  title?: string | null; // encrypted + contains
+  name?: string | null; // encrypted + equality
+  kind?: string | null; // plaintext metadata
+}
+
+export class EncSearchDocTable extends Table<EncSearchDoc> {
+  name = 'db_test_enc_search_doc';
+  columns: Table<EncSearchDoc>['columns'] = withRecordColumns<EncSearchDoc>({
+    scope: new StringColumn('scope', {}, 36),
+    title: new StringColumn('title', { encrypted: { searchable: 'contains' } }),
+    name: new StringColumn('name', { encrypted: { searchable: 'equality' } }),
+    kind: new StringColumn('kind', { encrypted: false }),
+  });
+}
+
+/** sortKey suite: the declared bounded-reveal native ORDER BY. */
+export interface EncSortedItem extends Record {
+  scope: string;
+  title?: string | null; // encrypted + sortKey(3)
+}
+
+export class EncSortedItemTable extends Table<EncSortedItem> {
+  name = 'db_test_enc_sorted_item';
+  columns: Table<EncSortedItem>['columns'] = withRecordColumns<EncSortedItem>({
+    scope: new StringColumn('scope', {}, 36),
+    title: new StringColumn('title', { encrypted: { searchable: 'contains', sortKey: { revealPrefix: 3 } } }),
+  });
+}
+
+/** Uniqueness suite: unique moves onto the equality fingerprint. */
+export interface EncUniqueTag extends Record {
+  scope: string;
+  name?: string | null; // encrypted + equality, unique
+}
+
+export class EncUniqueTagTable extends Table<EncUniqueTag> {
+  name = 'db_test_enc_unique_tag';
+  columns: Table<EncUniqueTag>['columns'] = withRecordColumns<EncUniqueTag>({
+    scope: new StringColumn('scope', {}, 36),
+    name: new StringColumn('name', { unique: { unique: true }, encrypted: { searchable: 'equality' } }),
+  });
+}
+
+/** Lifecycle-walker suite: encrypt / retokenize / decrypt / rotate transitions. */
+export interface EncWalkRow extends Record {
+  scope: string;
+  title?: string | null; // encrypted + contains
+  body?: string | null; // encrypted, no search
+}
+
+export class EncWalkRowTable extends Table<EncWalkRow> {
+  name = 'db_test_enc_walk_row';
+  columns: Table<EncWalkRow>['columns'] = withRecordColumns<EncWalkRow>({
+    scope: new StringColumn('scope', {}, 36),
+    title: new StringColumn('title', { encrypted: { searchable: 'contains' } }),
+    body: new StringColumn('body', { encrypted: {} }, 'MAX'),
+  });
+}
+
+/** Perf suite: encrypted-at-rest body vs a plaintext twin, 1k-row reads. */
+export interface EncPerfRow extends Record {
+  scope: string;
+  body?: string | null;
+}
+
+export class EncPerfRowTable extends Table<EncPerfRow> {
+  name = 'db_test_enc_perf_row';
+  columns: Table<EncPerfRow>['columns'] = withRecordColumns<EncPerfRow>({
+    scope: new StringColumn('scope', {}, 36),
+    body: new StringColumn('body', { encrypted: {} }, 'MAX'),
+  });
+}
+
+export class PlainPerfRowTable extends Table<EncPerfRow> {
+  name = 'db_test_plain_perf_row';
+  columns: Table<EncPerfRow>['columns'] = withRecordColumns<EncPerfRow>({
+    scope: new StringColumn('scope', {}, 36),
+    body: new StringColumn('body', { encrypted: false }, 'MAX'),
+  });
+}

package/test/util/referenceAdoptWidthTestTables.ts

@@ -0,0 +1,23 @@
+import { Record, Reference, ReferenceColumn, Table, withRecordColumns } from '@proteinjs/db';
+
+/**
+ * The registered face of the width-adoption suite's table (ReferenceColumnAdoptWidth.test.ts):
+ * the retyped declaration — a reference adopting the column's pre-existing STRING(255) width.
+ * Declared here (not inline in the suite) because Db statement generation resolves tables by
+ * NAME through the reflection registry; the suite's string-era and stock-width variants of the
+ * same physical table stay inline there, reaching only instance-passed TableManager APIs.
+ */
+
+export interface AdoptWidthRecord extends Record {
+  invitedBy?: Reference<Record> | null;
+}
+
+export const ADOPT_WIDTH_TABLE_NAME = 'db_test_reference_adopt_width';
+export const ADOPT_WIDTH_TARGET_TABLE_NAME = 'db_test_reference_adopt_target';
+
+export class ReferenceAdoptWidthTestTable extends Table<AdoptWidthRecord> {
+  name = ADOPT_WIDTH_TABLE_NAME;
+  columns = withRecordColumns<AdoptWidthRecord>({
+    invitedBy: new ReferenceColumn<Record>('invited_by', ADOPT_WIDTH_TARGET_TABLE_NAME, false, { maxLength: 255 }),
+  });
+}