scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-06 02:48:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

test(proxy): type the delete-batch transaction timeout capture

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
Devin AI 2026-09-26 04:00:44 +00:00
parent 4836b598c0
commit 64be9089fd
1 changed files with 8 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
tests/test_litellm/proxy/test_spend_log_cleanup.py
View file

@ -210,24 +210,22 @@ async def test_should_delete_spend_logs():
@pytest.mark.asyncio
async def test_delete_batch_transaction_timeout_covers_statement_timeout():
tx_kwargs = []
async def test_delete_batch_transaction_timeout_covers_statement_timeout() -> None:
transaction: Final = MagicMock()
transaction.execute_raw = AsyncMock(side_effect=[0, 0, 3])
@asynccontextmanager
async def tx(**kwargs):
tx_kwargs.append(kwargs)
transaction = MagicMock()
transaction.execute_raw = AsyncMock(side_effect=[0, 0, 3])
async def tx(timeout: timedelta):
yield transaction
prisma_client = MagicMock()
prisma_client.db.tx = tx
cleaner = SpendLogCleanup(general_settings={"maximum_spend_logs_retention_period": "30d"})
prisma_client: Final = MagicMock()
prisma_client.db.tx = MagicMock(side_effect=tx)
cleaner: Final = SpendLogCleanup(general_settings={"maximum_spend_logs_retention_period": "30d"})
cleaner.batch_timeout_seconds = 12
await cleaner._execute_delete_batch(prisma_client, "DELETE FROM table", datetime.now(), _far_deadline())
assert tx_kwargs[0]["timeout"] >= timedelta(seconds=12)
assert prisma_client.db.tx.call_args.kwargs["timeout"] >= timedelta(seconds=12)
@pytest.mark.asyncio