xiaoxia-saas/apps/web/src/test/hooks/useCloneProgress.test.tsx

import { describe, expect, it, vi, beforeEach } from "vitest"
import { renderHook, act } from "@testing-library/react"
import { useCloneProgress } from "@/hooks/useCloneProgress"

// mock API
const mockGetVoiceClones = vi.fn()
vi.mock("@/api/voice-clone", () => ({
  getVoiceClones: (...args: unknown[]) => mockGetVoiceClones(...args),
  VoiceCloneStatus: { READY: "ready" },
}))

vi.useFakeTimers()

describe("useCloneProgress", () => {
  beforeEach(() => {
    vi.clearAllMocks()
    vi.clearAllTimers()
  })

  it("should initial load clones", async () => {
    const mockClones = [{ id: "1", name: "克隆1", status: "ready" }]
    mockGetVoiceClones.mockResolvedValue(mockClones)

    const { result } = renderHook(() => useCloneProgress())

    // 等待初始加载
    await vi.waitFor(() => {
      expect(result.current.clones).toEqual(mockClones)
    })
    expect(result.current.loading).toBe(false)
    expect(result.current.hasProcessing).toBe(false)
  })

  it("should have loading state during fetch", async () => {
    mockGetVoiceClones.mockResolvedValue([])

    const { result } = renderHook(() => useCloneProgress())

    expect(result.current.loading).toBe(true)

    await vi.waitFor(() => {
      expect(result.current.loading).toBe(false)
    })
  })

  it("should add clone via addClone", async () => {
    mockGetVoiceClones.mockResolvedValue([])

    const { result } = renderHook(() => useCloneProgress())

    await vi.waitFor(() => expect(result.current.loading).toBe(false))

    act(() => {
      result.current.addClone({ id: "new-1", name: "新克隆", status: "processing" })
    })

    expect(result.current.clones.length).toBe(1)
    expect(result.current.clones[0].id).toBe("new-1")
  })

  it("should remove clone via removeClone", async () => {
    mockGetVoiceClones.mockResolvedValue([
      { id: "1", name: "克隆1", status: "ready" },
      { id: "2", name: "克隆2", status: "ready" },
    ])

    const { result } = renderHook(() => useCloneProgress())

    await vi.waitFor(() => expect(result.current.clones.length).toBe(2))

    act(() => {
      result.current.removeClone("1")
    })

    expect(result.current.clones.length).toBe(1)
    expect(result.current.clones[0].id).toBe("2")
  })

  it("should update clone via updateClone", async () => {
    mockGetVoiceClones.mockResolvedValue([{ id: "1", name: "旧名字", status: "processing" }])

    const { result } = renderHook(() => useCloneProgress())

    await vi.waitFor(() => expect(result.current.clones.length).toBe(1))

    act(() => {
      result.current.updateClone({ id: "1", name: "新名字", status: "ready" })
    })

    expect(result.current.clones[0].name).toBe("新名字")
    expect(result.current.clones[0].status).toBe("ready")
  })

  it("should refresh clones manually", async () => {
    let callCount = 0
    mockGetVoiceClones.mockImplementation(() => {
      callCount++
      return Promise.resolve([{ id: `${callCount}`, name: `克隆${callCount}`, status: "ready" }])
    })

    const { result } = renderHook(() => useCloneProgress())

    await vi.waitFor(() => expect(result.current.clones[0]?.id).toBe("1"))

    await act(async () => {
      await result.current.refresh()
    })

    expect(mockGetVoiceClones).toHaveBeenCalledTimes(2)
  })

  it("should not start polling when all clones are ready", async () => {
    mockGetVoiceClones.mockResolvedValue([{ id: "1", name: "克隆1", status: "ready" }])

    renderHook(() => useCloneProgress())

    await vi.waitFor(() => expect(mockGetVoiceClones).toHaveBeenCalledTimes(1))

    // 快进 10 秒，ready 状态不应该轮询
    vi.advanceTimersByTime(10000)

    // 应该只调用了初始的那一次
    expect(mockGetVoiceClones).toHaveBeenCalledTimes(1)
  })

  it("hasProcessing should be true when there are processing clones", async () => {
    mockGetVoiceClones.mockResolvedValue([{ id: "1", name: "克隆1", status: "processing" }])

    const { result } = renderHook(() => useCloneProgress())

    await vi.waitFor(() => {
      expect(result.current.hasProcessing).toBe(true)
    })
  })
})