Files
ThothII/backend/test/pi-process-manager.test.ts
T
marcopanandClaude Sonnet 4.6 c63b2bd126 fix(backend): idempotent spawnFor + identity-checked exit + unified SSE re-emit shape
- spawnFor now tears down any existing runtime for the same session id before
  the cap check, so resume/respawn neither leaks the old child nor falsely hits
  maxPiProcesses
- exit handler is identity-checked (captures rt) so a stale child's late exit
  cannot evict a newer runtime
- SSE pending re-emit now sends the full ClientEvent shape
  { type: "ui_request", ui_request } to match hub.publish live events
- tests: same-id respawn replaces runtime (count 1); old child exit does not
  evict new runtime; sse-hub re-emit asserts unified shape

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
2026-06-27 21:53:12 +02:00

71 lines
3.4 KiB
TypeScript

import { test, expect } from "vitest";
import { spawn } from "node:child_process";
import path from "node:path";
import { fileURLToPath } from "node:url";
import { PiProcessManager } from "../src/pi/pi-process-manager.js";
import { loadConfig } from "../src/config.js";
const __dirname = path.dirname(fileURLToPath(import.meta.url));
const FAKE = path.resolve(__dirname, "../../harness/tests/fake_pi/fake_pi_rpc.mjs");
const SCRIPT = path.resolve(__dirname, "../../harness/tests/fake_pi/scripts/f1_disambiguation.json");
test("spawnFor avvia un runtime e il bridge emette il widget F1", async () => {
const cfg = loadConfig({ THT_HARNESS_DIR: "../harness" });
const mgr = new PiProcessManager(cfg, { spawnFn: () => spawn("node", [FAKE, SCRIPT]) as any });
const rt = await mgr.spawnFor("2026-06-27-100000-x", {});
const widget = await new Promise<any>((res) => rt.bridge.onClientEvent((e) => e.type === "ui_request" && res(e)));
expect(widget.ui_request.widget).toBe("select");
mgr.teardown("2026-06-27-100000-x");
expect(mgr.count()).toBe(0);
});
test("l'exit del child rimuove il runtime dalla mappa (exit handler)", async () => {
const cfg = loadConfig({ THT_HARNESS_DIR: "../harness" });
const mgr = new PiProcessManager(cfg, { spawnFn: () => spawn("node", [FAKE, SCRIPT]) as any });
const rt = await mgr.spawnFor("exit-test", {});
expect(mgr.count()).toBe(1);
// Cause the child to exit on its own and await the 'exit' event (no teardown call).
const exited = new Promise<void>((res) => rt.child.on("exit", () => res()));
rt.child.kill();
await exited;
// Let the manager's registered exit handler run.
await new Promise((res) => setImmediate(res));
expect(mgr.count()).toBe(0);
expect(mgr.get("exit-test")).toBeUndefined();
});
test("spawnFor sullo STESSO id uccide il vecchio child e sostituisce il runtime (count resta 1)", async () => {
const cfg = loadConfig({ THT_HARNESS_DIR: "../harness" });
const mgr = new PiProcessManager(cfg, { spawnFn: () => spawn("node", [FAKE, SCRIPT]) as any });
const first = await mgr.spawnFor("dup-id", {});
expect(mgr.count()).toBe(1);
const firstExited = new Promise<void>((res) => first.child.on("exit", () => res()));
const second = await mgr.spawnFor("dup-id", {});
await firstExited; // the old child was killed by the idempotent respawn
expect(mgr.count()).toBe(1);
expect(mgr.get("dup-id")).toBe(second);
expect(second).not.toBe(first);
mgr.teardown("dup-id");
});
test("l'exit del VECCHIO child non elimina il nuovo runtime (exit identity-checked)", async () => {
const cfg = loadConfig({ THT_HARNESS_DIR: "../harness" });
const mgr = new PiProcessManager(cfg, { spawnFn: () => spawn("node", [FAKE, SCRIPT]) as any });
const first = await mgr.spawnFor("respawn-id", {});
const second = await mgr.spawnFor("respawn-id", {});
// The old child's exit handler fires after the respawn; it must NOT evict `second`.
await new Promise((res) => setImmediate(res));
expect(mgr.get("respawn-id")).toBe(second);
expect(mgr.count()).toBe(1);
void first;
mgr.teardown("respawn-id");
});
test("oltre maxPiProcesses solleva errore", async () => {
const cfg = { ...loadConfig({}), maxPiProcesses: 1 };
const mgr = new PiProcessManager(cfg, { spawnFn: () => spawn("node", [FAKE, SCRIPT]) as any });
await mgr.spawnFor("a", {});
await expect(mgr.spawnFor("b", {})).rejects.toThrow(/max/i);
mgr.teardown("a");
});