test(api): cover assistant notification inbox tools

2026-03-31 23:56:35 +02:00
parent 8c9e43512f
commit 25ce562d17
3 changed files with 187 additions and 0 deletions
@@ -0,0 +1,34 @@
+import { vi } from "vitest";
+import { SystemRole } from "@capakraken/shared";
+import type { ToolContext } from "../router/assistant-tools.js";
+
+export function createToolContext(
+  db: Record<string, unknown>,
+  userRole: SystemRole = SystemRole.USER,
+): ToolContext {
+  return {
+    db: db as ToolContext["db"],
+    userId: "user_1",
+    userRole,
+    permissions: new Set(),
+    session: {
+      user: { email: "user@example.com", name: "Assistant User", image: null },
+      expires: "2026-03-29T00:00:00.000Z",
+    },
+    dbUser: {
+      id: "user_1",
+      systemRole: userRole,
+      permissionOverrides: null,
+    },
+    roleDefaults: null,
+  };
+}
+
+export function withUserLookup(db: Record<string, unknown>, userId = "user_1") {
+  return {
+    user: {
+      findUnique: vi.fn().mockResolvedValue({ id: userId }),
+    },
+    ...db,
+  };
+}