shoc-frontend-new/src/test/app/(protected)/workorders/media-upload-modal-api.test.tsx

import { render, screen, waitFor } from "@testing-library/react";
import { QueryClient, QueryClientProvider } from "@tanstack/react-query";
import { beforeEach, describe, expect, it, vi } from "vitest";
import { MediaUploadModal } from "@/app/(protected)/workorders/_components/media/media-upload-modal";
import type { WorkOrderTableRow } from "@/domain/work-orders/types/work-order-table-row";

const uploadMutateAsync = vi.fn();
const deleteMutate = vi.fn();
const categorizeMutate = vi.fn();

vi.mock("@/domain/work-orders/use-cases/use-work-order-detail", () => ({
  useWorkOrderBoardDetail: () => ({
    data: {
      info: { status: "Scheduled" },
      media: [
        {
          id: "11",
          fileName: "before.jpg",
          url: "https://example.com/before.jpg",
          category: "Before",
          isLegacy: false,
        },
      ],
      comments: [],
      audit: [],
      completion: {},
    },
    isLoading: false,
  }),
}));

vi.mock("@/domain/work-orders/use-cases/use-work-order-mutations", () => ({
  useUploadWorkOrderMedia: () => ({
    mutate: vi.fn(),
    mutateAsync: uploadMutateAsync,
    isPending: false,
  }),
  useUpdateWorkOrderMediaCategory: () => ({ mutate: categorizeMutate, isPending: false }),
  useDeleteWorkOrderMedia: () => ({ mutate: deleteMutate, isPending: false }),
}));

function baseRow(): WorkOrderTableRow {
  return {
    id: 7,
    woNumber: "WO-7",
    site: "SEA1",
    type: "Emergency",
    dispatcherId: "u1",
    dispatcherName: "Dispatcher",
    scheduledOn: "2026-06-01",
    completedDate: "",
    targetWeek: "",
    dayGroup: "",
    dayLabel: "",
    location: "Seattle",
    locationId: "loc-1",
    pm: "",
    serviceNotes: "",
    extraServices: [],
    vendorId: "",
    company: "",
    tech: "",
    techPhone: "",
    apptStart: "",
    apptEnd: "",
    apptTime: "",
    status: "Scheduled",
    docStatus: "",
    pocName: "",
    pocPhone: "",
    pocNotes: "",
    dueDate: "",
    rescheduleCount: 0,
    carriedOver: 0,
    originalWeek: "",
    originalDate: "",
    isPastDue: false,
    scheduleWeekOnly: false,
    mediaCount: 1,
    title: "Emergency",
    priority: "Sev 1",
    primaryDispatchId: null,
    rowVersion: "",
    dispatchRowVersion: "",
    flagColor: null,
    upliftSummary: { hasUplift: false, pendingCount: 0 },
  };
}

describe("MediaUploadModal API wiring", () => {
  beforeEach(() => {
    uploadMutateAsync.mockReset();
    deleteMutate.mockReset();
    categorizeMutate.mockReset();
  });

  it("renders persisted media from board detail (refresh-safe)", async () => {
    const client = new QueryClient({
      defaultOptions: { queries: { retry: false }, mutations: { retry: false } },
    });

    render(
      <QueryClientProvider client={client}>
        <MediaUploadModal row={baseRow()} open onOpenChange={vi.fn()} />
      </QueryClientProvider>,
    );

    await waitFor(() => {
      expect(screen.getByText("before.jpg")).toBeInTheDocument();
    });
    expect(screen.getByText("Photos & Videos")).toBeInTheDocument();
    expect(screen.getByText(/categorize after upload/i)).toBeInTheDocument();
    expect(screen.getByLabelText(/category for before\.jpg/i)).toBeInTheDocument();
  });
});