basicmachines-co-basic-memory/tests/test_config.py

"""Tests for the Basic Memory configuration system."""

from pathlib import Path
from tempfile import TemporaryDirectory

import pytest

from basic_memory.config import BasicMemoryConfig, ConfigManager, DATA_DIR_NAME, CONFIG_FILE_NAME


class TestBasicMemoryConfig:
    """Test the BasicMemoryConfig pydantic model."""

    def test_default_values(self):
        """Test that default values are set correctly."""
        config = BasicMemoryConfig()
        assert "main" in config.projects
        assert config.default_project == "main"

    def test_model_post_init(self):
        """Test that model_post_init ensures valid configuration."""
        # Test with empty projects
        config = BasicMemoryConfig(projects={}, default_project="nonexistent")
        assert "main" in config.projects
        assert config.default_project == "main"

        # Test with invalid default project
        config = BasicMemoryConfig(
            projects={"project1": "/path/to/project1"}, default_project="nonexistent"
        )
        assert "main" in config.projects
        assert config.default_project == "main"

    def test_custom_values(self):
        """Test with custom values."""
        config = BasicMemoryConfig(
            projects={"project1": "/path/to/project1"}, default_project="project1"
        )
        assert config.projects["project1"] == "/path/to/project1"
        assert config.default_project == "project1"
        # Main should still be added automatically
        assert "main" in config.projects


class TestConfigManager:
    """Test the ConfigManager class."""

    @pytest.fixture
    def temp_home(self, monkeypatch):
        """Create a temporary directory for testing."""
        with TemporaryDirectory() as tempdir:
            temp_home = Path(tempdir)
            monkeypatch.setattr(Path, "home", lambda: temp_home)
            yield temp_home

    def test_init_creates_config_dir(self, temp_home):
        """Test that init creates the config directory."""
        config_manager = ConfigManager()
        assert config_manager.config_dir.exists()
        assert config_manager.config_dir == temp_home / ".basic-memory"

    def test_init_creates_default_config(self, temp_home):
        """Test that init creates a default config if none exists."""
        config_manager = ConfigManager()
        assert config_manager.config_file.exists()
        assert "main" in config_manager.projects
        assert config_manager.default_project == "main"

    def test_save_and_load_config(self, temp_home):
        """Test saving and loading configuration."""
        config_manager = ConfigManager()
        # Add a project
        config_manager.add_project("test", str(temp_home / "test-project"))
        # Set as default
        config_manager.set_default_project("test")

        # Create a new manager to load from file
        new_manager = ConfigManager()
        assert "test" in new_manager.projects
        assert new_manager.default_project == "test"
        assert Path(new_manager.projects["test"]) == temp_home / "test-project"

    def test_get_project_path(self, temp_home):
        """Test getting a project path."""
        config_manager = ConfigManager()
        config_manager.add_project("test", str(temp_home / "test-project"))

        # Get by name
        path = config_manager.get_project_path("test")
        assert path == temp_home / "test-project"

        # Get default
        path = config_manager.get_project_path()
        assert path == temp_home / "basic-memory"

        # Project does not exist
        with pytest.raises(ValueError):
            config_manager.get_project_path("nonexistent")

    def test_environment_variable(self, temp_home, monkeypatch):
        """Test using environment variable to select project."""
        config_manager = ConfigManager()
        config_manager.add_project("env_test", str(temp_home / "env-test-project"))

        # Set environment variable
        monkeypatch.setenv("BASIC_MEMORY_PROJECT", "env_test")

        # Get project without specifying name
        path = config_manager.get_project_path()
        assert path == temp_home / "env-test-project"

    def test_remove_project(self, temp_home):
        """Test removing a project."""
        config_manager = ConfigManager()
        config_manager.add_project("test", str(temp_home / "test-project"))

        # Remove project
        config_manager.remove_project("test")
        assert "test" not in config_manager.projects

        # Cannot remove default project
        with pytest.raises(ValueError):
            config_manager.remove_project("main")

        # Cannot remove nonexistent project
        with pytest.raises(ValueError):
            config_manager.remove_project("nonexistent")

    def test_load_invalid_config(self, temp_home):
        """Test loading invalid configuration."""
        # Create invalid config file
        config_dir = temp_home / DATA_DIR_NAME
        config_dir.mkdir(parents=True, exist_ok=True)
        config_file = config_dir / CONFIG_FILE_NAME
        config_file.write_text("invalid json")

        # Load config
        config_manager = ConfigManager()

        # Should have default config
        assert "main" in config_manager.projects
        assert config_manager.default_project == "main"

    def test_save_config_error(self, temp_home, monkeypatch):
        """Test error when saving configuration."""
        # Create config manager
        config_manager = ConfigManager()

        # Make write_text raise an exception
        def mock_write_text(content):
            raise PermissionError("Permission denied")

        monkeypatch.setattr(Path, "write_text", mock_write_text)

        # Should not raise exception
        config_manager.save_config(config_manager.config)