OpenHands/frontend/__tests__/hooks/mutation/use-invite-members-batch.test.tsx

import { renderHook, waitFor } from "@testing-library/react";
import { describe, expect, it, vi, beforeEach } from "vitest";
import { QueryClient, QueryClientProvider } from "@tanstack/react-query";
import { useInviteMembersBatch } from "#/hooks/mutation/use-invite-members-batch";
import { useSelectedOrganizationStore } from "#/stores/selected-organization-store";

// Mock the useRevalidator hook from react-router
vi.mock("react-router", () => ({
  useRevalidator: () => ({
    revalidate: vi.fn(),
  }),
}));

describe("useInviteMembersBatch", () => {
  beforeEach(() => {
    useSelectedOrganizationStore.setState({ organizationId: null });
  });

  it("should throw an error when organizationId is null", async () => {
    const queryClient = new QueryClient({
      defaultOptions: {
        mutations: {
          retry: false,
        },
      },
    });

    const { result } = renderHook(() => useInviteMembersBatch(), {
      wrapper: ({ children }) => (
        <QueryClientProvider client={queryClient}>{children}</QueryClientProvider>
      ),
    });

    // Attempt to mutate without organizationId
    result.current.mutate({ emails: ["test@example.com"] });

    // Should fail with an error about missing organizationId
    await waitFor(() => {
      expect(result.current.isError).toBe(true);
    });

    expect(result.current.error).toBeInstanceOf(Error);
    expect(result.current.error?.message).toBe("Organization ID is required");
  });
});