From 45c22081ee45da0d477ca307cafd855ff69972e4 Mon Sep 17 00:00:00 2001
From: Sameer Kankute <sameer@berri.ai>
Date: Wed, 29 Apr 2026 17:47:27 +0530
Subject: [PATCH] Fix ui unit test

---
 .../src/components/view_logs/index.test.tsx   | 28 +++++++++++++------
 1 file changed, 19 insertions(+), 9 deletions(-)

diff --git a/ui/litellm-dashboard/src/components/view_logs/index.test.tsx b/ui/litellm-dashboard/src/components/view_logs/index.test.tsx
index 427c55c92bb..937844e1c1d 100644
--- a/ui/litellm-dashboard/src/components/view_logs/index.test.tsx
+++ b/ui/litellm-dashboard/src/components/view_logs/index.test.tsx
@@ -7,15 +7,25 @@ import type { Row } from "@tanstack/react-table";
 import { renderWithProviders } from "../../../tests/test-utils";
 
 const mockHandleFilterResetFromHook = vi.fn();
-vi.mock("./log_filter_logic", () => ({
-  useLogFilterLogic: vi.fn(() => ({
-    filters: {},
-    filteredLogs: { data: [], total: 0, page: 1, page_size: 50, total_pages: 1 },
-    allTeams: [],
-    handleFilterChange: vi.fn(),
-    handleFilterReset: mockHandleFilterResetFromHook,
-  })),
-}));
+vi.mock("./log_filter_logic", async (importOriginal) => {
+  const actual = await importOriginal<typeof import("./log_filter_logic")>();
+  return {
+    ...actual,
+    useLogFilterLogic: vi.fn(() => ({
+      filters: {},
+      filteredLogs: {
+        data: [],
+        total: 0,
+        page: 1,
+        page_size: 50,
+        total_pages: 1,
+      },
+      allTeams: [],
+      handleFilterChange: vi.fn(),
+      handleFilterReset: mockHandleFilterResetFromHook,
+    })),
+  };
+});
 
 vi.mock("../networking", async (importOriginal) => {
   const actual = await importOriginal<typeof import("../networking")>();