mirror of
https://github.com/DayuanJiang/next-ai-draw-io.git
synced 2026-10-12 20:49:53 +08:00
dayuan.jiang
1333ed68c6
- tests/e2e/lib: openApp(page, { path, config }), MODEL_CONFIGS_KEY,
savedModelConfigs, settingsDialog, chunkedReplies, sse() and
toolCallEvents() replace about 70 copies of goto + wait for iframe,
three local openApp functions, 14 window.fetch overrides and the
Node-side SSE serialisers; unused fixtures removed
- delete theme, save, history, model-config, upload and copy-paste
specs (the copy-button test moves to chat.spec.ts); trim
iframe-interaction, history-restore, keyboard and smoke to tests that
can fail; file-upload asserts on the preview image
- settings-review-fixes.spec.ts split by topic into settings-bedrock,
settings-tabs-sync and settings-providers
- unit tests use vi.stubEnv with unstubEnvs: true instead of
hand-written process.env save and restore loops
87 lines
3.1 KiB
TypeScript
87 lines
3.1 KiB
TypeScript
import { expect, test } from "@playwright/test"
|
|
import { openApp } from "./lib/fixtures"
|
|
|
|
// One provider with three models; the test endpoint answers each differently
|
|
const CONFIG = {
|
|
version: 1,
|
|
providers: [
|
|
{
|
|
id: "p1",
|
|
provider: "glm",
|
|
apiKey: "test-key",
|
|
models: [
|
|
{ id: "m1", modelId: "model-ok" },
|
|
{ id: "m2", modelId: "model-no-tools" },
|
|
{ id: "m3", modelId: "model-broken" },
|
|
],
|
|
},
|
|
],
|
|
}
|
|
|
|
test("the Test button checks all models at once and shows each result", async ({
|
|
page,
|
|
}) => {
|
|
const started: string[] = []
|
|
await page.route("**/api/validate-model", async (route) => {
|
|
const { modelId } = route.request().postDataJSON()
|
|
started.push(modelId)
|
|
// Answer only once all three requests arrived: they run in parallel
|
|
while (started.length < 3) await new Promise((r) => setTimeout(r, 50))
|
|
const answers: Record<string, object> = {
|
|
"model-ok": { valid: true, responseTime: 1234 },
|
|
"model-no-tools": {
|
|
valid: true,
|
|
responseTime: 800,
|
|
warning:
|
|
"Connected, but the model answered without calling a tool.",
|
|
},
|
|
"model-broken": { valid: false, error: "Model not found" },
|
|
}
|
|
await route.fulfill({ json: answers[modelId] })
|
|
})
|
|
await openApp(page, { config: CONFIG })
|
|
|
|
await page.getByTestId("model-selector").click()
|
|
await page.getByText("Configure Models...").click()
|
|
const dialog = page.locator('[role="dialog"]')
|
|
await dialog.getByTestId("provider-row-p1").click()
|
|
await dialog.getByTestId("test-models").click()
|
|
|
|
await expect(
|
|
dialog.getByText("answered without calling a tool"),
|
|
).toBeVisible({ timeout: 15000 })
|
|
await expect(dialog.getByText("Model not found")).toBeVisible()
|
|
await expect(dialog.getByText("Works · 1.2 s")).toBeVisible()
|
|
expect(started.sort()).toEqual([
|
|
"model-broken",
|
|
"model-no-tools",
|
|
"model-ok",
|
|
])
|
|
})
|
|
|
|
test("the key link and the base URL cleanup", async ({ page }) => {
|
|
await openApp(page, { config: CONFIG })
|
|
await page.getByTestId("model-selector").click()
|
|
await page.getByText("Configure Models...").click()
|
|
const dialog = page.locator('[role="dialog"]')
|
|
await dialog.getByTestId("provider-row-p1").click()
|
|
|
|
await expect(
|
|
dialog.getByRole("link", { name: "Get API key" }),
|
|
).toHaveAttribute(
|
|
"href",
|
|
"https://open.bigmodel.cn/usercenter/proj-mgmt/apikeys",
|
|
)
|
|
// The base URL is under "More options"
|
|
await dialog.getByRole("button", { name: /^More options/ }).click()
|
|
const baseUrl = dialog.locator("#base-url")
|
|
await baseUrl.fill("https://proxy.example.com/v4/chat/completions/")
|
|
await baseUrl.blur()
|
|
await expect(baseUrl).toHaveValue("https://proxy.example.com/v4")
|
|
await expect(
|
|
dialog.getByText(
|
|
"Requests go to https://proxy.example.com/v4/chat/completions",
|
|
),
|
|
).toBeVisible()
|
|
})
|