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(tests): restore litellm.callbacks in TestSpendLogsPayload setup/teardown

Browse source
This commit is contained in:
Ishaan Jaffer 2026-02-21 11:27:19 -08:00
parent 7c8f2274ca
commit 8bf5601ceb
1 changed files with 6 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
tests/test_litellm/proxy/spend_tracking/test_spend_management_endpoints.py
View file

@ -1211,6 +1211,12 @@ async def _wait_for_mock_call(mock, timeout=10, interval=0.1):
class TestSpendLogsPayload:
def setup_method(self):
self._original_callbacks = litellm.callbacks[:]
def teardown_method(self):
litellm.callbacks = self._original_callbacks
@pytest.mark.asyncio
async def test_spend_logs_payload_e2e(self):
litellm.callbacks = [_ProxyDBLogger(message_logging=False)]