From a0f2536e8b18018ca993f3a75bebb5c0c3ca3d47 Mon Sep 17 00:00:00 2001
From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
Date: Mon, 31 Aug 2026 19:49:39 +0000
Subject: [PATCH] test(mcp): stub premium check in key-guardrail regression
 test

Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
---
 .../_experimental/mcp_server/test_mcp_server_manager.py      | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_server_manager.py b/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_server_manager.py
index 7a936461823..9edfe9b0291 100644
--- a/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_server_manager.py
+++ b/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_server_manager.py
@@ -10914,7 +10914,10 @@ class TestKeyAttachedGuardrailsReachTheMCPPreCallHook:
             seen.update(data)
             return None
 
-        with patch.object(proxy_logging_obj, "pre_call_hook", side_effect=capture_pre_call_hook):
+        with (
+            patch("litellm.proxy.utils._premium_user_check"),
+            patch.object(proxy_logging_obj, "pre_call_hook", side_effect=capture_pre_call_hook),
+        ):
             await MCPServerManager().pre_call_tool_check(
                 name="echo",
                 arguments={"text": "hi"},