phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix test_team_update_redis

Browse source
This commit is contained in:
Ishaan Jaff 2024-08-07 15:37:02 -07:00
parent be0984009b
commit 2c3e068435
1 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

8
litellm/tests/test_proxy_server.py
View file

@ -31,7 +31,7 @@ logging.basicConfig(
format="%(asctime)s - %(levelname)s - %(message)s", format="%(asctime)s - %(levelname)s - %(message)s",
) )
from unittest.mock import AsyncMock, MagicMock, patch from unittest.mock import AsyncMock, patch
from fastapi import FastAPI from fastapi import FastAPI
@ -757,7 +757,7 @@ async def test_team_update_redis():
with patch.object( with patch.object(
proxy_logging_obj.internal_usage_cache.redis_cache, proxy_logging_obj.internal_usage_cache.redis_cache,
"async_set_cache", "async_set_cache",
new=MagicMock(), new=AsyncMock(),
) as mock_client: ) as mock_client:
await _cache_team_object( await _cache_team_object(
team_id="1234", team_id="1234",
@ -766,7 +766,7 @@ async def test_team_update_redis():
proxy_logging_obj=proxy_logging_obj, proxy_logging_obj=proxy_logging_obj,
) )
mock_client.assert_called_once() mock_client.assert_called()
@pytest.mark.asyncio @pytest.mark.asyncio
@ -794,7 +794,7 @@ async def test_get_team_redis(client_no_auth):
user_api_key_cache=DualCache(), user_api_key_cache=DualCache(),
parent_otel_span=None, parent_otel_span=None,
proxy_logging_obj=proxy_logging_obj, proxy_logging_obj=proxy_logging_obj,
prisma_client=MagicMock(), prisma_client=AsyncMock(),
) )
except Exception as e: except Exception as e:
pass pass