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
47 lines
1.5 KiB
TypeScript
47 lines
1.5 KiB
TypeScript
// @vitest-environment node
|
|
// The real Bedrock SDK, with the server's own AWS keys in the environment
|
|
import { afterEach, beforeEach, describe, expect, it, vi } from "vitest"
|
|
import { getAIModel } from "@/lib/ai-providers"
|
|
|
|
const prompt = [
|
|
{ role: "user" as const, content: [{ type: "text" as const, text: "hi" }] },
|
|
]
|
|
let sent: string[] = []
|
|
|
|
beforeEach(() => {
|
|
vi.stubEnv("AWS_ACCESS_KEY_ID", "AKIASERVERKEY")
|
|
vi.stubEnv("AWS_SECRET_ACCESS_KEY", "server-secret")
|
|
vi.stubEnv("AWS_BEARER_TOKEN_BEDROCK", "")
|
|
sent = []
|
|
vi.stubGlobal("fetch", async (_url: unknown, init?: RequestInit) => {
|
|
sent.push(new Headers(init?.headers).get("authorization") ?? "")
|
|
return new Response("{}", { status: 400 })
|
|
})
|
|
})
|
|
afterEach(() => {
|
|
vi.unstubAllGlobals()
|
|
})
|
|
|
|
const run = async (apiKey: string) => {
|
|
const { model } = getAIModel({
|
|
provider: "bedrock",
|
|
modelId: "amazon.nova-lite-v1:0",
|
|
apiKey,
|
|
awsRegion: "us-east-1",
|
|
})
|
|
await (model as any).doGenerate({ prompt }).catch(() => {})
|
|
}
|
|
|
|
describe("a Bedrock API key from the user", () => {
|
|
it("never lets the request fall back to the server's AWS keys", async () => {
|
|
// The SDK drops a key of spaces
|
|
await run(" ")
|
|
expect(sent.some((auth) => auth.includes("AKIASERVERKEY"))).toBe(false)
|
|
})
|
|
|
|
it("is sent as the bearer token", async () => {
|
|
await run("user-key")
|
|
expect(sent).toEqual(["Bearer user-key"])
|
|
})
|
|
})
|