next-ai-draw-io/tests/unit/app-menu.test.ts

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

vi.mock("electron", () => ({
    app: {
        isPackaged: true,
        getName: () => "app",
        getVersion: () => "1",
        getLocale: () => "en",
    },
    BrowserWindow: { getFocusedWindow: () => null },
    dialog: {},
    Menu: { buildFromTemplate: () => ({}), setApplicationMenu: () => {} },
    shell: {},
}))

// The saved current preset, and restarts that wait until the test ends them
const state = vi.hoisted(() => ({
    current: "A" as string | null,
    restarts: [] as Array<{ resolve: () => void; reject: (e: Error) => void }>,
}))
vi.mock("@/electron/main/config-manager", () => ({
    applyPresetToEnv: (id: string) => {
        state.current = id
        return { AI_PROVIDER: id }
    },
    getAllPresets: () => [],
    getCurrentPresetId: () => state.current,
    setCurrentPreset: (id: string | null) => {
        state.current = id
        return true
    },
}))
vi.mock("@/electron/main/next-server", () => ({
    restartNextServer: () =>
        new Promise<void>((resolve, reject) =>
            state.restarts.push({ resolve, reject }),
        ),
}))
vi.mock("@/electron/main/menu-i18n", () => ({
    getMenuTranslations: () => new Proxy({}, { get: () => "x" }),
    getPreferredLocale: () => "en",
}))
vi.mock("@/electron/main/settings-window", () => ({
    showSettingsWindow: () => {},
}))

import { switchPreset } from "@/electron/main/app-menu"

beforeEach(() => {
    state.current = "A"
    state.restarts = []
})

describe("switchPreset", () => {
    it("keeps a preset chosen while a failed switch was restarting", async () => {
        const toB = switchPreset("B").catch(() => {})
        const toC = switchPreset("C")
        // B's restart fails after the user already picked C
        state.restarts[0].reject(new Error("timed out"))
        await new Promise((r) => setTimeout(r, 0))
        for (const r of state.restarts.slice(1)) r.resolve()
        await toB
        await toC
        expect(state.current).toBe("C")
    })
})