pi-mono/packages/coding-agent/test/experimental-cli-resolution.test.ts

import { describe, expect, test, vi } from "vitest";
import { cli } from "../src/cli/experimental/cli.ts";

const UNSUPPORTED_SERVER_OPTIONS = "The experimental server command does not support existing CLI options yet";
const UNSUPPORTED_CLIENT_OPTIONS = "The experimental client command does not support existing CLI options yet";

describe("experimental CLI command composition", () => {
	test("requires an experimental subcommand", () => {
		expect(cli.parse([])).toEqual({
			ok: false,
			errors: ["Expected experimental command: server or client"],
		});
	});

	test.each([
		["server", "--help", UNSUPPORTED_SERVER_OPTIONS],
		["server", "--version", UNSUPPORTED_SERVER_OPTIONS],
		["client", "--help", UNSUPPORTED_CLIENT_OPTIONS],
		["client", "--version", UNSUPPORTED_CLIENT_OPTIONS],
	] as const)("rejects unsupported %s %s handling", (command, option, error) => {
		expect(cli.parse([command, option])).toEqual({ ok: false, errors: [error] });
	});

	test("passes server options to the command action", async () => {
		const runServer = vi.fn(() => undefined);
		const result = await cli.execute(
			[
				"server",
				"--server-id",
				"00000000-0000-4000-8000-000000000001",
				"--session-dir",
				"./sessions",
				"--provider",
				"anthropic",
				"--model",
				"claude-sonnet-4-5",
			],
			{ runServer, runClient: vi.fn(() => undefined) },
		);

		const command = {
			command: "server" as const,
			serverId: "00000000-0000-4000-8000-000000000001",
			sessionDir: "./sessions",
			provider: "anthropic",
			model: "claude-sonnet-4-5",
		};
		expect(result).toEqual({ ok: true, command });
		expect(runServer).toHaveBeenCalledWith(command);
	});

	test.each(["server", "client"] as const)("executes the parsed %s command", async (name) => {
		const context = {
			runServer: vi.fn(() => undefined),
			runClient: vi.fn(() => undefined),
		};
		const result = await cli.execute([name], context);

		expect(result).toEqual({ ok: true, command: { command: name } });
		expect(context.runServer).toHaveBeenCalledTimes(name === "server" ? 1 : 0);
		expect(context.runClient).toHaveBeenCalledTimes(name === "client" ? 1 : 0);
	});
});