From 133bd508ca1433635c9e61ee5fdf184ab446251b Mon Sep 17 00:00:00 2001
From: lewis <lewis@cobley.io>
Date: Fri, 16 Oct 2026 22:23:16 -0400
Subject: [PATCH] test(ui): expect the change source in customer MCP tool
 permission tests

---
 .../mcp_server_management/MCPToolPermissions.test.tsx         | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/ui/litellm-dashboard/src/components/mcp_server_management/MCPToolPermissions.test.tsx b/ui/litellm-dashboard/src/components/mcp_server_management/MCPToolPermissions.test.tsx
index 716c9a7a839..0a213446af1 100644
--- a/ui/litellm-dashboard/src/components/mcp_server_management/MCPToolPermissions.test.tsx
+++ b/ui/litellm-dashboard/src/components/mcp_server_management/MCPToolPermissions.test.tsx
@@ -49,7 +49,7 @@ describe("MCPToolPermissions", () => {
 
     await user.click(screen.getByRole("checkbox", { name: "delete_record" }));
 
-    expect(onChange).toHaveBeenCalledExactlyOnceWith({ [mockServerId]: ["list_records"] });
+    expect(onChange).toHaveBeenCalledExactlyOnceWith({ [mockServerId]: ["list_records"] }, "edit");
   });
 
   it("defaults a newly added edit server to non-delete tools without narrowing stored servers", async () => {
@@ -78,7 +78,7 @@ describe("MCPToolPermissions", () => {
     rerender(<MCPToolPermissions {...props} selectedServers={[mockServerId, newServerId]} />);
 
     await waitFor(() => {
-      expect(onChange).toHaveBeenCalledExactlyOnceWith({ [newServerId]: ["list_records"] });
+      expect(onChange).toHaveBeenCalledExactlyOnceWith({ [newServerId]: ["list_records"] }, "default");
     });
   });