import { beforeEach, describe, expect, it, vi } from "vitest";
import type { FlockRunQuoteResult } from "@/workers/rpa/flock/types";

const directMock = vi.fn<() => Promise<FlockRunQuoteResult>>();
const visualMock = vi.fn<() => Promise<FlockRunQuoteResult>>();
const customerLoginMock = vi.fn();
const hasCredMock = vi.fn();

vi.mock("@/lib/flock/direct-quote", () => ({
  fetchFlockQuotesViaDirect: () => directMock(),
}));

vi.mock("@/workers/rpa/flock/visual-chain", () => ({
  runFlockVisualChain: () => visualMock(),
}));

vi.mock("@/workers/rpa/flock/worker-session", () => ({
  runFlockQuoteWithWorkerSession: () => visualMock(),
  shouldRotateFlockSession: () => false,
}));

vi.mock("@/lib/flock/exclusive-lock", () => ({
  withFlockExclusiveLock: async (
    _owner: string,
    fn: () => Promise<FlockRunQuoteResult>,
  ) => fn(),
}));

vi.mock("@/modules/customer/provider-credentials", () => ({
  getCustomerProviderLogin: (...args: unknown[]) => customerLoginMock(...args),
  hasCustomerProviderCredential: (...args: unknown[]) => hasCredMock(...args),
}));

describe("runFlockQuoteRpa customer login forces DOM", () => {
  beforeEach(() => {
    directMock.mockReset();
    visualMock.mockReset();
    customerLoginMock.mockReset();
    hasCredMock.mockReset();
    hasCredMock.mockResolvedValue(false);
    process.env.RPA_MOCK_MODE = "false";
    process.env.FLOCK_QUOTE_MODE = "direct_then_dom";
    delete process.env.FLOCK_LOGIN_EMAIL;
    delete process.env.FLOCK_LOGIN_PASSWORD;
    process.env.FLOCK_WORKER_REUSE_SESSION = "false";
  });

  const input = {
    pickupZip: "94102",
    deliveryZip: "90212",
    pickupDate: "07/20/2026",
    palletCount: 4,
    totalWeightLb: 2800,
    lengthIn: 48,
    widthIn: 40,
    heightIn: 48,
  };

  it("有客户账密时跳过 Direct，走 DOM", async () => {
    customerLoginMock.mockResolvedValue({
      email: "cust@example.com",
      password: "secret",
    });
    visualMock.mockResolvedValue({
      ok: true,
      quotes: [{ tier: "standard", totalUsd: 100 } as FlockRunQuoteResult["quotes"][0]],
    });

    const { runFlockQuoteRpa } = await import("@/workers/rpa/flock/run-quote");
    const result = await runFlockQuoteRpa(input, { customerId: "CUST_001" });

    expect(result.ok).toBe(true);
    expect(directMock).not.toHaveBeenCalled();
    expect(visualMock).toHaveBeenCalled();
    expect(customerLoginMock).toHaveBeenCalledWith("CUST_001", "flock");
  });

  it("库中已配账密但解密失败时不回落 Direct", async () => {
    customerLoginMock.mockResolvedValue(null);
    hasCredMock.mockResolvedValue(true);

    const { runFlockQuoteRpa } = await import("@/workers/rpa/flock/run-quote");
    const result = await runFlockQuoteRpa(input, { customerId: "CUST_001" });

    expect(result.ok).toBe(false);
    expect(result.errorMessage).toMatch(/解密失败|禁止改走 Direct/i);
    expect(directMock).not.toHaveBeenCalled();
    expect(visualMock).not.toHaveBeenCalled();
  });

  it("无客户账密时走 Direct", async () => {
    customerLoginMock.mockResolvedValue(null);
    hasCredMock.mockResolvedValue(false);
    directMock.mockResolvedValue({
      ok: true,
      quotes: [{ tier: "standard", totalUsd: 200 } as FlockRunQuoteResult["quotes"][0]],
    });

    const { runFlockQuoteRpa } = await import("@/workers/rpa/flock/run-quote");
    const result = await runFlockQuoteRpa(input, { customerId: "CUST_001" });

    expect(result.ok).toBe(true);
    expect(directMock).toHaveBeenCalled();
  });
});