scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-07 08:26:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 36

Fix bulk update tests

Browse source
This commit is contained in:
Sameer Kankute 2026-04-14 20:46:21 +05:30
parent 3d567c34dd
commit f6e526c5be
No known key found for this signature in database
1 changed files with 14 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
tests/test_litellm/proxy/management_endpoints/test_key_management_endpoints.py
View file

@ -5385,9 +5385,15 @@ async def test_bulk_update_keys_success(monkeypatch):
) as mock_hash:
mock_hash.side_effect = ["hashed-key-1", "hashed-key-2"]
def _hash_for_bulk_success(token: str) -> str:
return {
"test-key-1": "hashed-key-1",
"test-key-2": "hashed-key-2",
}[token]
with patch(
"litellm.proxy.management_endpoints.key_management_endpoints._hash_token_if_needed",
side_effect=["hashed-key-1", "hashed-key-2"],
side_effect=_hash_for_bulk_success,
):
with patch(
"litellm.proxy.management_endpoints.key_management_endpoints.KeyManagementEventHooks.async_key_updated_hook"
@ -5511,9 +5517,15 @@ async def test_bulk_update_keys_partial_failures(monkeypatch):
) as mock_hash:
mock_hash.return_value = "hashed-key-1"
def _hash_for_bulk_partial(token: str) -> str:
return {
"test-key-1": "hashed-key-1",
"non-existent-key": "hashed-non-existent-key",
}[token]
with patch(
"litellm.proxy.management_endpoints.key_management_endpoints._hash_token_if_needed",
side_effect=["hashed-key-1", "hashed-non-existent-key"],
side_effect=_hash_for_bulk_partial,
):
with patch(
"litellm.proxy.management_endpoints.key_management_endpoints.KeyManagementEventHooks.async_key_updated_hook"