Vito
763e6b37b1
* feat(history): normalize bounded execution evidence (#209) * fix(history): bound lifecycle kind checks before hashing (#209) * feat(history): persist execution attempts with restore fencing (#209) * test(api): bootstrap standalone backup restore fixtures (#209) * feat(history): isolate bounded asynchronous ledger writes (#209) * feat(history): observe actual queue attempts without altering execution (#209) * fix(database): honor package-scoped configuration (#211) * test(history): exercise real cached and headless executions (#209) * fix(history): reject unsupported uniqueness and isolate package fixtures (#209) * test(history): support Python 3.10 package fixture cleanup (#209)
42 lines
2.4 KiB
JavaScript
42 lines
2.4 KiB
JavaScript
import { execFileSync } from 'node:child_process';
|
|
import { copyFile, mkdtemp, rm } from 'node:fs/promises';
|
|
import { tmpdir } from 'node:os';
|
|
import { createServer } from 'node:net';
|
|
import { join } from 'node:path';
|
|
import { fileURLToPath } from 'node:url';
|
|
import { launchSharedHost as launchComfyHost, preparePromptManagerPlugin } from '../../tools/vendor/comfyui-host.mjs';
|
|
import runtime from '../../tools/vendor/comfyui-runtime.json' with { type: 'json' };
|
|
|
|
const repositoryRoot = fileURLToPath(new URL('../../', import.meta.url));
|
|
|
|
export async function launchExecutionHost({ checkout = process.env.PM_COMFY_ROOT,
|
|
repository = process.env.PM_EXECUTION_REPOSITORY || repositoryRoot,
|
|
python = process.env.PM_TEST_PYTHON || 'python', port = 8196, pins = runtime,
|
|
logPath = join(repositoryRoot, 'tools/vendor/test-results/executions/host.log'), forwardOutput = true } = {}) {
|
|
if (port !== 8196) throw new Error('Execution tests reserve managed loopback port 8196');
|
|
await new Promise((resolve, reject) => {
|
|
const probe = createServer();
|
|
probe.once('error', error => reject(new Error('Execution loopback port 8196 is already in use', { cause: error })));
|
|
probe.listen({ host: '127.0.0.1', port, exclusive: true }, () => probe.close(error => error ? reject(error) : resolve()));
|
|
});
|
|
const fixtureDirectory = await mkdtemp(join(tmpdir(), 'pm-execution-fixture-'));
|
|
let host;
|
|
try {
|
|
await copyFile(new URL('./fixtures/ledger_nodes.py', import.meta.url), join(fixtureDirectory, '__init__.py'));
|
|
execFileSync('git', ['init', '--quiet', fixtureDirectory]);
|
|
execFileSync('git', ['-C', fixtureDirectory, 'add', '__init__.py']);
|
|
host = await launchComfyHost({ checkout, repository, python, port, pins, logPath, forwardOutput,
|
|
pluginName: 'promptmanager', preparePlugin: preparePromptManagerPlugin,
|
|
additionalPlugins: [{ pluginName: 'pm_execution_fixture', repository: fixtureDirectory }],
|
|
});
|
|
} catch (error) {
|
|
await rm(fixtureDirectory, { recursive: true, force: true });
|
|
throw error;
|
|
}
|
|
host.finished.finally(() => rm(fixtureDirectory, { recursive: true, force: true })).catch(() => {});
|
|
return { host, fixtureDirectory, databasePath: join(host.stateDirectory, 'prompts.db'), async dispose() {
|
|
try { await host.stop(); }
|
|
finally { await rm(fixtureDirectory, { recursive: true, force: true }); }
|
|
} };
|
|
}
|