94 lines
4.1 KiB
Python
94 lines
4.1 KiB
Python
"""Tests for configuration and canonical path boundaries."""
|
|
|
|
import os
|
|
from pathlib import Path
|
|
|
|
import atocore.config as config
|
|
|
|
|
|
def test_settings_resolve_canonical_directories(tmp_path, monkeypatch):
|
|
monkeypatch.setenv("ATOCORE_DATA_DIR", str(tmp_path / "data"))
|
|
monkeypatch.setenv("ATOCORE_VAULT_SOURCE_DIR", str(tmp_path / "vault-source"))
|
|
monkeypatch.setenv("ATOCORE_DRIVE_SOURCE_DIR", str(tmp_path / "drive-source"))
|
|
monkeypatch.setenv("ATOCORE_LOG_DIR", str(tmp_path / "logs"))
|
|
monkeypatch.setenv("ATOCORE_BACKUP_DIR", str(tmp_path / "backups"))
|
|
monkeypatch.setenv(
|
|
"ATOCORE_PROJECT_REGISTRY_PATH", str(tmp_path / "config" / "project-registry.json")
|
|
)
|
|
|
|
settings = config.Settings()
|
|
|
|
assert settings.db_path == (tmp_path / "data" / "db" / "atocore.db").resolve()
|
|
assert settings.chroma_path == (tmp_path / "data" / "chroma").resolve()
|
|
assert settings.cache_path == (tmp_path / "data" / "cache").resolve()
|
|
assert settings.tmp_path == (tmp_path / "data" / "tmp").resolve()
|
|
assert settings.resolved_vault_source_dir == (tmp_path / "vault-source").resolve()
|
|
assert settings.resolved_drive_source_dir == (tmp_path / "drive-source").resolve()
|
|
assert settings.resolved_log_dir == (tmp_path / "logs").resolve()
|
|
assert settings.resolved_backup_dir == (tmp_path / "backups").resolve()
|
|
assert settings.resolved_run_dir == (tmp_path / "run").resolve()
|
|
assert settings.resolved_project_registry_path == (
|
|
tmp_path / "config" / "project-registry.json"
|
|
).resolve()
|
|
|
|
|
|
def test_settings_keep_legacy_db_path_when_present(tmp_path, monkeypatch):
|
|
data_dir = tmp_path / "data"
|
|
data_dir.mkdir()
|
|
legacy_db = data_dir / "atocore.db"
|
|
legacy_db.write_text("", encoding="utf-8")
|
|
monkeypatch.setenv("ATOCORE_DATA_DIR", str(data_dir))
|
|
|
|
settings = config.Settings()
|
|
|
|
assert settings.db_path == legacy_db.resolve()
|
|
|
|
|
|
def test_ranking_weights_are_tunable_via_env(monkeypatch):
|
|
monkeypatch.setenv("ATOCORE_RANK_PROJECT_MATCH_BOOST", "3.5")
|
|
monkeypatch.setenv("ATOCORE_RANK_PROJECT_SCOPE_FILTER", "false")
|
|
monkeypatch.setenv("ATOCORE_RANK_PROJECT_SCOPE_CANDIDATE_MULTIPLIER", "6")
|
|
monkeypatch.setenv("ATOCORE_RANK_QUERY_TOKEN_STEP", "0.12")
|
|
monkeypatch.setenv("ATOCORE_RANK_QUERY_TOKEN_CAP", "1.5")
|
|
monkeypatch.setenv("ATOCORE_RANK_PATH_HIGH_SIGNAL_BOOST", "1.25")
|
|
monkeypatch.setenv("ATOCORE_RANK_PATH_LOW_SIGNAL_PENALTY", "0.5")
|
|
|
|
settings = config.Settings()
|
|
|
|
assert settings.rank_project_match_boost == 3.5
|
|
assert settings.rank_project_scope_filter is False
|
|
assert settings.rank_project_scope_candidate_multiplier == 6
|
|
assert settings.rank_query_token_step == 0.12
|
|
assert settings.rank_query_token_cap == 1.5
|
|
assert settings.rank_path_high_signal_boost == 1.25
|
|
assert settings.rank_path_low_signal_penalty == 0.5
|
|
|
|
|
|
def test_ensure_runtime_dirs_creates_machine_dirs_only(tmp_path, monkeypatch):
|
|
monkeypatch.setenv("ATOCORE_DATA_DIR", str(tmp_path / "data"))
|
|
monkeypatch.setenv("ATOCORE_VAULT_SOURCE_DIR", str(tmp_path / "vault-source"))
|
|
monkeypatch.setenv("ATOCORE_DRIVE_SOURCE_DIR", str(tmp_path / "drive-source"))
|
|
monkeypatch.setenv("ATOCORE_LOG_DIR", str(tmp_path / "logs"))
|
|
monkeypatch.setenv("ATOCORE_BACKUP_DIR", str(tmp_path / "backups"))
|
|
monkeypatch.setenv(
|
|
"ATOCORE_PROJECT_REGISTRY_PATH", str(tmp_path / "config" / "project-registry.json")
|
|
)
|
|
|
|
original_settings = config.settings
|
|
try:
|
|
config.settings = config.Settings()
|
|
config.ensure_runtime_dirs()
|
|
|
|
assert config.settings.db_path.parent.exists()
|
|
assert config.settings.chroma_path.exists()
|
|
assert config.settings.cache_path.exists()
|
|
assert config.settings.tmp_path.exists()
|
|
assert config.settings.resolved_log_dir.exists()
|
|
assert config.settings.resolved_backup_dir.exists()
|
|
assert config.settings.resolved_run_dir.exists()
|
|
assert config.settings.resolved_project_registry_path.parent.exists()
|
|
assert not config.settings.resolved_vault_source_dir.exists()
|
|
assert not config.settings.resolved_drive_source_dir.exists()
|
|
finally:
|
|
config.settings = original_settings
|