89 lines
2.6 KiB
TypeScript
89 lines
2.6 KiB
TypeScript
// @vitest-environment jsdom
|
|
import React from "react";
|
|
import { act } from "react";
|
|
import { createRoot } from "react-dom/client";
|
|
import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
|
|
|
|
vi.mock("next-intl", () => ({
|
|
useTranslations: () => (key: string) => key,
|
|
}));
|
|
|
|
const { default: ModelSelectModal } = await import("@/shared/components/ModelSelectModal");
|
|
|
|
const containers: HTMLElement[] = [];
|
|
|
|
async function renderModal(props: Partial<React.ComponentProps<typeof ModelSelectModal>> = {}) {
|
|
const container = document.createElement("div");
|
|
document.body.appendChild(container);
|
|
containers.push(container);
|
|
|
|
const root = createRoot(container);
|
|
await act(async () => {
|
|
root.render(
|
|
<ModelSelectModal
|
|
isOpen={true}
|
|
onClose={() => {}}
|
|
onSelect={() => {}}
|
|
showCombos={false}
|
|
activeProviders={[]}
|
|
{...props}
|
|
/>
|
|
);
|
|
});
|
|
|
|
await act(async () => {});
|
|
return container;
|
|
}
|
|
|
|
beforeEach(() => {
|
|
(
|
|
globalThis as typeof globalThis & { IS_REACT_ACT_ENVIRONMENT?: boolean }
|
|
).IS_REACT_ACT_ENVIRONMENT = true;
|
|
vi.stubGlobal(
|
|
"fetch",
|
|
vi.fn(async (url: string) => {
|
|
if (url === "/api/provider-nodes") {
|
|
return { ok: true, json: async () => ({ nodes: [] }) };
|
|
}
|
|
if (url === "/api/provider-models") {
|
|
return { ok: true, json: async () => ({ models: {} }) };
|
|
}
|
|
if (url === "/api/combos") {
|
|
return { ok: true, json: async () => ({ combos: [] }) };
|
|
}
|
|
return { ok: true, json: async () => ({}) };
|
|
})
|
|
);
|
|
});
|
|
|
|
afterEach(() => {
|
|
while (containers.length > 0) {
|
|
containers.pop()?.remove();
|
|
}
|
|
document.body.innerHTML = "";
|
|
vi.unstubAllGlobals();
|
|
});
|
|
|
|
describe("ModelSelectModal zero-config providers", () => {
|
|
it("shows OpenCode Free models when explicitly included without an active connection", async () => {
|
|
const container = await renderModal({ alwaysIncludeProviders: ["opencode"] });
|
|
|
|
expect(container.textContent).toContain("OpenCode Free");
|
|
expect(container.textContent).toContain("Big Pickle");
|
|
});
|
|
|
|
it("does not show OpenCode Free by default without an active connection", async () => {
|
|
const container = await renderModal();
|
|
|
|
expect(container.textContent).not.toContain("OpenCode Free");
|
|
expect(container.textContent).not.toContain("Big Pickle");
|
|
});
|
|
|
|
it("treats null explicit provider lists as empty", async () => {
|
|
const container = await renderModal({ alwaysIncludeProviders: null });
|
|
|
|
expect(container.textContent).not.toContain("OpenCode Free");
|
|
expect(container.textContent).not.toContain("Big Pickle");
|
|
});
|
|
});
|