scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-14 23:21:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 102

test(proxy): cover all branches of spend_logs_metadata merge helper

Browse source
This commit is contained in:
Devin AI 2026-07-17 15:52:14 +00:00
parent eb8d69291a
commit 46c8d0f29c
1 changed files with 22 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
tests/test_litellm/proxy/test_litellm_pre_call_utils.py
View file

@ -281,6 +281,28 @@ async def test_add_litellm_data_to_request_merges_org_level_spend_logs_metadata(
assert spend_logs_metadata["shared"] == "key"
def test_merge_spend_logs_metadata_branches():
"""Directly exercise every branch of the shared merge helper."""
from litellm.proxy.litellm_pre_call_utils import LiteLLMProxyRequestSetup
# No spend_logs_metadata on the source -> no-op
data = {"metadata": {}}
LiteLLMProxyRequestSetup._merge_spend_logs_metadata({}, data, "metadata")
assert "spend_logs_metadata" not in data["metadata"]
# Non-dict spend_logs_metadata is ignored
LiteLLMProxyRequestSetup._merge_spend_logs_metadata({"spend_logs_metadata": "nope"}, data, "metadata")
assert "spend_logs_metadata" not in data["metadata"]
# No existing target -> assigned directly
LiteLLMProxyRequestSetup._merge_spend_logs_metadata({"spend_logs_metadata": {"a": 1}}, data, "metadata")
assert data["metadata"]["spend_logs_metadata"] == {"a": 1}
# Existing target -> merge without overriding present keys
LiteLLMProxyRequestSetup._merge_spend_logs_metadata({"spend_logs_metadata": {"a": 2, "b": 3}}, data, "metadata")
assert data["metadata"]["spend_logs_metadata"] == {"a": 1, "b": 3}
@pytest.mark.asyncio
async def test_add_litellm_data_to_request_strips_admin_injection_slots():
"""User-supplied user_api_key_metadata / user_api_key_team_metadata /