From 8d2fbbf8b112dca35337e790db5f4cd16b92bdd6 Mon Sep 17 00:00:00 2001 From: Russell Ballestrini Date: Mon, 5 Jan 2026 14:51:51 -0500 Subject: [PATCH] Fix test_stats_endpoint async context manager mock --- tests/unit/test_serp.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/tests/unit/test_serp.py b/tests/unit/test_serp.py index b1bd739..359b36b 100644 --- a/tests/unit/test_serp.py +++ b/tests/unit/test_serp.py @@ -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: