AgentHub/server/__tests__/run-session-bind.test.js

/**
 * @file Tests that a dashboard pane learns which session it holds: every pane
 * is started with `AGENTHUB_RUN_ID` in its tmux env, and a hook carrying that
 * `run_id` binds the hook's session to the run (`dashboard_runs.session_id`),
 * following `/clear` to a new session; a `run_id` that is not a dashboard pane
 * name, or names no run, changes nothing.
 * @author Nguyễn Ngọc Trí Vĩ <vinnt@smartgift.vn>
 */

const { describe, it, before, after } = require("node:test");
const assert = require("node:assert/strict");
const fs = require("node:fs");
const os = require("node:os");
const path = require("node:path");
const http = require("node:http");

const ROOT = fs.realpathSync(fs.mkdtempSync(path.join(os.tmpdir(), "run-bind-")));
process.env.DASHBOARD_DB_PATH = path.join(ROOT, "t.db");
process.env.CLAUDE_HOME = path.join(ROOT, "claude");
process.env.DASHBOARD_DATA_DIR = path.join(ROOT, "data");
process.env.DASHBOARD_REMOTE_SYNC_MS = "0";
process.env.DASHBOARD_LIVENESS_PROBE = "0";

const { createApp } = require("../index");
const dashboardRuns = require("../lib/dashboard-runs");
const tmux = require("../lib/tmux");
const pty = require("../lib/pty-run");

let server;
let base;

function postHook(body) {
  return new Promise((resolve, reject) => {
    const data = Buffer.from(JSON.stringify(body));
    const req = http.request(
      `${base}/api/hooks/event`,
      { method: "POST", headers: { "content-type": "application/json" } },
      (res) => {
        res.resume();
        res.on("end", () => resolve(res.statusCode));
      }
    );
    req.on("error", reject);
    req.end(data);
  });
}

before(async () => {
  fs.mkdirSync(process.env.CLAUDE_HOME, { recursive: true });
  server = http.createServer(createApp());
  await new Promise((r) => server.listen(0, "127.0.0.1", r));
  base = `http://127.0.0.1:${server.address().port}`;
});

after(() => {
  server?.close();
  fs.rmSync(ROOT, { recursive: true, force: true });
});

describe("run ↔ session binding", () => {
  it("starts every pane with AGENTHUB_RUN_ID naming that pane", () => {
    let argv = null;
    tmux.__setExecImpl((args) => {
      if (args[0] === "has-session") throw Object.assign(new Error("gone"), { status: 1 });
      if (args[0] === "new-session") argv = args;
      return "";
    });
    pty.spawnRun({ laneId: 77, slot: 2, cwd: ROOT });
    tmux.__reset();
    const env = argv.filter((a, i) => argv[i - 1] === "-e");
    assert.ok(env.includes("AGENTHUB_RUN_ID=agenthub-lane-77-2"), env.join(" "));
  });

  it("binds a hook's session to the run it names, and follows a new session", async () => {
    dashboardRuns.recordRun({
      id: "agenthub-lane-5-1",
      mode: "interactive",
      cwd: ROOT,
      status: "running",
      startedAt: Date.now(),
    });
    const hook = (session_id, run_id) =>
      postHook({ hook_type: "SessionStart", data: { session_id, cwd: ROOT }, run_id });
    assert.equal(await hook("sess-aaaa-1111", "agenthub-lane-5-1"), 200);
    assert.equal(dashboardRuns.getRun("agenthub-lane-5-1").session_id, "sess-aaaa-1111");
    // `/clear` in the pane: a new session that is the pane's now.
    await postHook({
      hook_type: "SessionStart",
      data: { session_id: "sess-bbbb-2222", cwd: ROOT, source: "clear" },
      run_id: "agenthub-lane-5-1",
    });
    assert.equal(dashboardRuns.getRun("agenthub-lane-5-1").session_id, "sess-bbbb-2222");
  });

  it("keeps the pane's session when a nested CLI inside it starts its own", async () => {
    // A `claude -p` the agent runs through Bash inherits AGENTHUB_RUN_ID.
    for (const hook_type of ["SessionStart", "UserPromptSubmit", "Stop"]) {
      await postHook({
        hook_type,
        data: { session_id: "sess-nest-9999", cwd: ROOT, source: "startup" },
        run_id: "agenthub-lane-5-1",
      });
    }
    assert.equal(dashboardRuns.getRun("agenthub-lane-5-1").session_id, "sess-bbbb-2222");
  });

  it("follows any new session on a non-claude pane, whose hooks carry no reliable source", async () => {
    dashboardRuns.recordRun({
      id: "agenthub-lane-6-1",
      mode: "interactive",
      cwd: ROOT,
      status: "running",
      startedAt: Date.now(),
    });
    const hook = (session_id) =>
      postHook({
        hook_type: "UserPromptSubmit",
        tool: "codex",
        data: { session_id, cwd: ROOT, prompt: "hi" },
        run_id: "agenthub-lane-6-1",
      });
    await hook("codex-thread-aaaa");
    assert.equal(dashboardRuns.getRun("agenthub-lane-6-1").session_id, "codex-thread-aaaa");
    await hook("codex-thread-bbbb");
    assert.equal(dashboardRuns.getRun("agenthub-lane-6-1").session_id, "codex-thread-bbbb");
  });

  it("ignores a run_id that is not a dashboard pane, or names no run", async () => {
    await postHook({
      hook_type: "SessionStart",
      data: { session_id: "sess-cccc-3333", cwd: ROOT },
      run_id: "main",
    });
    await postHook({
      hook_type: "SessionStart",
      data: { session_id: "sess-cccc-3333", cwd: ROOT },
      run_id: "agenthub-lane-999",
    });
    assert.equal(dashboardRuns.getRun("main"), null);
    assert.equal(dashboardRuns.getRun("agenthub-lane-999"), null);
    assert.equal(dashboardRuns.getRun("agenthub-lane-5-1").session_id, "sess-bbbb-2222");
  });
});