mirror of
https://github.com/bytedance/deer-flow.git
synced 2026-11-13 06:31:47 +00:00
9699f04cdc
* feat(projects): add visible project memory with extraction controls * fix(projects): avoid replaying committed memory extractions * chore(ci): retrigger project memory checks --------- Co-authored-by: Willem Jiang <willem.jiang@gmail.com>
70 lines
2.2 KiB
Python
70 lines
2.2 KiB
Python
"""Migration tests for 0019_projects and 0020_threads_meta_project_id."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import pytest
|
|
import sqlalchemy as sa
|
|
|
|
from deerflow.persistence.engine import close_engine, init_engine
|
|
from deerflow.persistence.migrations import _helpers # noqa: F401 (ensures helpers importable)
|
|
|
|
pytestmark = pytest.mark.asyncio
|
|
|
|
|
|
async def _fresh_db(tmp_path):
|
|
url = f"sqlite+aiosqlite:///{tmp_path / 'test.db'}"
|
|
await init_engine("sqlite", url=url, sqlite_dir=str(tmp_path))
|
|
return url
|
|
|
|
|
|
async def test_0019_creates_projects_table(tmp_path):
|
|
await _fresh_db(tmp_path)
|
|
try:
|
|
from deerflow.persistence.engine import get_engine
|
|
|
|
engine = get_engine()
|
|
async with engine.connect() as conn:
|
|
|
|
def _inspect(sync_conn):
|
|
inspector = sa.inspect(sync_conn)
|
|
assert inspector.has_table("projects")
|
|
cols = {c["name"]: c for c in inspector.get_columns("projects")}
|
|
assert set(cols) == {
|
|
"id",
|
|
"user_id",
|
|
"name",
|
|
"instructions",
|
|
"memory",
|
|
"presentation",
|
|
"status",
|
|
"created_at",
|
|
"updated_at",
|
|
}
|
|
index_names = {i["name"] for i in inspector.get_indexes("projects")}
|
|
assert "ix_projects_user_id" in index_names
|
|
assert "ix_projects_status" in index_names
|
|
|
|
await conn.run_sync(_inspect)
|
|
finally:
|
|
await close_engine()
|
|
|
|
|
|
async def test_0020_adds_project_id_column(tmp_path):
|
|
await _fresh_db(tmp_path)
|
|
try:
|
|
from deerflow.persistence.engine import get_engine
|
|
|
|
engine = get_engine()
|
|
async with engine.connect() as conn:
|
|
|
|
def _inspect(sync_conn):
|
|
inspector = sa.inspect(sync_conn)
|
|
cols = {c["name"] for c in inspector.get_columns("threads_meta")}
|
|
assert "project_id" in cols
|
|
index_names = {i["name"] for i in inspector.get_indexes("threads_meta")}
|
|
assert "ix_threads_meta_project_id" in index_names
|
|
|
|
await conn.run_sync(_inspect)
|
|
finally:
|
|
await close_engine()
|