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

test(proxy): cover concurrent Presidio session use

Browse source
This commit is contained in:
Vamsi-2581 2026-10-20 21:00:59 +05:30
parent b68ac32580
commit 181d841e29
1 changed files with 21 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
tests/unit/proxy/guardrails/guardrail_hooks/test_presidio.py
View file

@ -5070,3 +5070,24 @@ def test_get_session_iterator_reuses_one_session_per_background_loop(
assert len(sessions) == 10
assert all(session is sessions[0] for session in sessions)
assert presidio_guardrail._http_session is None
@pytest.mark.asyncio
async def test_get_session_iterator_does_not_hold_lock_while_session_is_in_use(
presidio_guardrail: OPTIONAL_PresidioPIIMasking,
) -> None:
both_entered = asyncio.Event()
active_users = 0
async def use_session() -> None:
nonlocal active_users
async with presidio_guardrail._get_session_iterator():
active_users += 1
if active_users == 2:
both_entered.set()
await asyncio.wait_for(both_entered.wait(), timeout=1)
try:
await asyncio.gather(use_session(), use_session())
finally:
await presidio_guardrail._close_http_session()