From de9802578baf443ace3a379f5613ee02e2ae51b4 Mon Sep 17 00:00:00 2001 From: yuneng-jiang Date: Fri, 23 Jan 2026 17:15:50 -0800 Subject: [PATCH] Fixing tests --- .../Settings/RouterSettings/Fallbacks/Fallbacks.test.tsx | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/ui/litellm-dashboard/src/components/Settings/RouterSettings/Fallbacks/Fallbacks.test.tsx b/ui/litellm-dashboard/src/components/Settings/RouterSettings/Fallbacks/Fallbacks.test.tsx index bc6f6a4856c..8a7119aa4a3 100644 --- a/ui/litellm-dashboard/src/components/Settings/RouterSettings/Fallbacks/Fallbacks.test.tsx +++ b/ui/litellm-dashboard/src/components/Settings/RouterSettings/Fallbacks/Fallbacks.test.tsx @@ -1,7 +1,7 @@ import { render, screen, waitFor } from "@testing-library/react"; import userEvent from "@testing-library/user-event"; import { beforeEach, describe, expect, it, vi } from "vitest"; -import Fallbacks from "./fallbacks"; +import Fallbacks from "./Fallbacks"; import * as networkingModule from "../../../networking"; import * as fetchModelsModule from "../../../playground/llm_calls/fetch_models"; @@ -108,7 +108,7 @@ describe("Fallbacks", () => { return buttons[buttons.length - 1]; } const clickableElements = lastCell.querySelectorAll("[class*='cursor-pointer'], button"); - return Array.from(clickableElements).find((el) => + return Array.from(clickableElements).find((el) => el.className.includes("red") || el.className.includes("hover:text-red") ) || clickableElements[clickableElements.length - 1]; }; @@ -202,7 +202,7 @@ describe("Fallbacks", () => { await waitFor(() => { expect(networkingModule.setCallbacksCall).toHaveBeenCalled(); - const callArgs = networkingModule.setCallbacksCall.mock.calls[0]; + const callArgs = (networkingModule.setCallbacksCall as any).mock.calls[0]; expect(callArgs[0]).toBe(mockAccessToken); expect(callArgs[1].router_settings.fallbacks).toHaveLength(1); });