xinyin025/LangBot
1
0
Fork 0
mirror of https://github.com/langbot-app/LangBot.git synced 2026-09-26 19:36:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

test: include legacy pipeline dependency in identity migration fixtures

Browse Source
This commit is contained in:
RockChinQ
2026-09-21 11:19:03 +00:00
parent 27871e4363
commit 73f26bd0a3
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/integration/persistence/test_rag_document_identity.py
+4 -1
View File
@@ -13,6 +13,7 @@ from sqlalchemy.ext.asyncio import create_async_engine
from langbot.pkg.api.http.context import ExecutionContext
from langbot.pkg.entity.persistence.base import Base
from langbot.pkg.entity.persistence.pipeline import LegacyPipeline
from langbot.pkg.entity.persistence.rag import File, KnowledgeBase
from langbot.pkg.entity.persistence.user import User
from langbot.pkg.entity.persistence.workspace import Workspace
@@ -74,10 +75,12 @@ async def database(request, tmp_path):
async def create_schema(engine, *, legacy=False):
# Only the fixture-owned dependency closure; never all imported application tables.
# Full-head upgrades also create the manual migration journal, whose scoped
# foreign key requires the historical legacy_pipelines table.
async with engine.begin() as conn:
await conn.run_sync(
lambda sync: Base.metadata.create_all(
sync, tables=[User.__table__, Workspace.__table__, KnowledgeBase.__table__]
sync, tables=[User.__table__, Workspace.__table__, LegacyPipeline.__table__, KnowledgeBase.__table__]
)
)
if legacy: