Skip to content

Commit 8c084e2

Browse files
committed
test(mcp): cover async ASGI database override
Signed-off-by: phernandez <paul@basicmachines.co>
1 parent a29d747 commit 8c084e2

1 file changed

Lines changed: 16 additions & 4 deletions

File tree

tests/mcp/test_async_client_modes.py

Lines changed: 16 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -94,8 +94,12 @@ async def fake_get_or_create_db(db_path):
9494
fastapi_app.state.session_maker = previous_session_maker
9595

9696

97+
@pytest.mark.parametrize("async_override", [False, True])
9798
@pytest.mark.asyncio
98-
async def test_get_client_uses_existing_local_asgi_database_override(config_manager):
99+
async def test_get_client_uses_existing_local_asgi_database_override(
100+
config_manager,
101+
async_override,
102+
):
99103
"""Local ASGI routing honors FastAPI test dependency overrides."""
100104
from basic_memory.api.app import app as fastapi_app
101105
from basic_memory.deps import get_engine_factory
@@ -113,9 +117,17 @@ async def test_get_client_uses_existing_local_asgi_database_override(config_mana
113117
session_maker = object()
114118
calls = []
115119

116-
def override_engine_factory():
117-
calls.append("override")
118-
return engine, session_maker
120+
if async_override:
121+
122+
async def override_engine_factory():
123+
calls.append("override")
124+
return engine, session_maker
125+
126+
else:
127+
128+
def override_engine_factory():
129+
calls.append("override")
130+
return engine, session_maker
119131

120132
fastapi_app.dependency_overrides[get_engine_factory] = override_engine_factory
121133

0 commit comments

Comments
 (0)