test(api): cover assistant comment tools

2026-04-01 00:30:23 +02:00
parent f6c252be34
commit 53158dc60d
5 changed files with 476 additions and 0 deletions
@@ -0,0 +1,115 @@
+import { beforeEach, describe, expect, it, vi } from "vitest";
+import { SystemRole } from "@capakraken/shared";
+
+import { createToolContext, executeTool } from "./assistant-tools-comments-test-helpers.js";
+
+describe("assistant comment mutation tools", () => {
+  beforeEach(() => {
+    vi.clearAllMocks();
+  });
+
+  it("routes comment creation and resolution through the real comment router path", async () => {
+    const commentFindUnique = vi.fn().mockResolvedValue({
+      id: "comment_1",
+      authorId: "user_1",
+      entityType: "estimate",
+      entityId: "est_1",
+    });
+    const db = {
+      estimate: {
+        findUnique: vi.fn().mockResolvedValue({ id: "est_1" }),
+      },
+      comment: {
+        findUnique: commentFindUnique,
+        create: vi.fn().mockResolvedValue({
+          id: "comment_created",
+          body: "Please review this estimate.",
+          resolved: false,
+          createdAt: new Date("2026-03-29T11:00:00.000Z"),
+          author: { id: "user_1", name: "Assistant User", email: "assistant@example.com", image: null },
+        }),
+        update: vi.fn().mockResolvedValue({
+          id: "comment_1",
+          body: "Initial note",
+          resolved: true,
+          createdAt: new Date("2026-03-29T09:00:00.000Z"),
+          author: { id: "user_1", name: "Assistant User", email: "assistant@example.com", image: null },
+        }),
+      },
+      notification: {
+        create: vi.fn(),
+      },
+      auditLog: {
+        create: vi.fn().mockResolvedValue({ id: "audit_1" }),
+      },
+    };
+    const ctx = createToolContext(db, { userRole: SystemRole.CONTROLLER });
+
+    const createResult = await executeTool(
+      "create_comment",
+      JSON.stringify({
+        entityType: "estimate",
+        entityId: "est_1",
+        body: "Please review this estimate.",
+      }),
+      ctx,
+    );
+    const resolveResult = await executeTool(
+      "resolve_comment",
+      JSON.stringify({ commentId: "comment_1", resolved: true }),
+      ctx,
+    );
+
+    expect(db.estimate.findUnique).toHaveBeenNthCalledWith(1, {
+      where: { id: "est_1" },
+      select: { id: true },
+    });
+    expect(db.estimate.findUnique).toHaveBeenNthCalledWith(2, {
+      where: { id: "est_1" },
+      select: { id: true },
+    });
+    expect(db.comment.create).toHaveBeenCalledWith({
+      data: {
+        entityType: "estimate",
+        entityId: "est_1",
+        authorId: "user_1",
+        body: "Please review this estimate.",
+        mentions: [],
+      },
+      include: {
+        author: { select: { id: true, name: true, email: true, image: true } },
+      },
+    });
+    expect(commentFindUnique).toHaveBeenCalledWith({
+      where: { id: "comment_1" },
+      select: { id: true, authorId: true, entityType: true, entityId: true },
+    });
+    expect(db.comment.update).toHaveBeenCalledWith({
+      where: { id: "comment_1" },
+      data: { resolved: true },
+      include: {
+        author: { select: { id: true, name: true, email: true, image: true } },
+      },
+    });
+    expect(JSON.parse(createResult.content)).toEqual({
+      id: "comment_created",
+      author: "Assistant User",
+      body: "Please review this estimate.",
+      createdAt: "2026-03-29T11:00:00.000Z",
+    });
+    expect(createResult.action).toEqual({
+      type: "invalidate",
+      scope: ["comment"],
+    });
+    expect(JSON.parse(resolveResult.content)).toEqual({
+      id: "comment_1",
+      resolved: true,
+      author: "Assistant User",
+      body: "Initial note",
+    });
+    expect(resolveResult.action).toEqual({
+      type: "invalidate",
+      scope: ["comment"],
+    });
+  });
+});