From 5fa30814de6736339b8cef9313bc74cf15dbc314 Mon Sep 17 00:00:00 2001 From: 72004 Date: Mon, 24 Aug 2026 22:54:14 +0500 Subject: [PATCH] test(prometheus): suppress TQ008 on new spend-backfill tests 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. --- .../integrations/test_prometheus_user_team_metrics.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_litellm/integrations/test_prometheus_user_team_metrics.py b/tests/test_litellm/integrations/test_prometheus_user_team_metrics.py index a3be1f2637b..c08d43188b8 100644 --- a/tests/test_litellm/integrations/test_prometheus_user_team_metrics.py +++ b/tests/test_litellm/integrations/test_prometheus_user_team_metrics.py @@ -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",