co-mono/packages/pi-teams/src/adapters/wezterm-adapter.test.ts

/**
 * WezTerm Adapter Tests
 */

import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
import * as terminalAdapter from "../utils/terminal-adapter";
import { WezTermAdapter } from "./wezterm-adapter";

describe("WezTermAdapter", () => {
	let adapter: WezTermAdapter;
	let mockExecCommand: ReturnType<typeof vi.spyOn>;

	beforeEach(() => {
		adapter = new WezTermAdapter();
		mockExecCommand = vi.spyOn(terminalAdapter, "execCommand");
		delete process.env.WEZTERM_PANE;
		delete process.env.TMUX;
		delete process.env.ZELLIJ;
		process.env.WEZTERM_PANE = "0";
	});

	afterEach(() => {
		vi.clearAllMocks();
	});

	describe("name", () => {
		it("should have the correct name", () => {
			expect(adapter.name).toBe("WezTerm");
		});
	});

	describe("detect", () => {
		it("should detect when WEZTERM_PANE is set", () => {
			mockExecCommand.mockReturnValue({ stdout: "version 1.0", stderr: "", status: 0 });
			expect(adapter.detect()).toBe(true);
		});
	});

	describe("spawn", () => {
		it("should spawn first pane to the right with 50%", () => {
			// Mock getPanes finding only current pane
			mockExecCommand.mockImplementation((_bin: string, args: string[]) => {
				if (args.includes("list")) {
					return {
						stdout: JSON.stringify([{ pane_id: 0, tab_id: 0 }]),
						stderr: "",
						status: 0,
					};
				}
				if (args.includes("split-pane")) {
					return { stdout: "1", stderr: "", status: 0 };
				}
				return { stdout: "", stderr: "", status: 0 };
			});

			const result = adapter.spawn({
				name: "test-agent",
				cwd: "/home/user/project",
				command: "pi --agent test",
				env: { PI_AGENT_ID: "test-123" },
			});

			expect(result).toBe("wezterm_1");
			expect(mockExecCommand).toHaveBeenCalledWith(
				expect.stringContaining("wezterm"),
				expect.arrayContaining(["cli", "split-pane", "--right", "--percent", "50"]),
			);
		});

		it("should spawn subsequent panes by splitting the sidebar", () => {
			// Mock getPanes finding current pane (0) and sidebar pane (1)
			mockExecCommand.mockImplementation((_bin: string, args: string[]) => {
				if (args.includes("list")) {
					return {
						stdout: JSON.stringify([
							{ pane_id: 0, tab_id: 0 },
							{ pane_id: 1, tab_id: 0 },
						]),
						stderr: "",
						status: 0,
					};
				}
				if (args.includes("split-pane")) {
					return { stdout: "2", stderr: "", status: 0 };
				}
				return { stdout: "", stderr: "", status: 0 };
			});

			const result = adapter.spawn({
				name: "agent2",
				cwd: "/home/user/project",
				command: "pi",
				env: {},
			});

			expect(result).toBe("wezterm_2");
			// 1 sidebar pane already exists, so percent should be floor(100/(1+1)) = 50%
			expect(mockExecCommand).toHaveBeenCalledWith(
				expect.stringContaining("wezterm"),
				expect.arrayContaining(["cli", "split-pane", "--bottom", "--pane-id", "1", "--percent", "50"]),
			);
		});
	});
});