test(api): cover assistant master data reads

2026-04-01 00:20:19 +02:00
parent 083857f19f
commit 7f9ee92516
7 changed files with 636 additions and 0 deletions
@@ -0,0 +1,75 @@
+import { beforeEach, describe, expect, it, vi } from "vitest";
+import { countPlanningEntries } from "@capakraken/application";
+import { SystemRole } from "@capakraken/shared";
+
+vi.mock("@capakraken/application", async (importOriginal) => {
+  const actual = await importOriginal<typeof import("@capakraken/application")>();
+  return {
+    ...actual,
+    countPlanningEntries: vi.fn().mockResolvedValue({ countsByRoleId: new Map() }),
+    getDashboardBudgetForecast: vi.fn().mockResolvedValue([]),
+    getDashboardPeakTimes: vi.fn().mockResolvedValue([]),
+    listAssignmentBookings: vi.fn().mockResolvedValue([]),
+  };
+});
+
+import { executeTool } from "../router/assistant-tools.js";
+import { createToolContext } from "./assistant-tools-master-data-read-test-helpers.js";
+
+describe("assistant master data calculation rule read tools", () => {
+  beforeEach(() => {
+    vi.clearAllMocks();
+    vi.mocked(countPlanningEntries).mockResolvedValue({ countsByRoleId: new Map() });
+  });
+
+  it("routes calculation rule reads through their backing router", async () => {
+    const db = {
+      calculationRule: {
+        findMany: vi.fn().mockResolvedValue([
+          {
+            id: "calc_1",
+            name: "Rush Fee",
+            description: "Adds cost for rush work",
+            isActive: true,
+            triggerType: "RUSH",
+            orderType: null,
+            costEffect: "INCREASE",
+            costReductionPercent: null,
+            chargeabilityEffect: "NONE",
+            priority: 90,
+            project: { id: "proj_1", name: "Falcon", shortCode: "FAL" },
+          },
+        ]),
+      },
+    };
+    const ctx = createToolContext(db, { userRole: SystemRole.CONTROLLER });
+
+    const calculationRulesResult = await executeTool("list_calculation_rules", "{}", ctx);
+
+    expect(db.calculationRule.findMany).toHaveBeenCalledWith({
+      orderBy: [{ priority: "desc" }, { name: "asc" }],
+      include: {
+        project: {
+          select: {
+            id: true,
+            shortCode: true,
+            name: true,
+            status: true,
+            endDate: true,
+          },
+        },
+      },
+    });
+
+    expect(JSON.parse(calculationRulesResult.content)).toEqual([
+      expect.objectContaining({
+        id: "calc_1",
+        name: "Rush Fee",
+        project: expect.objectContaining({
+          id: "proj_1",
+          shortCode: "FAL",
+        }),
+      }),
+    ]);
+  });
+});