sync update from LAN repo (source dhk-1 @ 3b1ec5b4)
This commit is contained in:
1 parent
da6521b6cb
commit
264e2200d6
995 files changed
+148105
-13399
No files matched your search
@@ -0,0 +1,31 @@
|
||||
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()
|
||||
Reference in new issue
Block a user