import { describe, expect, test } from "bun:test"; import { createModel, deleteModel, disableModel, enableModel, fetchModel, fetchModelList, updateModel, } from "../../../src/web/hooks/use-models"; import { installFetchMock, jsonResponse } from "../test-utils"; const MODEL = { capabilities: ["text"] as Array<"text">, contextLength: null, createdAt: "2024-01-01T00:00:00.000Z", enabled: true, id: "m1", maxOutputTokens: null, modelId: "gpt-4o", name: "GPT-4o", providerId: "pv1", updatedAt: "2024-01-01T00:00:00.000Z", }; async function expectRejectsWithMessage(action: () => Promise, message: string) { try { await action(); throw new Error("expected rejection"); } catch (error) { expect(error).toBeInstanceOf(Error); expect((error as Error).message).toBe(message); } } function jsonBody(body: BodyInit | null | undefined): unknown { return JSON.parse(typeof body === "string" ? body : "{}"); } describe("use-models request helpers", () => { test("fetchModelList 按协议拼接 query 参数(含 providerId)", async () => { const calls = installFetchMock(() => jsonResponse({ items: [MODEL], page: 1, pageSize: 20, total: 1 })); const result = await fetchModelList({ keyword: "GPT", page: 1, pageSize: 20, providerId: "pv1" }); expect(result.items).toHaveLength(1); expect(calls[0]?.method).toBe("GET"); expect(calls[0]?.url).toContain("providerId=pv1"); expect(calls[0]?.url).toContain("keyword=GPT"); }); test("模型 CRUD 与 enable/disable 使用正确 method、URL 与 body", async () => { const calls = installFetchMock((call) => { if (call.method === "DELETE") return new Response(null, { status: 204 }); return jsonResponse( { model: MODEL }, { status: call.method === "POST" && call.url === "/api/models" ? 201 : 200 }, ); }); await createModel({ capabilities: ["text"], modelId: "gpt-4o", name: "GPT-4o", providerId: "pv1", }); await updateModel("m1", { name: "GPT-4o Mini" }); await enableModel("m1"); await disableModel("m1"); await deleteModel("m1"); await fetchModel("m1"); expect(calls.map((call) => `${call.method} ${call.url}`)).toEqual([ "POST /api/models", "PATCH /api/models/m1", "POST /api/models/m1/enable", "POST /api/models/m1/disable", "DELETE /api/models/m1", "GET /api/models/m1", ]); expect(jsonBody(calls[0]?.body)).toEqual({ capabilities: ["text"], modelId: "gpt-4o", name: "GPT-4o", providerId: "pv1", }); expect(jsonBody(calls[1]?.body)).toEqual({ name: "GPT-4o Mini" }); }); test("错误响应优先使用后端 error 字段", async () => { installFetchMock(() => jsonResponse({ error: "模型名称已存在" }, { status: 409 })); await expectRejectsWithMessage( () => createModel({ capabilities: ["text"], modelId: "gpt-4o", name: "重复", providerId: "pv1" }), "模型名称已存在", ); }); test("非 JSON 错误响应回退到 HTTP 状态", async () => { installFetchMock(() => new Response("broken", { status: 500 })); await expectRejectsWithMessage(() => fetchModel("m-missing"), "HTTP 500"); }); });