rayhpeng
125 lines
3.7 KiB
Python
125 lines
3.7 KiB
Python
from __future__ import annotations
|
|
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
from pathlib import Path
|
|
from types import SimpleNamespace
|
|
|
|
import pytest
|
|
|
|
os.environ.setdefault("DEER_FLOW_CONFIG_PATH", str(Path(__file__).resolve().parents[2] / "config.example.yaml"))
|
|
|
|
from store.config.storage_config import StorageConfig
|
|
from store.persistence.factory import _create_database_url, storage_config_from_database_config
|
|
|
|
|
|
def test_database_sqlite_config_maps_to_storage_config(tmp_path):
|
|
database = SimpleNamespace(
|
|
backend="sqlite",
|
|
sqlite_dir=str(tmp_path),
|
|
echo_sql=True,
|
|
pool_size=9,
|
|
)
|
|
|
|
storage = storage_config_from_database_config(database)
|
|
|
|
assert storage == StorageConfig(
|
|
driver="sqlite",
|
|
sqlite_dir=str(tmp_path),
|
|
echo_sql=True,
|
|
pool_size=9,
|
|
)
|
|
assert storage.sqlite_storage_path == str(tmp_path / "deerflow.db")
|
|
|
|
|
|
def test_database_memory_config_is_not_a_storage_backend():
|
|
database = SimpleNamespace(backend="memory")
|
|
|
|
with pytest.raises(ValueError, match="Unsupported database backend"):
|
|
storage_config_from_database_config(database)
|
|
|
|
|
|
def test_database_postgres_config_preserves_url_and_pool_options():
|
|
database = SimpleNamespace(
|
|
backend="postgres",
|
|
postgres_url="postgresql://user:pass@db.example:5544/deerflow",
|
|
echo_sql=True,
|
|
pool_size=11,
|
|
)
|
|
|
|
storage = storage_config_from_database_config(database)
|
|
url = _create_database_url(storage)
|
|
|
|
assert storage.driver == "postgres"
|
|
assert storage.database_url == "postgresql://user:pass@db.example:5544/deerflow"
|
|
assert storage.username == "user"
|
|
assert storage.password == "pass"
|
|
assert storage.host == "db.example"
|
|
assert storage.port == 5544
|
|
assert storage.db_name == "deerflow"
|
|
assert storage.echo_sql is True
|
|
assert storage.pool_size == 11
|
|
assert url.drivername == "postgresql+asyncpg"
|
|
assert url.database == "deerflow"
|
|
|
|
|
|
def test_mysql_database_url_is_normalized_to_async_driver():
|
|
storage = StorageConfig(
|
|
driver="mysql",
|
|
database_url="mysql://user:pass@db.example:3306/deerflow",
|
|
)
|
|
|
|
url = _create_database_url(storage)
|
|
|
|
assert url.drivername == "mysql+aiomysql"
|
|
assert url.database == "deerflow"
|
|
|
|
|
|
def test_mysql_async_database_url_is_preserved():
|
|
storage = StorageConfig(
|
|
driver="mysql",
|
|
database_url="mysql+asyncmy://user:pass@db.example:3306/deerflow",
|
|
)
|
|
|
|
url = _create_database_url(storage)
|
|
|
|
assert url.drivername == "mysql+asyncmy"
|
|
assert url.database == "deerflow"
|
|
|
|
|
|
def test_database_postgres_requires_url():
|
|
database = SimpleNamespace(backend="postgres", postgres_url="")
|
|
|
|
with pytest.raises(ValueError, match="database.postgres_url is required"):
|
|
storage_config_from_database_config(database)
|
|
|
|
|
|
def test_unsupported_database_backend_rejected():
|
|
database = SimpleNamespace(backend="oracle")
|
|
|
|
with pytest.raises(ValueError, match="Unsupported database backend"):
|
|
storage_config_from_database_config(database)
|
|
|
|
|
|
def test_storage_models_import_without_config_file(tmp_path):
|
|
env = os.environ.copy()
|
|
env["DEER_FLOW_CONFIG_PATH"] = str(tmp_path / "missing-config.yaml")
|
|
|
|
result = subprocess.run(
|
|
[
|
|
sys.executable,
|
|
"-c",
|
|
"from store.persistence.base_model import UniversalText, id_key; "
|
|
"from store.repositories.models import RunEvent; "
|
|
"print(UniversalText.__name__, RunEvent.__tablename__, id_key)",
|
|
],
|
|
check=False,
|
|
capture_output=True,
|
|
env=env,
|
|
text=True,
|
|
)
|
|
|
|
assert result.returncode == 0, result.stderr
|
|
assert "UniversalText run_events" in result.stdout
|