From 81b223138aaf4ca59bcbfefb96983b377eeb5a5b Mon Sep 17 00:00:00 2001
From: yuneng-jiang <yuneng.jiang@gmail.com>
Date: Sat, 7 Mar 2026 23:50:06 -0800
Subject: [PATCH] fix(test): update agents.test.tsx for AgentsPanel table
 refactor
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

AgentsPanel no longer renders AgentCardGrid — it uses a Table directly.
The two tests that looked for agent-card-grid are updated to instead verify
the Actions column header appears for admins and is absent for non-admins.
---
 .../src/components/agents.test.tsx                 | 14 ++++++++------
 1 file changed, 8 insertions(+), 6 deletions(-)

diff --git a/ui/litellm-dashboard/src/components/agents.test.tsx b/ui/litellm-dashboard/src/components/agents.test.tsx
index 8914f8ad6bc..3b0f4341da8 100644
--- a/ui/litellm-dashboard/src/components/agents.test.tsx
+++ b/ui/litellm-dashboard/src/components/agents.test.tsx
@@ -20,6 +20,8 @@ vi.mock("./agents/agent_card_grid", () => ({
   ),
 }));
 
+// Note: agents.tsx no longer uses AgentCardGrid — it renders a Table directly.
+
 vi.mock("./agents/agent_info", () => ({
   default: () => <div data-testid="agent-info" />,
 }));
@@ -54,19 +56,19 @@ describe("AgentsPanel", () => {
     expect(screen.queryByText("+ Add New Agent")).not.toBeInTheDocument();
   });
 
-  it("should pass isAdmin=true to AgentCardGrid for admin role", async () => {
+  it("should show Actions column header for admin role", async () => {
     render(<AgentsPanel accessToken="test-token" userRole="Admin" />);
     await waitFor(() => {
-      const grid = screen.getByTestId("agent-card-grid");
-      expect(grid).toHaveAttribute("data-is-admin", "true");
+      expect(screen.getByRole("columnheader", { name: /actions/i })).toBeInTheDocument();
     });
   });
 
-  it("should pass isAdmin=false to AgentCardGrid for internal user role", async () => {
+  it("should not show Actions column header for internal user role", async () => {
     render(<AgentsPanel accessToken="test-token" userRole="Internal User" />);
     await waitFor(() => {
-      const grid = screen.getByTestId("agent-card-grid");
-      expect(grid).toHaveAttribute("data-is-admin", "false");
+      expect(screen.queryByRole("columnheader", { name: /actions/i })).not.toBeInTheDocument();
+      // confirm table is rendered (not still loading)
+      expect(screen.getByRole("table")).toBeInTheDocument();
     });
   });