From f805e15f7b40ee0b1879ffdc94acb3a9f3bfa218 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Tue, 1 Apr 2025 08:39:41 -0700 Subject: [PATCH] test_get_enforced_params_for_service_account_settings --- tests/litellm/proxy/test_litellm_pre_call_utils.py | 14 ++++++++------ 1 file changed, 8 insertions(+), 6 deletions(-) diff --git a/tests/litellm/proxy/test_litellm_pre_call_utils.py b/tests/litellm/proxy/test_litellm_pre_call_utils.py index 5124b6d2595..b671f715067 100644 --- a/tests/litellm/proxy/test_litellm_pre_call_utils.py +++ b/tests/litellm/proxy/test_litellm_pre_call_utils.py @@ -44,18 +44,20 @@ def test_get_enforced_params_for_service_account_settings(): service_account_token = UserAPIKeyAuth( api_key="test-key", metadata={"service_account_id": "test-service-account"} ) - general_settings_with_service = { - "enforced_params": ["user"], + general_settings_with_service_account_settings = { "service_account_settings": {"enforced_params": ["metadata.service"]}, } result = _get_enforced_params( - general_settings=general_settings_with_service, + general_settings=general_settings_with_service_account_settings, user_api_key_dict=service_account_token, ) - assert result == ["user", "metadata.service"] + assert result == ["metadata.service"] - regular_token = UserAPIKeyAuth(api_key="test-key", metadata={}) + regular_token = UserAPIKeyAuth( + api_key="test-key", metadata={"enforced_params": ["user"]} + ) result = _get_enforced_params( - general_settings=general_settings_with_service, user_api_key_dict=regular_token + general_settings=general_settings_with_service_account_settings, + user_api_key_dict=regular_token, ) assert result == ["user"]