61 lines
1.8 KiB
TypeScript
61 lines
1.8 KiB
TypeScript
import { beforeEach, describe, expect, it, vi } from "vitest";
|
|
|
|
import {
|
|
batchQuickAssignPermissions,
|
|
batchQuickAssignRole,
|
|
buildBaseProject,
|
|
buildBaseResource,
|
|
createBatchQuickAssignInput,
|
|
createToolContext,
|
|
executeTool,
|
|
} from "./assistant-tools-timeline-batch-quick-assign-test-helpers.js";
|
|
|
|
describe("assistant timeline batch quick-assign tools", () => {
|
|
beforeEach(() => {
|
|
vi.clearAllMocks();
|
|
});
|
|
|
|
it("batch quick-assigns timeline resources through the real timeline router mutation", async () => {
|
|
const db = {
|
|
project: {
|
|
findUnique: vi.fn().mockResolvedValue(buildBaseProject()),
|
|
},
|
|
resource: {
|
|
findUnique: vi
|
|
.fn()
|
|
.mockImplementation(async ({ where }: { where: { id: string } }) => buildBaseResource(where.id)),
|
|
},
|
|
assignment: {
|
|
findMany: vi.fn().mockResolvedValue([]),
|
|
create: vi
|
|
.fn()
|
|
.mockResolvedValueOnce({ id: "assignment_batch_1" })
|
|
.mockResolvedValueOnce({ id: "assignment_batch_2" }),
|
|
},
|
|
auditLog: {
|
|
create: vi.fn().mockResolvedValue({}),
|
|
},
|
|
vacation: {
|
|
findMany: vi.fn().mockResolvedValue([]),
|
|
},
|
|
$transaction: vi.fn(async (callback: (tx: unknown) => unknown) => callback(db)),
|
|
};
|
|
const ctx = createToolContext(db, batchQuickAssignPermissions, batchQuickAssignRole);
|
|
|
|
const result = await executeTool(
|
|
"batch_quick_assign_timeline_resources",
|
|
JSON.stringify(createBatchQuickAssignInput()),
|
|
ctx,
|
|
);
|
|
|
|
expect(result.action).toEqual({ type: "invalidate", scope: ["allocation", "timeline", "project"] });
|
|
expect(JSON.parse(result.content)).toEqual(
|
|
expect.objectContaining({
|
|
success: true,
|
|
count: 2,
|
|
}),
|
|
);
|
|
expect(db.assignment.create).toHaveBeenCalledTimes(2);
|
|
});
|
|
});
|