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();
  });
});