supabase-test 0.0.1 → 0.0.4

package/manager.d.ts

@@ -1,25 +0,0 @@
-import { Pool } from 'pg';
-import { PgConfig } from 'pg-env';
-import { PgTestClient, PgTestClientOpts } from './test-client';
-export declare class PgTestConnector {
-    private static instance;
-    private readonly clients;
-    private readonly pgPools;
-    private readonly seenDbConfigs;
-    private readonly pendingConnects;
-    private verbose;
-    private shuttingDown;
-    private constructor();
-    static getInstance(verbose?: boolean): PgTestConnector;
-    private poolKey;
-    private dbKey;
-    beginTeardown(): void;
-    private registerConnect;
-    private awaitPendingConnects;
-    getPool(config: PgConfig): Pool;
-    getClient(config: PgConfig, opts?: Partial<PgTestClientOpts>): PgTestClient;
-    closeAll(): Promise<void>;
-    close(): void;
-    drop(config: PgConfig): void;
-    kill(client: PgTestClient): Promise<void>;
-}

package/manager.js

@@ -1,140 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.PgTestConnector = void 0;
-const logger_1 = require("@launchql/logger");
-const pg_1 = require("pg");
-const pg_env_1 = require("pg-env");
-const admin_1 = require("./admin");
-const test_client_1 = require("./test-client");
-const log = new logger_1.Logger('test-connector');
-const SYS_EVENTS = ['SIGTERM'];
-const end = (pool) => {
-    try {
-        if (pool.ended || pool.ending) {
-            log.warn('⚠️ pg pool already ended or ending');
-            return;
-        }
-        pool.end();
-    }
-    catch (err) {
-        log.error('❌ pg pool termination error:', err);
-    }
-};
-class PgTestConnector {
-    static instance;
-    clients = new Set();
-    pgPools = new Map();
-    seenDbConfigs = new Map();
-    pendingConnects = new Set();
-    verbose = false;
-    shuttingDown = false;
-    constructor(verbose = false) {
-        this.verbose = verbose;
-        SYS_EVENTS.forEach((event) => {
-            process.on(event, () => {
-                log.info(`⏹ Received ${event}, closing all connections...`);
-                this.closeAll();
-            });
-        });
-    }
-    static getInstance(verbose = false) {
-        if (!PgTestConnector.instance) {
-            PgTestConnector.instance = new PgTestConnector(verbose);
-        }
-        return PgTestConnector.instance;
-    }
-    poolKey(config) {
-        return `${config.user}@${config.host}:${config.port}/${config.database}`;
-    }
-    dbKey(config) {
-        return `${config.host}:${config.port}/${config.database}`;
-    }
-    beginTeardown() {
-        this.shuttingDown = true;
-    }
-    registerConnect(p) {
-        this.pendingConnects.add(p);
-        p.finally(() => this.pendingConnects.delete(p));
-    }
-    async awaitPendingConnects() {
-        const arr = Array.from(this.pendingConnects);
-        if (arr.length) {
-            await Promise.allSettled(arr);
-        }
-    }
-    getPool(config) {
-        const key = this.poolKey(config);
-        if (!this.pgPools.has(key)) {
-            const pool = new pg_1.Pool(config);
-            this.pgPools.set(key, pool);
-            log.info(`📘 Created new pg pool: ${key}`);
-        }
-        return this.pgPools.get(key);
-    }
-    getClient(config, opts = {}) {
-        if (this.shuttingDown) {
-            throw new Error('PgTestConnector is shutting down; no new clients allowed');
-        }
-        const client = new test_client_1.PgTestClient(config, {
-            trackConnect: (p) => this.registerConnect(p),
-            ...opts
-        });
-        this.clients.add(client);
-        const key = this.dbKey(config);
-        this.seenDbConfigs.set(key, config);
-        log.info(`🔌 New PgTestClient connected to ${config.database}`);
-        return client;
-    }
-    async closeAll() {
-        this.beginTeardown();
-        await this.awaitPendingConnects();
-        log.info('🧹 Closing all PgTestClients...');
-        await Promise.all(Array.from(this.clients).map(async (client) => {
-            try {
-                await client.close();
-                log.success(`✅ Closed client for ${client.config.database}`);
-            }
-            catch (err) {
-                log.error(`❌ Error closing PgTestClient for ${client.config.database}:`, err);
-            }
-        }));
-        this.clients.clear();
-        log.info('🧯 Disposing pg pools...');
-        for (const [key, pool] of this.pgPools.entries()) {
-            log.debug(`🧯 Disposing pg pool [${key}]`);
-            end(pool);
-        }
-        this.pgPools.clear();
-        log.info('🗑️ Dropping seen databases...');
-        await Promise.all(Array.from(this.seenDbConfigs.values()).map(async (config) => {
-            try {
-                const rootPg = (0, pg_env_1.getPgEnvOptions)();
-                const admin = new admin_1.DbAdmin({ ...config, user: rootPg.user, password: rootPg.password }, this.verbose);
-                admin.drop();
-                log.warn(`🧨 Dropped database: ${config.database}`);
-            }
-            catch (err) {
-                log.error(`❌ Failed to drop database ${config.database}:`, err);
-            }
-        }));
-        this.seenDbConfigs.clear();
-        log.success('✅ All PgTestClients closed, pools disposed, databases dropped.');
-        this.pendingConnects.clear();
-        this.shuttingDown = false;
-    }
-    close() {
-        this.closeAll();
-    }
-    drop(config) {
-        const key = this.dbKey(config);
-        const admin = new admin_1.DbAdmin(config, this.verbose);
-        admin.drop();
-        log.warn(`🧨 Dropped database: ${config.database}`);
-        this.seenDbConfigs.delete(key);
-    }
-    async kill(client) {
-        await client.close();
-        this.drop(client.config);
-    }
-}
-exports.PgTestConnector = PgTestConnector;

package/roles.d.ts

@@ -1,17 +0,0 @@
-import { PgTestConnectionOptions, RoleMapping } from '@launchql/types';
-/**
- * Default role mapping configuration
- */
-export declare const DEFAULT_ROLE_MAPPING: Required<RoleMapping>;
-/**
- * Get resolved role mapping with defaults
- */
-export declare const getRoleMapping: (options?: PgTestConnectionOptions) => Required<RoleMapping>;
-/**
- * Get role name by key with fallback to default mapping
- */
-export declare const getRoleName: (roleKey: keyof Omit<RoleMapping, "default">, options?: PgTestConnectionOptions) => string;
-/**
- * Get default role name
- */
-export declare const getDefaultRole: (options?: PgTestConnectionOptions) => string;

package/roles.js

@@ -1,38 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.getDefaultRole = exports.getRoleName = exports.getRoleMapping = exports.DEFAULT_ROLE_MAPPING = void 0;
-/**
- * Default role mapping configuration
- */
-exports.DEFAULT_ROLE_MAPPING = {
-    anonymous: 'anonymous',
-    authenticated: 'authenticated',
-    administrator: 'administrator',
-    default: 'anonymous'
-};
-/**
- * Get resolved role mapping with defaults
- */
-const getRoleMapping = (options) => {
-    return {
-        ...exports.DEFAULT_ROLE_MAPPING,
-        ...(options?.roles || {})
-    };
-};
-exports.getRoleMapping = getRoleMapping;
-/**
- * Get role name by key with fallback to default mapping
- */
-const getRoleName = (roleKey, options) => {
-    const mapping = (0, exports.getRoleMapping)(options);
-    return mapping[roleKey];
-};
-exports.getRoleName = getRoleName;
-/**
- * Get default role name
- */
-const getDefaultRole = (options) => {
-    const mapping = (0, exports.getRoleMapping)(options);
-    return mapping.default;
-};
-exports.getDefaultRole = getDefaultRole;

package/seed/adapters.d.ts

@@ -1,4 +0,0 @@
-import { SeedAdapter, SeedContext } from './types';
-export declare function sqlfile(files: string[]): SeedAdapter;
-export declare function fn(fn: (ctx: SeedContext) => Promise<void>): SeedAdapter;
-export declare function compose(adapters: SeedAdapter[]): SeedAdapter;

package/seed/adapters.js

@@ -1,28 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.sqlfile = sqlfile;
-exports.fn = fn;
-exports.compose = compose;
-function sqlfile(files) {
-    return {
-        seed(ctx) {
-            for (const file of files) {
-                ctx.admin.loadSql(file, ctx.config.database);
-            }
-        }
-    };
-}
-function fn(fn) {
-    return {
-        seed: fn
-    };
-}
-function compose(adapters) {
-    return {
-        async seed(ctx) {
-            for (const adapter of adapters) {
-                await adapter.seed(ctx);
-            }
-        }
-    };
-}

package/seed/csv.d.ts

@@ -1,9 +0,0 @@
-import { PgTestClient } from '../test-client';
-import { SeedAdapter } from './types';
-interface CsvSeedMap {
-    [tableName: string]: string;
-}
-export declare function csv(tables: CsvSeedMap): SeedAdapter;
-export declare function copyCsvIntoTable(pg: PgTestClient, table: string, filePath: string): Promise<void>;
-export declare function exportTableToCsv(pg: PgTestClient, table: string, filePath: string): Promise<void>;
-export {};

package/seed/csv.js

@@ -1,49 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.csv = csv;
-exports.copyCsvIntoTable = copyCsvIntoTable;
-exports.exportTableToCsv = exportTableToCsv;
-const promises_1 = require("node:stream/promises");
-const logger_1 = require("@launchql/logger");
-const fs_1 = require("fs");
-const pg_copy_streams_1 = require("pg-copy-streams");
-const log = new logger_1.Logger('csv');
-function csv(tables) {
-    return {
-        async seed(ctx) {
-            for (const [table, filePath] of Object.entries(tables)) {
-                if (!(0, fs_1.existsSync)(filePath)) {
-                    throw new Error(`CSV file not found: ${filePath}`);
-                }
-                log.info(`📥 Seeding "${table}" from ${filePath}`);
-                await copyCsvIntoTable(ctx.pg, table, filePath);
-            }
-        }
-    };
-}
-async function copyCsvIntoTable(pg, table, filePath) {
-    const client = pg.client;
-    const stream = client.query((0, pg_copy_streams_1.from)(`COPY ${table} FROM STDIN WITH CSV HEADER`));
-    const source = (0, fs_1.createReadStream)(filePath);
-    try {
-        await (0, promises_1.pipeline)(source, stream);
-        log.success(`✅ Successfully seeded "${table}"`);
-    }
-    catch (err) {
-        log.error(`❌ COPY failed for "${table}": ${err.message}`);
-        throw err;
-    }
-}
-async function exportTableToCsv(pg, table, filePath) {
-    const client = pg.client;
-    const stream = client.query((0, pg_copy_streams_1.to)(`COPY ${table} TO STDOUT WITH CSV HEADER`));
-    const target = (0, fs_1.createWriteStream)(filePath);
-    try {
-        await (0, promises_1.pipeline)(stream, target);
-        log.success(`✅ Exported "${table}" to ${filePath}`);
-    }
-    catch (err) {
-        log.error(`❌ Failed to export "${table}": ${err.message}`);
-        throw err;
-    }
-}

package/seed/index.d.ts

@@ -1,16 +0,0 @@
-import { compose, fn, sqlfile } from './adapters';
-import { csv } from './csv';
-import { json } from './json';
-import { launchql } from './launchql';
-import { sqitch } from './sqitch';
-export * from './csv';
-export * from './types';
-export declare const seed: {
-    launchql: typeof launchql;
-    sqitch: typeof sqitch;
-    json: typeof json;
-    csv: typeof csv;
-    compose: typeof compose;
-    fn: typeof fn;
-    sqlfile: typeof sqlfile;
-};

package/seed/index.js

@@ -1,33 +0,0 @@
-"use strict";
-var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    var desc = Object.getOwnPropertyDescriptor(m, k);
-    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
-      desc = { enumerable: true, get: function() { return m[k]; } };
-    }
-    Object.defineProperty(o, k2, desc);
-}) : (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    o[k2] = m[k];
-}));
-var __exportStar = (this && this.__exportStar) || function(m, exports) {
-    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.seed = void 0;
-const adapters_1 = require("./adapters");
-const csv_1 = require("./csv");
-const json_1 = require("./json");
-const launchql_1 = require("./launchql");
-const sqitch_1 = require("./sqitch");
-__exportStar(require("./csv"), exports);
-__exportStar(require("./types"), exports);
-exports.seed = {
-    launchql: launchql_1.launchql,
-    sqitch: sqitch_1.sqitch,
-    json: json_1.json,
-    csv: csv_1.csv,
-    compose: adapters_1.compose,
-    fn: adapters_1.fn,
-    sqlfile: adapters_1.sqlfile
-};

package/seed/json.d.ts

@@ -1,6 +0,0 @@
-import { SeedAdapter } from './types';
-interface JsonSeedMap {
-    [table: string]: Record<string, any>[];
-}
-export declare function json(data: JsonSeedMap): SeedAdapter;
-export {};

package/seed/json.js

@@ -1,21 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.json = json;
-function json(data) {
-    return {
-        async seed(ctx) {
-            const { pg } = ctx;
-            for (const [table, rows] of Object.entries(data)) {
-                if (!Array.isArray(rows) || rows.length === 0)
-                    continue;
-                const columns = Object.keys(rows[0]);
-                const placeholders = columns.map((_, i) => `$${i + 1}`).join(', ');
-                const sql = `INSERT INTO ${table} (${columns.join(', ')}) VALUES (${placeholders})`;
-                for (const row of rows) {
-                    const values = columns.map((c) => row[c]);
-                    await pg.query(sql, values);
-                }
-            }
-        }
-    };
-}

package/seed/launchql.d.ts

@@ -1,2 +0,0 @@
-import { SeedAdapter } from './types';
-export declare function launchql(cwd?: string, cache?: boolean): SeedAdapter;

package/seed/launchql.js

@@ -1,22 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.launchql = launchql;
-const core_1 = require("@launchql/core");
-const env_1 = require("@launchql/env");
-function launchql(cwd, cache = false) {
-    return {
-        async seed(ctx) {
-            const proj = new core_1.LaunchQLPackage(cwd ?? ctx.connect.cwd);
-            if (!proj.isInModule())
-                return;
-            await proj.deploy((0, env_1.getEnvOptions)({
-                pg: ctx.config,
-                deployment: {
-                    fast: true,
-                    usePlan: true,
-                    cache
-                }
-            }), proj.getModuleName());
-        }
-    };
-}

package/seed/sqitch.d.ts

@@ -1,2 +0,0 @@
-import { SeedAdapter } from './types';
-export declare function sqitch(cwd?: string): SeedAdapter;

package/seed/sqitch.js

@@ -1,20 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.sqitch = sqitch;
-const core_1 = require("@launchql/core");
-const env_1 = require("@launchql/env");
-function sqitch(cwd) {
-    return {
-        async seed(ctx) {
-            const proj = new core_1.LaunchQLPackage(cwd ?? ctx.connect.cwd);
-            if (!proj.isInModule())
-                return;
-            await proj.deploy((0, env_1.getEnvOptions)({
-                pg: ctx.config,
-                deployment: {
-                    fast: false
-                }
-            }), proj.getModuleName(), true);
-        }
-    };
-}

package/seed/types.d.ts

@@ -1,13 +0,0 @@
-import { PgTestConnectionOptions } from '@launchql/types';
-import { PgConfig } from 'pg-env';
-import { DbAdmin } from '../admin';
-import { PgTestClient } from '../test-client';
-export interface SeedContext {
-    connect: PgTestConnectionOptions;
-    admin: DbAdmin;
-    config: PgConfig;
-    pg: PgTestClient;
-}
-export interface SeedAdapter {
-    seed(ctx: SeedContext): Promise<void> | void;
-}

package/seed/types.js

@@ -1,2 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });

package/stream.d.ts

@@ -1,2 +0,0 @@
-import { PgConfig } from 'pg-env';
-export declare function streamSql(config: PgConfig, sql: string): Promise<void>;

package/stream.js

@@ -1,46 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.streamSql = streamSql;
-const child_process_1 = require("child_process");
-const pg_env_1 = require("pg-env");
-const stream_1 = require("stream");
-function setArgs(config) {
-    const args = [
-        '-U', config.user,
-        '-h', config.host,
-        '-d', config.database
-    ];
-    if (config.port) {
-        args.push('-p', String(config.port));
-    }
-    return args;
-}
-// Converts a string to a readable stream (replaces streamify-string)
-function stringToStream(text) {
-    const stream = new stream_1.Readable({
-        read() {
-            this.push(text);
-            this.push(null);
-        }
-    });
-    return stream;
-}
-async function streamSql(config, sql) {
-    const args = setArgs(config);
-    return new Promise((resolve, reject) => {
-        const sqlStream = stringToStream(sql);
-        const proc = (0, child_process_1.spawn)('psql', args, {
-            env: (0, pg_env_1.getSpawnEnvWithPg)(config)
-        });
-        sqlStream.pipe(proc.stdin);
-        proc.on('close', (code) => {
-            resolve();
-        });
-        proc.on('error', (error) => {
-            reject(error);
-        });
-        proc.stderr.on('data', (data) => {
-            reject(new Error(data.toString()));
-        });
-    });
-}

package/test-client.d.ts

@@ -1,49 +0,0 @@
-import { Client, QueryResult } from 'pg';
-import { PgConfig } from 'pg-env';
-import { AuthOptions, PgTestConnectionOptions } from '@launchql/types';
-export type PgTestClientOpts = {
-    deferConnect?: boolean;
-    trackConnect?: (p: Promise<any>) => void;
-} & Partial<PgTestConnectionOptions>;
-export declare class PgTestClient {
-    config: PgConfig;
-    client: Client;
-    private opts;
-    private ctxStmts;
-    private contextSettings;
-    private _ended;
-    private connectPromise;
-    constructor(config: PgConfig, opts?: PgTestClientOpts);
-    private ensureConnected;
-    close(): Promise<void>;
-    begin(): Promise<void>;
-    savepoint(name?: string): Promise<void>;
-    rollback(name?: string): Promise<void>;
-    commit(): Promise<void>;
-    beforeEach(): Promise<void>;
-    afterEach(): Promise<void>;
-    setContext(ctx: Record<string, string | null>): void;
-    /**
-     * Set authentication context for the current session.
-     * Configures role and user ID using cascading defaults from options → opts.auth → RoleMapping.
-     */
-    auth(options?: AuthOptions): void;
-    /**
-     * Commit current transaction to make data visible to other connections, then start fresh transaction.
-     * Maintains test isolation by creating a savepoint and reapplying session context.
-     */
-    publish(): Promise<void>;
-    /**
-     * Clear all session context variables and reset to default anonymous role.
-     */
-    clearContext(): void;
-    any<T = any>(query: string, values?: any[]): Promise<T[]>;
-    one<T = any>(query: string, values?: any[]): Promise<T>;
-    oneOrNone<T = any>(query: string, values?: any[]): Promise<T | null>;
-    many<T = any>(query: string, values?: any[]): Promise<T[]>;
-    manyOrNone<T = any>(query: string, values?: any[]): Promise<T[]>;
-    none(query: string, values?: any[]): Promise<void>;
-    result(query: string, values?: any[]): Promise<import('pg').QueryResult>;
-    query<T = any>(query: string, values?: any[]): Promise<QueryResult<T>>;
-    ctxQuery(): Promise<void>;
-}