next-ai-draw-io/tests/unit/chat-route-edgeone.test.ts

// @vitest-environment node
import { afterEach, beforeEach, describe, expect, it, vi } from "vitest"

vi.mock("@/lib/dynamo-quota-manager", () => ({
    isQuotaEnabled: () => false,
    checkAndIncrementRequest: async () => ({ allowed: true }),
    recordTokenUsage: async () => {},
}))

import { POST as chat } from "@/app/api/chat/route"

const ENV = ["AI_MODELS_CONFIG", "AI_PROVIDER", "AI_MODEL"]
const saved: Record<string, string | undefined> = {}
const calls: Array<{ url: string; headers: Headers }> = []

beforeEach(() => {
    for (const k of ENV) saved[k] = process.env[k]
    delete process.env.AI_PROVIDER
    delete process.env.AI_MODEL
    calls.length = 0
    vi.stubGlobal(
        "fetch",
        vi.fn(async (url: string, init: RequestInit) => {
            calls.push({ url: String(url), headers: new Headers(init.headers) })
            throw new Error("no network in tests")
        }),
    )
})

afterEach(() => {
    for (const k of ENV) {
        if (saved[k] === undefined) delete process.env[k]
        else process.env[k] = saved[k]
    }
    vi.unstubAllGlobals()
})

describe("EdgeOne as a server model", () => {
    it("calls the site's Edge AI function with the cookies", async () => {
        // An admin panel or ai-models.json provider: the client sends the
        // provider name's slug, not "edgeone"
        process.env.AI_MODELS_CONFIG = JSON.stringify({
            providers: [
                {
                    name: "Edge Pages",
                    provider: "edgeone",
                    models: ["@tx/deepseek-ai/deepseek-v3-0324"],
                },
            ],
        })
        const res = await chat(
            new Request("http://localhost/api/chat", {
                method: "POST",
                headers: {
                    "Content-Type": "application/json",
                    "x-ai-provider": "edge-pages",
                    "x-selected-model-id":
                        "server:edge-pages:@tx/deepseek-ai/deepseek-v3-0324",
                    cookie: "eo_token=t; eo_time=1",
                },
                body: JSON.stringify({
                    messages: [
                        {
                            id: "u1",
                            role: "user",
                            parts: [{ type: "text", text: "Draw two boxes" }],
                        },
                    ],
                    xml: "",
                }),
            }),
        )
        await res.text()
        expect(calls[0]?.url).toBe(
            "http://localhost/api/edgeai/chat/completions",
        )
        expect(calls[0]?.headers.get("cookie")).toBe("eo_token=t; eo_time=1")
    })
})

const send = (headers: Record<string, string>) =>
    chat(
        new Request("http://localhost/api/chat", {
            method: "POST",
            headers: { "Content-Type": "application/json", ...headers },
            body: JSON.stringify({
                messages: [
                    {
                        id: "u1",
                        role: "user",
                        parts: [{ type: "text", text: "Draw two boxes" }],
                    },
                ],
                xml: "",
            }),
        }),
    ).then((r) => r.text())

describe("EdgeOne endpoints", () => {
    it("works when the deployment names EdgeOne only in AI_PROVIDER", async () => {
        process.env.AI_PROVIDER = "edgeone"
        process.env.AI_MODEL = "@tx/deepseek-ai/deepseek-v3-0324"
        await send({})
        expect(calls[0]?.url).toBe(
            "http://localhost/api/edgeai/chat/completions",
        )
    })

    it("always calls the site's own function, whatever URL the request names", async () => {
        // Another host would get the user's EdgeOne cookies
        await send({
            "x-ai-provider": "edgeone",
            "x-ai-model": "@tx/deepseek-ai/deepseek-v3-0324",
            "x-ai-base-url": "https://elsewhere.example/api/edgeai",
            cookie: "eo_token=t",
        })
        expect(calls[0]?.url).toBe(
            "http://localhost/api/edgeai/chat/completions",
        )
    })

    it("keeps the deployment's base path", async () => {
        const savedPath = process.env.NEXT_PUBLIC_BASE_PATH
        process.env.NEXT_PUBLIC_BASE_PATH = "/draw"
        try {
            await send({
                "x-ai-provider": "edgeone",
                "x-ai-model": "@tx/deepseek-ai/deepseek-v3-0324",
            })
            expect(calls[0]?.url).toBe(
                "http://localhost/draw/api/edgeai/chat/completions",
            )
        } finally {
            if (savedPath === undefined)
                delete process.env.NEXT_PUBLIC_BASE_PATH
            else process.env.NEXT_PUBLIC_BASE_PATH = savedPath
        }
    })
})