// @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")
    })
})