From f703af662197b63ce27b49750bb6a5f29556eddc Mon Sep 17 00:00:00 2001
From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
Date: Thu, 5 Nov 2026 00:37:02 +0000
Subject: [PATCH] test(health): key guardrails on test_model turns are an
 enterprise path, run that unit test as a premium user

---
 tests/unit/proxy/health_endpoints/test_health_endpoints.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/tests/unit/proxy/health_endpoints/test_health_endpoints.py b/tests/unit/proxy/health_endpoints/test_health_endpoints.py
index bd4917ba4c0..5b1f4223e52 100644
--- a/tests/unit/proxy/health_endpoints/test_health_endpoints.py
+++ b/tests/unit/proxy/health_endpoints/test_health_endpoints.py
@@ -6355,7 +6355,10 @@ async def test_test_model_turns_pick_up_the_keys_guardrails_and_the_pinned_team_
         hooks=hooks,
     )
 
-    with pytest.raises(HTTPException, match="stopped before the provider"):
+    with (
+        patch("litellm.proxy.proxy_server.premium_user", True),
+        pytest.raises(HTTPException, match="stopped before the provider"),
+    ):
         await call(CHAT_COMPLETIONS_SCRIPT, CHAT_COMPLETIONS.turn_1, "turn-a")
 
     assert hooks.guardrails == (["key-prompt-guard", "pii-mask"],)