Wikipedia MCP Server

""" Tests for country/locale support functionality. """ import pytest from unittest.mock import patch, Mock from wikipedia_mcp.wikipedia_client import WikipediaClient from wikipedia_mcp.server import create_server class TestCountryToLanguageMapping: """Test country code to language code mapping functionality.""" def test_resolve_country_to_language_basic_codes(self): """Test basic country code resolution.""" client = WikipediaClient() test_cases = [ ('US', 'en'), ('CN', 'zh-hans'), ('TW', 'zh-tw'), ('JP', 'ja'), ('DE', 'de'), ('FR', 'fr'), ('ES', 'es'), ('IT', 'it'), ('RU', 'ru'), ('BR', 'pt'), ] for country, expected_lang in test_cases: result = client._resolve_country_to_language(country) assert result == expected_lang, f"Failed for {country}: expected {expected_lang}, got {result}" def test_resolve_country_to_language_full_names(self): """Test country full name resolution.""" client = WikipediaClient() test_cases = [ ('United States', 'en'), ('China', 'zh-hans'), ('Taiwan', 'zh-tw'), ('Japan', 'ja'), ('Germany', 'de'), ('France', 'fr'), ('Spain', 'es'), ('Italy', 'it'), ('Russia', 'ru'), ('Brazil', 'pt'), ] for country, expected_lang in test_cases: result = client._resolve_country_to_language(country) assert result == expected_lang, f"Failed for {country}: expected {expected_lang}, got {result}" def test_resolve_country_case_insensitive(self): """Test that country resolution is case insensitive.""" client = WikipediaClient() test_cases = [ ('us', 'en'), ('Us', 'en'), ('US', 'en'), ('uS', 'en'), ('china', 'zh-hans'), ('CHINA', 'zh-hans'), ('China', 'zh-hans'), ] for country, expected_lang in test_cases: result = client._resolve_country_to_language(country) assert result == expected_lang, f"Failed for {country}: expected {expected_lang}, got {result}" def test_resolve_country_invalid_code(self): """Test error handling for invalid country codes.""" client = WikipediaClient() with pytest.raises(ValueError) as exc_info: client._resolve_country_to_language('INVALID') error_msg = str(exc_info.value) assert 'Unsupported country/locale' in error_msg assert 'INVALID' in error_msg assert 'Supported country codes include:' in error_msg assert '--language parameter' in error_msg def test_resolve_country_whitespace_handling(self): """Test that whitespace is properly handled.""" client = WikipediaClient() test_cases = [ (' US ', 'en'), ('\tJP\t', 'ja'), (' China ', 'zh-hans'), ] for country, expected_lang in test_cases: result = client._resolve_country_to_language(country) assert result == expected_lang, f"Failed for '{country}': expected {expected_lang}, got {result}" class TestWikipediaClientCountrySupport: """Test WikipediaClient initialization with country codes.""" def test_client_init_with_country_code(self): """Test initializing client with country code.""" client = WikipediaClient(country='TW') assert client.original_input == 'TW' assert client.input_type == 'country' assert client.resolved_language == 'zh-tw' assert client.base_language == 'zh' assert client.language_variant == 'zh-tw' assert client.api_url == 'https://zh.wikipedia.org/w/api.php' def test_client_init_with_country_name(self): """Test initializing client with country full name.""" client = WikipediaClient(country='Japan') assert client.original_input == 'Japan' assert client.input_type == 'country' assert client.resolved_language == 'ja' assert client.base_language == 'ja' assert client.language_variant is None assert client.api_url == 'https://ja.wikipedia.org/w/api.php' def test_client_init_with_language_fallback(self): """Test that language parameter still works when country is not provided.""" client = WikipediaClient(language='zh-hans') assert client.original_input == 'zh-hans' assert client.input_type == 'language' assert client.resolved_language == 'zh-hans' assert client.base_language == 'zh' assert client.language_variant == 'zh-hans' def test_client_init_country_overrides_language(self): """Test that country parameter overrides language parameter.""" client = WikipediaClient(language='en', country='JP') assert client.original_input == 'JP' assert client.input_type == 'country' assert client.resolved_language == 'ja' assert client.base_language == 'ja' assert client.language_variant is None def test_client_init_invalid_country(self): """Test error handling for invalid country during initialization.""" with pytest.raises(ValueError) as exc_info: WikipediaClient(country='INVALID') error_msg = str(exc_info.value) assert 'Unsupported country/locale: \'INVALID\'' in error_msg def test_client_init_with_country_and_cache(self): """Test initializing client with country and caching enabled.""" client = WikipediaClient(country='US', enable_cache=True) assert client.original_input == 'US' assert client.input_type == 'country' assert client.resolved_language == 'en' assert client.enable_cache is True class TestServerCountryIntegration: """Test server creation with country support.""" def test_create_server_with_country(self): """Test creating server with country parameter.""" server = create_server(country='TW') assert server is not None assert server.name == "Wikipedia" def test_create_server_with_country_and_cache(self): """Test creating server with country and caching.""" server = create_server(country='Japan', enable_cache=True) assert server is not None assert server.name == "Wikipedia" def test_create_server_country_overrides_language(self): """Test that country parameter overrides language in server creation.""" server = create_server(language='en', country='CN') assert server is not None assert server.name == "Wikipedia" def test_create_server_invalid_country(self): """Test error handling for invalid country in server creation.""" with pytest.raises(ValueError): create_server(country='INVALID') class TestCountryAPIIntegration: """Test API integration with country codes.""" @patch('requests.get') def test_search_with_country_code(self, mock_get): """Test that search works correctly with country-resolved language.""" # Setup mock response for Chinese Wikipedia mock_response = Mock() mock_response.raise_for_status.return_value = None mock_response.json.return_value = { 'query': { 'search': [ { 'title': '中華民國', 'snippet': 'Taiwan search result', 'pageid': 123, 'wordcount': 1000, 'timestamp': '2024-01-01T00:00:00Z' } ] } } mock_get.return_value = mock_response # Test search with Taiwan country code client = WikipediaClient(country='TW') results = client.search('Taiwan', limit=5) # Verify API was called with correct language mock_get.assert_called_once() call_args = mock_get.call_args assert call_args[0][0] == 'https://zh.wikipedia.org/w/api.php' params = call_args[1]['params'] assert params['variant'] == 'zh-tw' # Should use Taiwan variant assert params['srsearch'] == 'Taiwan' # Verify results assert len(results) == 1 assert results[0]['title'] == '中華民國' class TestCountryMappingCompleteness: """Test the completeness and consistency of country mappings.""" def test_country_mapping_structure(self): """Test that country mappings have proper structure.""" client = WikipediaClient() for country, language in client.COUNTRY_TO_LANGUAGE.items(): assert isinstance(country, str), f"Country key must be string: {country}" assert isinstance(language, str), f"Language value must be string: {language}" assert len(country) > 0, f"Country cannot be empty: {country}" assert len(language) > 0, f"Language cannot be empty: {language}" def test_major_countries_covered(self): """Test that major countries are covered in the mapping.""" client = WikipediaClient() # Major countries that should be supported major_countries = [ 'US', 'CN', 'JP', 'DE', 'FR', 'UK', 'IN', 'BR', 'RU', 'CA', 'AU', 'IT', 'ES', 'KR', 'MX', 'ID', 'TR', 'SA', 'TH', 'TW' ] for country in major_countries: assert country in client.COUNTRY_TO_LANGUAGE, f"Major country missing: {country}" def test_language_variants_consistency(self): """Test that country mappings are consistent with language variants.""" client = WikipediaClient() # Countries that should map to language variants variant_mappings = [ ('CN', 'zh-hans'), ('TW', 'zh-tw'), ('HK', 'zh-hk'), ('MO', 'zh-mo'), ('SG', 'zh-sg'), ('MY', 'zh-my'), ] for country, expected_variant in variant_mappings: assert client.COUNTRY_TO_LANGUAGE[country] == expected_variant def test_english_speaking_countries(self): """Test that major English-speaking countries map to English.""" client = WikipediaClient() english_countries = ['US', 'UK', 'CA', 'AU', 'NZ', 'IE', 'ZA'] for country in english_countries: assert client.COUNTRY_TO_LANGUAGE[country] == 'en', f"{country} should map to 'en'" def test_no_duplicate_country_names(self): """Test that there are no conflicting country names.""" client = WikipediaClient() # Check for potential conflicts (case-insensitive) seen_countries = set() conflicts = [] for country in client.COUNTRY_TO_LANGUAGE.keys(): country_lower = country.lower() if country_lower in seen_countries: conflicts.append(country) seen_countries.add(country_lower) assert len(conflicts) == 0, f"Found conflicting country entries: {conflicts}"