83 lines
2.5 KiB
TypeScript
83 lines
2.5 KiB
TypeScript
import { PermissionKey } from "@capakraken/shared";
|
|
import { beforeEach, describe, expect, it, vi } from "vitest";
|
|
|
|
vi.mock("@capakraken/application", async (importOriginal) => {
|
|
const actual = await importOriginal<typeof import("@capakraken/application")>();
|
|
return {
|
|
...actual,
|
|
getDashboardBudgetForecast: vi.fn().mockResolvedValue([]),
|
|
listAssignmentBookings: vi.fn().mockResolvedValue([]),
|
|
};
|
|
});
|
|
|
|
vi.mock("../lib/audit.js", () => ({
|
|
createAuditEntry: vi.fn().mockResolvedValue(undefined),
|
|
}));
|
|
|
|
import { executeTool } from "../router/assistant-tools.js";
|
|
import { createToolContext } from "./assistant-tools-holiday-capacity-test-helpers.js";
|
|
|
|
describe("assistant holiday-aware capacity tool", () => {
|
|
beforeEach(() => {
|
|
vi.clearAllMocks();
|
|
});
|
|
|
|
it("finds capacity with local holidays respected", async () => {
|
|
const db = {
|
|
resource: {
|
|
findMany: vi.fn().mockResolvedValue([
|
|
{
|
|
id: "res_by",
|
|
displayName: "Bavaria",
|
|
eid: "BY-1",
|
|
fte: 1,
|
|
availability: { monday: 8, tuesday: 8, wednesday: 8, thursday: 8, friday: 8 },
|
|
countryId: "country_de",
|
|
federalState: "BY",
|
|
metroCityId: null,
|
|
country: { code: "DE" },
|
|
metroCity: null,
|
|
areaRole: { name: "Consultant" },
|
|
chapter: "CGI",
|
|
assignments: [],
|
|
},
|
|
{
|
|
id: "res_hh",
|
|
displayName: "Hamburg",
|
|
eid: "HH-1",
|
|
fte: 1,
|
|
availability: { monday: 8, tuesday: 8, wednesday: 8, thursday: 8, friday: 8 },
|
|
countryId: "country_de",
|
|
federalState: "HH",
|
|
metroCityId: null,
|
|
country: { code: "DE" },
|
|
metroCity: null,
|
|
areaRole: { name: "Consultant" },
|
|
chapter: "CGI",
|
|
assignments: [],
|
|
},
|
|
]),
|
|
},
|
|
vacation: {
|
|
findMany: vi.fn().mockResolvedValue([]),
|
|
},
|
|
};
|
|
const ctx = createToolContext(db, [PermissionKey.VIEW_PLANNING]);
|
|
|
|
const result = await executeTool(
|
|
"find_capacity",
|
|
JSON.stringify({ startDate: "2026-01-06", endDate: "2026-01-06", minHoursPerDay: 1 }),
|
|
ctx,
|
|
);
|
|
|
|
const parsed = JSON.parse(result.content) as {
|
|
results: Array<{ name: string; availableHours: number; availableHoursPerDay: number }>;
|
|
};
|
|
|
|
expect(parsed.results).toHaveLength(1);
|
|
expect(parsed.results[0]).toEqual(
|
|
expect.objectContaining({ name: "Hamburg", availableHours: 8, availableHoursPerDay: 8 }),
|
|
);
|
|
});
|
|
});
|