clanker-agent/packages/pi-teams/src/adapters/tmux-adapter.test.ts

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

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

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

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

  it("detects tmux in headless runtimes when the binary is available", () => {
    mockExecCommand.mockReturnValue({
      stdout: "tmux 3.4",
      stderr: "",
      status: 0,
    });

    expect(adapter.detect()).toBe(true);
    expect(mockExecCommand).toHaveBeenCalledWith("tmux", ["-V"]);
  });

  it("does not detect tmux in GUI terminals just because the binary exists", () => {
    process.env.COLORTERM = "truecolor";
    mockExecCommand.mockReturnValue({
      stdout: "tmux 3.4",
      stderr: "",
      status: 0,
    });

    expect(adapter.detect()).toBe(false);
    expect(mockExecCommand).not.toHaveBeenCalled();
  });

  it("creates a detached team session when not already inside tmux", () => {
    mockExecCommand.mockImplementation((_bin: string, args: string[]) => {
      if (args[0] === "has-session") {
        return { stdout: "", stderr: "missing", status: 1 };
      }
      if (args[0] === "new-session") {
        return { stdout: "%1\n", stderr: "", status: 0 };
      }
      return { stdout: "", stderr: "", status: 0 };
    });

    expect(
      adapter.spawn({
        name: "worker",
        cwd: "/tmp/project",
        command: "pi",
        env: { PI_TEAM_NAME: "demo", PI_AGENT_NAME: "worker" },
      }),
    ).toBe("%1");

    expect(mockExecCommand).toHaveBeenCalledWith(
      "tmux",
      expect.arrayContaining(["new-session", "-d", "-s", "pi-teams-demo"]),
    );
  });

  it("splits an existing detached session when not already inside tmux", () => {
    mockExecCommand.mockImplementation((_bin: string, args: string[]) => {
      if (args[0] === "has-session") {
        return { stdout: "", stderr: "", status: 0 };
      }
      if (args[0] === "split-window") {
        return { stdout: "%2\n", stderr: "", status: 0 };
      }
      return { stdout: "", stderr: "", status: 0 };
    });

    expect(
      adapter.spawn({
        name: "worker",
        cwd: "/tmp/project",
        command: "pi",
        env: { PI_TEAM_NAME: "demo", PI_AGENT_NAME: "worker" },
      }),
    ).toBe("%2");

    expect(mockExecCommand).toHaveBeenCalledWith(
      "tmux",
      expect.arrayContaining(["split-window", "-t", "pi-teams-demo:0"]),
    );
  });

  it("checks pane liveness by pane id", () => {
    mockExecCommand.mockReturnValue({
      stdout: "%1\n%7\n",
      stderr: "",
      status: 0,
    });

    expect(adapter.isAlive("%7")).toBe(true);
    expect(mockExecCommand).toHaveBeenCalledWith("tmux", [
      "list-panes",
      "-a",
      "-F",
      "#{pane_id}",
    ]);
  });
});