"""Tests for configuration module."""
import os
import tempfile
from pathlib import Path
import pytest
import yaml
from homelab_mcp.core.config import (
Config,
LoggingConfig,
ServerConfig,
ServicesConfig,
load_config,
get_config,
set_config,
)
class TestServerConfig:
"""Tests for ServerConfig model."""
def test_default_values(self):
"""Test default server configuration values."""
config = ServerConfig()
assert config.host == "0.0.0.0"
assert config.port == 6971
assert config.transport == "streamable-http"
def test_custom_values(self):
"""Test custom server configuration values."""
config = ServerConfig(host="127.0.0.1", port=8080, transport="sse")
assert config.host == "127.0.0.1"
assert config.port == 8080
assert config.transport == "sse"
class TestLoggingConfig:
"""Tests for LoggingConfig model."""
def test_default_values(self):
"""Test default logging configuration values."""
config = LoggingConfig()
assert config.level == "INFO"
assert "%(asctime)s" in config.format
def test_custom_level(self):
"""Test custom log level."""
config = LoggingConfig(level="DEBUG")
assert config.level == "DEBUG"
class TestServicesConfig:
"""Tests for ServicesConfig model."""
def test_all_services_disabled_by_default(self):
"""Test that all services are disabled by default."""
config = ServicesConfig()
assert config.nginx_proxy_manager.enabled is False
assert config.pihole.enabled is False
assert config.uptime_kuma.enabled is False
assert config.ups_nut.enabled is False
assert config.portainer.enabled is False
assert config.adguard_home.enabled is False
assert config.technitium.enabled is False
assert config.netbox.enabled is False
class TestConfig:
"""Tests for root Config model."""
def test_default_config(self):
"""Test default root configuration."""
config = Config()
assert isinstance(config.server, ServerConfig)
assert isinstance(config.services, ServicesConfig)
assert isinstance(config.logging, LoggingConfig)
def test_nested_config(self):
"""Test nested configuration from dict."""
data = {
"server": {"port": 9000},
"services": {
"pihole": {"enabled": True, "url": "http://localhost:80"}
},
"logging": {"level": "DEBUG"},
}
config = Config(**data)
assert config.server.port == 9000
assert config.services.pihole.enabled is True
assert config.services.pihole.url == "http://localhost:80"
assert config.logging.level == "DEBUG"
class TestLoadConfig:
"""Tests for load_config function."""
def test_load_nonexistent_file_returns_defaults(self):
"""Test that loading nonexistent file returns default config."""
config = load_config("/nonexistent/path/config.yaml")
assert config.server.port == 6971
assert config.services.pihole.enabled is False
def test_load_from_yaml_file(self):
"""Test loading configuration from YAML file."""
with tempfile.NamedTemporaryFile(mode="w", suffix=".yaml", delete=False) as f:
yaml.dump({
"server": {"port": 7777},
"services": {"pihole": {"enabled": True}},
}, f)
f.flush()
try:
config = load_config(f.name)
assert config.server.port == 7777
assert config.services.pihole.enabled is True
finally:
os.unlink(f.name)
def test_load_from_env_var(self, monkeypatch):
"""Test loading configuration from HOMELAB_MCP_CONFIG env var."""
with tempfile.NamedTemporaryFile(mode="w", suffix=".yaml", delete=False) as f:
yaml.dump({"server": {"port": 8888}}, f)
f.flush()
try:
monkeypatch.setenv("HOMELAB_MCP_CONFIG", f.name)
config = load_config()
assert config.server.port == 8888
finally:
os.unlink(f.name)
class TestGlobalConfig:
"""Tests for global config get/set."""
def test_set_and_get_config(self):
"""Test setting and getting global config."""
custom_config = Config(server=ServerConfig(port=5555))
set_config(custom_config)
retrieved = get_config()
assert retrieved.server.port == 5555
