AgentHub/server/__tests__/lane-profile.test.js

/**
 * @file Tests for profile parsing and hook environment setup.
 * @author Nguyễn Ngọc Trí Vĩ <vinnt@smartgift.vn>
 */

const os = require("node:os");
const pathMod = require("node:path");
const fsMod = require("node:fs");

const SUITE_ROOT = fsMod.mkdtempSync(pathMod.join(os.tmpdir(), "agenthub-profile-"));
process.env.DASHBOARD_DB_PATH = pathMod.join(SUITE_ROOT, "dashboard.db");
process.env.LANES_ROOT = pathMod.join(SUITE_ROOT, "lanes");
process.env.AGENTHUB_SECRETS_PATH = pathMod.join(SUITE_ROOT, "secrets.env");

const { describe, it, after } = require("node:test");
const assert = require("node:assert/strict");

const lanesLib = require("../lib/lanes");
const slots = require("../lib/lane-slots");
const profileLib = require("../lib/lane-profile");

after(() => fsMod.rmSync(SUITE_ROOT, { recursive: true, force: true }));

let laneSeq = 0;
/** A managed lane row backed by a real directory. */
function makeLane(over = {}) {
  laneSeq += 1;
  const cwd = pathMod.join(SUITE_ROOT, `profile-lane-cwd-${laneSeq}`);
  fsMod.mkdirSync(cwd, { recursive: true });
  return lanesLib.createLane({
    title: `profile-lane ${laneSeq}`,
    cwd,
    kind: "managed",
    source_repo: cwd,
    ...over,
  });
}

/** Write a profile into a repo directory. `hooks` maps name -> script body. */
function writeProfile(root, envText, hooks = {}) {
  const dir = pathMod.join(root, ".agenthub", "profile");
  fsMod.mkdirSync(pathMod.join(dir, "hooks"), { recursive: true });
  fsMod.writeFileSync(pathMod.join(dir, "profile.env"), envText);
  for (const [name, body] of Object.entries(hooks)) {
    const file = pathMod.join(dir, "hooks", `${name}.sh`);
    fsMod.writeFileSync(file, body);
    fsMod.chmodSync(file, 0o755);
  }
  return dir;
}

describe("parseQcBootEnv", () => {
  it("parses space-separated KEY=value pairs", () => {
    assert.deepEqual(profileLib.parseQcBootEnv("MOCK_PAYMENTS=1 STUB_EMAIL=1"), {
      MOCK_PAYMENTS: "1",
      STUB_EMAIL: "1",
    });
  });

  it("returns an empty object for an empty or missing declaration", () => {
    assert.deepEqual(profileLib.parseQcBootEnv(""), {});
    assert.deepEqual(profileLib.parseQcBootEnv(undefined), {});
  });

  it("ignores a malformed token with no =", () => {
    assert.deepEqual(profileLib.parseQcBootEnv("GOOD=1 malformed"), { GOOD: "1" });
  });
});

describe("hookEnv extraEnv", () => {
  it("merges extraEnv on top of everything else, including profile.env", () => {
    const lane = makeLane();
    writeProfile(lane.cwd, "PORTS=api\n");
    slots.allocateSlot(lane.id);
    const current = lanesLib.getLane(lane.id);
    const profile = profileLib.resolveProfile(current);
    const env = profileLib.hookEnv(current, profile, { PORTS: "overridden" });
    assert.equal(env.PORTS, "overridden");
    slots.releaseSlot(lane.id);
  });

  it("defaults extraEnv to nothing when omitted (existing callers unaffected)", () => {
    const lane = makeLane();
    writeProfile(lane.cwd, "PORTS=api\n");
    slots.allocateSlot(lane.id);
    const current = lanesLib.getLane(lane.id);
    const profile = profileLib.resolveProfile(current);
    const env = profileLib.hookEnv(current, profile);
    assert.equal(env.LANE_ID, String(current.id));
    slots.releaseSlot(lane.id);
  });
});

describe("resolveProfile — E2 declarations", () => {
  it("defaults MIGRATIONS_DIR to empty and GENERATED_MERGE_PATHS to an empty array", () => {
    const lane = makeLane();
    writeProfile(lane.cwd, "PORTS=api\n");
    const profile = profileLib.resolveProfile(lanesLib.getLane(lane.id));
    assert.equal(profile.env.MIGRATIONS_DIR, "");
    assert.deepEqual(profile.generatedMergePaths, []);
  });

  it("parses declared MIGRATIONS_DIR and GENERATED_MERGE_PATHS", () => {
    const lane = makeLane();
    writeProfile(
      lane.cwd,
      'PORTS=api\nMIGRATIONS_DIR="db/migrations"\nGENERATED_MERGE_PATHS="api/openapi.json api/client.ts"\n'
    );
    const profile = profileLib.resolveProfile(lanesLib.getLane(lane.id));
    assert.equal(profile.env.MIGRATIONS_DIR, "db/migrations");
    assert.deepEqual(profile.generatedMergePaths, ["api/openapi.json", "api/client.ts"]);
  });
});

describe("isIntegrationEnabled", () => {
  it("returns false when integrations.env doesn't exist at all", () => {
    const lane = makeLane();
    writeProfile(lane.cwd, "PORTS=api\n");
    assert.equal(profileLib.isIntegrationEnabled(lanesLib.getLane(lane.id), "tracker"), false);
  });

  it("returns true when <NAME>_ENABLED=1 is set", () => {
    const lane = makeLane();
    writeProfile(lane.cwd, "PORTS=api\n");
    fsMod.writeFileSync(
      pathMod.join(lane.cwd, ".agenthub", "profile", "integrations.env"),
      "TRACKER_ENABLED=1\nTRACKER_PROJECT=demo\n"
    );
    assert.equal(profileLib.isIntegrationEnabled(lanesLib.getLane(lane.id), "tracker"), true);
  });

  it("returns false when the flag is 0 or absent from an existing file", () => {
    const lane = makeLane();
    writeProfile(lane.cwd, "PORTS=api\n");
    fsMod.writeFileSync(
      pathMod.join(lane.cwd, ".agenthub", "profile", "integrations.env"),
      "DEV_QC_ENABLED=0\n"
    );
    assert.equal(profileLib.isIntegrationEnabled(lanesLib.getLane(lane.id), "dev_qc"), false);
    assert.equal(profileLib.isIntegrationEnabled(lanesLib.getLane(lane.id), "ci_wait"), false);
  });
});