Multi subscriptions

__tests__/components/distribution-plan-tool/ReviewDistributionPlanTableSubscription.test.tsx

@@ -4,12 +4,10 @@ jest.mock("@/services/api/common-api", () => ({
   commonApiPost: jest.fn(),
 }));
 
-import {
-  download,
-  isSubscriptionsAdmin,
-  SubscriptionConfirm,
-} from "@/components/distribution-plan-tool/review-distribution-plan/table/ReviewDistributionPlanTableSubscription";
+import * as ReviewDistributionPlanTableSubscriptionModule from "@/components/distribution-plan-tool/review-distribution-plan/table/ReviewDistributionPlanTableSubscription";
 import { ApiIdentity } from "@/generated/models/ApiIdentity";
+const { download, isSubscriptionsAdmin, SubscriptionConfirm } =
+  ReviewDistributionPlanTableSubscriptionModule;
 
 jest.mock("react-bootstrap", () => ({
   __esModule: true,
@@ -59,7 +57,18 @@ describe("ReviewDistributionPlanTableSubscription utilities", () => {
   });
 });
 
-import { render, screen } from "@testing-library/react";
+import { AuthContext } from "@/components/auth/Auth";
+import { DistributionPlanToolContext } from "@/components/distribution-plan-tool/DistributionPlanToolContext";
+import { ReviewDistributionPlanTableItemType } from "@/components/distribution-plan-tool/review-distribution-plan/table/ReviewDistributionPlanTable";
+import { PUBLIC_SUBSCRIPTIONS_PHASE_ID } from "@/components/distribution-plan-tool/review-distribution-plan/table/constants";
+import { render, screen, waitFor } from "@testing-library/react";
+import userEvent from "@testing-library/user-event";
+const { SubscriptionLinks } = ReviewDistributionPlanTableSubscriptionModule;
+
+jest.mock("@/components/distribution-plan-tool/common/CircleLoader", () => ({
+  __esModule: true,
+  default: () => <div data-testid="circle-loader">Loading...</div>,
+}));
 
 test("SubscriptionConfirm extracts token id from plan name", () => {
   render(
@@ -74,3 +83,218 @@ test("SubscriptionConfirm extracts token id from plan name", () => {
   const input = screen.getByRole("spinbutton") as HTMLInputElement;
   expect(input.value).toBe("123");
 });
+
+test("SubscriptionConfirm displays token id as text when confirmedTokenId is provided", () => {
+  render(
+    <SubscriptionConfirm
+      title="t"
+      plan={{ id: "1", name: "Meme 123 drop" } as any}
+      show={true}
+      handleClose={jest.fn()}
+      confirmedTokenId="456"
+      onConfirm={jest.fn()}
+    />
+  );
+  expect(screen.queryByRole("spinbutton")).not.toBeInTheDocument();
+  expect(screen.getByText("456")).toBeInTheDocument();
+});
+
+test("SubscriptionConfirm shows input when confirmedTokenId is not provided", () => {
+  render(
+    <SubscriptionConfirm
+      title="t"
+      plan={{ id: "1", name: "Meme 123 drop" } as any}
+      show={true}
+      handleClose={jest.fn()}
+      onConfirm={jest.fn()}
+    />
+  );
+  const input = screen.getByRole("spinbutton") as HTMLInputElement;
+  expect(input).toBeInTheDocument();
+  expect(input.value).toBe("123");
+});
+
+describe("SubscriptionLinks", () => {
+  const mockSetToast = jest.fn();
+  const mockAuthCtx = {
+    connectedProfile: {
+      wallets: [{ wallet: "0x0187C9a182736ba18b44eE8134eE438374cf87DC" }],
+    },
+    setToast: mockSetToast,
+  };
+
+  const mockPlan = { id: "plan1", name: "Meme 123 drop" } as any;
+  const mockDistCtx = {
+    confirmedTokenId: null,
+  } as any;
+
+  beforeEach(() => {
+    jest.clearAllMocks();
+    mockFetch.mockResolvedValue({
+      airdrops: [],
+      airdrops_unconsolidated: [],
+      allowlists: [],
+    });
+    const mockURL = {
+      createObjectURL: jest.fn(() => "blob:mock-url"),
+      revokeObjectURL: jest.fn(),
+    };
+    globalThis.URL = mockURL as any;
+  });
+
+  afterEach(() => {
+    jest.restoreAllMocks();
+  });
+
+  it("renders Public Subscriptions button for public phase", () => {
+    const publicPhase = {
+      id: "phase1",
+      name: "public",
+      type: ReviewDistributionPlanTableItemType.PHASE,
+      phaseId: PUBLIC_SUBSCRIPTIONS_PHASE_ID,
+    } as any;
+
+    render(
+      <DistributionPlanToolContext.Provider value={mockDistCtx}>
+        <AuthContext.Provider value={mockAuthCtx as any}>
+          <SubscriptionLinks plan={mockPlan} phase={publicPhase} />
+        </AuthContext.Provider>
+      </DistributionPlanToolContext.Provider>
+    );
+
+    expect(screen.getByText("Public Subscriptions")).toBeInTheDocument();
+  });
+
+  it("shows confirm modal when Public Subscriptions button is clicked", async () => {
+    const user = userEvent.setup();
+    const publicPhase = {
+      id: "phase1",
+      name: "public",
+      type: ReviewDistributionPlanTableItemType.PHASE,
+      phaseId: PUBLIC_SUBSCRIPTIONS_PHASE_ID,
+    } as any;
+
+    render(
+      <DistributionPlanToolContext.Provider value={mockDistCtx}>
+        <AuthContext.Provider value={mockAuthCtx as any}>
+          <SubscriptionLinks plan={mockPlan} phase={publicPhase} />
+        </AuthContext.Provider>
+      </DistributionPlanToolContext.Provider>
+    );
+
+    await user.click(screen.getByText("Public Subscriptions"));
+
+    await waitFor(() => {
+      expect(
+        screen.getByText("Confirm Download Public Subscriptions Info")
+      ).toBeInTheDocument();
+    });
+  });
+
+  it("calls download and shows toast when Public Subscriptions is confirmed", async () => {
+    const user = userEvent.setup();
+
+    jest
+      .spyOn(
+        ReviewDistributionPlanTableSubscriptionModule,
+        "isSubscriptionsAdmin"
+      )
+      .mockReturnValue(true);
+    jest
+      .spyOn(ReviewDistributionPlanTableSubscriptionModule, "download")
+      .mockResolvedValue({
+        success: true,
+        message: "Download successful",
+      });
+
+    const publicPhase = {
+      id: "phase1",
+      name: "public",
+      type: ReviewDistributionPlanTableItemType.PHASE,
+      phaseId: PUBLIC_SUBSCRIPTIONS_PHASE_ID,
+    } as any;
+
+    render(
+      <DistributionPlanToolContext.Provider value={mockDistCtx}>
+        <AuthContext.Provider value={mockAuthCtx as any}>
+          <SubscriptionLinks plan={mockPlan} phase={publicPhase} />
+        </AuthContext.Provider>
+      </DistributionPlanToolContext.Provider>
+    );
+
+    await user.click(screen.getByText("Public Subscriptions"));
+
+    await waitFor(() => {
+      expect(
+        screen.getByText("Confirm Download Public Subscriptions Info")
+      ).toBeInTheDocument();
+    });
+
+    await user.click(screen.getByText("Looks good"));
+
+    await waitFor(() => {
+      expect(mockSetToast).toHaveBeenCalledWith({
+        type: "success",
+        message: "Download successful",
+      });
+    });
+  });
+
+  it("does not render for non-admin users", () => {
+    const nonAdminCtx = {
+      connectedProfile: {
+        wallets: [{ wallet: "0x123" }],
+      },
+      setToast: jest.fn(),
+    };
+
+    const publicPhase = {
+      id: "phase1",
+      name: "public",
+      type: ReviewDistributionPlanTableItemType.PHASE,
+      phaseId: PUBLIC_SUBSCRIPTIONS_PHASE_ID,
+    } as any;
+
+    const { container } = render(
+      <DistributionPlanToolContext.Provider value={mockDistCtx}>
+        <AuthContext.Provider value={nonAdminCtx as any}>
+          <SubscriptionLinks plan={mockPlan} phase={publicPhase} />
+        </AuthContext.Provider>
+      </DistributionPlanToolContext.Provider>
+    );
+
+    expect(container.innerHTML).toBe("");
+  });
+
+  it("displays token id as text when confirmedTokenId is provided in context", async () => {
+    const user = userEvent.setup();
+    const distCtxWithTokenId = {
+      confirmedTokenId: "789",
+    } as any;
+    const publicPhase = {
+      id: "phase1",
+      name: "public",
+      type: ReviewDistributionPlanTableItemType.PHASE,
+      phaseId: PUBLIC_SUBSCRIPTIONS_PHASE_ID,
+    } as any;
+
+    render(
+      <DistributionPlanToolContext.Provider value={distCtxWithTokenId}>
+        <AuthContext.Provider value={mockAuthCtx as any}>
+          <SubscriptionLinks plan={mockPlan} phase={publicPhase} />
+        </AuthContext.Provider>
+      </DistributionPlanToolContext.Provider>
+    );
+
+    await user.click(screen.getByText("Public Subscriptions"));
+
+    await waitFor(() => {
+      expect(
+        screen.getByText("Confirm Download Public Subscriptions Info")
+      ).toBeInTheDocument();
+    });
+
+    expect(screen.queryByRole("spinbutton")).not.toBeInTheDocument();
+    expect(screen.getByText("789")).toBeInTheDocument();
+  });
+});