Codex
84 lines
3.1 KiB
TypeScript
84 lines
3.1 KiB
TypeScript
import assert from "node:assert/strict";
|
|
import { execFile } from "node:child_process";
|
|
import { access, mkdtemp, rm } from "node:fs/promises";
|
|
import os from "node:os";
|
|
import path from "node:path";
|
|
import test from "node:test";
|
|
import { promisify } from "node:util";
|
|
|
|
const execFileAsync = promisify(execFile);
|
|
const bunCommand = process.env.HWLAB_TEST_BUN_COMMAND || process.env.HWLAB_BUN_COMMAND || (process.versions.bun ? process.execPath : "bun");
|
|
|
|
test("agent worker health and dry-run preserve session evidence and cleanup behavior", async () => {
|
|
const stateDir = await mkdtemp(path.join(os.tmpdir(), "hwlab-agent-worker-"));
|
|
try {
|
|
const health = await runWorkerJson(["health", "--state-dir", stateDir]);
|
|
assert.equal(health.serviceId, "hwlab-agent-worker");
|
|
assert.equal(health.ok, true);
|
|
assert.equal(health.status, "ok");
|
|
assert.equal(health.skills.status, "degraded");
|
|
|
|
await runManagerJson([
|
|
"create",
|
|
"--agent-session-id", "ags_worker",
|
|
"--project-id", "prj_worker",
|
|
"--skill-commit-id", "commit-worker",
|
|
"--skill-version", "2.0.0",
|
|
"--state-dir", stateDir
|
|
]);
|
|
|
|
const dryRun = await runWorkerJson([
|
|
"dry-run",
|
|
"--agent-session-id", "ags_worker",
|
|
"--skill-commit-id", "commit-worker",
|
|
"--skill-version", "2.0.0",
|
|
"--state-dir", stateDir
|
|
]);
|
|
assert.equal(dryRun.serviceId, "hwlab-agent-worker");
|
|
assert.equal(dryRun.dryRun, true);
|
|
assert.equal(dryRun.task.status, "completed");
|
|
assert.equal(dryRun.health.status, "ok");
|
|
assert.equal(dryRun.evidenceRecord.evidenceId, "evi_ags_worker");
|
|
assert.equal(dryRun.cleanup.completed, true);
|
|
assert.equal(dryRun.workspace.cleaned, true);
|
|
assert.equal(dryRun.workspace.existsAfterCleanup, false);
|
|
await assert.rejects(access(dryRun.workspace.path));
|
|
|
|
const evidence = await runManagerJson(["evidence", "--agent-session-id", "ags_worker", "--state-dir", stateDir]);
|
|
assert.equal(evidence.evidenceRecords.length, 1);
|
|
assert.equal(evidence.evidenceRecords[0].serviceId, "hwlab-agent-worker");
|
|
} finally {
|
|
await rm(stateDir, { recursive: true, force: true });
|
|
}
|
|
});
|
|
|
|
test("agent worker dry-run requires agent session id", async () => {
|
|
const result = await execFileAsync(bunCommand, ["run", "cmd/hwlab-agent-worker/main.ts", "dry-run"], {
|
|
cwd: process.cwd(),
|
|
maxBuffer: 1024 * 1024
|
|
}).then(
|
|
() => ({ code: 0, stderr: "" }),
|
|
(error) => ({ code: error.code, stderr: error.stderr })
|
|
);
|
|
assert.equal(result.code, 1);
|
|
const payload = JSON.parse(result.stderr);
|
|
assert.equal(payload.serviceId, "hwlab-agent-worker");
|
|
assert.match(payload.error.message, /--agent-session-id is required/u);
|
|
});
|
|
|
|
async function runManagerJson(args) {
|
|
const { stdout } = await execFileAsync(bunCommand, ["run", "cmd/hwlab-agent-mgr/main.ts", ...args], {
|
|
cwd: process.cwd(),
|
|
maxBuffer: 1024 * 1024
|
|
});
|
|
return JSON.parse(stdout);
|
|
}
|
|
|
|
async function runWorkerJson(args) {
|
|
const { stdout } = await execFileAsync(bunCommand, ["run", "cmd/hwlab-agent-worker/main.ts", ...args], {
|
|
cwd: process.cwd(),
|
|
maxBuffer: 1024 * 1024
|
|
});
|
|
return JSON.parse(stdout);
|
|
}
|