@rawsql-ts/ztd-cli 0.14.4 → 0.16.0

package/templates/dist/testkit-core/src/provider/TestkitProvider.d.ts

@@ -0,0 +1,57 @@
+import type { TableFixture } from '../types';
+/** Supported connection reuse strategies exposed by the provider. */
+export type ConnectionStrategy = 'shared' | 'perTest';
+/** Minimum subset of a connection required by the provider. */
+export type TestkitConnection = {
+    query(sql: string, values?: unknown[]): Promise<unknown>;
+};
+/** Custom hook invoked after a shared scenario to reset session state. */
+export type ConnectionResetHook<TConnection extends TestkitConnection> = (connection: TConnection) => Promise<void> | void;
+/**
+ * Options that determine how the shared connection is cleaned up between scenarios.
+ */
+export type ConnectionResetOption<TConnection extends TestkitConnection> = 'transaction' | 'none' | ConnectionResetHook<TConnection>;
+/**
+ * Configuration required to construct a `TestkitProvider`.
+ */
+export interface CreateTestkitProviderOptions<TConnection extends TestkitConnection, TResource> {
+    connectionFactory: () => Promise<TConnection>;
+    resourceFactory: (connection: TConnection, fixtures: TableFixture[]) => Promise<TResource> | TResource;
+    strategy?: ConnectionStrategy;
+    reset?: ConnectionResetOption<TConnection>;
+    releaseResource?: (resource: TResource) => Promise<void> | void;
+    disposeConnection?: (connection: TConnection) => Promise<void> | void;
+}
+type PerTestAccessor<TResource> = {
+    withRepositoryFixture<Result>(fixtures: TableFixture[], callback: (resource: TResource) => Promise<Result> | Result): Promise<Result>;
+};
+/**
+ * Manages connection recycling for ZTD repositories by wiring fixtures into
+ * shared or per-test sessions.
+ */
+export declare class TestkitProvider<TConnection extends TestkitConnection, TResource> {
+    private readonly options;
+    private sharedConnectionPromise?;
+    private sharedConnection?;
+    private closed;
+    private sharedQueue;
+    private readonly releaseResource;
+    private readonly disposeConnection;
+    constructor(options: CreateTestkitProviderOptions<TConnection, TResource>);
+    withRepositoryFixture<Result>(fixtures: TableFixture[], callback: (resource: TResource) => Promise<Result> | Result, overrides?: {
+        strategy?: ConnectionStrategy;
+    }): Promise<Result>;
+    perTest(): PerTestAccessor<TResource>;
+    close(): Promise<void>;
+    private runWithSharedConnection;
+    private runWithPerTestConnection;
+    private beginTransactionIfNeeded;
+    private executeReset;
+    private ensureSharedConnection;
+    private ensureNotClosed;
+}
+/**
+ * Instantiates a `TestkitProvider` with the given configuration.
+ */
+export declare function createTestkitProvider<TConnection extends TestkitConnection, TResource>(options: CreateTestkitProviderOptions<TConnection, TResource>): Promise<TestkitProvider<TConnection, TResource>>;
+export {};

package/templates/dist/testkit-core/src/provider/TestkitProvider.js

@@ -0,0 +1,149 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.TestkitProvider = void 0;
+exports.createTestkitProvider = createTestkitProvider;
+const DEFAULT_STRATEGY = 'shared';
+const DEFAULT_RESET = 'transaction';
+const defaultDisposeConnection = async (connection) => {
+    const closable = connection;
+    if (typeof closable.release === 'function') {
+        closable.release();
+        return;
+    }
+    if (typeof closable.end === 'function') {
+        await closable.end();
+    }
+};
+/**
+ * Manages connection recycling for ZTD repositories by wiring fixtures into
+ * shared or per-test sessions.
+ */
+class TestkitProvider {
+    constructor(options) {
+        var _a, _b;
+        this.options = options;
+        this.closed = false;
+        this.sharedQueue = Promise.resolve();
+        this.releaseResource = (_a = options.releaseResource) !== null && _a !== void 0 ? _a : (() => undefined);
+        this.disposeConnection =
+            (_b = options.disposeConnection) !== null && _b !== void 0 ? _b : (async (connection) => defaultDisposeConnection(connection));
+    }
+    withRepositoryFixture(fixtures, callback, overrides) {
+        var _a, _b;
+        this.ensureNotClosed();
+        const strategy = (_b = (_a = overrides === null || overrides === void 0 ? void 0 : overrides.strategy) !== null && _a !== void 0 ? _a : this.options.strategy) !== null && _b !== void 0 ? _b : DEFAULT_STRATEGY;
+        if (strategy === 'perTest') {
+            // Use a fresh connection for every invocation to keep session state isolated.
+            return this.runWithPerTestConnection(fixtures, callback);
+        }
+        // The shared path reuses a single connection and applies the configured reset.
+        return this.runWithSharedConnection(fixtures, callback);
+    }
+    perTest() {
+        return {
+            withRepositoryFixture: (fixtures, callback) => this.withRepositoryFixture(fixtures, callback, { strategy: 'perTest' }),
+        };
+    }
+    async close() {
+        if (this.closed) {
+            return;
+        }
+        this.closed = true;
+        // Ensure no shared scenario is still running before cleanup.
+        await this.sharedQueue.catch(() => undefined);
+        if (!this.sharedConnectionPromise) {
+            return;
+        }
+        // Wait for the shared connection to settle before disposing it.
+        const connection = await this.sharedConnectionPromise;
+        await this.disposeConnection(connection);
+    }
+    async runWithSharedConnection(fixtures, callback) {
+        const runScenario = async () => {
+            var _a;
+            const connection = await this.ensureSharedConnection();
+            const resetOption = (_a = this.options.reset) !== null && _a !== void 0 ? _a : DEFAULT_RESET;
+            const transactionStarted = await this.beginTransactionIfNeeded(connection, resetOption);
+            let resource;
+            try {
+                // Create the resource tied to this shared connection before invoking the callback.
+                resource = await this.options.resourceFactory(connection, fixtures);
+                return await callback(resource);
+            }
+            finally {
+                if (resource !== undefined) {
+                    await this.releaseResource(resource);
+                }
+                // Run the configured reset after the scenario so the connection stays clean.
+                await this.executeReset(connection, resetOption, transactionStarted);
+            }
+        };
+        // Serialize shared usage so scenarios never run in parallel.
+        const next = this.sharedQueue.catch(() => undefined);
+        const task = next.then(() => runScenario());
+        this.sharedQueue = task.then(() => undefined, () => undefined);
+        return task;
+    }
+    async runWithPerTestConnection(fixtures, callback) {
+        const connection = await this.options.connectionFactory();
+        try {
+            // Every invocation gets a new connection that is disposed afterward.
+            const resource = await this.options.resourceFactory(connection, fixtures);
+            try {
+                return await callback(resource);
+            }
+            finally {
+                await this.releaseResource(resource);
+            }
+        }
+        finally {
+            // Tear down the transient connection once the callback completes.
+            await this.disposeConnection(connection);
+        }
+    }
+    async beginTransactionIfNeeded(connection, resetOption) {
+        if (resetOption !== 'transaction') {
+            return false;
+        }
+        // Wrap the scenario in a transaction so it can be rolled back later.
+        await connection.query('BEGIN');
+        return true;
+    }
+    async executeReset(connection, resetOption, transactionStarted) {
+        if (resetOption === 'transaction' && transactionStarted) {
+            // Roll the prior transaction back to keep the shared connection clean.
+            await connection.query('ROLLBACK');
+            return;
+        }
+        if (typeof resetOption === 'function') {
+            // Invoke the custom reset hook after each shared scenario.
+            await resetOption(connection);
+        }
+    }
+    async ensureSharedConnection() {
+        if (this.sharedConnectionPromise) {
+            return this.sharedConnectionPromise;
+        }
+        // Cache the single shared connection so concurrent callers reuse it.
+        this.sharedConnectionPromise = (async () => {
+            const connection = await this.options.connectionFactory();
+            this.sharedConnection = connection;
+            return connection;
+        })();
+        return this.sharedConnectionPromise;
+    }
+    ensureNotClosed() {
+        if (this.closed) {
+            // Prevent operations after the provider has been shut down.
+            throw new Error('TestkitProvider is already closed.');
+        }
+    }
+}
+exports.TestkitProvider = TestkitProvider;
+/**
+ * Instantiates a `TestkitProvider` with the given configuration.
+ */
+async function createTestkitProvider(options) {
+    return new TestkitProvider(options);
+}
+//# sourceMappingURL=TestkitProvider.js.map

package/templates/dist/testkit-core/src/provider/TestkitProvider.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"TestkitProvider.js","sourceRoot":"","sources":["../../../../../../testkit-core/src/provider/TestkitProvider.ts"],"names":[],"mappings":";;;AA6OA,sDAOC;AApMD,MAAM,gBAAgB,GAAuB,QAAQ,CAAC;AACtD,MAAM,aAAa,GAA6C,aAAa,CAAC;AAE9E,MAAM,wBAAwB,GAAG,KAAK,EACpC,UAA6B,EACd,EAAE;IACjB,MAAM,QAAQ,GAAG,UAGf,CAAC;IACH,IAAI,OAAO,QAAQ,CAAC,OAAO,KAAK,UAAU,EAAE,CAAC;QAC3C,QAAQ,CAAC,OAAO,EAAE,CAAC;QACnB,OAAO;IACT,CAAC;IACD,IAAI,OAAO,QAAQ,CAAC,GAAG,KAAK,UAAU,EAAE,CAAC;QACvC,MAAM,QAAQ,CAAC,GAAG,EAAE,CAAC;IACvB,CAAC;AACH,CAAC,CAAC;AAEF;;;GAGG;AACH,MAAa,eAAe;IAgB1B,YAA6B,OAA6D;;QAA7D,YAAO,GAAP,OAAO,CAAsD;QAVlF,WAAM,GAAG,KAAK,CAAC;QACf,gBAAW,GAAkB,OAAO,CAAC,OAAO,EAAE,CAAC;QAUrD,IAAI,CAAC,eAAe,GAAG,MAAA,OAAO,CAAC,eAAe,mCAAI,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;QACpE,IAAI,CAAC,iBAAiB;YACpB,MAAA,OAAO,CAAC,iBAAiB,mCACzB,CAAC,KAAK,EAAE,UAAU,EAAE,EAAE,CAAC,wBAAwB,CAAC,UAAU,CAAC,CAAC,CAAC;IACjE,CAAC;IAEM,qBAAqB,CAC1B,QAAwB,EACxB,QAA2D,EAC3D,SAA6C;;QAE7C,IAAI,CAAC,eAAe,EAAE,CAAC;QACvB,MAAM,QAAQ,GAAG,MAAA,MAAA,SAAS,aAAT,SAAS,uBAAT,SAAS,CAAE,QAAQ,mCAAI,IAAI,CAAC,OAAO,CAAC,QAAQ,mCAAI,gBAAgB,CAAC;QAClF,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;YAC3B,8EAA8E;YAC9E,OAAO,IAAI,CAAC,wBAAwB,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;QAC3D,CAAC;QACD,+EAA+E;QAC/E,OAAO,IAAI,CAAC,uBAAuB,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;IAC1D,CAAC;IAEM,OAAO;QACZ,OAAO;YACL,qBAAqB,EAAE,CAAC,QAAQ,EAAE,QAAQ,EAAE,EAAE,CAC5C,IAAI,CAAC,qBAAqB,CAAC,QAAQ,EAAE,QAAQ,EAAE,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC;SAC1E,CAAC;IACJ,CAAC;IAEM,KAAK,CAAC,KAAK;QAChB,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,OAAO;QACT,CAAC;QACD,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;QACnB,6DAA6D;QAC7D,MAAM,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;QAC9C,IAAI,CAAC,IAAI,CAAC,uBAAuB,EAAE,CAAC;YAClC,OAAO;QACT,CAAC;QACD,gEAAgE;QAChE,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,uBAAuB,CAAC;QACtD,MAAM,IAAI,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;IAC3C,CAAC;IAEO,KAAK,CAAC,uBAAuB,CACnC,QAAwB,EACxB,QAA2D;QAE3D,MAAM,WAAW,GAAG,KAAK,IAAqB,EAAE;;YAC9C,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,sBAAsB,EAAE,CAAC;YACvD,MAAM,WAAW,GAAG,MAAA,IAAI,CAAC,OAAO,CAAC,KAAK,mCAAI,aAAa,CAAC;YACxD,MAAM,kBAAkB,GAAG,MAAM,IAAI,CAAC,wBAAwB,CAC5D,UAAU,EACV,WAAW,CACZ,CAAC;YACF,IAAI,QAA+B,CAAC;YACpC,IAAI,CAAC;gBACH,mFAAmF;gBACnF,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;gBACpE,OAAO,MAAM,QAAQ,CAAC,QAAQ,CAAC,CAAC;YAClC,CAAC;oBAAS,CAAC;gBACT,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;oBAC3B,MAAM,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC;gBACvC,CAAC;gBACD,6EAA6E;gBAC7E,MAAM,IAAI,CAAC,YAAY,CAAC,UAAU,EAAE,WAAW,EAAE,kBAAkB,CAAC,CAAC;YACvE,CAAC;QACH,CAAC,CAAC;QAEF,6DAA6D;QAC7D,MAAM,IAAI,GAAG,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;QACrD,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,WAAW,EAAE,CAAC,CAAC;QAC5C,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,IAAI,CAC1B,GAAG,EAAE,CAAC,SAAS,EACf,GAAG,EAAE,CAAC,SAAS,CAChB,CAAC;QACF,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,KAAK,CAAC,wBAAwB,CACpC,QAAwB,EACxB,QAA2D;QAE3D,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,iBAAiB,EAAE,CAAC;QAC1D,IAAI,CAAC;YACH,qEAAqE;YACrE,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;YAC1E,IAAI,CAAC;gBACH,OAAO,MAAM,QAAQ,CAAC,QAAQ,CAAC,CAAC;YAClC,CAAC;oBAAS,CAAC;gBACT,MAAM,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC;YACvC,CAAC;QACH,CAAC;gBAAS,CAAC;YACT,kEAAkE;YAClE,MAAM,IAAI,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;QAC3C,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,wBAAwB,CACpC,UAAuB,EACvB,WAA+C;QAE/C,IAAI,WAAW,KAAK,aAAa,EAAE,CAAC;YAClC,OAAO,KAAK,CAAC;QACf,CAAC;QACD,qEAAqE;QACrE,MAAM,UAAU,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAChC,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,KAAK,CAAC,YAAY,CACxB,UAAuB,EACvB,WAA+C,EAC/C,kBAA2B;QAE3B,IAAI,WAAW,KAAK,aAAa,IAAI,kBAAkB,EAAE,CAAC;YACxD,uEAAuE;YACvE,MAAM,UAAU,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;YACnC,OAAO;QACT,CAAC;QACD,IAAI,OAAO,WAAW,KAAK,UAAU,EAAE,CAAC;YACtC,2DAA2D;YAC3D,MAAM,WAAW,CAAC,UAAU,CAAC,CAAC;QAChC,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,sBAAsB;QAClC,IAAI,IAAI,CAAC,uBAAuB,EAAE,CAAC;YACjC,OAAO,IAAI,CAAC,uBAAuB,CAAC;QACtC,CAAC;QACD,qEAAqE;QACrE,IAAI,CAAC,uBAAuB,GAAG,CAAC,KAAK,IAAI,EAAE;YACzC,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,iBAAiB,EAAE,CAAC;YAC1D,IAAI,CAAC,gBAAgB,GAAG,UAAU,CAAC;YACnC,OAAO,UAAU,CAAC;QACpB,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,uBAAuB,CAAC;IACtC,CAAC;IAEO,eAAe;QACrB,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,4DAA4D;YAC5D,MAAM,IAAI,KAAK,CAAC,oCAAoC,CAAC,CAAC;QACxD,CAAC;IACH,CAAC;CACF;AAjKD,0CAiKC;AAED;;GAEG;AACI,KAAK,UAAU,qBAAqB,CAIzC,OAA6D;IAE7D,OAAO,IAAI,eAAe,CAAC,OAAO,CAAC,CAAC;AACtC,CAAC"}

package/templates/dist/testkit-core/src/rewriter/ResultSelectRewriter.d.ts

@@ -0,0 +1,43 @@
+import type { MissingFixtureStrategy, SqlFormatterOptions } from 'rawsql-ts';
+import type { FixtureResolver, TableRowsFixture } from '../types';
+import { TableNameResolver } from '../fixtures/TableNameResolver';
+interface RewriteResult {
+    sql: string;
+    fixturesApplied: string[];
+    sourceCommand: string | null;
+    isCountWrapper: boolean;
+}
+/** Rewrites CRUD/SELECT statements into fixture-backed SELECTs while ignoring unsupported DDL. */
+export declare class ResultSelectRewriter {
+    private readonly fixtures;
+    private readonly missingFixtureStrategy;
+    private readonly tableNameResolver?;
+    private readonly formatter;
+    private visitedComponents;
+    constructor(fixtures: FixtureResolver, missingFixtureStrategy?: MissingFixtureStrategy, formatterOptions?: SqlFormatterOptions, tableNameResolver?: TableNameResolver | undefined);
+    rewrite(sql: string, scopedFixtures?: TableRowsFixture[]): RewriteResult;
+    private prepareInputs;
+    private parseStatements;
+    private convertStatement;
+    private convertToResultSelect;
+    private injectFixtures;
+    private applyFixtureAliases;
+    private buildFixtureAliasMap;
+    private buildFixtureAlias;
+    private filterFixturesForQuery;
+    private collectReferencedTableKeys;
+    private getLookupCandidates;
+    private getPrimaryKey;
+    private normalizeParameters;
+    private restoreParameters;
+    private rewriteSchemaQualifiers;
+    private rewriteColumnReferencesInQuery;
+    private rewriteSourceExpression;
+    private rewriteSourceComponent;
+    private rewriteSqlComponent;
+    private rewriteValueComponent;
+    private rewriteSwitchCaseArgument;
+    private rewriteWindowFrameExpression;
+    private rewriteColumnReference;
+}
+export {};