@carlonicora/nextjs-jsonapi 1.131.1 → 1.131.2

package/src/features/user/contexts/__tests__/CurrentUserContext.spec.tsx

@@ -1,5 +1,5 @@
 import { describe, it, expect, vi, beforeEach } from "vitest";
-import { render } from "@testing-library/react";
+import { render, act } from "@testing-library/react";
 
 vi.mock("cookies-next", () => ({
   getCookie: vi.fn(() => undefined),
@@ -9,14 +9,54 @@ vi.mock("next/navigation", () => ({
   usePathname: () => "/",
 }));
 
+const socketMock = vi.hoisted(() => {
+  const handlers: Record<string, ((data: any) => void)[]> = {};
+  return {
+    socket: {
+      on: (event: string, handler: (data: any) => void) => {
+        (handlers[event] ??= []).push(handler);
+      },
+      off: (event: string, handler: (data: any) => void) => {
+        handlers[event] = (handlers[event] ?? []).filter((h) => h !== handler);
+      },
+    },
+    emit: (event: string, data: any) => {
+      (handlers[event] ?? []).forEach((h) => h(data));
+    },
+    reset: () => {
+      for (const key of Object.keys(handlers)) delete handlers[key];
+    },
+  };
+});
+
 vi.mock("../../../../contexts/SocketContext", () => ({
-  useSocketContext: () => ({ socket: null, isConnected: false }),
+  useSocketContext: () => ({ socket: socketMock.socket, isConnected: true }),
 }));
 
-vi.mock("../../../../core", () => ({
-  Modules: { User: { name: "users" } },
-  rehydrate: (_module: unknown, data: any) => data,
-}));
+vi.mock("../../../../core", () => {
+  // Mirrors the real Company model: public getters over private backing fields.
+  // The token overlay writes the backing fields, so a plain object would not
+  // exercise it.
+  class FakeCompany {
+    id = "";
+    monthlyTokens = 0;
+    _availableMonthlyTokens = 0;
+    _availableExtraTokens = 0;
+    get availableMonthlyTokens() {
+      return this._availableMonthlyTokens ?? 0;
+    }
+    get availableExtraTokens() {
+      return this._availableExtraTokens ?? 0;
+    }
+  }
+  return {
+    Modules: { User: { name: "users" } },
+    rehydrate: (_module: unknown, data: any) => ({
+      ...data,
+      company: data.company ? Object.assign(new FakeCompany(), data.company) : null,
+    }),
+  };
+});
 
 vi.mock("../../../../permissions", () => ({
   Action: { Read: "read", Write: "write", Create: "create", Update: "update", Delete: "delete" },
@@ -139,3 +179,106 @@ describe("CurrentUserProvider", () => {
     expect(getByTestId("current-user-id").textContent).toBe("new");
   });
 });
+
+describe("CurrentUserProvider token updates", () => {
+  const STORED_USER = {
+    id: "user-1",
+    roles: [],
+    modules: [],
+    company: {
+      id: "company-1",
+      monthlyTokens: 1000,
+      _availableMonthlyTokens: 900,
+      _availableExtraTokens: 50,
+    },
+  };
+
+  function CompanyConsumer() {
+    const { company } = useCurrentUserContext();
+    return (
+      <div data-testid="balances">
+        {company ? `${(company as any).availableMonthlyTokens}/${(company as any).availableExtraTokens}` : "null"}
+      </div>
+    );
+  }
+
+  beforeEach(() => {
+    localStorage.clear();
+    socketMock.reset();
+    vi.clearAllMocks();
+  });
+
+  it("patches balances from company:tokens_updated without refetching the user", async () => {
+    localStorage.setItem("user", JSON.stringify(STORED_USER));
+
+    const { getByTestId } = render(
+      <CurrentUserProvider>
+        <CompanyConsumer />
+      </CurrentUserProvider>,
+    );
+
+    await new Promise((resolve) => setTimeout(resolve, 50));
+    expect(getByTestId("balances").textContent).toBe("900/50");
+
+    await act(async () => {
+      socketMock.emit("company:tokens_updated", {
+        type: "company:tokens_updated",
+        companyId: "company-1",
+        availableMonthlyTokens: 750,
+        availableExtraTokens: 40,
+      });
+    });
+
+    expect(getByTestId("balances").textContent).toBe("750/40");
+    expect(UserService.findFullUser).not.toHaveBeenCalled();
+  });
+
+  it("ignores company:tokens_updated for a different company", async () => {
+    localStorage.setItem("user", JSON.stringify(STORED_USER));
+
+    const { getByTestId } = render(
+      <CurrentUserProvider>
+        <CompanyConsumer />
+      </CurrentUserProvider>,
+    );
+
+    await new Promise((resolve) => setTimeout(resolve, 50));
+
+    await act(async () => {
+      socketMock.emit("company:tokens_updated", {
+        type: "company:tokens_updated",
+        companyId: "someone-else",
+        availableMonthlyTokens: 1,
+        availableExtraTokens: 1,
+      });
+    });
+
+    expect(getByTestId("balances").textContent).toBe("900/50");
+  });
+
+  it("still refetches the full user on company:subscription_updated", async () => {
+    localStorage.setItem("user", JSON.stringify(STORED_USER));
+    vi.mocked(UserService.findFullUser).mockResolvedValue({
+      ...STORED_USER,
+      dehydrate: () => STORED_USER,
+    } as any);
+
+    render(
+      <CurrentUserProvider>
+        <CompanyConsumer />
+      </CurrentUserProvider>,
+    );
+
+    await new Promise((resolve) => setTimeout(resolve, 50));
+
+    await act(async () => {
+      socketMock.emit("company:subscription_updated", {
+        type: "company:subscription_updated",
+        companyId: "company-1",
+      });
+      await new Promise((resolve) => setTimeout(resolve, 50));
+    });
+
+    expect(UserService.findFullUser).toHaveBeenCalledTimes(1);
+  });
+});