From fbca24d8ebba8ff6dec3883282023bc2b10d70f5 Mon Sep 17 00:00:00 2001 From: Tin Chi Lo Date: Mon, 7 Sep 2026 12:59:47 -0700 Subject: [PATCH] test(ui): reject unexpected automatic tier models --- .../components/add_model/add_auto_router_tab.test.tsx | 9 ++------- 1 file changed, 2 insertions(+), 7 deletions(-) diff --git a/ui/litellm-dashboard/src/components/add_model/add_auto_router_tab.test.tsx b/ui/litellm-dashboard/src/components/add_model/add_auto_router_tab.test.tsx index f5e4763dcd7..014854ac712 100644 --- a/ui/litellm-dashboard/src/components/add_model/add_auto_router_tab.test.tsx +++ b/ui/litellm-dashboard/src/components/add_model/add_auto_router_tab.test.tsx @@ -70,11 +70,8 @@ const tierChips = (tier: string): HTMLElement => { }; const expectTierModel = (tier: string, model: string): void => { - expect(within(tierChips(tier)).getByLabelText(model)).toBeInTheDocument(); -}; - -const expectTierMissingModel = (tier: string, model: string): void => { - expect(within(tierChips(tier)).queryByLabelText(model)).not.toBeInTheDocument(); + const chips = within(tierChips(tier)).getAllByLabelText(/.+/, { selector: '[data-slot="combobox-chip"]' }); + expect(chips.map((chip) => chip.getAttribute("aria-label"))).toEqual([model]); }; const selectTemplate = async (label: string): Promise => { @@ -209,7 +206,6 @@ describe("AddAutoRouterTab", () => { expectTierModel("Medium", "claude-sonnet-5"); expectTierModel("Complex", "claude-opus-5"); expectTierModel("Reasoning", "claude-opus-5"); - expectTierMissingModel("Complex", "gpt-5.6-luna"); expect(toast.success).not.toHaveBeenCalledWith(expect.stringContaining("Configured with")); }); @@ -230,7 +226,6 @@ describe("AddAutoRouterTab", () => { expectTierModel("Medium", "claude-sonnet-5"); expectTierModel("Complex", "gpt-5.6-sol"); expectTierModel("Reasoning", "gpt-5.6-sol"); - expectTierMissingModel("Complex", "gpt-5.6-luna"); }); it("opens Detailed Configuration on the tiers automatic setup just filled in", async () => {