76 lines
2.4 KiB
TypeScript
76 lines
2.4 KiB
TypeScript
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",
|
|
}),
|
|
}),
|
|
]);
|
|
});
|
|
});
|