85 lines
2.3 KiB
TypeScript
85 lines
2.3 KiB
TypeScript
import { beforeEach, describe, expect, it, vi } from "vitest";
|
|
import { SystemRole } from "@capakraken/shared";
|
|
import {
|
|
createToolContext,
|
|
executeTool,
|
|
withUserLookup,
|
|
} from "./assistant-tools-task-workflow-test-helpers.js";
|
|
|
|
describe("assistant task workflow tools - status", () => {
|
|
beforeEach(() => {
|
|
vi.clearAllMocks();
|
|
});
|
|
|
|
it("maps taskId to the router id when updating task status and returns invalidate metadata", async () => {
|
|
const db = withUserLookup({
|
|
notification: {
|
|
findFirst: vi.fn().mockResolvedValue({
|
|
id: "task_1",
|
|
userId: "user_1",
|
|
assigneeId: "user_2",
|
|
}),
|
|
update: vi.fn().mockResolvedValue({
|
|
id: "task_1",
|
|
taskStatus: "DONE",
|
|
}),
|
|
},
|
|
});
|
|
const ctx = createToolContext(db, SystemRole.USER);
|
|
|
|
const result = await executeTool(
|
|
"update_task_status",
|
|
JSON.stringify({ taskId: "task_1", status: "DONE" }),
|
|
ctx,
|
|
);
|
|
|
|
expect(db.notification.findFirst).toHaveBeenCalledWith({
|
|
where: {
|
|
id: "task_1",
|
|
OR: [{ userId: "user_1" }, { assigneeId: "user_1" }],
|
|
category: { in: ["TASK", "APPROVAL"] },
|
|
},
|
|
});
|
|
expect(db.notification.update).toHaveBeenCalledWith({
|
|
where: { id: "task_1" },
|
|
data: {
|
|
taskStatus: "DONE",
|
|
completedAt: expect.any(Date),
|
|
completedBy: "user_1",
|
|
},
|
|
});
|
|
expect(result.action).toEqual({ type: "invalidate", scope: ["notification"] });
|
|
expect(result.data).toEqual(
|
|
expect.objectContaining({
|
|
success: true,
|
|
message: "Task status updated to DONE.",
|
|
task: expect.objectContaining({
|
|
id: "task_1",
|
|
taskStatus: "DONE",
|
|
}),
|
|
}),
|
|
);
|
|
});
|
|
|
|
it("returns a stable assistant error when updating a missing task", async () => {
|
|
const ctx = createToolContext(
|
|
withUserLookup({
|
|
notification: {
|
|
findFirst: vi.fn().mockResolvedValue(null),
|
|
},
|
|
}),
|
|
SystemRole.ADMIN,
|
|
);
|
|
|
|
const result = await executeTool(
|
|
"update_task_status",
|
|
JSON.stringify({ taskId: "task_missing", status: "DONE" }),
|
|
ctx,
|
|
);
|
|
|
|
expect(JSON.parse(result.content)).toEqual({
|
|
error: "Task not found with the given criteria.",
|
|
});
|
|
});
|
|
});
|