refactor(api): extract timeline allocation assignment procedures
This commit is contained in:
@@ -1,12 +1,6 @@
|
||||
import { beforeEach, describe, expect, it, vi } from "vitest";
|
||||
|
||||
vi.mock("@capakraken/application", () => ({
|
||||
buildSplitAllocationReadModel: vi.fn(),
|
||||
createAssignment: vi.fn(),
|
||||
}));
|
||||
|
||||
vi.mock("../sse/event-bus.js", () => ({
|
||||
emitAllocationCreated: vi.fn(),
|
||||
emitAllocationUpdated: vi.fn(),
|
||||
}));
|
||||
|
||||
@@ -14,24 +8,10 @@ vi.mock("../router/timeline-allocation-shift-support.js", () => ({
|
||||
applyTimelineBatchAllocationShift: vi.fn(),
|
||||
}));
|
||||
|
||||
import {
|
||||
buildSplitAllocationReadModel,
|
||||
createAssignment,
|
||||
} from "@capakraken/application";
|
||||
import {
|
||||
emitAllocationCreated,
|
||||
emitAllocationUpdated,
|
||||
} from "../sse/event-bus.js";
|
||||
import {
|
||||
createTimelineBatchQuickAssignments,
|
||||
createTimelineQuickAssignment,
|
||||
shiftTimelineAllocations,
|
||||
} from "../router/timeline-allocation-procedure-support.js";
|
||||
import { emitAllocationUpdated } from "../sse/event-bus.js";
|
||||
import { shiftTimelineAllocations } from "../router/timeline-allocation-procedure-support.js";
|
||||
import { applyTimelineBatchAllocationShift } from "../router/timeline-allocation-shift-support.js";
|
||||
|
||||
const buildSplitAllocationReadModelMock = vi.mocked(buildSplitAllocationReadModel);
|
||||
const createAssignmentMock = vi.mocked(createAssignment);
|
||||
const emitAllocationCreatedMock = vi.mocked(emitAllocationCreated);
|
||||
const emitAllocationUpdatedMock = vi.mocked(emitAllocationUpdated);
|
||||
const applyTimelineBatchAllocationShiftMock = vi.mocked(applyTimelineBatchAllocationShift);
|
||||
|
||||
@@ -40,104 +20,6 @@ describe("timeline allocation procedure support", () => {
|
||||
vi.clearAllMocks();
|
||||
});
|
||||
|
||||
it("creates a quick assignment and emits the allocation created event", async () => {
|
||||
const db = {
|
||||
$transaction: vi.fn(async (callback: (tx: unknown) => unknown) => callback(db)),
|
||||
} as never;
|
||||
|
||||
createAssignmentMock.mockResolvedValueOnce({ id: "assignment_1" } as never);
|
||||
buildSplitAllocationReadModelMock.mockReturnValueOnce({
|
||||
allocations: [
|
||||
{
|
||||
id: "allocation_1",
|
||||
projectId: "project_1",
|
||||
resourceId: "resource_1",
|
||||
},
|
||||
],
|
||||
} as never);
|
||||
|
||||
await expect(
|
||||
createTimelineQuickAssignment(db, {
|
||||
resourceId: "resource_1",
|
||||
projectId: "project_1",
|
||||
startDate: new Date("2026-04-01T00:00:00.000Z"),
|
||||
endDate: new Date("2026-04-05T00:00:00.000Z"),
|
||||
hoursPerDay: 8,
|
||||
role: "Team Member",
|
||||
status: "PROPOSED",
|
||||
source: "quickAssign",
|
||||
}),
|
||||
).resolves.toEqual({
|
||||
id: "allocation_1",
|
||||
projectId: "project_1",
|
||||
resourceId: "resource_1",
|
||||
});
|
||||
|
||||
expect(createAssignmentMock).toHaveBeenCalledOnce();
|
||||
expect(emitAllocationCreatedMock).toHaveBeenCalledWith({
|
||||
id: "allocation_1",
|
||||
projectId: "project_1",
|
||||
resourceId: "resource_1",
|
||||
});
|
||||
});
|
||||
|
||||
it("creates batch quick assignments and emits events for each result", async () => {
|
||||
const db = {
|
||||
$transaction: vi.fn(async (callback: (tx: unknown) => unknown) => callback(db)),
|
||||
} as never;
|
||||
|
||||
createAssignmentMock
|
||||
.mockResolvedValueOnce({
|
||||
id: "assignment_1",
|
||||
projectId: "project_1",
|
||||
resourceId: "resource_1",
|
||||
} as never)
|
||||
.mockResolvedValueOnce({
|
||||
id: "assignment_2",
|
||||
projectId: "project_2",
|
||||
resourceId: "resource_2",
|
||||
} as never);
|
||||
|
||||
await expect(
|
||||
createTimelineBatchQuickAssignments(db, {
|
||||
assignments: [
|
||||
{
|
||||
resourceId: "resource_1",
|
||||
projectId: "project_1",
|
||||
startDate: new Date("2026-04-01T00:00:00.000Z"),
|
||||
endDate: new Date("2026-04-05T00:00:00.000Z"),
|
||||
hoursPerDay: 8,
|
||||
role: "Team Member",
|
||||
status: "PROPOSED",
|
||||
source: "batchQuickAssign",
|
||||
},
|
||||
{
|
||||
resourceId: "resource_2",
|
||||
projectId: "project_2",
|
||||
startDate: new Date("2026-04-06T00:00:00.000Z"),
|
||||
endDate: new Date("2026-04-10T00:00:00.000Z"),
|
||||
hoursPerDay: 6,
|
||||
role: "Lead",
|
||||
status: "PROPOSED",
|
||||
source: "batchQuickAssign",
|
||||
},
|
||||
],
|
||||
}),
|
||||
).resolves.toEqual({ count: 2 });
|
||||
|
||||
expect(createAssignmentMock).toHaveBeenCalledTimes(2);
|
||||
expect(emitAllocationCreatedMock).toHaveBeenNthCalledWith(1, {
|
||||
id: "assignment_1",
|
||||
projectId: "project_1",
|
||||
resourceId: "resource_1",
|
||||
});
|
||||
expect(emitAllocationCreatedMock).toHaveBeenNthCalledWith(2, {
|
||||
id: "assignment_2",
|
||||
projectId: "project_2",
|
||||
resourceId: "resource_2",
|
||||
});
|
||||
});
|
||||
|
||||
it("applies allocation shifts and emits update events", async () => {
|
||||
applyTimelineBatchAllocationShiftMock.mockResolvedValueOnce([
|
||||
{
|
||||
|
||||
Reference in New Issue
Block a user