Files
AgentCompany/tests/unit/agent-hire-routing.test.ts
T

99 lines
3.0 KiB
TypeScript

import { mkdtemp, rm } from "node:fs/promises";
import { join } from "node:path";
import { tmpdir } from "node:os";
import { afterEach, describe, expect, it } from "vitest";
import { AppDatabase } from "../../src/main/database";
import { getHireRoutingValidationError, resolveHireReportsTo } from "../../src/main/agent-permissions";
const tempDirs: string[] = [];
async function createTempDir(prefix: string) {
const dir = await mkdtemp(join(tmpdir(), prefix));
tempDirs.push(dir);
return dir;
}
afterEach(async () => {
await Promise.all(tempDirs.splice(0).map((dir) => rm(dir, { recursive: true, force: true })));
});
describe("hire routing guards", () => {
it("defaults same-department hires to the requester and infers department leads for cross-department hires", async () => {
const profileDir = await createTempDir("agentcompany-hire-manager-");
const db = new AppDatabase(profileDir);
db.init();
const companyId = db.core.company.saveCompany({
name: "Routing Co",
description: "Hire routing regression coverage",
status: "active",
});
db.core.connectors.updateConnectorHealth({
id: "openai-codex",
status: "ready",
authState: "authenticated",
version: "test",
lastError: null,
});
const workspaceId = db.core.workspaces.saveWorkspace({
companyId,
name: "Primary",
localPath: profileDir,
repoUrl: "",
repoRef: "",
isPrimary: true,
});
const ceoId = db.core.agents.saveAgent({
companyId,
name: "CEO",
role: "CEO",
title: "Chief Executive Officer",
status: "active",
reportsTo: null,
connectorId: "openai-codex",
workspaceId,
model: null,
capabilities: "Lead the company",
budgetMonthlyUsd: 250,
department: "executive",
});
const hrId = db.core.agents.saveAgent({
companyId,
name: "HR Lead",
role: "HR Lead",
title: "Head of HR",
status: "active",
reportsTo: ceoId,
connectorId: "openai-codex",
workspaceId,
model: null,
capabilities: "Run hiring operations",
budgetMonthlyUsd: 180,
department: "hr",
});
const ctoId = db.core.agents.saveAgent({
companyId,
name: "CTO",
role: "CTO",
title: "Chief Technology Officer",
status: "active",
reportsTo: ceoId,
connectorId: "openai-codex",
workspaceId,
model: null,
capabilities: "Lead engineering",
budgetMonthlyUsd: 220,
department: "engineering",
});
expect(resolveHireReportsTo(db, companyId, hrId, undefined, "hr")).toBe(hrId);
expect(resolveHireReportsTo(db, companyId, hrId, undefined, "engineering")).toBe(ctoId);
expect(getHireRoutingValidationError(db, companyId, hrId, hrId, "engineering")).toBe(
"Selected manager must belong to the same department as the new hire, unless they are a top-level executive.",
);
expect(getHireRoutingValidationError(db, companyId, ctoId, ctoId, "engineering")).toBeNull();
db.close();
});
});