Linux MCP Server

"""Tests for logging configuration.""" import json import logging import os import tempfile from pathlib import Path from unittest.mock import patch import pytest from linux_mcp_server.logging_config import ( setup_logging, get_log_directory, JSONFormatter, StructuredFormatter, ) class TestGetLogDirectory: """Test log directory resolution.""" def test_default_log_directory(self): """Test default log directory is in user's home.""" log_dir = get_log_directory() assert log_dir.is_absolute() assert ".local/share/linux-mcp-server/logs" in str(log_dir) def test_custom_log_directory(self): """Test custom log directory from environment variable.""" with tempfile.TemporaryDirectory() as tmpdir: with patch.dict(os.environ, {"LINUX_MCP_LOG_DIR": tmpdir}): log_dir = get_log_directory() assert log_dir == Path(tmpdir) def test_log_directory_created(self): """Test that log directory is created if it doesn't exist.""" with tempfile.TemporaryDirectory() as tmpdir: log_path = Path(tmpdir) / "subdir" / "logs" with patch.dict(os.environ, {"LINUX_MCP_LOG_DIR": str(log_path)}): log_dir = get_log_directory() assert log_dir.exists() assert log_dir.is_dir() class TestSetupLogging: """Test logging setup.""" def test_setup_creates_log_files(self): """Test that setup creates both text and JSON log files.""" with tempfile.TemporaryDirectory() as tmpdir: with patch.dict(os.environ, {"LINUX_MCP_LOG_DIR": tmpdir}): setup_logging() # Log something logger = logging.getLogger("test") logger.info("Test message") # Check both log files exist text_log = Path(tmpdir) / "server.log" json_log = Path(tmpdir) / "server.json" assert text_log.exists() assert json_log.exists() def test_log_level_from_environment(self): """Test that log level can be set from environment variable.""" with tempfile.TemporaryDirectory() as tmpdir: with patch.dict(os.environ, { "LINUX_MCP_LOG_DIR": tmpdir, "LINUX_MCP_LOG_LEVEL": "DEBUG" }): setup_logging() # Root logger should be at DEBUG level root_logger = logging.getLogger() assert root_logger.level == logging.DEBUG def test_default_log_level_is_info(self): """Test that default log level is INFO.""" with tempfile.TemporaryDirectory() as tmpdir: with patch.dict(os.environ, {"LINUX_MCP_LOG_DIR": tmpdir}, clear=True): # Clear any existing LINUX_MCP_LOG_LEVEL os.environ.pop("LINUX_MCP_LOG_LEVEL", None) setup_logging() root_logger = logging.getLogger() assert root_logger.level == logging.INFO class TestStructuredFormatter: """Test structured log formatter.""" def test_format_basic_message(self): """Test formatting a basic log message.""" formatter = StructuredFormatter( '%(asctime)s | %(levelname)s | %(name)s | %(message)s', datefmt='%Y-%m-%d %H:%M:%S' ) record = logging.LogRecord( name="test_module", level=logging.INFO, pathname="", lineno=0, msg="Test message", args=(), exc_info=None, ) formatted = formatter.format(record) # Check format: TIMESTAMP | LEVEL | MODULE | MESSAGE parts = formatted.split(" | ") assert len(parts) == 4 assert parts[1] == "INFO" assert parts[2] == "test_module" assert parts[3] == "Test message" def test_format_with_extra_fields(self): """Test formatting with extra context fields.""" formatter = StructuredFormatter( '%(asctime)s | %(levelname)s | %(name)s | %(message)s', datefmt='%Y-%m-%d %H:%M:%S' ) record = logging.LogRecord( name="test_module", level=logging.INFO, pathname="", lineno=0, msg="Test message", args=(), exc_info=None, ) record.host = "server1.example.com" record.username = "admin" formatted = formatter.format(record) assert "host=server1.example.com" in formatted assert "username=admin" in formatted class TestJSONFormatter: """Test JSON log formatter.""" def test_format_basic_message(self): """Test formatting a basic log message as JSON.""" formatter = JSONFormatter(datefmt='%Y-%m-%dT%H:%M:%S') record = logging.LogRecord( name="test_module", level=logging.INFO, pathname="", lineno=0, msg="Test message", args=(), exc_info=None, ) formatted = formatter.format(record) data = json.loads(formatted) assert data["level"] == "INFO" assert data["logger"] == "test_module" assert data["message"] == "Test message" assert "timestamp" in data def test_format_with_extra_fields(self): """Test formatting with extra context fields as JSON.""" formatter = JSONFormatter(datefmt='%Y-%m-%dT%H:%M:%S') record = logging.LogRecord( name="test_module", level=logging.INFO, pathname="", lineno=0, msg="Test message", args=(), exc_info=None, ) record.host = "server1.example.com" record.username = "admin" record.exit_code = 0 formatted = formatter.format(record) data = json.loads(formatted) assert data["host"] == "server1.example.com" assert data["username"] == "admin" assert data["exit_code"] == 0 def test_format_with_exception(self): """Test formatting with exception information.""" import sys formatter = JSONFormatter(datefmt='%Y-%m-%dT%H:%M:%S') try: raise ValueError("Test error") except ValueError: exc_info = sys.exc_info() record = logging.LogRecord( name="test_module", level=logging.ERROR, pathname="", lineno=0, msg="Error occurred", args=(), exc_info=exc_info, ) formatted = formatter.format(record) data = json.loads(formatted) assert "exception" in data assert "ValueError: Test error" in data["exception"]