next-ai-draw-io/tests/unit/admin-providers-route.test.ts

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