engineering/pig.py
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix test_stats_endpoint async context manager mock

Browse source
This commit is contained in:
Russell Ballestrini 2026-01-05 14:51:51 -05:00
parent 3303554a8a
commit 8d2fbbf8b1
1 changed files with 7 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

8
tests/unit/test_serp.py
View file

@ -248,7 +248,13 @@ class TestAsyncEndpoints:
"videos": 50,
"audio": 10
})
mock_db.session = MagicMock()
# Mock session as async context manager
mock_session = MagicMock()
mock_session.execute = AsyncMock(return_value=MagicMock(scalar=MagicMock(return_value=0)))
mock_db.session = MagicMock(return_value=MagicMock(
__aenter__=AsyncMock(return_value=mock_session),
__aexit__=AsyncMock(return_value=None)
))
transport = ASGITransport(app=app)
async with AsyncClient(transport=transport, base_url="http://test") as client: