From 7c936f154d481340c8f9d96f04ab6d24dc288199 Mon Sep 17 00:00:00 2001
From: Yuneng Jiang <yuneng@berri.ai>
Date: Wed, 12 Aug 2026 16:53:47 -0700
Subject: [PATCH] test(ui): preserve data utility exports in usage tests

---
 .../src/components/activity_metrics.test.tsx        | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/ui/litellm-dashboard/src/components/activity_metrics.test.tsx b/ui/litellm-dashboard/src/components/activity_metrics.test.tsx
index 21d66991bb5..cd37ea42a90 100644
--- a/ui/litellm-dashboard/src/components/activity_metrics.test.tsx
+++ b/ui/litellm-dashboard/src/components/activity_metrics.test.tsx
@@ -48,11 +48,14 @@ vi.mock("antd", () => {
   };
 });
 
-vi.mock("@/utils/dataUtils", () => ({
-  formatNumberWithCommas: (value: number, decimals?: number) => {
-    return value.toFixed(decimals || 0);
-  },
-}));
+vi.mock("@/utils/dataUtils", async (importOriginal) => {
+  const actual = await importOriginal<typeof import("@/utils/dataUtils")>();
+
+  return {
+    ...actual,
+    formatNumberWithCommas: (value: number, decimals?: number) => value.toFixed(decimals || 0),
+  };
+});
 
 vi.mock("@/utils/teamUtils", () => ({
   resolveTeamAliasFromTeamID: (teamID: string, teams: any[]) => {