tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-06-17 22:48:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #24824 from BerriAI/litellm_fix_test_redis_and_azure_mock

Browse Source 
[Fix] Use correct Redis env vars and fix Azure AD token test mocking
This commit is contained in:
Krrish Dholakia
2026-03-30 19:12:45 -07:00
committed by GitHub
parent 7954d1a11f e69dfbc07d
commit 40074fffe1
2 changed files with 10 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/local_testing/test_caching_ssl.py
+4 -4
View File
@@ -69,10 +69,10 @@ def test_caching_router():
]
litellm.cache = Cache(
type="redis",
host="os.environ/REDIS_HOST_2",
port="os.environ/REDIS_PORT_2",
password="os.environ/REDIS_PASSWORD_2",
ssl="os.environ/REDIS_SSL_2",
host="os.environ/REDIS_HOST",
port="os.environ/REDIS_PORT",
password="os.environ/REDIS_PASSWORD",
ssl="os.environ/REDIS_SSL",
)
router = Router(
model_list=model_list,
tests/local_testing/test_router_client_init.py
+6 -12
View File
@@ -71,9 +71,7 @@ def test_router_init_with_neither_api_key_nor_azure_service_principal_with_secre
@patch("azure.identity.get_bearer_token_provider")
@patch("azure.identity.ClientSecretCredential")
@patch("litellm.secret_managers.get_azure_ad_token_provider.os")
def test_router_init_azure_service_principal_with_secret_with_environment_variables(
mocked_os_lib: MagicMock,
mocked_credential: MagicMock,
mocked_get_bearer_token_provider: MagicMock,
monkeypatch,
@@ -86,21 +84,17 @@ def test_router_init_azure_service_principal_with_secret_with_environment_variab
and environment variables.
"""
monkeypatch.delenv("AZURE_AI_API_KEY", raising=False)
monkeypatch.delenv("AZURE_OPENAI_API_KEY", raising=False)
litellm.enable_azure_ad_token_refresh = True
# mock the token provider function
mocked_func_generating_token = MagicMock(return_value="test_token")
mocked_get_bearer_token_provider.return_value = mocked_func_generating_token
# mock the environment variables with mocked credentials
environment_variables_expected_to_use = {
"AZURE_CLIENT_ID": "test_client_id",
"AZURE_CLIENT_SECRET": "test_client_secret",
"AZURE_TENANT_ID": "test_tenant_id",
}
mocked_environ = PropertyMock(return_value=environment_variables_expected_to_use)
# Because of the way mock attributes are stored you can’t directly attach a PropertyMock to a mock object.
# https://docs.python.org/3.11/library/unittest.mock.html#unittest.mock.PropertyMock
type(mocked_os_lib).environ = mocked_environ
# set environment variables with mocked credentials using monkeypatch
# so both common_utils._resolve_env_var and get_azure_ad_token_provider see them
monkeypatch.setenv("AZURE_CLIENT_ID", "test_client_id")
monkeypatch.setenv("AZURE_CLIENT_SECRET", "test_client_secret")
monkeypatch.setenv("AZURE_TENANT_ID", "test_tenant_id")
# define the model list
model_list = [