54e680f284
One owner for "may heavy GPU work start": numbers-only samples (full-screen
state, sustained 3D use, host free VRAM) in, { inUse, since, checkedAt,
probeOk } out; fail closed at start and when the probe goes stale.
Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
65 lines
2.9 KiB
JavaScript
65 lines
2.9 KiB
JavaScript
// Boots the real daemon and observes the gate where it is decided: the HTTP edge. A fresh daemon is
|
|
// in use until a clear hold passes, so a new job is refused whatever the host is doing. Needs the
|
|
// daemon's own dependencies (ws, cultcache-ts), so it runs on a machine that has them.
|
|
import assert from "node:assert/strict";
|
|
import { spawn } from "node:child_process";
|
|
import { mkdtemp, readdir, rm } from "node:fs/promises";
|
|
import os from "node:os";
|
|
import path from "node:path";
|
|
import test from "node:test";
|
|
import { fileURLToPath } from "node:url";
|
|
|
|
const daemonPath = path.join(path.dirname(fileURLToPath(import.meta.url)), "vili-daemon.mjs");
|
|
const port = 18000 + Math.floor(Math.random() * 1000);
|
|
|
|
async function bootDaemon(stateRoot) {
|
|
const child = spawn(process.execPath, [daemonPath, "--host", "127.0.0.1", "--port", String(port), "--state-root", stateRoot], {
|
|
stdio: ["ignore", "pipe", "pipe"],
|
|
windowsHide: true,
|
|
});
|
|
let output = "";
|
|
child.stdout.on("data", (chunk) => { output += chunk; });
|
|
child.stderr.on("data", (chunk) => { output += chunk; });
|
|
const deadline = Date.now() + 30000;
|
|
while (!output.includes("Vili listening")) {
|
|
if (child.exitCode !== null || Date.now() > deadline) throw new Error(`daemon did not start: ${output.slice(-500)}`);
|
|
await new Promise((resolve) => setTimeout(resolve, 100));
|
|
}
|
|
return child;
|
|
}
|
|
|
|
const post = (body) => fetch(`http://127.0.0.1:${port}/motion/generate`, {
|
|
method: "POST",
|
|
headers: { "content-type": "application/json" },
|
|
body: JSON.stringify(body),
|
|
});
|
|
|
|
test("a new job is refused 503 gpu-busy with retry-after, creating nothing; validation stays 400", async () => {
|
|
const stateRoot = await mkdtemp(path.join(os.tmpdir(), "vili-gate-"));
|
|
const daemon = await bootDaemon(stateRoot);
|
|
try {
|
|
const invalid = await post({});
|
|
assert.equal(invalid.status, 400, "request-shape errors come ahead of the gate");
|
|
|
|
const refused = await post({ prompt: "a person waves hello, never seen before" });
|
|
assert.equal(refused.status, 503);
|
|
assert.equal(refused.headers.get("retry-after"), "60");
|
|
const body = await refused.json();
|
|
assert.equal(body.ok, false);
|
|
assert.equal(body.code, "gpu-busy");
|
|
assert.equal(body.source, "admission");
|
|
assert.equal(body.gpu.inUse, true);
|
|
assert.deepEqual(Object.keys(body.gpu).sort(), ["checkedAt", "inUse", "probeOk", "since"]);
|
|
assert.equal(body.jobId, undefined);
|
|
|
|
assert.deepEqual(await readdir(path.join(stateRoot, "jobs")), [], "no job file");
|
|
assert.equal((await readdir(stateRoot)).includes("artifacts"), false, "no artifact directory");
|
|
|
|
const state = await (await fetch(`http://127.0.0.1:${port}/operator-state`)).json();
|
|
assert.deepEqual(Object.keys(state.gpu).sort(), ["checkedAt", "inUse", "probeOk", "since"]);
|
|
assert.equal(state.gpu.inUse, true);
|
|
} finally {
|
|
daemon.kill();
|
|
await rm(stateRoot, { recursive: true, force: true });
|
|
}
|
|
});
|