xiaoxia-saas/apps/web/src/test/pages/editing-planner/components/GenerationProgressModal.test.tsx

import { describe, expect, it, vi } from "vitest"
import { render } from "@testing-library/react"
import GenerationProgressModal from "@/pages/editing-planner/components/GenerationProgressModal"

const baseProps = {
  open: true,
  voiceoverDuration: null,
  estimatedDuration: 60,
  onDurationChange: vi.fn(),
  onGenerate: vi.fn(),
  task: null,
  submitting: false,
  onCancel: vi.fn(),
  onRetry: vi.fn(),
  onClose: vi.fn(),
}

describe("GenerationProgressModal", () => {
  it("should render setup phase", () => {
    const { container } = render(<GenerationProgressModal {...baseProps} phase="setup" />)
    expect(container).toBeTruthy()
  })

  it("should render progress phase without task", () => {
    const { container } = render(<GenerationProgressModal {...baseProps} phase="progress" />)
    expect(container).toBeTruthy()
  })

  it("should render progress phase with task data", () => {
    const { container } = render(
      <GenerationProgressModal
        {...baseProps}
        phase="progress"
        task={
          {
            id: "task-123",
            status: "generating_video",
            progress: 50,
            current_step: "generating_video",
            user_message: "正在生成视频",
          } as any
        }
      />,
    )
    expect(container).toBeTruthy()
  })

  it("should render completed phase", () => {
    const { container } = render(
      <GenerationProgressModal
        {...baseProps}
        phase="completed"
        task={{ id: "task-1", status: "completed", progress: 100 } as any}
      />,
    )
    expect(container).toBeTruthy()
  })

  it("should render failed phase with retry", () => {
    const { container } = render(
      <GenerationProgressModal
        {...baseProps}
        phase="failed"
        task={
          {
            id: "task-1",
            status: "failed",
            progress: 30,
            error_message: "生成失败",
            retryable: true,
          } as any
        }
      />,
    )
    expect(container).toBeTruthy()
  })

  it("should render failed phase without retry", () => {
    const { container } = render(
      <GenerationProgressModal
        {...baseProps}
        phase="failed"
        task={
          {
            id: "task-1",
            status: "failed",
            progress: 30,
            retryable: false,
          } as any
        }
      />,
    )
    expect(container).toBeTruthy()
  })

  it("should not render when closed", () => {
    const { container } = render(
      <GenerationProgressModal {...baseProps} phase="setup" open={false} />,
    )
    expect(container).toBeTruthy()
  })
})