added payload check in LibreTranslator.translate and added tests in test_libre.py

deep_translator/libre.py

@@ -10,7 +10,8 @@ from .constants import BASE_URLS,LIBRE_LANGUAGES_TO_CODES, LIBRE_CODES_TO_LANGUA
 from .exceptions import (ServerException,
                         TranslationNotFound,
                         LanguageNotSupportedException,
-                        AuthorizationException)
+                        AuthorizationException,
+                        NotValidPayload)
 
 
 class LibreTranslator(BaseTranslator):
@@ -31,7 +32,10 @@ class LibreTranslator(BaseTranslator):
             raise ServerException(401)
         self.__base_url = base_url
         self.api_key = api_key
-        self.source = self._map_language_to_code(source)
+        if source == "auto":
+            self.source = "auto"
+        else:
+            self.source = self._map_language_to_code(source)
         self.target = self._map_language_to_code(target)
 
 
@@ -62,7 +66,10 @@ class LibreTranslator(BaseTranslator):
         @param language: a string for 1 language
         @return: bool or raise an Exception
         """
-        return language == 'auto' or language in self._languages.keys() or language in self._languages.values()
+        if language == 'auto' or language in self._languages.keys() or language in self._languages.values():
+            return True
+        else:
+            raise LanguageNotSupportedException(language)
     
     def translate(self, text, **kwargs):
         """
@@ -71,6 +78,9 @@ class LibreTranslator(BaseTranslator):
         @return: str: translated text
         """
          # Create the request parameters.
+        if type(text) != str or text == "":
+            raise NotValidPayload(text)
+
         translate_endpoint = 'translate'
         params = {
             "q": text,
@@ -95,4 +105,3 @@ class LibreTranslator(BaseTranslator):
             raise TranslationNotFound(text)
         # Process and return the response.
         return res['translatedText']
-    

tests/test_libre.py

@@ -0,0 +1,49 @@
+#!/usr/bin/env python
+
+"""Tests for `deep_translator` package."""
+
+import pytest
+from deep_translator import exceptions, LibreTranslator
+
+
+@pytest.fixture
+def libre():
+    return LibreTranslator(source="en", target='fr')
+
+
+def test_content(libre):
+    """Sample pytest test function with the pytest fixture as an argument."""
+    # from bs4 import BeautifulSoup
+    # assert 'GitHub' in BeautifulSoup(response.content).title.string
+    assert libre.translate(text='good') is not None
+
+
+def test_inputs():
+    with pytest.raises(exceptions.LanguageNotSupportedException):
+        LibreTranslator(source="", target="")
+
+    with pytest.raises(exceptions.LanguageNotSupportedException):
+        LibreTranslator(source="auto", target="nothing")
+
+    l1 = LibreTranslator("en", "fr")
+    l2 = LibreTranslator("english", "french")
+    assert l1.source == l2.source
+    assert l1.target == l2.target
+
+
+def test_payload(libre):
+    with pytest.raises(exceptions.NotValidPayload):
+        libre.translate("")
+
+    with pytest.raises(exceptions.NotValidPayload):
+        libre.translate(123)
+
+    with pytest.raises(exceptions.NotValidPayload):
+        libre.translate({})
+
+    with pytest.raises(exceptions.NotValidPayload):
+        libre.translate([])
+
+
+def test_one_character_words():
+    assert LibreTranslator(source='es', target='en').translate('y') == 'and'