fixing test

ui/litellm-dashboard/src/components/mcp_tools/MCPPermissionManagement.test.tsx

@@ -44,7 +44,10 @@ const renderWithForm = (props = {}) => {
   it("should default allow_all_keys switch to unchecked for new servers", async () => {
     renderWithForm();
     await expandPanel();
-    const toggle = screen.getByRole("switch");
+    // Find the switch associated with "Allow All LiteLLM Keys" text
+    // The first switch in the component is for allow_all_keys
+    const switches = screen.getAllByRole("switch");
+    const toggle = switches[0];
     expect(toggle).toHaveAttribute("aria-checked", "false");
   });
 
@@ -62,7 +65,10 @@ const renderWithForm = (props = {}) => {
     });
 
     const user = await expandPanel();
-    const toggle = screen.getByRole("switch");
+    // Find the switch associated with "Allow All LiteLLM Keys" text
+    // The first switch in the component is for allow_all_keys
+    const switches = screen.getAllByRole("switch");
+    const toggle = switches[0];
     expect(toggle).toHaveAttribute("aria-checked", "true");
 
     await user.click(toggle);

ui/litellm-dashboard/src/components/mcp_tools/mcp_servers.test.tsx

@@ -11,6 +11,10 @@ vi.mock("../networking", () => ({
   fetchMCPServerHealth: vi.fn(),
   deleteMCPServer: vi.fn(),
   getProxyBaseUrl: vi.fn().mockReturnValue("http://localhost:4000"),
+  fetchMCPClientIp: vi.fn().mockResolvedValue(null),
+  getGeneralSettingsCall: vi.fn().mockResolvedValue([]),
+  updateConfigFieldSetting: vi.fn().mockResolvedValue(undefined),
+  deleteConfigFieldSetting: vi.fn().mockResolvedValue(undefined),
 }));
 
 // Mock NotificationsManager