mirror of
https://github.com/DayuanJiang/next-ai-draw-io.git
synced 2026-10-04 08:47:45 +08:00
dayuan.jiang
528b6e54c8
- Shared checkAccessCode for validate-diagram, validate-model, parse-url, verify-access-code - parse-url: 5 MB streamed body limit; validate-diagram: 5 MB image limit - validate-model refuses redirects when private URLs are blocked - Admin settings state shared across module instances via globalThis - Server model ids: unique slugs (non-ASCII names encoded), duplicates rejected - Panel Bedrock credentials stored as ADMIN_AWS_* so the DynamoDB client keeps its own - Locale redirect keeps basePath and query; EdgeOne function drops open CORS and checks the access code - Providers payload reports whether .env sets a default model
72 lines
2.2 KiB
TypeScript
72 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 } 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-"))
|
|
process.env.SETTINGS_FILE = path.join(tmpDir, "settings.json")
|
|
process.env.ADMIN_PASSWORD = "pw"
|
|
process.env.AI_MODELS_CONFIG_PATH = path.join(tmpDir, "none.json")
|
|
_resetForTests()
|
|
})
|
|
|
|
afterEach(() => {
|
|
_resetForTests()
|
|
delete process.env.SETTINGS_FILE
|
|
delete process.env.ADMIN_PASSWORD
|
|
delete process.env.AI_MODELS_CONFIG_PATH
|
|
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 () => {
|
|
process.env.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)
|
|
})
|
|
})
|