Adding providers tests

Author: vbossica

python_gpt_po/tests/providers/__init__.py

@@ -0,0 +1,30 @@
+import pytest
+from unittest.mock import MagicMock
+
+from python_gpt_po.models.provider_clients import ProviderClients
+from python_gpt_po.services.providers.anthropic_provider import AnthropicProvider
+
+@pytest.fixture
+def mock_provider_clients() -> ProviderClients:
+    """Mock provider clients for testing."""
+    clients = ProviderClients()
+    clients.anthropic_client = MagicMock()
+    clients.anthropic_client.api_key = "sk-ant-mock-key"
+    return clients
+
+def test_translate(mock_provider_clients: ProviderClients) -> None:
+    """Test bulk translation with Anthropic."""
+    # Setup mock response
+    mock_chatcompletion = MagicMock()
+    mock_chatcompletion.content = [MagicMock()]
+    mock_chatcompletion.content[0].text = '["Bonjour", "Monde", "Bienvenue dans notre application", "Au revoir"]'
+    mock_provider_clients.anthropic_client.messages.create.return_value = mock_chatcompletion
+
+    provider = AnthropicProvider()
+    translations = provider.translate(
+        provider_clients=mock_provider_clients,
+        model="gpt-4",
+        content="['Hello', 'World', 'Welcome to our application', 'Goodbye']"
+    )
+
+    assert translations == '["Bonjour", "Monde", "Bienvenue dans notre application", "Au revoir"]'

python_gpt_po/tests/providers/test_anthropic_provider.py

@@ -0,0 +1,30 @@
+import pytest
+from unittest.mock import MagicMock
+
+from python_gpt_po.models.provider_clients import ProviderClients
+from python_gpt_po.services.providers.azure_openai_provider import AzureOpenAIProvider
+
+@pytest.fixture
+def mock_provider_clients() -> ProviderClients:
+    """Mock provider clients for testing."""
+    clients = ProviderClients()
+    clients.azure_openai_client = MagicMock()
+    clients.azure_openai_client.api_key = "sk-aoi-mock-key"
+    return clients
+
+def test_translate(mock_provider_clients: ProviderClients) -> None:
+    """Test bulk translation with Azure OpenAI."""
+    # Setup mock response
+    mock_chatcompletion = MagicMock()
+    mock_chatcompletion.choices = [MagicMock()]
+    mock_chatcompletion.choices[0].message.content = '["Bonjour", "Monde", "Bienvenue dans notre application", "Au revoir"]'
+    mock_provider_clients.azure_openai_client.chat.completions.create.return_value = mock_chatcompletion
+
+    provider = AzureOpenAIProvider()
+    translations = provider.translate(
+        provider_clients=mock_provider_clients,
+        model="gpt-4",
+        content="['Hello', 'World', 'Welcome to our application', 'Goodbye']"
+    )
+
+    assert translations == '["Bonjour", "Monde", "Bienvenue dans notre application", "Au revoir"]'

python_gpt_po/tests/providers/test_azure_openai_provider.py

@@ -0,0 +1,41 @@
+import pytest
+from unittest.mock import MagicMock, patch
+
+from python_gpt_po.models.provider_clients import ProviderClients
+from python_gpt_po.services.providers.deepseek_provider import DeepSeekProvider
+
+DEEPSEEK_TRANSLATION_RESPONSE = {
+    "choices": [
+        {
+            "message": {
+                "content": "```json\n[\"Bonjour\", \"Monde\", \"Bienvenue dans notre application\", \"Au revoir\"]\n```"
+            }
+        }
+    ]
+}
+
+@pytest.fixture
+def mock_provider_clients() -> ProviderClients:
+    """Mock provider clients for testing."""
+    clients = ProviderClients()
+    clients.deepseek_api_key = "sk-deepseek-mock-key"
+    clients.deepseek_base_url = "https://api.deepseek.com/v1"
+    return clients
+
+@patch('python_gpt_po.services.providers.deepseek_provider.requests.post')
+def test_translate(mock_post: MagicMock, mock_provider_clients: ProviderClients) -> None:
+    """Test translation with DeepSeek."""
+    # Setup mock response
+    mock_response = MagicMock()
+    mock_response.json.return_value = DEEPSEEK_TRANSLATION_RESPONSE
+    mock_post.return_value = mock_response
+
+    provider = DeepSeekProvider()
+    translations = provider.translate(
+        provider_clients=mock_provider_clients,
+        model="deepseek-chat",
+        content="['Hello', 'World', 'Welcome to our application', 'Goodbye']"
+    )
+
+    print(type(translations))
+    assert translations == '```json\n["Bonjour", "Monde", "Bienvenue dans notre application", "Au revoir"]\n```'

python_gpt_po/tests/providers/test_deepseek_provider.py

@@ -0,0 +1,29 @@
+import pytest
+from unittest.mock import MagicMock
+
+from python_gpt_po.models.provider_clients import ProviderClients
+from python_gpt_po.services.providers.openai_provider import OpenAIProvider
+
+@pytest.fixture
+def mock_provider_clients() -> ProviderClients:
+    """Mock provider clients for testing."""
+    clients = ProviderClients()
+    clients.openai_client = MagicMock()
+    return clients
+
+def test_translate(mock_provider_clients: ProviderClients) -> None:
+    """Test bulk translation with OpenAI."""
+    # Setup mock response
+    mock_chatcompletion = MagicMock()
+    mock_chatcompletion.choices = [MagicMock()]
+    mock_chatcompletion.choices[0].message.content = '["Bonjour", "Monde", "Bienvenue dans notre application", "Au revoir"]'
+    mock_provider_clients.openai_client.chat.completions.create.return_value = mock_chatcompletion
+
+    provider = OpenAIProvider()
+    translations = provider.translate(
+        provider_clients=mock_provider_clients,
+        model="gpt-4",
+        content="['Hello', 'World', 'Welcome to our application', 'Goodbye']"
+    )
+
+    assert translations == '["Bonjour", "Monde", "Bienvenue dans notre application", "Au revoir"]'

python_gpt_po/tests/providers/test_openai_provider.py

@@ -16,6 +16,7 @@
 from python_gpt_po.services.model_manager import ModelManager
 from python_gpt_po.services.po_file_handler import POFileHandler
 from python_gpt_po.services.translation_service import TranslationService
+from providers.test_deepseek_provider import DEEPSEEK_TRANSLATION_RESPONSE
 
 logging.basicConfig(level=logging.INFO)
 
@@ -99,38 +100,6 @@
     ]
 }
 
-# Translation responses for different providers
-OPENAI_TRANSLATION_RESPONSE = {
-    "choices": [
-        {
-            "message": {
-                "content": '["Bonjour", "Monde", "Bienvenue dans notre application", "Au revoir"]'
-            }
-        }
-    ]
-}
-
-AZURE_OPENAI_TRANSLATION_RESPONSE = OPENAI_TRANSLATION_RESPONSE
-
-ANTHROPIC_TRANSLATION_RESPONSE = {
-    "content": [
-        {
-            "text": '["Bonjour", "Monde", "Bienvenue dans notre application", "Au revoir"]'
-        }
-    ]
-}
-
-DEEPSEEK_TRANSLATION_RESPONSE = {
-    "choices": [
-        {
-            "message": {
-                "content": "```json\n[\"Bonjour\", \"Monde\", \"Bienvenue dans notre application\", \"Au revoir\"]\n```"
-            }
-        }
-    ]
-}
-
-
 @pytest.fixture
 def temp_po_file(tmp_path: str) -> str:
     """Create a temporary PO file for testing."""
@@ -227,7 +196,7 @@ def translation_service_deepseek(translation_config_deepseek: TranslationConfig)
 
 
 @patch('requests.get')
-def test_get_openai_models(mock_get, mock_provider_clients: ProviderClients):
+def test_get_openai_models(mock_get: MagicMock, mock_provider_clients: ProviderClients):
     """Test getting OpenAI models."""
     # Setup mock response
     mock_response = MagicMock()
@@ -249,7 +218,7 @@ def test_get_openai_models(mock_get, mock_provider_clients: ProviderClients):
 
 
 @patch('requests.get')
-def test_get_ayure_openai_models(mock_get, mock_provider_clients: ProviderClients):
+def test_get_ayure_openai_models(mock_get: MagicMock, mock_provider_clients: ProviderClients):
     """Test getting OpenAI models."""
     # Setup mock response
     mock_response = MagicMock()
@@ -415,7 +384,7 @@ def test_clean_json_response(translation_service_deepseek: TranslationService):
 
 
 @patch('polib.pofile')
-def test_process_po_file_all_providers(mock_pofile,
+def test_process_po_file_all_providers(mock_pofile: MagicMock,
                                        translation_service_openai: TranslationService,
                                        translation_service_anthropic: TranslationService,
                                        translation_service_deepseek: TranslationService,
@@ -462,7 +431,7 @@ def test_process_po_file_all_providers(mock_pofile,
 
 
 @patch('python_gpt_po.services.po_file_handler.POFileHandler.disable_fuzzy_translations')
-def test_fuzzy_flag_handling(mock_disable_fuzzy, translation_service_openai: TranslationService, temp_po_file):
+def test_fuzzy_flag_handling(mock_disable_fuzzy: MagicMock, translation_service_openai: TranslationService, temp_po_file: MagicMock):
     """Test handling of fuzzy translations."""
     # Enable fuzzy flag
     translation_service_openai.config.flags.fuzzy = True
@@ -520,7 +489,7 @@ def test_validation_model_connection_all_providers(
 
 @patch('os.walk')
 @patch('polib.pofile')
-def test_scan_and_process_po_files(mock_pofile, mock_walk, translation_service_openai: TranslationService):
+def test_scan_and_process_po_files(mock_pofile: MagicMock, mock_walk: MagicMock, translation_service_openai: TranslationService):
     """Test scanning and processing PO files."""
     # Setup mock directory structure
     mock_walk.return_value = [

python_gpt_po/tests/test_multi_provider.py

@@ -28,7 +28,7 @@ def fixture_mock_openai_client():
 
 
 @pytest.fixture(name='translation_config')
-def fixture_translation_config(mock_openai_client):
+def fixture_translation_config(mock_openai_client: MagicMock):
     """
     Fixture to create a TranslationConfig instance.
     """
@@ -50,7 +50,7 @@ def fixture_translation_config(mock_openai_client):
 
 
 @pytest.fixture(name='translation_service')
-def fixture_translation_service(translation_config):
+def fixture_translation_service(translation_config: MagicMock):
     """
     Fixture to create a TranslationService instance.
     """
@@ -65,14 +65,14 @@ def fixture_mock_po_file_handler():
     return MagicMock(spec=POFileHandler)
 
 
-def test_validate_openai_connection(translation_service):
+def test_validate_openai_connection(translation_service: MagicMock):
     """Test to validate the connection."""
     # The new method is validate_provider_connection instead of validate_openai_connection
     assert translation_service.validate_provider_connection() is True
 
 
 @patch('python_gpt_po.services.po_file_handler.POFileHandler')
-def test_process_po_file(mock_po_file_handler_class, translation_service, tmp_path):
+def test_process_po_file(mock_po_file_handler_class: MagicMock, translation_service: MagicMock, tmp_path: MagicMock):
     """
     Test the process_po_file method.
     """
@@ -131,7 +131,7 @@ def mock_prepare(po_file_path, languages):
         translation_service._prepare_po_file = original_prepare
 
 
-def test_translate_bulk(translation_service, tmp_path):
+def test_translate_bulk(translation_service: MagicMock, tmp_path: MagicMock):
     """Test the bulk translation functionality."""
     texts_to_translate = ["HR", "TENANT", "HEALTHCARE", "TRANSPORT", "SERVICES"]
     po_file_path = str(tmp_path / "django.po")
@@ -145,7 +145,7 @@ def test_translate_bulk(translation_service, tmp_path):
     assert translated_texts == ["HR", "Inquilino", "Salud", "Transporte", "Servicios"]
 
 
-def test_translate_single(translation_service):
+def test_translate_single(translation_service: MagicMock):
     """Test the single translation functionality."""
     text_to_translate = "HEALTHCARE"