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)
48 lines
2.6 KiB
JavaScript
48 lines
2.6 KiB
JavaScript
import { execFile } from 'node:child_process';
|
|
import { promisify } from 'node:util';
|
|
import { fileURLToPath } from 'node:url';
|
|
import { access } from 'node:fs/promises';
|
|
import { join } from 'node:path';
|
|
import assert from 'node:assert/strict';
|
|
import { test as base, expect } from '../../tools/vendor/node_modules/@playwright/test/index.mjs';
|
|
import { launchExecutionHost } from './execution-host.mjs';
|
|
import runtime from '../../tools/vendor/comfyui-runtime.json' with { type: 'json' };
|
|
|
|
export { launchExecutionHost };
|
|
const run = promisify(execFile);
|
|
const reader = fileURLToPath(new URL('./read_execution_history.py', import.meta.url));
|
|
|
|
export async function readExecutionRuns(databasePath, jobId, sessionId) {
|
|
const args = [reader, databasePath, jobId];
|
|
if (sessionId !== undefined) args.push('--session-id', sessionId);
|
|
const { stdout } = await run(process.env.PM_TEST_PYTHON || 'python', args, { timeout: 5000, maxBuffer: 1024 * 1024 });
|
|
return JSON.parse(stdout);
|
|
}
|
|
|
|
export const test = base.extend({
|
|
execution: [async ({ playwright }, use, workerInfo) => {
|
|
const execution = await launchExecutionHost({
|
|
logPath: fileURLToPath(new URL(`../../tools/vendor/test-results/executions/${workerInfo.project.name}-host.log`, import.meta.url)),
|
|
});
|
|
let request;
|
|
try {
|
|
request = await playwright.request.newContext({ baseURL: 'http://127.0.0.1:8196/' });
|
|
await expect.poll(async () => {
|
|
if (execution.host.child.exitCode !== null) throw new Error('Owned execution host exited before readiness');
|
|
try { return (await request.get('/system_stats')).ok(); } catch { return false; }
|
|
}, { timeout: 90000 }).toBe(true);
|
|
const info = (await (await request.get('/system_stats')).json()).system;
|
|
expect(info.comfyui_version).toBe(runtime.serverVersion);
|
|
expect(info.pytorch_version).toBe(`${runtime.torchVersion}+cpu`);
|
|
expect(info.required_frontend_version).toBe(runtime.frontendVersion);
|
|
expect((await request.get('/prompt_manager/stats')).ok()).toBe(true);
|
|
await access(execution.databasePath);
|
|
await assert.rejects(access(join(execution.host.stateDirectory, 'custom_nodes/promptmanager/prompts.db')), { code: 'ENOENT' });
|
|
await use({ ...execution, request });
|
|
} finally {
|
|
try { if (request) await request.post('/pm_execution_fixture/gate/release', { timeout: 5000 }); }
|
|
finally { try { await request?.dispose(); } finally { await execution.dispose(); } }
|
|
}
|
|
}, { scope: 'worker', timeout: 120000 }],
|
|
});
|