test(api): cover assistant demand tools

2026-04-01 00:29:07 +02:00
parent 40bf22a01a
commit e1228244e9
6 changed files with 750 additions and 0 deletions
@@ -0,0 +1,126 @@
+import { beforeEach, describe, expect, it, vi } from "vitest";
+import { PermissionKey, SystemRole } from "@capakraken/shared";
+import {
+  createToolContext,
+  executeTool,
+} from "./assistant-tools-demand-create-test-helpers.js";
+
+describe("assistant demand create tool - success", () => {
+  beforeEach(() => {
+    vi.clearAllMocks();
+  });
+
+  it("routes demand creation through the real allocation router path and writes an audit log", async () => {
+    const auditCreate = vi.fn().mockResolvedValue({ id: "audit_1" });
+    const notificationCreate = vi.fn().mockResolvedValue({ id: "task_1" });
+    const demandCreate = vi.fn().mockResolvedValue({
+      id: "demand_1",
+      projectId: "project_1",
+      roleId: "role_1",
+      role: null,
+      headcount: 2,
+      hoursPerDay: 6,
+      percentage: 75,
+      budgetCents: 0,
+      status: "PROPOSED",
+      metadata: {},
+      startDate: new Date("2026-05-01T00:00:00.000Z"),
+      endDate: new Date("2026-05-15T00:00:00.000Z"),
+      project: { id: "project_1", name: "Project One", shortCode: "PROJ-1" },
+      roleEntity: { id: "role_1", name: "Designer", color: "#00AAFF" },
+    });
+    const tx = {
+      project: {
+        findUnique: vi.fn().mockResolvedValue({
+          id: "project_1",
+          name: "Project One",
+          shortCode: "PROJ-1",
+          budgetCents: 0,
+        }),
+      },
+      demandRequirement: { create: demandCreate },
+      auditLog: { create: auditCreate },
+    };
+    const transaction = vi.fn().mockImplementation(
+      async (callback: (inner: typeof tx) => Promise<unknown>) => callback(tx),
+    );
+    const projectFindFirst = vi.fn().mockResolvedValue({
+      id: "project_1",
+      name: "Project One",
+      shortCode: "PROJ-1",
+    });
+    const projectFindUnique = vi.fn().mockResolvedValue({
+      id: "project_1",
+      name: "Project One",
+      shortCode: "PROJ-1",
+      budgetCents: 0,
+    });
+    const roleFindFirst = vi.fn().mockResolvedValue({ id: "role_1", name: "Designer" });
+    const roleFindUnique = vi.fn().mockResolvedValue({ id: "role_1", name: "Designer" });
+    const ctx = createToolContext(
+      {
+        project: {
+          findFirst: projectFindFirst,
+          findUnique: projectFindUnique,
+        },
+        role: {
+          findFirst: roleFindFirst,
+          findUnique: roleFindUnique,
+        },
+        assignment: {
+          findMany: vi.fn().mockResolvedValue([]),
+        },
+        demandRequirement: {
+          findMany: vi.fn().mockResolvedValue([]),
+        },
+        user: {
+          findMany: vi.fn().mockResolvedValue([{ id: "manager_1" }]),
+        },
+        notification: {
+          findFirst: vi.fn().mockResolvedValue(null),
+          create: notificationCreate,
+        },
+        $transaction: transaction,
+      },
+      {
+        userRole: SystemRole.ADMIN,
+        permissions: [PermissionKey.MANAGE_ALLOCATIONS],
+      },
+    );
+
+    const result = await executeTool(
+      "create_demand",
+      JSON.stringify({
+        projectId: "PROJ-1",
+        roleName: "Designer",
+        headcount: 2,
+        hoursPerDay: 6,
+        startDate: "2026-05-01",
+        endDate: "2026-05-15",
+      }),
+      ctx,
+    );
+
+    expect(JSON.parse(result.content)).toEqual(
+      expect.objectContaining({
+        success: true,
+        message:
+          "Created demand: Designer × 2 for Project One (PROJ-1), 6h/day, 2026-05-01 to 2026-05-15",
+        demandId: "demand_1",
+      }),
+    );
+    expect(demandCreate).toHaveBeenCalledWith(
+      expect.objectContaining({
+        data: expect.objectContaining({
+          projectId: "project_1",
+          roleId: "role_1",
+          headcount: 2,
+          hoursPerDay: 6,
+          percentage: 75,
+        }),
+      }),
+    );
+    expect(auditCreate).toHaveBeenCalledTimes(1);
+    expect(notificationCreate).toHaveBeenCalledTimes(1);
+  });
+});