test_config.py

import os
from unittest.mock import patch
from greptimedb_mcp_server.config import Config


def test_config_default_values():
    """
    Test default configuration values
    """
    with patch.dict(os.environ, {}, clear=True):
        with patch("sys.argv", ["script_name"]):
            config = Config.from_env_arguments()

            assert config.host == "localhost"
            assert config.port == 4002
            assert config.database == "public"
            assert config.user == ""
            assert config.password == ""
            assert config.time_zone == ""
            assert config.http_protocol == "http"
            assert config.mask_enabled is True
            assert config.mask_patterns == ""
            assert config.transport == "stdio"
            assert config.listen_host == "0.0.0.0"
            assert config.listen_port == 8080


def test_config_env_variables():
    """
    Test configuration via environment variables
    """
    env_vars = {
        "GREPTIMEDB_HOST": "test-host",
        "GREPTIMEDB_PORT": "5432",
        "GREPTIMEDB_DATABASE": "test_db",
        "GREPTIMEDB_USER": "test_user",
        "GREPTIMEDB_PASSWORD": "test_password",
        "GREPTIMEDB_TIMEZONE": "test_tz",
        "GREPTIMEDB_HTTP_PROTOCOL": "https",
        "GREPTIMEDB_MASK_ENABLED": "false",
        "GREPTIMEDB_MASK_PATTERNS": "phone,address",
        "GREPTIMEDB_TRANSPORT": "streamable-http",
        "GREPTIMEDB_LISTEN_HOST": "127.0.0.1",
        "GREPTIMEDB_LISTEN_PORT": "3000",
    }

    with patch.dict(os.environ, env_vars):
        with patch("sys.argv", ["script_name"]):
            config = Config.from_env_arguments()

            assert config.host == "test-host"
            assert config.port == 5432
            assert config.database == "test_db"
            assert config.user == "test_user"
            assert config.password == "test_password"
            assert config.time_zone == "test_tz"
            assert config.http_protocol == "https"
            assert config.mask_enabled is False
            assert config.mask_patterns == "phone,address"
            assert config.transport == "streamable-http"
            assert config.listen_host == "127.0.0.1"
            assert config.listen_port == 3000


def test_config_cli_arguments():
    """
    Test configuration via command-line arguments
    """
    cli_args = [
        "script_name",
        "--host",
        "cli-host",
        "--port",
        "9999",
        "--database",
        "cli_db",
        "--user",
        "cli_user",
        "--password",
        "cli_password",
        "--timezone",
        "cli_tz",
        "--http-protocol",
        "https",
        "--mask-enabled",
        "false",
        "--mask-patterns",
        "custom1,custom2",
        "--transport",
        "sse",
        "--listen-host",
        "192.168.1.1",
        "--listen-port",
        "9090",
    ]

    with patch.dict(os.environ, {}, clear=True):
        with patch("sys.argv", cli_args):
            config = Config.from_env_arguments()

            assert config.host == "cli-host"
            assert config.port == 9999
            assert config.database == "cli_db"
            assert config.user == "cli_user"
            assert config.password == "cli_password"
            assert config.time_zone == "cli_tz"
            assert config.http_protocol == "https"
            assert config.mask_enabled is False
            assert config.mask_patterns == "custom1,custom2"
            assert config.transport == "sse"
            assert config.listen_host == "192.168.1.1"
            assert config.listen_port == 9090


def test_config_precedence():
    """
    Test configuration precedence (CLI arguments override environment variables)
    """
    env_vars = {
        "GREPTIMEDB_HOST": "env-host",
        "GREPTIMEDB_PORT": "6666",
        "GREPTIMEDB_DATABASE": "env_db",
        "GREPTIMEDB_USER": "env_user",
        "GREPTIMEDB_PASSWORD": "env_password",
        "GREPTIMEDB_TIMEZONE": "env_tz",
        "GREPTIMEDB_HTTP_PROTOCOL": "http",
        "GREPTIMEDB_MASK_ENABLED": "true",
        "GREPTIMEDB_MASK_PATTERNS": "env_pattern",
        "GREPTIMEDB_TRANSPORT": "stdio",
        "GREPTIMEDB_LISTEN_HOST": "env-listen-host",
        "GREPTIMEDB_LISTEN_PORT": "1111",
    }

    cli_args = [
        "script_name",
        "--host",
        "cli-host",
        "--port",
        "9999",
        "--database",
        "cli_db",
        "--user",
        "cli_user",
        "--password",
        "cli_password",
        "--timezone",
        "cli_tz",
        "--http-protocol",
        "https",
        "--mask-enabled",
        "false",
        "--mask-patterns",
        "cli_pattern",
        "--transport",
        "streamable-http",
        "--listen-host",
        "cli-listen-host",
        "--listen-port",
        "2222",
    ]

    with patch.dict(os.environ, env_vars):
        with patch("sys.argv", cli_args):
            config = Config.from_env_arguments()

            assert config.host == "cli-host"
            assert config.port == 9999
            assert config.database == "cli_db"
            assert config.user == "cli_user"
            assert config.password == "cli_password"
            assert config.time_zone == "cli_tz"
            assert config.http_protocol == "https"
            assert config.mask_enabled is False
            assert config.mask_patterns == "cli_pattern"
            assert config.transport == "streamable-http"
            assert config.listen_host == "cli-listen-host"
            assert config.listen_port == 2222