AgentHub/server/__tests__/dashboard-runs.test.js

/**
 * @file Tests for the persisted run history (`server/lib/dashboard-runs.js`).
 * A run's id is its tmux pane name, which every Start in that pane reuses, so
 * a new run must ARCHIVE the previous row under its own id instead of
 * overwriting it; and a row is flagged dead only when its tmux session is
 * really gone — panes outlive a server restart.
 * @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 ROOT = fs.mkdtempSync(path.join(os.tmpdir(), "dashboard-runs-"));
process.env.DASHBOARD_DB_PATH = path.join(ROOT, "t.db");

let runs;
before(() => {
  runs = require("../lib/dashboard-runs");
});
after(() => fs.rmSync(ROOT, { recursive: true, force: true }));

const start = (id, extra = {}) =>
  runs.recordRun({ id, mode: "interactive", cwd: "/r", status: "running", laneId: 4, ...extra });

describe("dashboard-runs", () => {
  it("keeps the previous run of a pane as history when the pane starts again", () => {
    start("agenthub-lane-4", { startedAt: Date.parse("2026-10-01T10:00:00Z"), sessionId: "s-old" });
    start("agenthub-lane-4", { startedAt: Date.parse("2026-10-02T10:00:00Z"), sessionId: "s-new" });
    const current = runs.getRun("agenthub-lane-4");
    assert.equal(current.session_id, "s-new");
    const history = runs.listRuns({ laneId: 4 });
    assert.equal(history.length, 2);
    const old = history.find((r) => r.session_id === "s-old");
    assert.notEqual(old.id, "agenthub-lane-4");
    assert.ok(old.id.startsWith("agenthub-lane-4@"), old.id);
    // A pane that is being replaced is not running any more.
    assert.equal(old.status, "abandoned");
    assert.ok(old.ended_at);
  });

  it("narrows a lane's history to one tool", () => {
    start("agenthub-lane-8", { laneId: 8, tool: "claude" });
    start("agenthub-lane-8-1", { laneId: 8, tool: "shell" });
    const shells = runs.listRuns({ laneId: 8, tool: "shell" });
    assert.deepEqual(
      shells.map((r) => r.id),
      ["agenthub-lane-8-1"]
    );
    assert.equal(runs.listRuns({ laneId: 8 }).length, 2);
  });

  it("forgets a closed Terminal tab and nothing of another tool", () => {
    start("agenthub-lane-9", { laneId: 9, tool: "claude" });
    start("agenthub-lane-9-1", { laneId: 9, tool: "shell" });
    assert.equal(runs.forgetShellRun("agenthub-lane-9"), false);
    assert.ok(runs.getRun("agenthub-lane-9"));
    assert.equal(runs.forgetShellRun("agenthub-lane-9-1"), true);
    assert.equal(runs.getRun("agenthub-lane-9-1"), null);
  });

  it("drops every closed Terminal tab of a lane, live ones untouched", () => {
    for (let i = 1; i <= 5; i++) {
      start(`agenthub-lane-11-${i}`, {
        laneId: 11,
        tool: "shell",
        startedAt: Date.parse(`2026-10-0${i}T10:00:00Z`),
      });
    }
    start("agenthub-lane-11", { laneId: 11, tool: "claude", startedAt: Date.parse("2026-09-01") });
    // Pane 5 still runs; every other one has closed.
    runs.reconcileOrphans((id) => id === "agenthub-lane-11-5");
    const shells = runs.listRuns({ laneId: 11, tool: "shell" }).map((r) => r.id);
    assert.deepEqual(shells, ["agenthub-lane-11-5"]);
    assert.ok(runs.getRun("agenthub-lane-11"), "another tool's row stays");
  });

  it("flags as abandoned only the running rows whose pane is gone", () => {
    start("agenthub-lane-5", { laneId: 5 });
    start("agenthub-lane-6", { laneId: 6 });
    const n = runs.reconcileOrphans((id) => id !== "agenthub-lane-6");
    assert.equal(n, 1);
    assert.equal(runs.getRun("agenthub-lane-5").status, "running");
    assert.equal(runs.getRun("agenthub-lane-6").status, "abandoned");
  });

  it("gives a hookless tool's session the lane of the run it started in", () => {
    const { db } = require("../db");
    start("agenthub-lane-7-1", {
      laneId: 7,
      cwd: "/w",
      tool: "antigravity",
      startedAt: Date.parse("2026-10-05T08:49:57.145Z"),
    });
    runs.patchRun({
      id: "agenthub-lane-7-1",
      status: "killed",
      endedAt: Date.parse("2026-10-05T08:50:10.740Z"),
    });
    const add = db.prepare(
      "INSERT INTO sessions (id, name, status, cwd, started_at, tool) VALUES (?, 'x', 'completed', ?, ?, ?)"
    );
    add.run("in", "/w", "2026-10-05T08:50:01Z", "antigravity");
    add.run("before", "/w", "2026-10-05T08:40:00Z", "antigravity");
    add.run("after", "/w", "2026-10-05T09:00:00Z", "antigravity");
    add.run("elsewhere", "/x", "2026-10-05T08:50:01Z", "antigravity");
    add.run("other-tool", "/w", "2026-10-05T08:50:01Z", "codex");
    add.run("claude", "/w", "2026-10-05T08:50:01Z", "claude");
    runs.claimToolSessions();
    const lane = (id) => db.prepare("SELECT lane_id FROM sessions WHERE id = ?").get(id).lane_id;
    assert.equal(lane("in"), 7);
    for (const id of ["before", "after", "elsewhere", "other-tool", "claude"]) {
      assert.equal(lane(id), null, id);
    }
    // One session at a time also becomes its run's session.
    assert.equal(runs.getRun("agenthub-lane-7-1").session_id, null);
    runs.claimToolSessions("in");
    assert.equal(runs.getRun("agenthub-lane-7-1").session_id, "in");
  });
});