furyhawk/deer-flow
1
0
Fork 0
mirror of https://github.com/bytedance/deer-flow.git synced 2026-05-23 16:35:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(storage): address user repository test comment

Browse Source
This commit is contained in:
rayhpeng
2026-05-13 13:03:06 +08:00
parent 3c0b42d836
commit def2a3ad79
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/tests/test_storage_user_repository.py
+2 -3
View File
@@ -11,10 +11,9 @@ import pytest
os.environ.setdefault("DEER_FLOW_CONFIG_PATH", str(Path(__file__).resolve().parents[2] / "config.example.yaml")) os.environ.setdefault("DEER_FLOW_CONFIG_PATH", str(Path(__file__).resolve().parents[2] / "config.example.yaml"))
import store.repositories.models # noqa: F401
from sqlalchemy.ext.asyncio import AsyncSession, async_sessionmaker, create_async_engine from sqlalchemy.ext.asyncio import AsyncSession, async_sessionmaker, create_async_engine
from store.persistence import MappedBase
from store.repositories import UserCreate, UserNotFoundError, build_user_repository from store.repositories import UserCreate, UserNotFoundError, build_user_repository
from store.repositories.models import User as UserModel
@asynccontextmanager @asynccontextmanager
@@ -22,7 +21,7 @@ async def _session_factory(tmp_path) -> AsyncGenerator[async_sessionmaker[AsyncS
db_path = tmp_path / "storage-users.db" db_path = tmp_path / "storage-users.db"
engine = create_async_engine(f"sqlite+aiosqlite:///{db_path}") engine = create_async_engine(f"sqlite+aiosqlite:///{db_path}")
async with engine.begin() as conn: async with engine.begin() as conn:
await conn.run_sync(MappedBase.metadata.create_all) await conn.run_sync(UserModel.metadata.create_all)
try: try:
yield async_sessionmaker(engine, expire_on_commit=False) yield async_sessionmaker(engine, expire_on_commit=False)