mirror of
https://github.com/bytedance/deer-flow.git
synced 2026-05-22 07:56:48 +00:00
feat(storage): add storage package base
This commit is contained in:
@@ -0,0 +1,76 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import os
|
||||
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_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)
|
||||
Reference in New Issue
Block a user