72 lines
2.3 KiB
TypeScript
72 lines
2.3 KiB
TypeScript
import { describe, expect, it, vi } from "vitest";
|
|
import { SystemRole } from "@capakraken/shared";
|
|
|
|
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([]),
|
|
};
|
|
});
|
|
|
|
import { executeTool } from "../router/assistant-tools.js";
|
|
import { createToolContext, withUserLookup } from "./assistant-tools-notification-test-helpers.js";
|
|
|
|
describe("assistant notification inbox mutation tools", () => {
|
|
it("marks all unread notifications as read through the notification router", async () => {
|
|
const db = {
|
|
user: {
|
|
findUnique: vi.fn().mockResolvedValue({ id: "user_1" }),
|
|
},
|
|
notification: {
|
|
updateMany: vi.fn().mockResolvedValue({ count: 3 }),
|
|
},
|
|
};
|
|
const ctx = createToolContext(db, SystemRole.CONTROLLER);
|
|
|
|
const result = await executeTool("mark_notification_read", "{}", ctx);
|
|
|
|
expect(db.notification.updateMany).toHaveBeenCalledWith({
|
|
where: { userId: "user_1", readAt: null },
|
|
data: { readAt: expect.any(Date) },
|
|
});
|
|
expect(JSON.parse(result.content)).toEqual(
|
|
expect.objectContaining({
|
|
success: true,
|
|
message: "All unread notifications marked as read.",
|
|
}),
|
|
);
|
|
});
|
|
|
|
it("deletes notifications through the router and invalidates notification queries", async () => {
|
|
const db = withUserLookup({
|
|
notification: {
|
|
findFirst: vi.fn().mockResolvedValue({
|
|
id: "notif_9",
|
|
userId: "user_1",
|
|
category: "NOTIFICATION",
|
|
senderId: "user_mgr",
|
|
}),
|
|
delete: vi.fn().mockResolvedValue({ id: "notif_9" }),
|
|
},
|
|
});
|
|
const ctx = createToolContext(db, SystemRole.USER);
|
|
|
|
const result = await executeTool(
|
|
"delete_notification",
|
|
JSON.stringify({ id: "notif_9" }),
|
|
ctx,
|
|
);
|
|
|
|
expect(db.notification.delete).toHaveBeenCalledWith({ where: { id: "notif_9" } });
|
|
expect(result.action).toEqual({ type: "invalidate", scope: ["notification"] });
|
|
expect(result.data).toEqual({
|
|
success: true,
|
|
id: "notif_9",
|
|
message: "Deleted notification notif_9.",
|
|
});
|
|
});
|
|
});
|