test(api): cover assistant broadcast sends
This commit is contained in:
+98
@@ -0,0 +1,98 @@
|
||||
import { beforeEach, describe, expect, it, vi } from "vitest";
|
||||
import { SystemRole } from "@capakraken/shared";
|
||||
|
||||
import { executeTool } from "./assistant-tools-broadcast-send-test-helpers.js";
|
||||
import { createToolContext } from "./assistant-tools-notification-test-helpers.js";
|
||||
|
||||
describe("assistant broadcast send recipient fan-out errors", () => {
|
||||
beforeEach(() => {
|
||||
vi.clearAllMocks();
|
||||
});
|
||||
|
||||
it("returns a stable assistant error when broadcast recipient user is missing", async () => {
|
||||
const ctx = createToolContext(
|
||||
{
|
||||
notificationBroadcast: {
|
||||
create: vi.fn().mockResolvedValue({
|
||||
id: "broadcast_missing_user",
|
||||
title: "Office update",
|
||||
targetType: "user",
|
||||
}),
|
||||
},
|
||||
notification: {
|
||||
create: vi.fn().mockRejectedValue({
|
||||
code: "P2003",
|
||||
message: "Foreign key constraint failed",
|
||||
meta: { field_name: "Notification_userId_fkey" },
|
||||
}),
|
||||
},
|
||||
},
|
||||
SystemRole.MANAGER,
|
||||
);
|
||||
|
||||
const result = await executeTool(
|
||||
"send_broadcast",
|
||||
JSON.stringify({
|
||||
title: "Office update",
|
||||
targetType: "user",
|
||||
targetValue: "user_missing",
|
||||
}),
|
||||
ctx,
|
||||
);
|
||||
|
||||
expect(JSON.parse(result.content)).toEqual({
|
||||
error: "Broadcast recipient user not found with the given criteria.",
|
||||
});
|
||||
});
|
||||
|
||||
it("returns a stable assistant error when broadcast fan-out loses recipient rows inside the router transaction", async () => {
|
||||
const recipientMissingTx = {
|
||||
notificationBroadcast: {
|
||||
create: vi.fn().mockResolvedValue({
|
||||
id: "broadcast_missing_recipient",
|
||||
title: "Office update",
|
||||
targetType: "all",
|
||||
}),
|
||||
update: vi.fn(),
|
||||
},
|
||||
notification: {
|
||||
create: vi.fn().mockRejectedValue(
|
||||
Object.assign(new Error("Foreign key constraint failed"), {
|
||||
code: "P2003",
|
||||
meta: { field_name: "Notification_userId_fkey" },
|
||||
}),
|
||||
),
|
||||
},
|
||||
};
|
||||
const recipientMissingCtx = createToolContext(
|
||||
{
|
||||
user: {
|
||||
findMany: vi.fn().mockResolvedValue([{ id: "user_2" }]),
|
||||
},
|
||||
$transaction: vi.fn(async (callback: (db: typeof recipientMissingTx) => Promise<unknown>) =>
|
||||
callback(recipientMissingTx)),
|
||||
notificationBroadcast: {
|
||||
create: vi.fn(),
|
||||
update: vi.fn(),
|
||||
},
|
||||
notification: {
|
||||
create: vi.fn(),
|
||||
},
|
||||
},
|
||||
SystemRole.MANAGER,
|
||||
);
|
||||
|
||||
const recipientMissingResult = await executeTool(
|
||||
"send_broadcast",
|
||||
JSON.stringify({
|
||||
title: "Office update",
|
||||
targetType: "all",
|
||||
}),
|
||||
recipientMissingCtx,
|
||||
);
|
||||
|
||||
expect(JSON.parse(recipientMissingResult.content)).toEqual({
|
||||
error: "Broadcast recipient user not found with the given criteria.",
|
||||
});
|
||||
});
|
||||
});
|
||||
+163
@@ -0,0 +1,163 @@
|
||||
import { beforeEach, describe, expect, it, vi } from "vitest";
|
||||
import { SystemRole } from "@capakraken/shared";
|
||||
|
||||
import { executeTool } from "./assistant-tools-broadcast-send-test-helpers.js";
|
||||
import { createToolContext } from "./assistant-tools-notification-test-helpers.js";
|
||||
|
||||
describe("assistant broadcast send sender fan-out errors", () => {
|
||||
beforeEach(() => {
|
||||
vi.clearAllMocks();
|
||||
});
|
||||
|
||||
it("returns a stable assistant error when broadcast sender user is missing", async () => {
|
||||
const ctx = createToolContext(
|
||||
{
|
||||
notificationBroadcast: {
|
||||
create: vi.fn().mockResolvedValue({
|
||||
id: "broadcast_missing_sender",
|
||||
title: "Office update",
|
||||
targetType: "user",
|
||||
}),
|
||||
},
|
||||
notification: {
|
||||
create: vi.fn().mockRejectedValue({
|
||||
code: "P2003",
|
||||
message: "Foreign key constraint failed",
|
||||
meta: { field_name: "Notification_senderId_fkey" },
|
||||
}),
|
||||
},
|
||||
},
|
||||
SystemRole.MANAGER,
|
||||
);
|
||||
|
||||
const result = await executeTool(
|
||||
"send_broadcast",
|
||||
JSON.stringify({
|
||||
title: "Office update",
|
||||
targetType: "user",
|
||||
targetValue: "user_2",
|
||||
}),
|
||||
ctx,
|
||||
);
|
||||
|
||||
expect(JSON.parse(result.content)).toEqual({
|
||||
error: "Sender user not found with the given criteria.",
|
||||
});
|
||||
});
|
||||
|
||||
it("returns a stable assistant error when broadcast fan-out loses sender rows inside the router transaction", async () => {
|
||||
const senderMissingTx = {
|
||||
notificationBroadcast: {
|
||||
create: vi.fn().mockRejectedValue(
|
||||
Object.assign(new Error("Foreign key constraint failed"), {
|
||||
code: "P2003",
|
||||
meta: { field_name: "NotificationBroadcast_senderId_fkey" },
|
||||
}),
|
||||
),
|
||||
update: vi.fn(),
|
||||
},
|
||||
notification: {
|
||||
create: vi.fn(),
|
||||
},
|
||||
};
|
||||
const senderMissingCtx = createToolContext(
|
||||
{
|
||||
user: {
|
||||
findMany: vi.fn().mockResolvedValue([{ id: "user_2" }]),
|
||||
},
|
||||
$transaction: vi.fn(async (callback: (db: typeof senderMissingTx) => Promise<unknown>) =>
|
||||
callback(senderMissingTx)),
|
||||
notificationBroadcast: {
|
||||
create: vi.fn(),
|
||||
update: vi.fn(),
|
||||
},
|
||||
notification: {
|
||||
create: vi.fn(),
|
||||
},
|
||||
},
|
||||
SystemRole.MANAGER,
|
||||
);
|
||||
|
||||
const senderMissingResult = await executeTool(
|
||||
"send_broadcast",
|
||||
JSON.stringify({
|
||||
title: "Office update",
|
||||
targetType: "all",
|
||||
}),
|
||||
senderMissingCtx,
|
||||
);
|
||||
|
||||
expect(JSON.parse(senderMissingResult.content)).toEqual({
|
||||
error: "Sender user not found with the given criteria.",
|
||||
});
|
||||
});
|
||||
|
||||
it("returns a stable assistant error when broadcast creation fails because the sender user is missing", async () => {
|
||||
const ctx = createToolContext(
|
||||
{
|
||||
user: {
|
||||
findMany: vi.fn().mockResolvedValue([{ id: "user_2" }]),
|
||||
},
|
||||
notificationBroadcast: {
|
||||
create: vi.fn().mockRejectedValue(
|
||||
Object.assign(new Error("Foreign key constraint failed"), {
|
||||
code: "P2003",
|
||||
meta: { field_name: "NotificationBroadcast_senderId_fkey" },
|
||||
}),
|
||||
),
|
||||
},
|
||||
},
|
||||
SystemRole.MANAGER,
|
||||
);
|
||||
|
||||
const result = await executeTool(
|
||||
"send_broadcast",
|
||||
JSON.stringify({
|
||||
title: "Office update",
|
||||
targetType: "all",
|
||||
}),
|
||||
ctx,
|
||||
);
|
||||
|
||||
expect(JSON.parse(result.content)).toEqual({
|
||||
error: "Sender user not found with the given criteria.",
|
||||
});
|
||||
});
|
||||
|
||||
it("returns a stable assistant error when scheduled broadcast creation loses the sender user", async () => {
|
||||
const ctx = createToolContext(
|
||||
{
|
||||
user: {
|
||||
findMany: vi.fn().mockResolvedValue([{ id: "user_2" }]),
|
||||
},
|
||||
notificationBroadcast: {
|
||||
create: vi.fn().mockRejectedValue(
|
||||
Object.assign(new Error("Foreign key constraint failed"), {
|
||||
code: "P2003",
|
||||
meta: { field_name: "NotificationBroadcast_senderId_fkey" },
|
||||
}),
|
||||
),
|
||||
},
|
||||
notification: {
|
||||
create: vi.fn(),
|
||||
},
|
||||
},
|
||||
SystemRole.MANAGER,
|
||||
);
|
||||
|
||||
const result = await executeTool(
|
||||
"send_broadcast",
|
||||
JSON.stringify({
|
||||
title: "Office update",
|
||||
targetType: "all",
|
||||
scheduledAt: "2099-04-10T08:00:00.000Z",
|
||||
}),
|
||||
ctx,
|
||||
);
|
||||
|
||||
expect(JSON.parse(result.content)).toEqual({
|
||||
error: "Sender user not found with the given criteria.",
|
||||
});
|
||||
expect(ctx.db.notification.create).not.toHaveBeenCalled();
|
||||
});
|
||||
});
|
||||
+113
@@ -0,0 +1,113 @@
|
||||
import { beforeEach, describe, expect, it, vi } from "vitest";
|
||||
import { SystemRole } from "@capakraken/shared";
|
||||
|
||||
import { executeTool } from "./assistant-tools-broadcast-send-test-helpers.js";
|
||||
import { createToolContext } from "./assistant-tools-notification-test-helpers.js";
|
||||
|
||||
describe("assistant broadcast send finalization errors", () => {
|
||||
beforeEach(() => {
|
||||
vi.clearAllMocks();
|
||||
});
|
||||
|
||||
it("returns a stable assistant error when broadcast finalization loses the broadcast row", async () => {
|
||||
const txCreateBroadcast = vi.fn().mockResolvedValue({
|
||||
id: "broadcast_missing_after_create",
|
||||
title: "Office update",
|
||||
targetType: "all",
|
||||
});
|
||||
const txCreateNotification = vi.fn().mockResolvedValue({ id: "notification_2", userId: "user_2" });
|
||||
const txUpdateBroadcast = vi.fn().mockRejectedValue(
|
||||
Object.assign(new Error("Record to update not found"), {
|
||||
code: "P2025",
|
||||
meta: { modelName: "NotificationBroadcast" },
|
||||
}),
|
||||
);
|
||||
const tx = {
|
||||
notificationBroadcast: {
|
||||
create: txCreateBroadcast,
|
||||
update: txUpdateBroadcast,
|
||||
},
|
||||
notification: {
|
||||
create: txCreateNotification,
|
||||
},
|
||||
};
|
||||
const ctx = createToolContext(
|
||||
{
|
||||
user: {
|
||||
findMany: vi.fn().mockResolvedValue([{ id: "user_2" }]),
|
||||
},
|
||||
$transaction: vi.fn(async (callback: (db: typeof tx) => Promise<unknown>) => callback(tx)),
|
||||
notificationBroadcast: {
|
||||
create: vi.fn(),
|
||||
update: vi.fn(),
|
||||
},
|
||||
notification: {
|
||||
create: vi.fn(),
|
||||
},
|
||||
},
|
||||
SystemRole.MANAGER,
|
||||
);
|
||||
|
||||
const result = await executeTool(
|
||||
"send_broadcast",
|
||||
JSON.stringify({
|
||||
title: "Office update",
|
||||
targetType: "all",
|
||||
}),
|
||||
ctx,
|
||||
);
|
||||
|
||||
expect(JSON.parse(result.content)).toEqual({
|
||||
error: "Broadcast not found with the given criteria.",
|
||||
});
|
||||
});
|
||||
|
||||
it("returns a stable assistant error when broadcast finalization loses the row without prisma metadata", async () => {
|
||||
const tx = {
|
||||
notificationBroadcast: {
|
||||
create: vi.fn().mockResolvedValue({
|
||||
id: "broadcast_missing_after_create_generic",
|
||||
title: "Office update",
|
||||
targetType: "all",
|
||||
}),
|
||||
update: vi.fn().mockRejectedValue(
|
||||
Object.assign(new Error("Record to update not found"), {
|
||||
code: "P2025",
|
||||
}),
|
||||
),
|
||||
},
|
||||
notification: {
|
||||
create: vi.fn().mockResolvedValue({ id: "notification_2", userId: "user_2" }),
|
||||
},
|
||||
};
|
||||
const ctx = createToolContext(
|
||||
{
|
||||
user: {
|
||||
findMany: vi.fn().mockResolvedValue([{ id: "user_2" }]),
|
||||
},
|
||||
$transaction: vi.fn(async (callback: (db: typeof tx) => Promise<unknown>) => callback(tx)),
|
||||
notificationBroadcast: {
|
||||
create: vi.fn(),
|
||||
update: vi.fn(),
|
||||
},
|
||||
notification: {
|
||||
create: vi.fn(),
|
||||
},
|
||||
},
|
||||
SystemRole.MANAGER,
|
||||
);
|
||||
|
||||
const result = await executeTool(
|
||||
"send_broadcast",
|
||||
JSON.stringify({
|
||||
title: "Office update",
|
||||
targetType: "all",
|
||||
}),
|
||||
ctx,
|
||||
);
|
||||
|
||||
expect(JSON.parse(result.content)).toEqual({
|
||||
error: "Broadcast not found with the given criteria.",
|
||||
});
|
||||
});
|
||||
});
|
||||
@@ -0,0 +1,137 @@
|
||||
import { beforeEach, 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,
|
||||
approveEstimateVersion: vi.fn(),
|
||||
cloneEstimate: vi.fn(),
|
||||
commitDispoImportBatch: vi.fn(),
|
||||
countPlanningEntries: vi.fn().mockResolvedValue({ countsByRoleId: new Map() }),
|
||||
createEstimateExport: vi.fn(),
|
||||
createEstimatePlanningHandoff: vi.fn(),
|
||||
createEstimateRevision: vi.fn(),
|
||||
assessDispoImportReadiness: vi.fn(),
|
||||
loadResourceDailyAvailabilityContexts: vi.fn().mockResolvedValue(new Map()),
|
||||
getDashboardDemand: vi.fn().mockResolvedValue([]),
|
||||
getDashboardBudgetForecast: vi.fn().mockResolvedValue([]),
|
||||
getDashboardOverview: vi.fn(),
|
||||
getDashboardSkillGapSummary: vi.fn().mockResolvedValue({
|
||||
roleGaps: [],
|
||||
totalOpenPositions: 0,
|
||||
skillSupplyTop10: [],
|
||||
resourcesByRole: [],
|
||||
}),
|
||||
getDashboardProjectHealth: vi.fn().mockResolvedValue([]),
|
||||
getDashboardPeakTimes: vi.fn().mockResolvedValue([]),
|
||||
getDashboardTopValueResources: vi.fn().mockResolvedValue([]),
|
||||
getEstimateById: vi.fn(),
|
||||
listAssignmentBookings: vi.fn().mockResolvedValue([]),
|
||||
stageDispoImportBatch: vi.fn(),
|
||||
submitEstimateVersion: vi.fn(),
|
||||
updateEstimateDraft: vi.fn(),
|
||||
};
|
||||
});
|
||||
|
||||
import { executeTool } from "../router/assistant-tools.js";
|
||||
import { createToolContext } from "./assistant-tools-notification-test-helpers.js";
|
||||
|
||||
describe("assistant broadcast send tools", () => {
|
||||
beforeEach(() => {
|
||||
vi.clearAllMocks();
|
||||
});
|
||||
|
||||
it("creates and sends a broadcast through the notification router", async () => {
|
||||
const tx = {
|
||||
notificationBroadcast: {
|
||||
create: vi.fn().mockResolvedValue({
|
||||
id: "broadcast_1",
|
||||
title: "Office update",
|
||||
targetType: "user",
|
||||
createdAt: new Date("2026-03-30T09:00:00.000Z"),
|
||||
}),
|
||||
update: vi.fn().mockResolvedValue({
|
||||
id: "broadcast_1",
|
||||
recipientCount: 1,
|
||||
}),
|
||||
},
|
||||
notification: {
|
||||
create: vi.fn().mockResolvedValue({ id: "notification_2", userId: "user_2" }),
|
||||
},
|
||||
};
|
||||
const db = {
|
||||
notificationBroadcast: {
|
||||
create: vi.fn(),
|
||||
update: vi.fn(),
|
||||
},
|
||||
notification: {
|
||||
create: vi.fn(),
|
||||
},
|
||||
$transaction: vi.fn(async (callback: (db: typeof tx) => Promise<unknown>) => callback(tx)),
|
||||
};
|
||||
const ctx = createToolContext(db, SystemRole.ADMIN);
|
||||
|
||||
const result = await executeTool(
|
||||
"send_broadcast",
|
||||
JSON.stringify({
|
||||
title: "Office update",
|
||||
body: "New schedule",
|
||||
targetType: "user",
|
||||
targetValue: "user_2",
|
||||
category: "TASK",
|
||||
priority: "HIGH",
|
||||
channel: "in_app",
|
||||
dueDate: "2026-04-04T10:00:00.000Z",
|
||||
}),
|
||||
ctx,
|
||||
);
|
||||
|
||||
expect(tx.notificationBroadcast.create).toHaveBeenCalledWith({
|
||||
data: expect.objectContaining({
|
||||
senderId: "user_1",
|
||||
title: "Office update",
|
||||
body: "New schedule",
|
||||
category: "TASK",
|
||||
priority: "HIGH",
|
||||
channel: "in_app",
|
||||
targetType: "user",
|
||||
targetValue: "user_2",
|
||||
}),
|
||||
});
|
||||
expect(tx.notification.create).toHaveBeenCalledWith({
|
||||
data: expect.objectContaining({
|
||||
userId: "user_2",
|
||||
type: "BROADCAST_TASK",
|
||||
title: "Office update",
|
||||
body: "New schedule",
|
||||
category: "TASK",
|
||||
priority: "HIGH",
|
||||
channel: "in_app",
|
||||
sourceId: "broadcast_1",
|
||||
senderId: "user_1",
|
||||
taskStatus: "OPEN",
|
||||
dueDate: new Date("2026-04-04T10:00:00.000Z"),
|
||||
}),
|
||||
});
|
||||
expect(tx.notificationBroadcast.update).toHaveBeenCalledWith({
|
||||
where: { id: "broadcast_1" },
|
||||
data: expect.objectContaining({
|
||||
sentAt: expect.any(Date),
|
||||
recipientCount: 1,
|
||||
}),
|
||||
});
|
||||
expect(JSON.parse(result.content)).toEqual(
|
||||
expect.objectContaining({
|
||||
success: true,
|
||||
broadcastId: "broadcast_1",
|
||||
recipientCount: 1,
|
||||
message: 'Broadcast "Office update" created.',
|
||||
}),
|
||||
);
|
||||
expect(result.action).toEqual({
|
||||
type: "invalidate",
|
||||
scope: ["notification"],
|
||||
});
|
||||
});
|
||||
});
|
||||
@@ -0,0 +1,38 @@
|
||||
import { vi } from "vitest";
|
||||
|
||||
vi.mock("@capakraken/application", async (importOriginal) => {
|
||||
const actual = await importOriginal<typeof import("@capakraken/application")>();
|
||||
return {
|
||||
...actual,
|
||||
approveEstimateVersion: vi.fn(),
|
||||
cloneEstimate: vi.fn(),
|
||||
commitDispoImportBatch: vi.fn(),
|
||||
countPlanningEntries: vi.fn().mockResolvedValue({ countsByRoleId: new Map() }),
|
||||
createEstimateExport: vi.fn(),
|
||||
createEstimatePlanningHandoff: vi.fn(),
|
||||
createEstimateRevision: vi.fn(),
|
||||
assessDispoImportReadiness: vi.fn(),
|
||||
loadResourceDailyAvailabilityContexts: vi.fn().mockResolvedValue(new Map()),
|
||||
getDashboardDemand: vi.fn().mockResolvedValue([]),
|
||||
getDashboardBudgetForecast: vi.fn().mockResolvedValue([]),
|
||||
getDashboardOverview: vi.fn(),
|
||||
getDashboardSkillGapSummary: vi.fn().mockResolvedValue({
|
||||
roleGaps: [],
|
||||
totalOpenPositions: 0,
|
||||
skillSupplyTop10: [],
|
||||
resourcesByRole: [],
|
||||
}),
|
||||
getDashboardProjectHealth: vi.fn().mockResolvedValue([]),
|
||||
getDashboardPeakTimes: vi.fn().mockResolvedValue([]),
|
||||
getDashboardTopValueResources: vi.fn().mockResolvedValue([]),
|
||||
getEstimateById: vi.fn(),
|
||||
listAssignmentBookings: vi.fn().mockResolvedValue([]),
|
||||
stageDispoImportBatch: vi.fn(),
|
||||
submitEstimateVersion: vi.fn(),
|
||||
updateEstimateDraft: vi.fn(),
|
||||
};
|
||||
});
|
||||
|
||||
import { executeTool as executeAssistantTool } from "../router/assistant-tools.js";
|
||||
|
||||
export const executeTool = executeAssistantTool;
|
||||
@@ -0,0 +1,80 @@
|
||||
import { beforeEach, describe, expect, it, vi } from "vitest";
|
||||
import { SystemRole } from "@capakraken/shared";
|
||||
|
||||
import { executeTool } from "./assistant-tools-broadcast-send-test-helpers.js";
|
||||
import { createToolContext } from "./assistant-tools-notification-test-helpers.js";
|
||||
|
||||
describe("assistant broadcast send validation errors", () => {
|
||||
beforeEach(() => {
|
||||
vi.clearAllMocks();
|
||||
});
|
||||
|
||||
it("returns a stable assistant error when broadcast scheduledAt is invalid", async () => {
|
||||
const ctx = createToolContext({}, SystemRole.MANAGER);
|
||||
|
||||
const result = await executeTool(
|
||||
"send_broadcast",
|
||||
JSON.stringify({
|
||||
title: "Office update",
|
||||
targetType: "all",
|
||||
scheduledAt: "not-a-datetime",
|
||||
}),
|
||||
ctx,
|
||||
);
|
||||
|
||||
expect(JSON.parse(result.content)).toEqual({
|
||||
error: "Invalid scheduledAt: not-a-datetime",
|
||||
});
|
||||
});
|
||||
|
||||
it("returns a stable assistant error when a broadcast target resolves to no recipients", async () => {
|
||||
const create = vi.fn().mockResolvedValue({
|
||||
id: "broadcast_empty",
|
||||
title: "Office update",
|
||||
targetType: "user",
|
||||
});
|
||||
const ctx = createToolContext(
|
||||
{
|
||||
notificationBroadcast: {
|
||||
create,
|
||||
},
|
||||
},
|
||||
SystemRole.MANAGER,
|
||||
);
|
||||
|
||||
const result = await executeTool(
|
||||
"send_broadcast",
|
||||
JSON.stringify({
|
||||
title: "Office update",
|
||||
targetType: "user",
|
||||
}),
|
||||
ctx,
|
||||
);
|
||||
|
||||
expect(JSON.parse(result.content)).toEqual({
|
||||
error: "No recipients matched the broadcast target.",
|
||||
});
|
||||
expect(create).not.toHaveBeenCalled();
|
||||
});
|
||||
|
||||
it("returns a stable assistant error for scheduled broadcasts with task metadata", async () => {
|
||||
const ctx = createToolContext({}, SystemRole.MANAGER);
|
||||
|
||||
const result = await executeTool(
|
||||
"send_broadcast",
|
||||
JSON.stringify({
|
||||
title: "Approve later",
|
||||
targetType: "all",
|
||||
category: "TASK",
|
||||
taskAction: "approve_vacation:vac_1",
|
||||
dueDate: "2099-04-11T08:00:00.000Z",
|
||||
scheduledAt: "2099-04-10T08:00:00.000Z",
|
||||
}),
|
||||
ctx,
|
||||
);
|
||||
|
||||
expect(JSON.parse(result.content)).toEqual({
|
||||
error: "Scheduled broadcasts with task metadata are not supported yet.",
|
||||
});
|
||||
});
|
||||
});
|
||||
@@ -0,0 +1,104 @@
|
||||
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 } from "./assistant-tools-notification-test-helpers.js";
|
||||
|
||||
describe("assistant broadcast send tool", () => {
|
||||
it("returns a stable assistant error when a scheduled broadcast resolves to no recipients", async () => {
|
||||
const create = vi.fn();
|
||||
const db = {
|
||||
user: {
|
||||
findMany: vi.fn().mockResolvedValue([]),
|
||||
},
|
||||
notificationBroadcast: {
|
||||
create,
|
||||
},
|
||||
};
|
||||
const ctx = createToolContext(db, SystemRole.MANAGER);
|
||||
|
||||
const result = await executeTool(
|
||||
"send_broadcast",
|
||||
JSON.stringify({
|
||||
title: "Holiday planning update",
|
||||
targetType: "all",
|
||||
scheduledAt: "2099-04-10T08:00:00.000Z",
|
||||
}),
|
||||
ctx,
|
||||
);
|
||||
|
||||
expect(db.user.findMany).toHaveBeenCalledWith({
|
||||
select: { id: true },
|
||||
});
|
||||
expect(create).not.toHaveBeenCalled();
|
||||
expect(JSON.parse(result.content)).toEqual({
|
||||
error: "No recipients matched the broadcast target.",
|
||||
});
|
||||
});
|
||||
|
||||
it("returns invalidate metadata and broadcast identifiers for scheduled broadcasts", async () => {
|
||||
const scheduledAt = "2099-04-10T08:00:00.000Z";
|
||||
const create = vi.fn().mockResolvedValue({
|
||||
id: "broadcast_future",
|
||||
title: "Planned update",
|
||||
targetType: "all",
|
||||
scheduledAt: new Date(scheduledAt),
|
||||
});
|
||||
const db = {
|
||||
user: {
|
||||
findMany: vi.fn().mockResolvedValue([{ id: "user_2" }, { id: "user_3" }]),
|
||||
},
|
||||
notificationBroadcast: {
|
||||
create,
|
||||
update: vi.fn(),
|
||||
},
|
||||
notification: {
|
||||
create: vi.fn(),
|
||||
},
|
||||
};
|
||||
const ctx = createToolContext(db, SystemRole.ADMIN);
|
||||
|
||||
const result = await executeTool(
|
||||
"send_broadcast",
|
||||
JSON.stringify({
|
||||
title: "Planned update",
|
||||
targetType: "all",
|
||||
scheduledAt,
|
||||
}),
|
||||
ctx,
|
||||
);
|
||||
|
||||
expect(db.user.findMany).toHaveBeenCalledWith({
|
||||
select: { id: true },
|
||||
});
|
||||
expect(create).toHaveBeenCalledWith({
|
||||
data: expect.objectContaining({
|
||||
senderId: "user_1",
|
||||
title: "Planned update",
|
||||
targetType: "all",
|
||||
scheduledAt: new Date(scheduledAt),
|
||||
}),
|
||||
});
|
||||
expect(db.notification.create).not.toHaveBeenCalled();
|
||||
expect(db.notificationBroadcast.update).not.toHaveBeenCalled();
|
||||
expect(result.action).toEqual({ type: "invalidate", scope: ["notification"] });
|
||||
expect(JSON.parse(result.content)).toEqual(
|
||||
expect.objectContaining({
|
||||
success: true,
|
||||
broadcastId: "broadcast_future",
|
||||
recipientCount: 0,
|
||||
message: 'Broadcast "Planned update" created.',
|
||||
}),
|
||||
);
|
||||
});
|
||||
});
|
||||
Reference in New Issue
Block a user