shoc-frontend-new/src/test/app/(protected)/workorders/wizard-service-clears-vendor.test.tsx

import { fireEvent, render, screen } from "@testing-library/react";
import { describe, expect, it, vi } from "vitest";
import { WizardStepLocationServiceSelect } from "@/app/(protected)/workorders/_components/wizard/wizard-step-location-service-select";
import {
  EMPTY_WIZARD_DRAFT,
  type WorkOrderWizardDraft,
} from "@/domain/work-orders/types/work-order-wizard";

describe("WizardStepLocationServiceSelect — vendor clear on service change", () => {
  it("clears vendor fields when the primary service changes", () => {
    const onPatch = vi.fn();
    const draft: WorkOrderWizardDraft = {
      ...EMPTY_WIZARD_DRAFT,
      pm: "HVAC",
      vendorId: "9",
      vendorName: "Old Vendor",
      techPhone: "555-0100",
      vendorNotes: "Old company notes",
    };

    render(
      <WizardStepLocationServiceSelect
        draft={draft}
        services={[
          { id: "1", value: "HVAC" },
          { id: "2", value: "Plumbing" },
        ]}
        svcOpen={false}
        onSvcOpenChange={vi.fn()}
        svcSearch=""
        onSvcSearchChange={vi.fn()}
        onUpdate={vi.fn()}
        onPatch={onPatch}
      />,
    );

    fireEvent.click(screen.getByRole("button", { name: "HVAC" }));
    fireEvent.click(screen.getByRole("button", { name: "Plumbing" }));

    expect(onPatch).toHaveBeenCalledWith(
      expect.objectContaining({
        pm: "Plumbing",
        extraServices: [],
        vendorId: "",
        vendorName: "",
        techPhone: "",
        vendorCompany: "",
        vendorNotes: "",
        newTechnicians: [],
      }),
    );
  });

  it("replaces the selected service in Service & Notes without persisting extraServices", () => {
    const onPatch = vi.fn();
    const draft: WorkOrderWizardDraft = {
      ...EMPTY_WIZARD_DRAFT,
      pm: "HVAC",
      extraServices: ["Plumbing"],
      vendorId: "9",
      vendorName: "Old Vendor",
      techPhone: "555-0100",
      vendorNotes: "Old company notes",
    };

    render(
      <WizardStepLocationServiceSelect
        draft={draft}
        services={[
          { id: "1", value: "HVAC" },
          { id: "2", value: "Plumbing" },
          { id: "3", value: "Electrical" },
        ]}
        svcOpen={true}
        onSvcOpenChange={vi.fn()}
        svcSearch=""
        onSvcSearchChange={vi.fn()}
        onUpdate={vi.fn()}
        onPatch={onPatch}
      />,
    );

    fireEvent.click(screen.getByRole("button", { name: "Electrical" }));

    expect(onPatch).toHaveBeenCalledWith({
      pm: "Electrical",
      serviceId: "3",
      extraServices: [],
      vendorId: "",
      vendorName: "",
      techPhone: "",
      vendorCompany: "",
      vendorNotes: "",
      newTechnicians: [],
    });
  });

  it("clears the vendor company and staged inline technicians when the service changes", () => {
    const onPatch = vi.fn();
    const draft: WorkOrderWizardDraft = {
      ...EMPTY_WIZARD_DRAFT,
      pm: "HVAC",
      vendorCompany: "Vinewood LLC",
      newTechnicians: [{ name: "Jordan Lee", phone: "(314) 555-0199" }],
    };

    render(
      <WizardStepLocationServiceSelect
        draft={draft}
        services={[
          { id: "1", value: "HVAC" },
          { id: "2", value: "Plumbing" },
        ]}
        svcOpen={false}
        onSvcOpenChange={vi.fn()}
        svcSearch=""
        onSvcSearchChange={vi.fn()}
        onUpdate={vi.fn()}
        onPatch={onPatch}
      />,
    );

    fireEvent.click(screen.getByRole("button", { name: "HVAC" }));
    fireEvent.click(screen.getByRole("button", { name: "Plumbing" }));

    expect(onPatch).toHaveBeenCalledWith({
      pm: "Plumbing",
      serviceId: "2",
      extraServices: [],
      vendorId: "",
      vendorName: "",
      techPhone: "",
      vendorCompany: "",
      vendorNotes: "",
      newTechnicians: [],
    });
  });

  it("keeps vendor company and staged technicians when the service is unchanged", () => {
    const onPatch = vi.fn();
    const draft: WorkOrderWizardDraft = {
      ...EMPTY_WIZARD_DRAFT,
      pm: "HVAC",
      vendorCompany: "Vinewood LLC",
      newTechnicians: [{ name: "Jordan Lee", phone: "(314) 555-0199" }],
    };

    render(
      <WizardStepLocationServiceSelect
        draft={draft}
        services={[{ id: "1", value: "HVAC" }]}
        svcOpen={false}
        onSvcOpenChange={vi.fn()}
        svcSearch=""
        onSvcSearchChange={vi.fn()}
        onUpdate={vi.fn()}
        onPatch={onPatch}
      />,
    );

    fireEvent.click(screen.getByRole("button", { name: "HVAC" }));
    fireEvent.click(screen.getByRole("button", { name: "HVAC" }));

    expect(onPatch).toHaveBeenCalledWith({ pm: "HVAC", serviceId: "1", extraServices: [] });
  });
});