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

test(storage): update postgres extra expectation

Browse Source
This commit is contained in:
rayhpeng
2026-05-13 14:18:35 +08:00
parent def2a3ad79
commit d6b3a277a5
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/tests/test_checkpointer.py
+5 -2
View File
@@ -94,12 +94,15 @@ class TestHarnessPackaging:
"psycopg-pool>=3.3.0", "psycopg-pool>=3.3.0",
] ]
def test_workspace_pyproject_forwards_postgres_extra_to_harness(self): def test_workspace_pyproject_forwards_postgres_extra_to_storage_packages(self):
pyproject_path = Path(__file__).resolve().parents[1] / "pyproject.toml" pyproject_path = Path(__file__).resolve().parents[1] / "pyproject.toml"
data = tomllib.loads(pyproject_path.read_text()) data = tomllib.loads(pyproject_path.read_text())
optional_dependencies = data["project"]["optional-dependencies"] optional_dependencies = data["project"]["optional-dependencies"]
assert optional_dependencies["postgres"] == ["deerflow-harness[postgres]"] assert optional_dependencies["postgres"] == [
"deerflow-harness[postgres]",
"deerflow-storage[postgres]",
]
def test_postgres_missing_dependency_messages_recommend_package_extra(self): def test_postgres_missing_dependency_messages_recommend_package_extra(self):
assert "deerflow-harness[postgres]" in POSTGRES_INSTALL assert "deerflow-harness[postgres]" in POSTGRES_INSTALL