more UT

Author: jordimas

open_dubbing/translation.py

@@ -18,6 +18,7 @@
 
 from abc import ABC, abstractmethod
 from typing import Final, Mapping, Sequence
+from unittest.mock import MagicMock
 
 from open_dubbing import logger
 
@@ -153,3 +154,35 @@ def _add_translations(
         logger().debug(f"translation.translated_script. Input: {translated_script}")
         logger().debug(f"translation.translated_script. Returns: {pretty_data}")
         return updated_utterance_metadata
+
+    def test_translate_utterances():
+        mock_instance = MagicMock()
+        utterance_metadata = [
+            {"text": "Hello", "timestamp": 0.0},
+            {"text": "How are you?", "timestamp": 1.0},
+        ]
+
+        mock_instance._generate_script.return_value = "Hello\nHow are you?"
+        mock_instance._translate_script.return_value = "Hola\nCom estàs?"
+        mock_instance._add_translations.return_value = [
+            {"text": "Hello", "translation": "Hola", "timestamp": 0.0},
+            {"text": "How are you?", "translation": "Com estàs?", "timestamp": 1.0},
+        ]
+
+        result = mock_instance.translate_utterances(
+            utterance_metadata=utterance_metadata,
+            source_language="en",
+            target_language="ca",
+        )
+
+        # Assert
+        mock_instance._generate_script.assert_called_once_with(
+            utterance_metadata=utterance_metadata
+        )
+        mock_instance._translate_script.assert_called_once()
+        mock_instance._add_translations.assert_called_once()
+
+        assert result == [
+            {"text": "Hello", "translation": "Hola", "timestamp": 0.0},
+            {"text": "How are you?", "translation": "Com estàs?", "timestamp": 1.0},
+        ]