32 lines
1.2 KiB
Python
32 lines
1.2 KiB
Python
import asyncio
|
|
|
|
import pytest
|
|
from sqlalchemy import text
|
|
from sqlalchemy.ext.asyncio import async_sessionmaker, create_async_engine
|
|
|
|
from deerflow.persistence import engine as database
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_sync_bridge_uses_separate_factory_and_keeps_home_pool(tmp_path, monkeypatch):
|
|
engine = create_async_engine(f"sqlite+aiosqlite:///{tmp_path / 'loop.db'}")
|
|
factory = async_sessionmaker(engine, expire_on_commit=False)
|
|
monkeypatch.setattr(database, "_engine", engine)
|
|
monkeypatch.setattr(database, "_session_factory", factory)
|
|
monkeypatch.setattr(database, "_engine_loop", asyncio.get_running_loop())
|
|
|
|
async def read():
|
|
isolated = database.get_session_factory()
|
|
assert isolated is not factory
|
|
async with isolated() as session:
|
|
return (await session.execute(text("SELECT 42"))).scalar()
|
|
|
|
try:
|
|
for _ in range(4):
|
|
assert database.run_db_blocking(read()) == 42
|
|
assert database.get_session_factory() is factory
|
|
async with factory() as session:
|
|
assert (await session.execute(text("SELECT 1"))).scalar() == 1
|
|
finally:
|
|
await engine.dispose()
|