Q64: agent edits meet the same department limits as hiring

Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>

tests/unit/staffing-limits.test.ts

@@ -0,0 +1,36 @@
+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();
+  });
+});