test(api): cover notification and user edge cases

packages/api/src/__tests__/user-router.test.ts

@@ -0,0 +1,121 @@
+import { SystemRole } from "@capakraken/shared";
+import { describe, expect, it, vi } from "vitest";
+import { userRouter } from "../router/user.js";
+import { createCallerFactory } from "../trpc.js";
+
+const createCaller = createCallerFactory(userRouter);
+
+function createAdminCaller(db: Record<string, unknown>) {
+  return createCaller({
+    session: {
+      user: { email: "admin@example.com", name: "Admin", image: null },
+      expires: "2099-01-01T00:00:00.000Z",
+    },
+    db: db as never,
+    dbUser: {
+      id: "user_admin",
+      systemRole: SystemRole.ADMIN,
+      permissionOverrides: null,
+    },
+    roleDefaults: null,
+  });
+}
+
+describe("user.linkResource", () => {
+  it("returns NOT_FOUND when the target user does not exist", async () => {
+    const userFindUnique = vi.fn().mockResolvedValue(null);
+    const resourceFindUnique = vi.fn();
+    const updateMany = vi.fn();
+    const update = vi.fn();
+    const caller = createAdminCaller({
+      user: {
+        findUnique: userFindUnique,
+      },
+      resource: {
+        findUnique: resourceFindUnique,
+        updateMany,
+        update,
+      },
+    });
+
+    await expect(caller.linkResource({
+      userId: "missing_user",
+      resourceId: "resource_1",
+    })).rejects.toMatchObject({
+      code: "NOT_FOUND",
+      message: "User not found",
+    });
+
+    expect(userFindUnique).toHaveBeenCalledWith({
+      where: { id: "missing_user" },
+      select: { id: true },
+    });
+    expect(resourceFindUnique).not.toHaveBeenCalled();
+    expect(updateMany).not.toHaveBeenCalled();
+    expect(update).not.toHaveBeenCalled();
+  });
+
+  it("returns NOT_FOUND when the target resource does not exist", async () => {
+    const userFindUnique = vi.fn().mockResolvedValue({ id: "user_1" });
+    const resourceFindUnique = vi.fn().mockResolvedValue(null);
+    const updateMany = vi.fn();
+    const update = vi.fn();
+    const caller = createAdminCaller({
+      user: {
+        findUnique: userFindUnique,
+      },
+      resource: {
+        findUnique: resourceFindUnique,
+        updateMany,
+        update,
+      },
+    });
+
+    await expect(caller.linkResource({
+      userId: "user_1",
+      resourceId: "missing_resource",
+    })).rejects.toMatchObject({
+      code: "NOT_FOUND",
+      message: "Resource not found",
+    });
+
+    expect(resourceFindUnique).toHaveBeenCalledWith({
+      where: { id: "missing_resource" },
+      select: { id: true },
+    });
+    expect(updateMany).not.toHaveBeenCalled();
+    expect(update).not.toHaveBeenCalled();
+  });
+
+  it("unlinks existing assignments before linking the requested resource", async () => {
+    const userFindUnique = vi.fn().mockResolvedValue({ id: "user_1" });
+    const resourceFindUnique = vi.fn().mockResolvedValue({ id: "resource_1" });
+    const updateMany = vi.fn().mockResolvedValue({ count: 1 });
+    const update = vi.fn().mockResolvedValue({ id: "resource_1", userId: "user_1" });
+    const caller = createAdminCaller({
+      user: {
+        findUnique: userFindUnique,
+      },
+      resource: {
+        findUnique: resourceFindUnique,
+        updateMany,
+        update,
+      },
+    });
+
+    const result = await caller.linkResource({
+      userId: "user_1",
+      resourceId: "resource_1",
+    });
+
+    expect(result).toEqual({ success: true });
+    expect(updateMany).toHaveBeenCalledWith({
+      where: { userId: "user_1" },
+      data: { userId: null },
+    });
+    expect(update).toHaveBeenCalledWith({
+      where: { id: "resource_1" },
+      data: { userId: "user_1" },
+    });
+  });
+});