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

test(prometheus): suppress TQ008 on new spend-backfill tests

Browse source 
The two new _assemble_team_object tests stub get_team_object the same way
the sibling max_budget-fallback tests above them do; mark them
test-quality-ok so the test-quality budget gate passes.
This commit is contained in:
72004 2026-08-24 22:54:14 +05:00
parent 93e4ebd7f9
commit 5fa30814de
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
tests/test_litellm/integrations/test_prometheus_user_team_metrics.py
View file

@ -372,7 +372,7 @@ async def test_assemble_team_object_uses_db_spend_when_metadata_is_none(
db_team.spend = 902.60
db_team.budget_reset_at = None
with patch("litellm.proxy.auth.auth_checks.get_team_object") as mock_get_team:
with patch("litellm.proxy.auth.auth_checks.get_team_object") as mock_get_team: # test-quality-ok: stubbing the DB fetch, same pattern as the sibling max_budget fallback tests above
mock_get_team.return_value = db_team
team_object = await prometheus_logger._assemble_team_object(
team_id="team-1",
@ -399,7 +399,7 @@ async def test_assemble_team_object_does_not_override_metadata_spend(
db_team.spend = 9999.0
db_team.budget_reset_at = None
with patch("litellm.proxy.auth.auth_checks.get_team_object") as mock_get_team:
with patch("litellm.proxy.auth.auth_checks.get_team_object") as mock_get_team: # test-quality-ok: stubbing the DB fetch, same pattern as the sibling max_budget fallback tests above
mock_get_team.return_value = db_team
team_object = await prometheus_logger._assemble_team_object(
team_id="team-1",