Wikipedia MCP Server

"""
Tests for CLI country support.
"""

import pytest
import subprocess
import time
import signal
import os
from unittest.mock import patch
import sys


class TestCountryCLI:
    """Test CLI functionality with country codes."""

    def test_cli_country_argument_help(self):
        """Test that country argument appears in help."""
        result = subprocess.run(
            [sys.executable, "-m", "wikipedia_mcp", "--help"],
            capture_output=True,
            text=True,
        )

        assert result.returncode == 0
        assert "--country" in result.stdout
        assert "Country/locale code" in result.stdout
        assert "--list-countries" in result.stdout

    def test_cli_list_countries_functionality(self):
        """Test the --list-countries functionality."""
        result = subprocess.run(
            [sys.executable, "-m", "wikipedia_mcp", "--list-countries"],
            capture_output=True,
            text=True,
        )

        assert result.returncode == 0
        assert "Supported Country/Locale Codes:" in result.stdout
        assert "en: US, USA, United States, UK, GB" in result.stdout  # Updated expectation
        assert "zh-hans: CN, China" in result.stdout
        assert "zh-tw: TW, Taiwan" in result.stdout
        assert "Examples:" in result.stdout
        assert "wikipedia-mcp --country US" in result.stdout

    def test_cli_country_validation_error(self):
        """Test error handling for invalid country codes."""
        result = subprocess.run(
            [sys.executable, "-m", "wikipedia_mcp", "--country", "INVALID"],
            capture_output=True,
            text=True,
        )

        assert result.returncode == 0  # Should exit gracefully
        assert "Configuration error" in result.stderr
        assert "Unsupported country/locale: 'INVALID'" in result.stdout
        assert "Use --list-countries" in result.stdout

    def test_cli_country_and_language_conflict(self):
        """Test handling of conflicting country and language arguments."""
        result = subprocess.run(
            [
                sys.executable,
                "-m",
                "wikipedia_mcp",
                "--language",
                "ja",
                "--country",
                "US",
            ],
            capture_output=True,
            text=True,
        )

        assert result.returncode == 0  # Should exit gracefully
        assert "Cannot specify both --language and --country" in result.stdout

    def test_cli_country_start_timeout(self):
        """Test that server starts successfully with country code."""
        process = None
        try:
            process = subprocess.Popen(
                [
                    sys.executable,
                    "-m",
                    "wikipedia_mcp",
                    "--country",
                    "JP",
                    "--transport",
                    "sse",
                    "--port",
                    "8200",
                ],
                stdout=subprocess.PIPE,
                stderr=subprocess.PIPE,
                text=True,
            )

            # Give the server time to start
            time.sleep(2)

            # Check if process is still running (didn't crash)
            poll_result = process.poll()
            assert poll_result is None, "Server process should still be running"

        finally:
            if process and process.poll() is None:
                process.terminate()
                try:
                    process.wait(timeout=5)
                except subprocess.TimeoutExpired:
                    process.kill()

    def test_cli_country_short_flag(self):
        """Test that -c short flag works for country."""
        process = None
        try:
            process = subprocess.Popen(
                [
                    sys.executable,
                    "-m",
                    "wikipedia_mcp",
                    "-c",
                    "FR",
                    "--transport",
                    "sse",
                    "--port",
                    "8201",
                ],
                stdout=subprocess.PIPE,
                stderr=subprocess.PIPE,
                text=True,
            )

            # Give the server time to start
            time.sleep(2)

            # Check if process is still running
            poll_result = process.poll()
            assert poll_result is None, "Server process should still be running"

        finally:
            if process and process.poll() is None:
                process.terminate()
                try:
                    process.wait(timeout=5)
                except subprocess.TimeoutExpired:
                    process.kill()


class TestCLICountryIntegration:
    """Integration tests for CLI with country support."""

    def test_country_to_language_resolution_log(self):
        """Test that country resolution is properly logged."""
        process = None
        try:
            process = subprocess.Popen(
                [sys.executable, "-m", "wikipedia_mcp", "--country", "Taiwan"],
                stdout=subprocess.PIPE,
                stderr=subprocess.PIPE,
                text=True,
            )

            # Give the server time to start and log
            time.sleep(3)

            # Terminate and read output
            process.terminate()
            stdout, stderr = process.communicate(timeout=5)

            # Check that Taiwan country was properly logged
            assert "Starting Wikipedia MCP server with stdio transport for country: Taiwan" in stderr

        except subprocess.TimeoutExpired:
            if process:
                process.kill()
                stdout, stderr = process.communicate()

    def test_country_with_cache_option(self):
        """Test that country works with caching option."""
        process = None
        try:
            process = subprocess.Popen(
                [
                    sys.executable,
                    "-m",
                    "wikipedia_mcp",
                    "--country",
                    "DE",
                    "--enable-cache",
                    "--transport",
                    "sse",
                    "--port",
                    "8202",
                ],
                stdout=subprocess.PIPE,
                stderr=subprocess.PIPE,
                text=True,
            )

            # Give the server time to start
            time.sleep(2)

            # Check if process is still running
            poll_result = process.poll()
            assert poll_result is None, "Server with country and cache should start successfully"

        finally:
            if process and process.poll() is None:
                process.terminate()
                try:
                    process.wait(timeout=5)
                except subprocess.TimeoutExpired:
                    process.kill()

    def test_country_with_sse_transport(self):
        """Test that country works with SSE transport."""
        process = None
        try:
            process = subprocess.Popen(
                [
                    sys.executable,
                    "-m",
                    "wikipedia_mcp",
                    "--country",
                    "IT",
                    "--transport",
                    "sse",
                    "--port",
                    "8203",
                ],
                stdout=subprocess.PIPE,
                stderr=subprocess.PIPE,
                text=True,
            )

            # Give the server time to start
            time.sleep(2)

            # Check if process is still running
            poll_result = process.poll()
            assert poll_result is None, "Server with country and SSE transport should start successfully"

        finally:
            if process and process.poll() is None:
                process.terminate()
                try:
                    process.wait(timeout=5)
                except subprocess.TimeoutExpired:
                    process.kill()


class TestCLICountryExamples:
    """Test CLI with various country examples from documentation."""

    @pytest.mark.parametrize(
        "country,expected_language",
        [
            ("US", "en"),
            ("CN", "zh-hans"),
            ("TW", "zh-tw"),
            ("Japan", "ja"),
            ("Germany", "de"),
            ("france", "fr"),  # Test case insensitive
        ],
    )
    def test_cli_country_examples(self, country, expected_language):
        """Test CLI with different country examples."""
        process = None
        port = 8210 + hash(country) % 100  # Generate unique port for each test
        try:
            process = subprocess.Popen(
                [
                    sys.executable,
                    "-m",
                    "wikipedia_mcp",
                    "--country",
                    country,
                    "--transport",
                    "sse",
                    "--port",
                    str(port),
                ],
                stdout=subprocess.PIPE,
                stderr=subprocess.PIPE,
                text=True,
            )

            # Give the server time to start
            time.sleep(2)

            # Check if process is still running (successful start)
            poll_result = process.poll()
            assert poll_result is None, f"Server should start successfully with country: {country}"

        finally:
            if process and process.poll() is None:
                process.terminate()
                try:
                    process.wait(timeout=5)
                except subprocess.TimeoutExpired:
                    process.kill()