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

test: restore litellm.headers through monkeypatch instead of a raw global write

Browse source 
The raw assignment tripped the TQ005 test-quality budget; monkeypatch.setattr undoes it at teardown, which is what the rule asks for.
This commit is contained in:
Yucheng He 2026-09-04 17:49:47 -07:00
parent 396bcca423
commit 8971d7389f
1 changed files with 11 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -7888,23 +7888,21 @@ async def test_forward_spend_logs_metadata_reruns_after_a_pre_call_hook_edit(
@pytest.mark.asyncio
async def test_forward_spend_logs_metadata_keeps_globally_configured_headers():
async def test_forward_spend_logs_metadata_keeps_globally_configured_headers(
monkeypatch: pytest.MonkeyPatch,
):
"""
Providers read `headers or litellm.headers`, replacing rather than merging, so creating
`data["headers"]` here would silently drop `litellm_settings.headers` from the call.
"""
original = litellm.headers
litellm.headers = {"x-org-routing": "eu"}
try:
updated = await add_litellm_data_to_request(
data={"model": "gpt-4o", "messages": []},
request=_spend_logs_metadata_request(),
user_api_key_dict=_proxy_chain_auth(),
proxy_config=MagicMock(),
general_settings={"forward_spend_logs_metadata_to_llm_api": True},
)
finally:
litellm.headers = original
monkeypatch.setattr(litellm, "headers", {"x-org-routing": "eu"})
updated = await add_litellm_data_to_request(
data={"model": "gpt-4o", "messages": []},
request=_spend_logs_metadata_request(),
user_api_key_dict=_proxy_chain_auth(),
proxy_config=MagicMock(),
general_settings={"forward_spend_logs_metadata_to_llm_api": True},
)
assert updated["headers"]["x-org-routing"] == "eu"
assert json.loads(updated["headers"][SPEND_LOGS_METADATA_HEADER_NAME])["user_id"] == "U0099887"