auto-virtual-tryon-frontend/tests/features/orders/submit-workbench.test.tsx

import {
  fireEvent,
  render,
  screen,
  waitFor,
  within,
} from "@testing-library/react";
import { afterEach, beforeEach, expect, test, vi } from "vitest";

import { SubmitWorkbench } from "@/features/orders/submit-workbench";
import {
  GARMENT_LIBRARY_FIXTURES,
  MODEL_LIBRARY_FIXTURES,
  SCENE_LIBRARY_FIXTURES,
} from "@/lib/mock/libraries";

const { pushMock } = vi.hoisted(() => ({
  pushMock: vi.fn(),
}));

vi.mock("next/navigation", () => ({
  useRouter: () => ({
    push: pushMock,
  }),
}));

type LibraryType = "models" | "scenes" | "garments";

function createLibraryPayload(libraryType: LibraryType) {
  const fixtureMap = {
    models: MODEL_LIBRARY_FIXTURES,
    scenes: SCENE_LIBRARY_FIXTURES,
    garments: GARMENT_LIBRARY_FIXTURES,
  };

  return {
    mode: "placeholder",
    message: "资源库当前使用占位数据，真实后端接口尚未提供。",
    data: {
      items: fixtureMap[libraryType],
    },
  };
}

function createFetchMock({
  orderResponse,
}: {
  orderResponse?: Response;
} = {}) {
  return vi.fn(async (input: RequestInfo | URL, init?: RequestInit) => {
    const url = typeof input === "string" ? input : input.toString();

    if (url === "/api/libraries/models") {
      return new Response(JSON.stringify(createLibraryPayload("models")), {
        status: 200,
        headers: { "content-type": "application/json" },
      });
    }

    if (url === "/api/libraries/scenes") {
      return new Response(JSON.stringify(createLibraryPayload("scenes")), {
        status: 200,
        headers: { "content-type": "application/json" },
      });
    }

    if (url === "/api/libraries/garments") {
      return new Response(JSON.stringify(createLibraryPayload("garments")), {
        status: 200,
        headers: { "content-type": "application/json" },
      });
    }

    if (url === "/api/orders" && init?.method === "POST" && orderResponse) {
      return orderResponse;
    }

    throw new Error(`Unhandled fetch: ${url}`);
  });
}

beforeEach(() => {
  pushMock.mockReset();
});

afterEach(() => {
  vi.unstubAllGlobals();
});

test("forces low customers to use auto_basic and mid customers to use semi_pro", async () => {
  vi.stubGlobal("fetch", createFetchMock());

  render(<SubmitWorkbench />);

  const customerLevelSelect = await screen.findByLabelText("客户层级");
  const serviceModeSelect = screen.getByLabelText("服务模式");

  fireEvent.change(customerLevelSelect, {
    target: { value: "low" },
  });

  expect(serviceModeSelect).toHaveValue("auto_basic");
  expect(
    screen.getByRole("option", { name: "半人工专业处理 semi_pro" }),
  ).toBeDisabled();

  fireEvent.change(customerLevelSelect, {
    target: { value: "mid" },
  });

  expect(serviceModeSelect).toHaveValue("semi_pro");
  expect(
    screen.getByRole("option", { name: "自动基础处理 auto_basic" }),
  ).toBeDisabled();
});

test("preserves selected values when order submission fails", async () => {
  vi.stubGlobal(
    "fetch",
    createFetchMock({
      orderResponse: new Response(
        JSON.stringify({
          error: "BACKEND_UNAVAILABLE",
          message: "后端暂时不可用，请稍后重试。",
        }),
        {
          status: 502,
          headers: { "content-type": "application/json" },
        },
      ),
    }),
  );

  render(<SubmitWorkbench />);

  await screen.findByText("Ava / Studio");

  fireEvent.change(screen.getByLabelText("客户层级"), {
    target: { value: "low" },
  });
  fireEvent.change(screen.getByLabelText("模特资源"), {
    target: { value: "model-ava" },
  });
  fireEvent.change(screen.getByLabelText("场景资源"), {
    target: { value: "scene-loft" },
  });
  fireEvent.change(screen.getByLabelText("服装资源"), {
    target: { value: "garment-coat-01" },
  });

  const summaryCard = screen.getByRole("region", { name: "提单摘要" });
  expect(within(summaryCard).getByText("Ava / Studio")).toBeInTheDocument();
  expect(within(summaryCard).getByText("Loft Window")).toBeInTheDocument();
  expect(within(summaryCard).getByText("Structured Coat 01")).toBeInTheDocument();

  fireEvent.click(screen.getByRole("button", { name: "提交订单" }));

  expect(
    await screen.findByText("后端暂时不可用，请稍后重试。"),
  ).toBeInTheDocument();
  expect(screen.getByLabelText("客户层级")).toHaveValue("low");
  expect(screen.getByLabelText("服务模式")).toHaveValue("auto_basic");
  expect(screen.getByLabelText("模特资源")).toHaveValue("model-ava");
  expect(screen.getByLabelText("场景资源")).toHaveValue("scene-loft");
  expect(screen.getByLabelText("服装资源")).toHaveValue("garment-coat-01");
});

test("submits the selected resources, surfaces returned ids, and redirects to the order detail page", async () => {
  const fetchMock = createFetchMock({
    orderResponse: new Response(
      JSON.stringify({
        mode: "proxy",
        data: {
          orderId: 77,
          workflowId: "wf-77",
          status: "created",
        },
      }),
      {
        status: 201,
        headers: { "content-type": "application/json" },
      },
    ),
  });

  vi.stubGlobal("fetch", fetchMock);

  render(<SubmitWorkbench />);

  await screen.findByText("Ava / Studio");

  fireEvent.change(screen.getByLabelText("客户层级"), {
    target: { value: "low" },
  });
  fireEvent.change(screen.getByLabelText("模特资源"), {
    target: { value: "model-ava" },
  });
  fireEvent.change(screen.getByLabelText("场景资源"), {
    target: { value: "scene-loft" },
  });
  fireEvent.change(screen.getByLabelText("服装资源"), {
    target: { value: "garment-coat-01" },
  });

  fireEvent.click(screen.getByRole("button", { name: "提交订单" }));

  await waitFor(() => {
    expect(fetchMock).toHaveBeenCalledWith(
      "/api/orders",
      expect.objectContaining({
        method: "POST",
        body: JSON.stringify({
          customer_level: "low",
          service_mode: "auto_basic",
          model_id: 101,
          pose_id: 202,
          garment_asset_id: 303,
          scene_ref_asset_id: 404,
        }),
      }),
    );
  });

  expect(
    await screen.findByText("订单 #77 已创建，正在跳转到详情页。"),
  ).toBeInTheDocument();
  expect(screen.getByText("工作流 ID wf-77")).toBeInTheDocument();

  await waitFor(() => {
    expect(pushMock).toHaveBeenCalledWith("/orders/77");
  });
});