test(api): cover assistant estimate reads

2026-04-01 00:37:38 +02:00
parent c510eeae37
commit 80c31cc53f
5 changed files with 487 additions and 0 deletions
@@ -0,0 +1,52 @@
+import {
+  approveEstimateVersion,
+  cloneEstimate,
+  createEstimateExport,
+  createEstimatePlanningHandoff,
+  createEstimateRevision,
+  getEstimateById,
+  submitEstimateVersion,
+  updateEstimateDraft,
+} from "@capakraken/application";
+import { PermissionKey, SystemRole } from "@capakraken/shared";
+import { vi } from "vitest";
+
+import type { ToolContext } from "../router/assistant-tools.js";
+
+export function createToolContext(
+  db: Record<string, unknown>,
+  options?: {
+    permissions?: PermissionKey[];
+    userRole?: SystemRole;
+  },
+): ToolContext {
+  const userRole = options?.userRole ?? SystemRole.ADMIN;
+  return {
+    db: db as ToolContext["db"],
+    userId: "user_1",
+    userRole,
+    permissions: new Set(options?.permissions ?? []),
+    session: {
+      user: { email: "assistant@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 resetEstimateToolMocks() {
+  vi.clearAllMocks();
+  vi.mocked(approveEstimateVersion).mockReset();
+  vi.mocked(cloneEstimate).mockReset();
+  vi.mocked(createEstimateExport).mockReset();
+  vi.mocked(createEstimatePlanningHandoff).mockReset();
+  vi.mocked(createEstimateRevision).mockReset();
+  vi.mocked(getEstimateById).mockReset();
+  vi.mocked(submitEstimateVersion).mockReset();
+  vi.mocked(updateEstimateDraft).mockReset();
+}