scripts/test-postgres-live.test.mjs

import assert from 'node:assert/strict';
import { mkdir, mkdtemp, readFile, rm, writeFile } from 'node:fs/promises';
import os from 'node:os';
import path from 'node:path';
import { spawnSync } from 'node:child_process';
import { fileURLToPath } from 'node:url';
import { describe, it } from 'node:test';

const repoRoot = fileURLToPath(new URL('..', import.meta.url));
const scriptPath = fileURLToPath(new URL('./test-postgres-live.mjs', import.meta.url));

describe('live PostgreSQL test launcher cleanup', () => {
    it('removes the temporary PostgreSQL container when the test command fails', async () => {
        const tempDir = await mkdtemp(path.join(os.tmpdir(), 'pg-live-launcher-'));
        const logPath = path.join(tempDir, 'docker.log');
        const binDir = path.join(tempDir, 'bin');
        await mkdir(binDir);
        await writeFile(path.join(binDir, 'node'), buildFailingNodeShim(), { mode: 0o755 });
        await writeFile(path.join(binDir, 'docker'), buildDockerShim(logPath), { mode: 0o755 });

        try {
            const result = spawnSync(process.execPath, [scriptPath], {
                cwd: repoRoot,
                encoding: 'utf8',
                env: {
                    ...process.env,
                    PATH: `${binDir}${path.delimiter}${process.env.PATH || ''}`
                }
            });

            assert.equal(result.status, 37, `stdout:\n${result.stdout}\nstderr:\n${result.stderr}`);
            const dockerLog = await readFile(logPath, 'utf8');
            assert.match(dockerLog, /^run /m);
            assert.match(dockerLog, /^exec /m);
            assert.match(dockerLog, /^port /m);
            assert.match(dockerLog, /^rm -f gpt-image-agent-test-pg-/m);
        } finally {
            await rm(tempDir, { recursive: true, force: true });
        }
    });

    it('preserves the test exit code when a PostgreSQL URL is provided', async () => {
        const tempDir = await mkdtemp(path.join(os.tmpdir(), 'pg-live-launcher-'));
        const logPath = path.join(tempDir, 'docker.log');
        const binDir = path.join(tempDir, 'bin');
        await mkdir(binDir);
        await writeFile(path.join(binDir, 'node'), buildFailingNodeShim(), { mode: 0o755 });
        await writeFile(path.join(binDir, 'docker'), buildDockerShim(logPath), { mode: 0o755 });

        try {
            const result = spawnSync(process.execPath, [scriptPath], {
                cwd: repoRoot,
                encoding: 'utf8',
                env: {
                    ...process.env,
                    AGENT_POSTGRES_TEST_DATABASE_URL: 'postgres://agent_test:agent_test@127.0.0.1:55432/agent_test',
                    PATH: `${binDir}${path.delimiter}${process.env.PATH || ''}`
                }
            });

            assert.equal(result.status, 37, `stdout:\n${result.stdout}\nstderr:\n${result.stderr}`);
            await assert.rejects(readFile(logPath, 'utf8'), { code: 'ENOENT' });
        } finally {
            await rm(tempDir, { recursive: true, force: true });
        }
    });

    it('surfaces Docker startup stderr when the temporary container cannot start', async () => {
        const tempDir = await mkdtemp(path.join(os.tmpdir(), 'pg-live-launcher-'));
        const logPath = path.join(tempDir, 'docker.log');
        const binDir = path.join(tempDir, 'bin');
        await mkdir(binDir);
        await writeFile(path.join(binDir, 'node'), buildFailingNodeShim(), { mode: 0o755 });
        await writeFile(path.join(binDir, 'docker'), buildFailingDockerRunShim(logPath), { mode: 0o755 });

        try {
            const result = spawnSync(process.execPath, [scriptPath], {
                cwd: repoRoot,
                encoding: 'utf8',
                env: {
                    ...process.env,
                    PATH: `${binDir}${path.delimiter}${process.env.PATH || ''}`
                }
            });

            assert.equal(result.status, 42, `stdout:\n${result.stdout}\nstderr:\n${result.stderr}`);
            assert.match(result.stderr, /docker daemon unavailable/);
            const dockerLog = await readFile(logPath, 'utf8');
            assert.match(dockerLog, /^run /m);
            assert.match(dockerLog, /^rm -f gpt-image-agent-test-pg-/m);
        } finally {
            await rm(tempDir, { recursive: true, force: true });
        }
    });
});

function buildFailingNodeShim() {
    return `#!/bin/sh
exit 37
`;
}

function buildDockerShim(logPath) {
    return `#!/bin/sh
printf '%s\\n' "$*" >> ${JSON.stringify(logPath)}
case "$1" in
  run)
    exit 0
    ;;
  exec)
    exit 0
    ;;
  port)
    printf '127.0.0.1:55432\\n'
    exit 0
    ;;
  rm)
    exit 0
    ;;
  *)
    exit 2
    ;;
esac
`;
}

function buildFailingDockerRunShim(logPath) {
    return `#!/bin/sh
printf '%s\\n' "$*" >> ${JSON.stringify(logPath)}
case "$1" in
  run)
    printf 'docker daemon unavailable\\n' >&2
    exit 42
    ;;
  rm)
    exit 0
    ;;
  *)
    exit 2
    ;;
esac
`;
}