Hartmut/CapaKraken
1
0
Fork 0
Code Issues 10 Pull Requests 1 Actions Packages Projects Releases Wiki Activity

test(api): cover assistant timeline quick assign

Browse Source
This commit is contained in:
Hartmut Nörenberg
2026-04-01 00:34:18 +02:00
parent 2b8e1a1bf1
commit 705b570684
3 changed files with 310 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
packages/api/src/__tests__/assistant-tools-timeline-quick-assign-errors.test.ts
+153
View File
@@ -0,0 +1,153 @@
import { beforeEach, describe, expect, it, vi } from "vitest";
import { TRPCError } from "@trpc/server";
import {
buildBaseProject,
buildBaseResource,
createQuickAssignInput,
createToolContext,
executeTool,
quickAssignPermissions,
quickAssignRole,
} from "./assistant-tools-timeline-quick-assign-test-helpers.js";
describe("assistant timeline quick-assign error tools", () => {
beforeEach(() => {
vi.clearAllMocks();
});
it("returns a stable conflict error for quick-assign timeline mutations", async () => {
const db = {
project: {
findUnique: vi.fn().mockResolvedValue(buildBaseProject()),
},
resource: {
findUnique: vi.fn().mockResolvedValue(buildBaseResource()),
},
assignment: {
findMany: vi.fn().mockResolvedValue([]),
create: vi.fn().mockRejectedValue(
new TRPCError({
code: "CONFLICT",
message: "Resource is already assigned to this project with overlapping dates",
}),
),
},
vacation: {
findMany: vi.fn().mockResolvedValue([]),
},
auditLog: {
create: vi.fn().mockResolvedValue({}),
},
$transaction: vi.fn(async (callback: (tx: unknown) => unknown) => callback(db)),
};
const ctx = createToolContext(db, quickAssignPermissions, quickAssignRole);
const result = await executeTool(
"quick_assign_timeline_resource",
JSON.stringify(createQuickAssignInput()),
ctx,
);
expect(result.action).toBeUndefined();
expect(JSON.parse(result.content)).toEqual({
error: "Resource is already assigned to this project with overlapping dates",
});
});
it("returns stable not-found errors when quick-assign timeline targets disappear mid-mutation", async () => {
const cases = [
{
name: "missing project",
tx: {
project: {
findUnique: vi.fn().mockResolvedValue(null),
},
resource: {
findUnique: vi.fn().mockResolvedValue(buildBaseResource()),
},
assignment: {
findMany: vi.fn().mockResolvedValue([]),
create: vi.fn(),
},
},
expected: "Project not found with the given criteria.",
},
{
name: "missing resource",
tx: {
project: {
findUnique: vi.fn().mockResolvedValue(buildBaseProject()),
},
resource: {
findUnique: vi.fn().mockResolvedValue(null),
},
assignment: {
findMany: vi.fn().mockResolvedValue([]),
create: vi.fn(),
},
},
expected: "Resource not found with the given criteria.",
},
] as const;
for (const testCase of cases) {
const db = {
project: {
findUnique: vi.fn().mockResolvedValue(buildBaseProject()),
},
resource: {
findUnique: vi.fn().mockResolvedValue(buildBaseResource()),
},
assignment: {
findMany: vi.fn().mockResolvedValue([]),
create: vi.fn(),
},
vacation: {
findMany: vi.fn().mockResolvedValue([]),
},
auditLog: {
create: vi.fn().mockResolvedValue({}),
},
$transaction: vi.fn(async (callback: (tx: typeof testCase.tx) => unknown) => callback(testCase.tx)),
};
const ctx = createToolContext(db, quickAssignPermissions, quickAssignRole);
const result = await executeTool(
"quick_assign_timeline_resource",
JSON.stringify(createQuickAssignInput()),
ctx,
);
expect(JSON.parse(result.content)).toEqual({ error: testCase.expected });
}
});
it("returns a stable validation error for quick-assign timeline date ranges", async () => {
const ctx = createToolContext(
{
project: {
findUnique: vi.fn().mockResolvedValue(buildBaseProject()),
},
resource: {
findUnique: vi.fn().mockResolvedValue(buildBaseResource()),
},
},
quickAssignPermissions,
quickAssignRole,
);
const result = await executeTool(
"quick_assign_timeline_resource",
JSON.stringify(createQuickAssignInput({
startDate: "2026-03-20",
endDate: "2026-03-16",
})),
ctx,
);
expect(JSON.parse(result.content)).toEqual({
error: "End date must be after start date",
});
});
});
packages/api/src/__tests__/assistant-tools-timeline-quick-assign-success.test.ts
+66
View File
@@ -0,0 +1,66 @@
import { beforeEach, describe, expect, it, vi } from "vitest";
import {
buildBaseProject,
buildBaseResource,
createCreatedAssignment,
createQuickAssignInput,
createToolContext,
executeTool,
quickAssignPermissions,
quickAssignRole,
} from "./assistant-tools-timeline-quick-assign-test-helpers.js";
describe("assistant timeline quick-assign tools", () => {
beforeEach(() => {
vi.clearAllMocks();
});
it("quick-assigns a timeline resource through the real timeline router mutation", async () => {
const createdAssignment = createCreatedAssignment();
const db = {
project: {
findUnique: vi.fn().mockResolvedValue(buildBaseProject()),
},
resource: {
findUnique: vi.fn().mockResolvedValue(buildBaseResource()),
},
allocation: {
findMany: vi.fn().mockResolvedValue([]),
create: vi.fn(),
},
assignment: {
findMany: vi.fn().mockResolvedValue([]),
create: vi.fn().mockResolvedValue(createdAssignment),
},
vacation: {
findMany: vi.fn().mockResolvedValue([]),
},
auditLog: {
create: vi.fn().mockResolvedValue({}),
},
$transaction: vi.fn(async (callback: (tx: unknown) => unknown) => callback(db)),
};
const ctx = createToolContext(db, quickAssignPermissions, quickAssignRole);
const result = await executeTool(
"quick_assign_timeline_resource",
JSON.stringify(createQuickAssignInput()),
ctx,
);
expect(result.action).toEqual({ type: "invalidate", scope: ["allocation", "timeline", "project"] });
expect(JSON.parse(result.content)).toEqual(
expect.objectContaining({
success: true,
allocation: expect.objectContaining({
id: "assignment_quick_1",
projectId: "project_1",
resourceId: "resource_1",
hoursPerDay: 8,
}),
}),
);
expect(db.assignment.create).toHaveBeenCalled();
});
});
packages/api/src/__tests__/assistant-tools-timeline-quick-assign-test-helpers.ts
+91
View File
@@ -0,0 +1,91 @@
import { PermissionKey, SystemRole } from "@capakraken/shared";
import { vi } from "vitest";
vi.mock("@capakraken/application", async (importOriginal) => {
const actual = await importOriginal<typeof import("@capakraken/application")>();
return {
...actual,
getDashboardBudgetForecast: vi.fn().mockResolvedValue([]),
getDashboardPeakTimes: vi.fn().mockResolvedValue([]),
listAssignmentBookings: vi.fn().mockResolvedValue([]),
};
});
vi.mock("../sse/event-bus.js", () => ({
emitAllocationCreated: vi.fn(),
emitAllocationDeleted: vi.fn(),
emitAllocationUpdated: vi.fn(),
emitProjectShifted: vi.fn(),
}));
vi.mock("../lib/budget-alerts.js", () => ({
checkBudgetThresholds: vi.fn(),
}));
vi.mock("../lib/cache.js", () => ({
invalidateDashboardCache: vi.fn(),
}));
import { executeTool as executeAssistantTool } from "../router/assistant-tools.js";
export { createToolContext } from "./assistant-tools-advanced-timeline-test-helpers.js";
export {
buildBaseProject,
buildBaseResource,
} from "./assistant-tools-timeline-allocation-mutation-test-helpers.js";
export const quickAssignPermissions = [
PermissionKey.MANAGE_ALLOCATIONS,
PermissionKey.USE_ASSISTANT_ADVANCED_TOOLS,
];
export const quickAssignRole = SystemRole.MANAGER;
export function createQuickAssignInput(overrides: Record<string, unknown> = {}) {
return {
resourceIdentifier: "resource_1",
projectIdentifier: "project_1",
startDate: "2026-03-16",
endDate: "2026-03-20",
hoursPerDay: 8,
...overrides,
};
}
export function createCreatedAssignment(overrides: Record<string, unknown> = {}) {
return {
id: "assignment_quick_1",
demandRequirementId: null,
resourceId: "resource_1",
projectId: "project_1",
startDate: new Date("2026-03-16"),
endDate: new Date("2026-03-20"),
hoursPerDay: 8,
percentage: 100,
role: "Team Member",
roleId: null,
dailyCostCents: 40000,
status: "PROPOSED",
metadata: { source: "quickAssign" },
createdAt: new Date("2026-03-13"),
updatedAt: new Date("2026-03-13"),
resource: {
id: "resource_1",
displayName: "Alice",
eid: "E-001",
lcrCents: 5000,
},
project: {
id: "project_1",
name: "Gelddruckmaschine",
shortCode: "GDM",
status: "ACTIVE",
responsiblePerson: null,
},
roleEntity: null,
demandRequirement: null,
...overrides,
};
}
export const executeTool = executeAssistantTool;