mcp-server-tree-sitter

"""Tests for configuration loading from YAML files using DI.""" import os import tempfile import pytest import yaml from mcp_server_tree_sitter.config import ServerConfig from mcp_server_tree_sitter.di import get_container from tests.test_helpers import configure @pytest.fixture def temp_yaml_file(): """Create a temporary YAML file with test configuration.""" with tempfile.NamedTemporaryFile(suffix=".yaml", mode="w+", delete=False) as temp_file: test_config = { "cache": {"enabled": True, "max_size_mb": 256, "ttl_seconds": 3600}, "security": {"max_file_size_mb": 10, "excluded_dirs": [".git", "node_modules", "__pycache__", ".cache"]}, "language": {"auto_install": True, "default_max_depth": 7}, } yaml.dump(test_config, temp_file) temp_file.flush() temp_file_path = temp_file.name yield temp_file_path # Clean up the temporary file os.unlink(temp_file_path) def test_server_config_from_file(temp_yaml_file): """Test the ServerConfig.from_file method directly.""" # Print debug information print(f"Temporary YAML file created at: {temp_yaml_file}") with open(temp_yaml_file, "r") as f: print(f"File contents:\n{f.read()}") # Call from_file directly config = ServerConfig.from_file(temp_yaml_file) # Print the result for debugging print(f"ServerConfig from file: {config}") # Verify that the config object has the expected values assert config.cache.enabled is True assert config.cache.max_size_mb == 256 assert config.cache.ttl_seconds == 3600 assert config.security.max_file_size_mb == 10 assert ".git" in config.security.excluded_dirs assert config.language.auto_install is True assert config.language.default_max_depth == 7 def test_load_config_function_di(temp_yaml_file): """Test the config loading with DI container.""" # Print debug information print(f"Temporary YAML file created at: {temp_yaml_file}") # Get the container directly container = get_container() original_config = container.get_config() # Save original values to restore later original_cache_size = original_config.cache.max_size_mb original_security_size = original_config.security.max_file_size_mb original_depth = original_config.language.default_max_depth try: # Load config file using container's config manager container.config_manager.load_from_file(temp_yaml_file) config = container.get_config() # Verify that the config values were loaded correctly assert config.cache.max_size_mb == 256 assert config.security.max_file_size_mb == 10 assert config.language.default_max_depth == 7 finally: # Restore original values container.config_manager.update_value("cache.max_size_mb", original_cache_size) container.config_manager.update_value("security.max_file_size_mb", original_security_size) container.config_manager.update_value("language.default_max_depth", original_depth) def test_configure_helper(temp_yaml_file): """Test that the configure helper function properly loads values from a YAML file.""" # Print debug information print(f"Temporary YAML file created at: {temp_yaml_file}") print(f"File exists: {os.path.exists(temp_yaml_file)}") # Get container to save original values container = get_container() original_config = container.get_config() # Save original values to restore later original_cache_size = original_config.cache.max_size_mb original_security_size = original_config.security.max_file_size_mb original_depth = original_config.language.default_max_depth try: # Call the configure helper with the path to the temp file result = configure(config_path=temp_yaml_file) # Print the result for debugging print(f"Configure result: {result}") # Verify the returned configuration matches the expected values # Cache settings assert result["cache"]["enabled"] is True assert result["cache"]["max_size_mb"] == 256 assert result["cache"]["ttl_seconds"] == 3600 # Security settings assert result["security"]["max_file_size_mb"] == 10 assert ".git" in result["security"]["excluded_dirs"] # Language settings assert result["language"]["auto_install"] is True assert result["language"]["default_max_depth"] == 7 # Also verify the container's config was updated config = container.get_config() assert config.cache.max_size_mb == 256 assert config.security.max_file_size_mb == 10 assert config.language.default_max_depth == 7 finally: # Restore original values container.config_manager.update_value("cache.max_size_mb", original_cache_size) container.config_manager.update_value("security.max_file_size_mb", original_security_size) container.config_manager.update_value("language.default_max_depth", original_depth) def test_real_yaml_example_di(): """Test with a real-world example like the one in the issue.""" with tempfile.NamedTemporaryFile(suffix=".yaml", mode="w+", delete=False) as temp_file: # Copy the example from the issue temp_file.write("""cache: enabled: true max_size_mb: 256 ttl_seconds: 3600 security: max_file_size_mb: 10 excluded_dirs: - .git - node_modules - __pycache__ - .cache - .claude - .config - .idea - .llm-context - .local - .npm - .phpstorm_helpers - .tmp - .venv - .vscode - .w3m - admin/logs - cache - logs - tools/data_management/.error_codes_journal - tools/code_management/.patch_journal - runtime - vendor - venv - .aider* - .bash* - .claude-preferences.json - .codeiumignore - .continuerules - .env - .lesshst - .php_history - .python-version - .viminfo - .wget-hsts - .windsurfrules language: auto_install: true default_max_depth: 7 """) temp_file.flush() temp_file_path = temp_file.name try: # Get container to save original values container = get_container() original_config = container.get_config() # Save original values to restore later original_cache_size = original_config.cache.max_size_mb original_security_size = original_config.security.max_file_size_mb original_depth = original_config.language.default_max_depth try: # Call configure helper result = configure(config_path=temp_file_path) # Print the result for debugging print(f"Configure result: {result}") # Verify the returned configuration matches the expected values assert result["cache"]["max_size_mb"] == 256 assert result["security"]["max_file_size_mb"] == 10 assert ".claude" in result["security"]["excluded_dirs"] assert result["language"]["auto_install"] is True assert result["language"]["default_max_depth"] == 7 # Also verify the container's config was updated config = container.get_config() assert config.cache.max_size_mb == 256 assert config.security.max_file_size_mb == 10 assert config.language.default_max_depth == 7 finally: # Restore original values container.config_manager.update_value("cache.max_size_mb", original_cache_size) container.config_manager.update_value("security.max_file_size_mb", original_security_size) container.config_manager.update_value("language.default_max_depth", original_depth) finally: # Clean up the temporary file os.unlink(temp_file_path)