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()