pi-plan/test/herdr.test.ts

import { chmodSync, mkdtempSync, rmSync, writeFileSync } from "node:fs";
import { tmpdir } from "node:os";
import { join } from "node:path";
import { afterEach, describe, expect, it, vi } from "vitest";
import { closeSupervisorPane, openSupervisorPane, supervisorCommand, waitForSupervisorReady } from "../src/herdr.js";
import { createMailbox, readyMailbox } from "../src/mailbox.js";

function input(mailboxPath = "/repo/.pi/goals-supervision/worker/approval") {
	return {
		cwd: "/repo",
		sourceSessionFile: "/sessions/worker.jsonl",
		workerSessionId: "worker-12345678",
		planPath: "/repo/.pi/plan/worker-v1.md",
		approvalId: "approval-1",
		mailboxPath,
		extensionPath: "/repo/src/index.ts",
		model: "provider/supervisor",
	};
}

afterEach(() => vi.unstubAllEnvs());

describe("supervisor pane command", () => {
	it("forks the planning session with only pi-goals and its mailbox", () => {
		const command = supervisorCommand(input());
		expect(command).toContain("'PI_GOALS_ROLE=supervisor'");
		expect(command).toContain("'PI_GOALS_MAILBOX_PATH=/repo/.pi/goals-supervision/worker/approval'");
		expect(command).toContain("'pi' '--no-extensions' '-e' '/repo/src/index.ts'");
		expect(command).toContain("'--fork' '/sessions/worker.jsonl'");
		expect(command).toContain("'--model' 'provider/supervisor'");
		expect(command).not.toContain("pi-supervise");
		expect(command).not.toContain("pi-intercom");
	});

	it("waits for an explicit mailbox readiness receipt", async () => {
		const cwd = mkdtempSync(join(tmpdir(), "pi-goals-herdr-"));
		try {
			const mailbox = createMailbox(cwd, "worker", "approval", join(cwd, "plan.md"));
			await expect(waitForSupervisorReady(mailbox.path, 10)).rejects.toThrow("did not become ready");
			readyMailbox(mailbox.path);
			await expect(waitForSupervisorReady(mailbox.path, 10)).resolves.toBeUndefined();
		} finally {
			rmSync(cwd, { recursive: true, force: true });
		}
	});

	it("accepts Herdr's text version output and stale pane cleanup", async () => {
		const cwd = mkdtempSync(join(tmpdir(), "pi-goals-herdr-"));
		const bin = join(cwd, "herdr");
		writeFileSync(bin, `#!/bin/sh
if [ "$1" = "--version" ]; then echo "herdr 0.8.2"; exit 0; fi
if [ "$1" = "pane" ] && [ "$2" = "split" ]; then echo '{"pane_id":"new-pane"}'; exit 0; fi
if [ "$1" = "pane" ] && [ "$2" = "run" ]; then if [ "$HERDR_SMOKE_RUN_FAIL" = "1" ]; then echo "run failed" >&2; exit 1; fi; echo '{}'; exit 0; fi
if [ "$1" = "pane" ] && [ "$2" = "close" ]; then echo '{"error":{"code":"PANE_GONE"}}' >&2; exit 1; fi
exit 2
`);
		chmodSync(bin, 0o755);
		vi.stubEnv("HERDR_ENV", "1");
		vi.stubEnv("HERDR_BIN_PATH", bin);
		try {
			const mailbox = createMailbox(cwd, "worker", "approval", join(cwd, "plan.md"));
			readyMailbox(mailbox.path);
			await expect(openSupervisorPane({ ...input(mailbox.path), cwd })).resolves.toBe("new-pane");
			await expect(closeSupervisorPane("new-pane")).resolves.toBeUndefined();
			vi.stubEnv("HERDR_SMOKE_RUN_FAIL", "1");
			await expect(openSupervisorPane({ ...input(mailbox.path), cwd })).rejects.toThrow("run failed");
		} finally {
			rmSync(cwd, { recursive: true, force: true });
		}
	});
});