Files
OmniRoute/tests/unit/ui/cheaperInferenceSponsorBanner.test.tsx
Praveen K Palaniswamy 65e81158ab fix(ollama): route models by advertised capability (#11088)
Landed with the design call resolved per the owner's pick — **option 1**: the synced store is now endpoint-agnostic (persistDiscoveredModels and managedModelImport no longer drop non-chat models at write time), and chat selectability moved to read time (auto-pool expansion in autoStrategy applies filterChatSelectableModels; the models-route projection already had its chatOnly filter). Your discovery test now passes end-to-end (3/3): /api/show capabilities persist per connection and image/embedding requests route through the advertising host.

Reconciliation notes: conflicted areas merged onto the current tip (adobe discovery import, requestedModel preflight signature, resolvedProvider fast-path coexists with the synced-route override — explicit resolution wins); carried base-red drains (#10055 memoization, #11071 test variants) dropped as already-landed; the managed-model-import exclusion test was propagated to the new contract (image/video models persist; the read filter still hides them from chat pickers — pinned by a new assertion). Full battery: 205/206 focused (the one red is a confirmed periodic-timer timing flake on the loaded devbox — 20/20 isolated), autoCombo vitest 30/30, combo suites 46/46, gates + typecheck clean.

Thank you @yourspraveen — the capability probe + routing design was right; it just needed the store contract opened up. Fixes #11087.
2026-08-23 11:45:01 -03:00

93 lines
3.3 KiB
TypeScript

// @vitest-environment jsdom
/**
* CheaperInferenceSponsorBanner — render gate (localStorage dismissal), CTA
* pointing at our link.omniroute.online branded short link, and discreet
* partner-link note. Mirrors kimiSponsorBanner.test.tsx, minus the version gate
* (this banner is a durable partnership, not a time-boxed offer).
*/
import React from "react";
import { act } from "react";
import { createRoot } from "react-dom/client";
import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
const STORAGE_KEY = "omniroute-cheaperinference-sponsor-banner-dismissed-v1";
const DISMISS_EVENT = "omniroute:cheaperinference-sponsor-banner-dismissed";
const SHORT_URL = "https://link.omniroute.online/cheaper";
vi.mock("next-intl", () => ({ useTranslations: () => (k: string) => k }));
vi.mock("@/shared/components/ProviderIcon", () => ({ default: () => null }));
async function renderBanner(): Promise<HTMLDivElement> {
vi.resetModules();
const { default: CheaperInferenceSponsorBanner } =
await import("../../../src/app/(dashboard)/dashboard/CheaperInferenceSponsorBanner");
const container = document.createElement("div");
document.body.appendChild(container);
const root = createRoot(container);
act(() => {
root.render(<CheaperInferenceSponsorBanner />);
});
return container;
}
describe("CheaperInferenceSponsorBanner", () => {
beforeEach(() => {
(
globalThis as typeof globalThis & { IS_REACT_ACT_ENVIRONMENT?: boolean }
).IS_REACT_ACT_ENVIRONMENT = true;
localStorage.removeItem(STORAGE_KEY);
});
afterEach(() => {
document.body.innerHTML = "";
localStorage.removeItem(STORAGE_KEY);
});
it("renders with the CTA pointing at the branded short link", async () => {
const container = await renderBanner();
expect(container.textContent).toContain("title");
expect(container.textContent).toContain("cta");
const link = container.querySelector("a[href]");
expect(link).not.toBeNull();
expect(link?.getAttribute("href")).toBe(SHORT_URL);
expect(link?.getAttribute("target")).toBe("_blank");
expect(link?.getAttribute("rel")).toContain("noopener");
});
it("shows the discreet partner-link note near the CTA", async () => {
const container = await renderBanner();
expect(container.textContent).toContain("partnerLinkNote");
const link = container.querySelector("a[href]");
expect(link?.getAttribute("title")).toBe("partnerLinkNote");
});
it("hides after dismissal and stays hidden on re-render", async () => {
const first = await renderBanner();
const button = first.querySelector("button");
expect(button).not.toBeNull();
act(() => {
button?.click();
});
expect(localStorage.getItem(STORAGE_KEY)).toBe("true");
expect(first.textContent).not.toContain("title");
// a fresh render (simulating a later visit) stays hidden
const second = await renderBanner();
expect(second.textContent).not.toContain("title");
});
it("re-renders visible again only after the key is cleared", async () => {
const first = await renderBanner();
const button = first.querySelector("button");
act(() => {
button?.click();
});
expect(localStorage.getItem(STORAGE_KEY)).toBe("true");
localStorage.removeItem(STORAGE_KEY);
const second = await renderBanner();
expect(second.textContent).toContain("title");
});
});