// @vitest-environment jsdom import { act, renderHook } from "@testing-library/react"; import { afterEach, beforeEach, describe, expect, it, vi } from "vitest"; import { useContinue } from "./useContinue"; const post = vi.fn(); const toast = vi.fn(); vi.mock("@/lib/api/client", () => ({ api: { POST: (...a: unknown[]) => post(...a) }, })); vi.mock("@/components/Toast", () => ({ useToast: () => toast })); const GEN_PATH = "/projects/{project_id}/skills/{tool_key}/generate"; describe("useContinue", () => { beforeEach(() => { post.mockReset(); toast.mockReset(); }); afterEach(() => vi.clearAllMocks()); it("初始为 idle,无候选", () => { const { result } = renderHook(() => useContinue()); expect(result.current.status).toBe("idle"); expect(result.current.candidates).toEqual([]); }); it("续写:调 continue 生成器(读该章前文),产出一条候选,status=ready", async () => { post.mockResolvedValue({ data: { output_kind: "ContinuationResult", preview: { text: "续写正文一" } }, error: null, }); const { result } = renderHook(() => useContinue()); await act(async () => { await result.current.generate("p1", 3); }); expect(post).toHaveBeenCalledWith(GEN_PATH, { params: { path: { project_id: "p1", tool_key: "continue" } }, body: { brief: "", chapter_no: 3 }, }); expect(result.current.status).toBe("ready"); expect(result.current.candidates).toEqual(["续写正文一"]); }); it("多候选:再次生成累加候选,不覆盖前一条", async () => { post .mockResolvedValueOnce({ data: { preview: { text: "候选一" } }, error: null }) .mockResolvedValueOnce({ data: { preview: { text: "候选二" } }, error: null }); const { result } = renderHook(() => useContinue()); await act(async () => { await result.current.generate("p1", 1); }); await act(async () => { await result.current.generate("p1", 1); }); expect(result.current.candidates).toEqual(["候选一", "候选二"]); }); it("产出为空:不加候选、提示重试、status=ready", async () => { post.mockResolvedValue({ data: { preview: { text: "" } }, error: null }); const { result } = renderHook(() => useContinue()); await act(async () => { await result.current.generate("p1", 1); }); expect(result.current.candidates).toEqual([]); expect(result.current.status).toBe("ready"); expect(toast).toHaveBeenCalledWith(expect.any(String), "info"); }); it("后端返回 error:status=error、弹错误 toast、候选不变", async () => { post.mockResolvedValue({ data: null, error: { detail: "配额不足" } }); const { result } = renderHook(() => useContinue()); await act(async () => { await result.current.generate("p1", 1); }); expect(result.current.status).toBe("error"); expect(result.current.candidates).toEqual([]); expect(toast).toHaveBeenCalledWith(expect.any(String), "error"); }); it("请求抛异常:status=error、弹网络异常 toast", async () => { post.mockRejectedValue(new Error("network down")); const { result } = renderHook(() => useContinue()); await act(async () => { await result.current.generate("p1", 1); }); expect(result.current.status).toBe("error"); expect(toast).toHaveBeenCalledWith("续写请求异常,请检查网络。", "error"); }); it("reset 清回 idle 与空候选", async () => { post.mockResolvedValue({ data: { preview: { text: "候选" } }, error: null }); const { result } = renderHook(() => useContinue()); await act(async () => { await result.current.generate("p1", 1); }); act(() => result.current.reset()); expect(result.current.status).toBe("idle"); expect(result.current.candidates).toEqual([]); }); });