// @vitest-environment jsdom import { act, renderHook } from "@testing-library/react"; import { afterEach, beforeEach, describe, expect, it, vi } from "vitest"; import { useRefine } from "./useRefine"; // 后端客户端与 Toast 是 hook 的外部副作用边界,单测一律 mock。 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 REFINE_PATH = "/projects/{project_id}/chapters/{chapter_no}/refine"; describe("useRefine", () => { beforeEach(() => { post.mockReset(); toast.mockReset(); }); afterEach(() => vi.clearAllMocks()); it("初始为 idle,无版本、无原文", () => { const { result } = renderHook(() => useRefine()); expect(result.current.status).toBe("idle"); expect(result.current.versions).toEqual([]); expect(result.current.latest).toBeNull(); expect(result.current.original).toBeNull(); }); it("润色:调 /refine,压入一版,original=选中原文,status=ready", async () => { post.mockResolvedValue({ data: { original: "原文段", refined: "润色后的段落" }, error: null, }); const { result } = renderHook(() => useRefine()); let returned: unknown; await act(async () => { returned = await result.current.refine("p1", 3, "原文段"); }); expect(post).toHaveBeenCalledWith(REFINE_PATH, { params: { path: { project_id: "p1", chapter_no: 3 } }, body: { segment: "原文段" }, }); expect(result.current.status).toBe("ready"); expect(result.current.original).toBe("原文段"); expect(result.current.latest).toEqual({ segment: "原文段", instruction: "", refined: "润色后的段落", }); expect(result.current.versions).toHaveLength(1); // 追加断言:refine 返回本版产物(供 RefinePanel 结束后留痕)。 expect(returned).toEqual({ segment: "原文段", instruction: "", refined: "润色后的段落", }); }); it("润色带指令:instruction 透传后端", async () => { post.mockResolvedValue({ data: { original: "s", refined: "r" }, error: null }); const { result } = renderHook(() => useRefine()); await act(async () => { await result.current.refine("p1", 1, "原文", "更口语一点"); }); expect(post).toHaveBeenCalledWith(REFINE_PATH, { params: { path: { project_id: "p1", chapter_no: 1 } }, body: { segment: "原文", instruction: "更口语一点" }, }); }); it("再沟通:以上一版产出为输入 + 作者新意见,压入新版,original 不变", async () => { post .mockResolvedValueOnce({ data: { original: "原文", refined: "第一版" }, error: null }) .mockResolvedValueOnce({ data: { original: "第一版", refined: "第二版" }, error: null }); const { result } = renderHook(() => useRefine()); await act(async () => { await result.current.refine("p1", 2, "原文"); }); await act(async () => { await result.current.recommunicate("p1", 2, "再收紧节奏"); }); expect(post).toHaveBeenLastCalledWith(REFINE_PATH, { params: { path: { project_id: "p1", chapter_no: 2 } }, body: { segment: "第一版", instruction: "再收紧节奏" }, }); expect(result.current.versions).toHaveLength(2); expect(result.current.original).toBe("原文"); expect(result.current.latest?.refined).toBe("第二版"); }); it("无历史版本时再沟通为空操作(不打后端)", async () => { const { result } = renderHook(() => useRefine()); await act(async () => { await result.current.recommunicate("p1", 1, "改一下"); }); expect(post).not.toHaveBeenCalled(); }); it("后端返回 error:status=error、弹错误 toast、版本栈不变", async () => { post.mockResolvedValue({ data: null, error: { detail: "配额不足" } }); const { result } = renderHook(() => useRefine()); await act(async () => { await result.current.refine("p1", 1, "原文"); }); expect(result.current.status).toBe("error"); expect(result.current.versions).toEqual([]); expect(toast).toHaveBeenCalledWith(expect.any(String), "error"); }); it("请求抛异常:status=error、弹网络异常 toast", async () => { post.mockRejectedValue(new Error("network down")); const { result } = renderHook(() => useRefine()); await act(async () => { await result.current.refine("p1", 1, "原文"); }); expect(result.current.status).toBe("error"); expect(toast).toHaveBeenCalledWith("回炉请求异常,请检查网络。", "error"); }); it("reset 清回 idle 与空版本", async () => { post.mockResolvedValue({ data: { original: "s", refined: "r" }, error: null }); const { result } = renderHook(() => useRefine()); await act(async () => { await result.current.refine("p1", 1, "s"); }); act(() => result.current.reset()); expect(result.current.status).toBe("idle"); expect(result.current.versions).toEqual([]); expect(result.current.latest).toBeNull(); }); });