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
70 lines
2.2 KiB
TypeScript
70 lines
2.2 KiB
TypeScript
// @vitest-environment node
|
|
import fs from "fs"
|
|
import os from "os"
|
|
import path from "path"
|
|
import { afterEach, beforeEach, describe, expect, it, vi } from "vitest"
|
|
import { GET, PUT } from "@/app/api/admin/providers/route"
|
|
import { _resetForTests } from "@/lib/admin/settings"
|
|
|
|
let tmpDir: string
|
|
|
|
beforeEach(() => {
|
|
tmpDir = fs.mkdtempSync(path.join(os.tmpdir(), "admin-providers-route-"))
|
|
vi.stubEnv("SETTINGS_FILE", path.join(tmpDir, "settings.json"))
|
|
vi.stubEnv("ADMIN_PASSWORD", "pw")
|
|
vi.stubEnv("AI_MODELS_CONFIG_PATH", path.join(tmpDir, "none.json"))
|
|
_resetForTests()
|
|
})
|
|
|
|
afterEach(() => {
|
|
_resetForTests()
|
|
// Written by the panel default's save, not stubbed by a test
|
|
delete process.env.AI_MODEL
|
|
fs.rmSync(tmpDir, { recursive: true, force: true })
|
|
})
|
|
|
|
const headers = { "x-admin-password": "pw" }
|
|
|
|
async function saveDefaultPanelProvider() {
|
|
const res = await PUT(
|
|
new Request("http://localhost/api/admin/providers", {
|
|
method: "PUT",
|
|
headers: { ...headers, "Content-Type": "application/json" },
|
|
body: JSON.stringify({
|
|
providers: [
|
|
{
|
|
id: "p1",
|
|
provider: "openai",
|
|
apiKey: "sk-test",
|
|
models: ["gpt-panel"],
|
|
isDefault: true,
|
|
},
|
|
],
|
|
}),
|
|
}),
|
|
)
|
|
expect(res.status).toBe(200)
|
|
}
|
|
|
|
async function envHasDefaultModel(): Promise<boolean> {
|
|
const res = await GET(
|
|
new Request("http://localhost/api/admin/providers", { headers }),
|
|
)
|
|
return (await res.json()).envHasDefaultModel
|
|
}
|
|
|
|
describe("envHasDefaultModel", () => {
|
|
it("is true when .env sets AI_MODEL, even after a panel default", async () => {
|
|
vi.stubEnv("AI_MODEL", "gpt-env")
|
|
expect(await envHasDefaultModel()).toBe(true)
|
|
await saveDefaultPanelProvider()
|
|
expect(await envHasDefaultModel()).toBe(true)
|
|
})
|
|
|
|
it("ignores the AI_MODEL the panel default writes", async () => {
|
|
await saveDefaultPanelProvider()
|
|
expect(process.env.AI_MODEL).toBe("gpt-panel")
|
|
expect(await envHasDefaultModel()).toBe(false)
|
|
})
|
|
})
|