test: 接入 jsdom 测试栈, 把 15 个 React hooks 纳入 80% 覆盖率门禁
新增 jsdom + @testing-library/react, 为 lib/** 全部 use*.ts hooks 写 renderHook 单测: 生成/CRUD(world/character/generator/outline/accept/foreshadow/rules)、 SSE流(draftStream/reviewStream)、定时轮询(autosave/jobPoll/kimiOauth)、 文风与注入(refine/styleLearn/injection)。覆盖率 mock api 客户端+Toast, 不打真实 LLM。 vitest.config.ts 去掉 use*.ts 排除; 前端覆盖率 63%→95%。CLAUDE.md 同步说明。
This commit is contained in:
Yaojia Wang
216
apps/web/lib/toolbox/useGenerator.test.ts
Normal file
216
apps/web/lib/toolbox/useGenerator.test.ts
Normal file
@@ -0,0 +1,216 @@
|
||||
// @vitest-environment jsdom
|
||||
import { act, renderHook } from "@testing-library/react";
|
||||
import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
|
||||
|
||||
import { useGenerator } from "./useGenerator";
|
||||
import type { IngestBuildInput } from "./ingest";
|
||||
|
||||
// 后端客户端与 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 conflictEnvelope = {
|
||||
error: {
|
||||
code: "CONFLICT_UNRESOLVED",
|
||||
details: {
|
||||
conflicts: [{ type: "冲突", where: "world", refs: [], suggestion: "改写" }],
|
||||
conflict_count: 1,
|
||||
},
|
||||
},
|
||||
};
|
||||
|
||||
// 行式可入库产物（金手指 → world_entities）。
|
||||
const goldenInput: IngestBuildInput = {
|
||||
outputKind: "GoldenFingerResult",
|
||||
rows: [{ name: "吞噬之眼", mechanism: "吞噬", growth: "无限", limits: "反噬" }],
|
||||
};
|
||||
|
||||
describe("useGenerator", () => {
|
||||
beforeEach(() => {
|
||||
post.mockReset();
|
||||
toast.mockReset();
|
||||
});
|
||||
afterEach(() => vi.clearAllMocks());
|
||||
|
||||
it("初始为 idle、无预览、入库 idle、无冲突", () => {
|
||||
const { result } = renderHook(() => useGenerator());
|
||||
expect(result.current.genStatus).toBe("idle");
|
||||
expect(result.current.preview).toBeNull();
|
||||
expect(result.current.rawPreview).toBeUndefined();
|
||||
expect(result.current.outputKind).toBeNull();
|
||||
expect(result.current.ingestStatus).toBe("idle");
|
||||
expect(result.current.conflicts).toBeNull();
|
||||
expect(result.current.created).toEqual([]);
|
||||
});
|
||||
|
||||
// —— generate ——
|
||||
|
||||
it("生成成功：genStatus=preview，映射预览模型 + 记录 rawPreview/outputKind", async () => {
|
||||
const preview = { ideas: [{ premise: "废柴逆袭", hook: "扮猪吃虎" }] };
|
||||
post.mockResolvedValue({
|
||||
data: { output_kind: "IdeaListResult", preview },
|
||||
error: null,
|
||||
});
|
||||
|
||||
const { result } = renderHook(() => useGenerator());
|
||||
await act(async () => {
|
||||
await result.current.generate("p1", "idea", { brief: "脑洞" });
|
||||
});
|
||||
|
||||
expect(result.current.genStatus).toBe("preview");
|
||||
expect(result.current.outputKind).toBe("IdeaListResult");
|
||||
expect(result.current.rawPreview).toEqual(preview);
|
||||
expect(result.current.preview?.kind).toBe("IdeaListResult");
|
||||
expect(result.current.preview?.items[0]?.heading).toBe("废柴逆袭");
|
||||
expect(toast).not.toHaveBeenCalled();
|
||||
});
|
||||
|
||||
it("生成后端返回 error：genStatus=error 且弹错误 toast", async () => {
|
||||
post.mockResolvedValue({ data: null, error: { detail: "失败" } });
|
||||
const { result } = renderHook(() => useGenerator());
|
||||
await act(async () => {
|
||||
await result.current.generate("p1", "idea", { brief: "x" });
|
||||
});
|
||||
expect(result.current.genStatus).toBe("error");
|
||||
expect(toast).toHaveBeenCalledWith(expect.any(String), "error");
|
||||
});
|
||||
|
||||
it("生成请求抛异常：genStatus=error 且弹网络异常 toast", async () => {
|
||||
post.mockRejectedValue(new Error("network down"));
|
||||
const { result } = renderHook(() => useGenerator());
|
||||
await act(async () => {
|
||||
await result.current.generate("p1", "idea", { brief: "x" });
|
||||
});
|
||||
expect(result.current.genStatus).toBe("error");
|
||||
expect(toast).toHaveBeenCalledWith("生成请求异常，请检查网络。", "error");
|
||||
});
|
||||
|
||||
// —— ingest ——
|
||||
|
||||
it("入库不支持的产物（buildIngestRequest 返 null）：提示并返回 false", async () => {
|
||||
const { result } = renderHook(() => useGenerator());
|
||||
let ok = true;
|
||||
await act(async () => {
|
||||
ok = await result.current.ingest("p1", "opening", {
|
||||
outputKind: "OpeningResult",
|
||||
rows: [{ text: "正文" }],
|
||||
});
|
||||
});
|
||||
expect(ok).toBe(false);
|
||||
expect(toast).toHaveBeenCalledWith("该生成器不支持入库。", "error");
|
||||
expect(post).not.toHaveBeenCalled();
|
||||
});
|
||||
|
||||
it("入库空行：提示至少选择一项并返回 false", async () => {
|
||||
const { result } = renderHook(() => useGenerator());
|
||||
let ok = true;
|
||||
await act(async () => {
|
||||
ok = await result.current.ingest("p1", "golden", {
|
||||
outputKind: "GoldenFingerResult",
|
||||
rows: [],
|
||||
});
|
||||
});
|
||||
expect(ok).toBe(false);
|
||||
expect(toast).toHaveBeenCalledWith("请至少选择一项入库。", "error");
|
||||
expect(post).not.toHaveBeenCalled();
|
||||
});
|
||||
|
||||
it("入库成功：ingestStatus=done、记录 created、弹成功 toast，返回 true", async () => {
|
||||
post.mockResolvedValue({
|
||||
data: { created: ["w1", "w2"], table: "world_entities" },
|
||||
error: null,
|
||||
});
|
||||
const { result } = renderHook(() => useGenerator());
|
||||
let ok = false;
|
||||
await act(async () => {
|
||||
ok = await result.current.ingest("p1", "golden", goldenInput);
|
||||
});
|
||||
expect(ok).toBe(true);
|
||||
expect(result.current.ingestStatus).toBe("done");
|
||||
expect(result.current.created).toEqual(["w1", "w2"]);
|
||||
expect(result.current.conflicts).toBeNull();
|
||||
expect(toast).toHaveBeenCalledWith("已入库 2 项至 world_entities", "success");
|
||||
});
|
||||
|
||||
it("入库成功但 created 缺失：回退空数组并提示 0 项", async () => {
|
||||
post.mockResolvedValue({ data: { created: null, table: "world_entities" }, error: null });
|
||||
const { result } = renderHook(() => useGenerator());
|
||||
await act(async () => {
|
||||
await result.current.ingest("p1", "golden", goldenInput);
|
||||
});
|
||||
expect(result.current.created).toEqual([]);
|
||||
expect(toast).toHaveBeenCalledWith("已入库 0 项至 world_entities", "success");
|
||||
});
|
||||
|
||||
it("入库成功且有越权写表：额外弹 info toast", async () => {
|
||||
post.mockResolvedValue({
|
||||
data: { created: ["w1"], table: "world_entities", rejected_tables: ["secrets"] },
|
||||
error: null,
|
||||
});
|
||||
const { result } = renderHook(() => useGenerator());
|
||||
await act(async () => {
|
||||
await result.current.ingest("p1", "golden", goldenInput);
|
||||
});
|
||||
expect(toast).toHaveBeenCalledWith("越权写表被丢弃：secrets", "info");
|
||||
});
|
||||
|
||||
it("入库 409 冲突：ingestStatus=conflict 并暴露 conflicts，返回 false", async () => {
|
||||
post.mockResolvedValue({ data: null, error: conflictEnvelope });
|
||||
const { result } = renderHook(() => useGenerator());
|
||||
let ok = true;
|
||||
await act(async () => {
|
||||
ok = await result.current.ingest("p1", "golden", goldenInput);
|
||||
});
|
||||
expect(ok).toBe(false);
|
||||
expect(result.current.ingestStatus).toBe("conflict");
|
||||
expect(result.current.conflicts?.conflictCount).toBe(1);
|
||||
expect(result.current.conflicts?.conflicts).toHaveLength(1);
|
||||
});
|
||||
|
||||
it("入库非冲突 error：ingestStatus=error 且弹错误 toast，返回 false", async () => {
|
||||
post.mockResolvedValue({ data: null, error: { error: { code: "LLM_UNAVAILABLE" } } });
|
||||
const { result } = renderHook(() => useGenerator());
|
||||
let ok = true;
|
||||
await act(async () => {
|
||||
ok = await result.current.ingest("p1", "golden", goldenInput);
|
||||
});
|
||||
expect(ok).toBe(false);
|
||||
expect(result.current.ingestStatus).toBe("error");
|
||||
expect(toast).toHaveBeenCalledWith(expect.any(String), "error");
|
||||
});
|
||||
|
||||
it("入库请求抛异常：ingestStatus=error 且弹网络异常 toast，返回 false", async () => {
|
||||
post.mockRejectedValue(new Error("boom"));
|
||||
const { result } = renderHook(() => useGenerator());
|
||||
let ok = true;
|
||||
await act(async () => {
|
||||
ok = await result.current.ingest("p1", "golden", goldenInput);
|
||||
});
|
||||
expect(ok).toBe(false);
|
||||
expect(result.current.ingestStatus).toBe("error");
|
||||
expect(toast).toHaveBeenCalledWith("入库请求异常，请检查网络。", "error");
|
||||
});
|
||||
|
||||
// —— reset ——
|
||||
|
||||
it("reset 清回初始态", async () => {
|
||||
post.mockResolvedValue({
|
||||
data: { output_kind: "IdeaListResult", preview: { ideas: [{ premise: "p" }] } },
|
||||
error: null,
|
||||
});
|
||||
const { result } = renderHook(() => useGenerator());
|
||||
await act(async () => {
|
||||
await result.current.generate("p1", "idea", { brief: "x" });
|
||||
});
|
||||
act(() => result.current.reset());
|
||||
expect(result.current.genStatus).toBe("idle");
|
||||
expect(result.current.preview).toBeNull();
|
||||
expect(result.current.rawPreview).toBeUndefined();
|
||||
expect(result.current.outputKind).toBeNull();
|
||||
expect(result.current.ingestStatus).toBe("idle");
|
||||
expect(result.current.conflicts).toBeNull();
|
||||
expect(result.current.created).toEqual([]);
|
||||
});
|
||||
});
|
||||
Reference in New Issue
Block a user