upd tests

Author: Andrei

tests/test_datadome_slider.py

@@ -16,6 +16,16 @@ def test_captcha_handler_exist(self):
     def test_aio_captcha_handler_exist(self):
         assert "aio_captcha_handler" in DatadomeSlider.__dict__.keys()
 
+    def test_wrong_captcha_type(self):
+        with pytest.raises(ValueError):
+            DatadomeSlider(
+                api_key=self.API_KEY,
+                captcha_type=self.get_random_string(36),
+                websiteURL=websiteURL,
+                captchaUrl=captchaUrl,
+                userAgent=self.get_random_string(36),
+            )
+
 
 class TestDatadomeSlider(BaseTest):
     """

tests/test_image_to_text.py

@@ -1,5 +1,7 @@
 import base64
 
+import pytest
+
 from tests.conftest import BaseTest
 from python3_capsolver.core.enum import ResponseStatusEnm
 from python3_capsolver.image_to_text import ImageToText
@@ -9,18 +11,24 @@
     img_data = img_file.read()
 
 
-class TestImageToText(BaseTest):
-    image_body = base64.b64encode(img_data).decode("utf-8")
-    """
-    Success tests
-    """
-
+class TestImageToTextBase(BaseTest):
     def test_captcha_handler_exist(self):
         assert "captcha_handler" in ImageToText.__dict__.keys()
 
     def test_aio_captcha_handler_exist(self):
         assert "aio_captcha_handler" in ImageToText.__dict__.keys()
 
+    def test_wrong_captcha_type(self):
+        with pytest.raises(ValueError):
+            ImageToText(api_key=self.API_KEY, captcha_type=self.get_random_string())
+
+
+class TestImageToText(BaseTest):
+    image_body = base64.b64encode(img_data).decode("utf-8")
+    """
+    Success tests
+    """
+
     def test_solve_image(self):
         resp = ImageToText(api_key=self.API_KEY).captcha_handler(body=self.image_body)
         assert isinstance(resp, CaptchaResponseSer)

tests/test_mt_captcha.py

@@ -19,6 +19,16 @@ def test_captcha_handler_exist(self):
     def test_aio_captcha_handler_exist(self):
         assert "aio_captcha_handler" in MtCaptcha.__dict__.keys()
 
+    def test_wrong_captcha_type(self):
+        with pytest.raises(ValueError):
+            MtCaptcha(
+                api_key=self.API_KEY,
+                captcha_type=self.get_random_string(),
+                websiteKey=websiteKey,
+                proxy=proxy,
+                websiteURL=websiteURL,
+            )
+
     @pytest.mark.parametrize("captcha_type", captcha_types)
     def test_no_website_url(self, captcha_type: str):
         with pytest.raises(TypeError):