import { describe, expect, it } from "vitest";
import { PUBLIC_COMMAND_NAMES } from "../src/cli/command-registry.js";
import {
classifyOwnedSessionWorkerInvocation,
createOwnedWorkerLaunchSpec,
createRpcRecoveryArgs,
isOwnedSessionWorkerProcess,
} from "../src/cli/owned-session-worker.js";
describe("owned session worker CLI routing", () => {
it("leaves resident interactive and non-session operations in the frontend", () => {
expect(classifyOwnedSessionWorkerInvocation([], true, {})).toBeUndefined();
expect(classifyOwnedSessionWorkerInvocation(["--mode", "daemon"], false, {})).toBeUndefined();
expect(classifyOwnedSessionWorkerInvocation(["--help"], false, {})).toBeUndefined();
expect(classifyOwnedSessionWorkerInvocation(["--version"], false, {})).toBeUndefined();
expect(classifyOwnedSessionWorkerInvocation(["--list-models"], false, {})).toBeUndefined();
expect(classifyOwnedSessionWorkerInvocation(["--export", "session.jsonl"], false, {})).toBeUndefined();
expect(classifyOwnedSessionWorkerInvocation(["help"], false, {})).toBeUndefined();
expect(classifyOwnedSessionWorkerInvocation(["help", "me", "fix", "this"], false, {})).toBe("print");
expect(classifyOwnedSessionWorkerInvocation(["daemon", "list"], false, {})).toBeUndefined();
for (const command of PUBLIC_COMMAND_NAMES) {
expect(classifyOwnedSessionWorkerInvocation([command], false, {})).toBeUndefined();
}
});
it("does not recursively route an owned worker", () => {
expect(isOwnedSessionWorkerProcess({ PRIME_AGENT_INTERNAL_OWNED_WORKER: "1" })).toBe(true);
expect(isOwnedSessionWorkerProcess({})).toBe(false);
expect(
classifyOwnedSessionWorkerInvocation(["--mode", "rpc"], true, {
PRIME_AGENT_INTERNAL_OWNED_WORKER: "1",
}),
).toBeUndefined();
});
it("preserves the current runtime flags and CLI entrypoint", () => {
expect(createOwnedWorkerLaunchSpec(["--mode", "rpc"], "/node", ["--loader", "tsx"], "/cli.ts")).toEqual({
command: "/node",
args: ["--loader", "tsx", "/cli.ts", "--mode", "rpc"],
});
});
it("restarts RPC against the exact persisted session without changing public flags", () => {
expect(
createRpcRecoveryArgs(
["--mode", "rpc", "--continue", "--resume", "/old.jsonl", "--model", "openai/gpt-5"],
"/current.jsonl",
),
).toEqual(["--mode", "rpc", "--model", "openai/gpt-5", "--resume", "/current.jsonl"]);
});
});