refactor: 简化模型管理,移除启用/禁用,优化测试和布局
- 移除供应商/模型启用禁用能力,清理DB schema/migration/API/前端 - 供应商测试改为Base URL连通性+/models探测 - 新增POST /api/models/test模型连接测试 - 新增GET /api/providers/options专用供应商选项接口 - 统一工具栏为ModelsToolbar,参考项目管理布局 - 模型弹窗优化:默认能力、响应式3列标签、并排数值 - 前后端正整数校验、供应商下拉loading/error/empty状态 - 表格列宽统一,操作列/名称列固定宽度
This commit is contained in:
@@ -2,38 +2,27 @@ import { fireEvent, screen, waitFor } from "@testing-library/react";
|
||||
import { describe, expect, mock, test } from "bun:test";
|
||||
import { createElement } from "react";
|
||||
|
||||
import type { Model, Provider } from "../../../src/shared/api";
|
||||
import type { Model, ProviderOption } from "../../../src/shared/api";
|
||||
|
||||
import { ModelTable } from "../../../src/web/pages/models/components/ModelTable";
|
||||
import { renderWithProviders } from "../test-utils";
|
||||
|
||||
const ENABLED_PROVIDER: Provider = {
|
||||
apiKey: "sk-test",
|
||||
baseUrl: "https://api.openai.com/v1",
|
||||
createdAt: "2024-01-01T00:00:00.000Z",
|
||||
enabled: true,
|
||||
const OPENAI_PROVIDER: ProviderOption = {
|
||||
id: "pv1",
|
||||
name: "OpenAI",
|
||||
type: "openai",
|
||||
updatedAt: "2024-01-01T00:00:00.000Z",
|
||||
};
|
||||
|
||||
const DISABLED_PROVIDER: Provider = {
|
||||
apiKey: "sk-off",
|
||||
baseUrl: "https://api.deepseek.com/v1",
|
||||
createdAt: "2024-01-01T00:00:00.000Z",
|
||||
enabled: false,
|
||||
const DEEPSEEK_PROVIDER: ProviderOption = {
|
||||
id: "pv2",
|
||||
name: "DeepSeek",
|
||||
type: "openai-compatible",
|
||||
updatedAt: "2024-01-01T00:00:00.000Z",
|
||||
};
|
||||
|
||||
const ENABLED_MODEL: Model = {
|
||||
capabilities: ["text", "reasoning"],
|
||||
contextLength: 128000,
|
||||
createdAt: "2024-01-01T00:00:00.000Z",
|
||||
enabled: true,
|
||||
id: "m1",
|
||||
maxOutputTokens: 4096,
|
||||
modelId: "gpt-4o",
|
||||
@@ -46,7 +35,6 @@ const DISABLED_MODEL: Model = {
|
||||
capabilities: ["text"],
|
||||
contextLength: null,
|
||||
createdAt: "2024-01-01T00:00:00.000Z",
|
||||
enabled: false,
|
||||
id: "m2",
|
||||
maxOutputTokens: null,
|
||||
modelId: "deepseek-chat",
|
||||
@@ -67,51 +55,45 @@ describe("ModelTable", () => {
|
||||
data: { items: [ENABLED_MODEL, DISABLED_MODEL], page: 1, pageSize: 20, total: 2 },
|
||||
loading: false,
|
||||
onDelete: () => Promise.resolve(),
|
||||
onDisable: () => Promise.resolve(),
|
||||
onEdit: () => undefined,
|
||||
onEnable: () => Promise.resolve(),
|
||||
onPageChange: () => undefined,
|
||||
page: 1,
|
||||
pageSize: 20,
|
||||
providers: [ENABLED_PROVIDER, DISABLED_PROVIDER],
|
||||
providers: [OPENAI_PROVIDER, DEEPSEEK_PROVIDER],
|
||||
}),
|
||||
);
|
||||
|
||||
expect(screen.getByText("GPT-4o")).not.toBeNull();
|
||||
expect(screen.getByText("gpt-4o")).not.toBeNull();
|
||||
expect(screen.getByText("DeepSeek Chat")).not.toBeNull();
|
||||
expect(screen.getByText("OpenAI")).not.toBeNull();
|
||||
expect(screen.getByText("DeepSeek")).not.toBeNull();
|
||||
expect(screen.queryByText("状态")).toBeNull();
|
||||
expect(screen.queryByRole("button", { name: /启用|禁用/ })).toBeNull();
|
||||
});
|
||||
|
||||
test("模型表格操作触发 enable/disable/delete", async () => {
|
||||
const onDisable = mock(() => Promise.resolve());
|
||||
const onEnable = mock(() => Promise.resolve());
|
||||
test("模型表格操作触发 edit/delete", async () => {
|
||||
const onDelete = mock(() => Promise.resolve());
|
||||
const onEdit = mock(() => undefined);
|
||||
|
||||
renderWithProviders(
|
||||
createElement(ModelTable, {
|
||||
data: { items: [ENABLED_MODEL, DISABLED_MODEL], page: 1, pageSize: 20, total: 2 },
|
||||
loading: false,
|
||||
onDelete,
|
||||
onDisable,
|
||||
onEdit: () => undefined,
|
||||
onEnable,
|
||||
onEdit,
|
||||
onPageChange: () => undefined,
|
||||
page: 1,
|
||||
pageSize: 20,
|
||||
providers: [ENABLED_PROVIDER, DISABLED_PROVIDER],
|
||||
providers: [OPENAI_PROVIDER, DEEPSEEK_PROVIDER],
|
||||
}),
|
||||
);
|
||||
|
||||
const disableButtons = screen.getAllByRole("button", { name: /禁用/ });
|
||||
fireEvent.click(disableButtons[0]!);
|
||||
await waitFor(() => expect(screen.getByText("确认禁用此模型?")).not.toBeNull());
|
||||
clickLatestConfirmButton();
|
||||
await waitFor(() => expect(onDisable).toHaveBeenCalledWith("m1"));
|
||||
fireEvent.click(screen.getAllByRole("button", { name: /编辑/ })[0]!);
|
||||
expect(onEdit).toHaveBeenCalledWith(ENABLED_MODEL);
|
||||
|
||||
const enableButtons = screen.getAllByRole("button", { name: /启用/ });
|
||||
fireEvent.click(enableButtons[0]!);
|
||||
await waitFor(() => expect(onEnable).toHaveBeenCalledWith("m2"));
|
||||
fireEvent.click(screen.getAllByRole("button", { name: /删除/ })[0]!);
|
||||
await waitFor(() => expect(screen.getByText("确认删除此模型?")).not.toBeNull());
|
||||
clickLatestConfirmButton();
|
||||
await waitFor(() => expect(onDelete).toHaveBeenCalledWith("m1"));
|
||||
});
|
||||
});
|
||||
|
||||
Reference in New Issue
Block a user