From bbc120095e2dcc9477f76d647d01490699861114 Mon Sep 17 00:00:00 2001
From: yuneng-jiang <yuneng.jiang@gmail.com>
Date: Wed, 18 Mar 2026 12:48:01 -0700
Subject: [PATCH] address greptile review feedback (greploop iteration 3)

- Remove Ant Design CSS class selector coupling in ExportFormatSelector test
- Lift mock fns out of TestTable component body to enable callback assertions

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
---
 .../components/AIHub/AgentHubTableColumns.test.tsx | 14 +++++++++++---
 .../ExportFormatSelector.test.tsx                  |  4 +---
 2 files changed, 12 insertions(+), 6 deletions(-)

diff --git a/ui/litellm-dashboard/src/components/AIHub/AgentHubTableColumns.test.tsx b/ui/litellm-dashboard/src/components/AIHub/AgentHubTableColumns.test.tsx
index 703a24d8d6..083e67c297 100644
--- a/ui/litellm-dashboard/src/components/AIHub/AgentHubTableColumns.test.tsx
+++ b/ui/litellm-dashboard/src/components/AIHub/AgentHubTableColumns.test.tsx
@@ -22,9 +22,17 @@ const mockAgent: AgentHubData = {
   is_public: true,
 };
 
-function TestTable({ data, publicPage = false }: { data: AgentHubData[]; publicPage?: boolean }) {
-  const showModal = vi.fn();
-  const copyToClipboard = vi.fn();
+function TestTable({
+  data,
+  publicPage = false,
+  showModal = vi.fn(),
+  copyToClipboard = vi.fn(),
+}: {
+  data: AgentHubData[];
+  publicPage?: boolean;
+  showModal?: ReturnType<typeof vi.fn>;
+  copyToClipboard?: ReturnType<typeof vi.fn>;
+}) {
   const columns = getAgentHubTableColumns(showModal, copyToClipboard, publicPage);
   const table = useReactTable({ data, columns, getCoreRowModel: getCoreRowModel() });
 
diff --git a/ui/litellm-dashboard/src/components/EntityUsageExport/ExportFormatSelector.test.tsx b/ui/litellm-dashboard/src/components/EntityUsageExport/ExportFormatSelector.test.tsx
index a88f9cb116..d20d24992f 100644
--- a/ui/litellm-dashboard/src/components/EntityUsageExport/ExportFormatSelector.test.tsx
+++ b/ui/litellm-dashboard/src/components/EntityUsageExport/ExportFormatSelector.test.tsx
@@ -27,9 +27,7 @@ describe("ExportFormatSelector", () => {
     // Open the Ant Design Select dropdown
     await user.click(screen.getByText("CSV (Excel, Google Sheets)"));
     // Select JSON option from the dropdown
-    const jsonOption = await screen.findByText("JSON (includes metadata)", {
-      selector: ".ant-select-item-option-content",
-    });
+    const jsonOption = await screen.findByText("JSON (includes metadata)");
     await user.click(jsonOption);
     expect(onChange).toHaveBeenCalledWith("json", expect.anything());
   });