From 4411562a0f98626bfc42c3e91804b511b25ad700 Mon Sep 17 00:00:00 2001
From: ryan-crabbe-berri <ryan@berri.ai>
Date: Sat, 29 Aug 2026 10:32:29 -0700
Subject: [PATCH] fix(ui): satisfy lint and the modelInfoCall arity in the
 models hook tests

---
 .../src/app/(dashboard)/hooks/models/useModels.test.ts          | 2 ++
 .../src/components/shared/table_cells/status_badge.test.tsx     | 2 +-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/ui/litellm-dashboard/src/app/(dashboard)/hooks/models/useModels.test.ts b/ui/litellm-dashboard/src/app/(dashboard)/hooks/models/useModels.test.ts
index 411e8402e11..7231c126a63 100644
--- a/ui/litellm-dashboard/src/app/(dashboard)/hooks/models/useModels.test.ts
+++ b/ui/litellm-dashboard/src/app/(dashboard)/hooks/models/useModels.test.ts
@@ -118,6 +118,7 @@ describe("useModelsInfo", () => {
       // exclude_auto_routers defaults off: only the Models + Endpoints table opts in, so
       // every other consumer of this hook keeps seeing auto-routers.
       false,
+      undefined,
     );
     expect(modelInfoCall).toHaveBeenCalledTimes(1);
   });
@@ -145,6 +146,7 @@ describe("useModelsInfo", () => {
       // exclude_auto_routers defaults off: only the Models + Endpoints table opts in, so
       // every other consumer of this hook keeps seeing auto-routers.
       false,
+      undefined,
     );
   });
 
diff --git a/ui/litellm-dashboard/src/components/shared/table_cells/status_badge.test.tsx b/ui/litellm-dashboard/src/components/shared/table_cells/status_badge.test.tsx
index 9eddbff1950..916a637f31b 100644
--- a/ui/litellm-dashboard/src/components/shared/table_cells/status_badge.test.tsx
+++ b/ui/litellm-dashboard/src/components/shared/table_cells/status_badge.test.tsx
@@ -48,7 +48,7 @@ describe("StatusBadge", () => {
     render(<StatusBadge tone="info" label="gpt-4.1" href="/models-and-endpoints?model_group=gpt-4.1" />);
     const link = screen.getByRole("link", { name: "gpt-4.1" });
     expect(link).toHaveAttribute("href", "/models-and-endpoints?model_group=gpt-4.1");
-    expect(link.className).toContain("text-info");
+    expect(link).toHaveClass("text-info");
     await user.click(link);
     expect(push).toHaveBeenCalledWith("/models-and-endpoints?model_group=gpt-4.1");
   });