ZacharyZhang-NYandClaude Opus 5.5
1367a8900e
Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
37 lines
2.3 KiB
TypeScript
37 lines
2.3 KiB
TypeScript
import { randomUUID } from "node:crypto";
|
|
import { tmpdir } from "node:os";
|
|
import { join } from "node:path";
|
|
import { describe, expect, test } from "vitest";
|
|
import { DatabaseConnection } from "../../src/main/database/connection";
|
|
import { OrganizationService } from "../../src/main/services/organization-service";
|
|
|
|
describe("staffing limits on edits", () => {
|
|
test("moving an agent or raising its budget obeys the same department limits as hiring", () => {
|
|
const connection = new DatabaseConnection(join(tmpdir(), `ac-staffing-${randomUUID()}.db`));
|
|
const organization = new OrganizationService(connection.db);
|
|
let orgId = "";
|
|
connection.transaction(() => {
|
|
orgId = organization.createOrganization({ organizationName: "Acme", employeeName: "Dana", mission: "m", goal: "g" });
|
|
});
|
|
connection.db.prepare("update hiring_policies set max_headcount = 1, monthly_budget_limit_cents = 5000 where org_id = ? and department_name = 'Design'").run(orgId);
|
|
connection.db.prepare("update hiring_policies set monthly_budget_limit_cents = 3000 where org_id = ? and department_name = 'Sales'").run(orgId);
|
|
const hire = (name: string, department: string, budgetCents: number) => organization.createAgent({
|
|
orgId, name, role: "Staff", department, modelProviderId: null, modelId: null, budgetCents, managerAgentId: null,
|
|
});
|
|
hire("Dee", "Design", 1000);
|
|
const sam = hire("Sam", "Sales", 1000);
|
|
const edit = (department: string, budgetCents: number) => organization.updateAgent({
|
|
orgId, agentId: sam, role: "Staff", title: "Staff", roleLevel: "staff", department, budgetCents, modelProviderId: null, modelId: null,
|
|
});
|
|
|
|
expect(() => edit("Design", 1000)).toThrow("Design is at its headcount limit (1).");
|
|
expect(() => edit("Sales", 3500)).toThrow("Sales would budget $35.00 a month, over its $30.00 limit.");
|
|
edit("Sales", 3000);
|
|
// A department already over its limit still allows edits that do not add to it.
|
|
connection.db.prepare("update hiring_policies set monthly_budget_limit_cents = 100 where org_id = ? and department_name = 'Sales'").run(orgId);
|
|
edit("Sales", 2000);
|
|
expect(connection.db.prepare("select department, budget_cents from agents where id = ?").get(sam)).toEqual({ department: "Sales", budget_cents: 2000 });
|
|
connection.close();
|
|
});
|
|
});
|