Update test_mt_captcha.py

Author: Andrei

tests/test_mt_captcha.py

@@ -1,7 +1,7 @@
 import pytest
 
 from tests.conftest import BaseTest
-from python3_capsolver.core.enum import ResponseStatusEnm
+from python3_capsolver.core.enum import MtCaptchaTypeEnm, ResponseStatusEnm
 from python3_capsolver.mt_captcha import MtCaptcha
 from python3_capsolver.core.serializer import CaptchaResponseSer
 
@@ -11,56 +11,65 @@
 
 
 class TestMtCaptchaBase(BaseTest):
+    captcha_types = (MtCaptchaTypeEnm.MtCaptchaTask, MtCaptchaTypeEnm.MtCaptchaTaskProxyLess)
+
     def test_captcha_handler_exist(self):
         assert "captcha_handler" in MtCaptcha.__dict__.keys()
 
     def test_aio_captcha_handler_exist(self):
         assert "aio_captcha_handler" in MtCaptcha.__dict__.keys()
 
-    def test_no_website_url(self):
-        with pytest.raises(TypeError):
-            MtCaptcha(api_key=self.API_KEY, websiteKey=websiteKey, proxy=proxy)
-
-    def test_no_website_key(self):
+    @pytest.mark.parametrize("captcha_type", captcha_types)
+    def test_no_website_url(self, captcha_type: str):
         with pytest.raises(TypeError):
-            MtCaptcha(api_key=self.API_KEY, websiteURL=websiteURL, proxy=proxy)
+            MtCaptcha(api_key=self.API_KEY, captcha_type=captcha_type, websiteKey=websiteKey, proxy=proxy)
 
-    def test_no_proxy(self):
+    @pytest.mark.parametrize("captcha_type", captcha_types)
+    def test_no_website_key(self, captcha_type: str):
         with pytest.raises(TypeError):
-            MtCaptcha(api_key=self.API_KEY, websiteURL=websiteURL, websiteKey=websiteKey)
+            MtCaptcha(api_key=self.API_KEY, captcha_type=captcha_type, websiteURL=websiteURL, proxy=proxy)
 
 
 class TestMtCaptcha(BaseTest):
+    captcha_types = (MtCaptchaTypeEnm.MtCaptchaTask, MtCaptchaTypeEnm.MtCaptchaTaskProxyLess)
     """
     Success tests
     """
 
     @pytest.mark.parametrize("proxy_type", BaseTest.proxyTypes)
-    def test_params(self, proxy_type: str):
-        MtCaptcha(api_key=self.API_KEY, websiteURL=websiteURL, websiteKey=websiteKey, proxy=proxy)
+    @pytest.mark.parametrize("captcha_type", captcha_types)
+    def test_params(self, proxy_type: str, captcha_type: str):
+        MtCaptcha(
+            api_key=self.API_KEY, captcha_type=captcha_type, websiteURL=websiteURL, websiteKey=websiteKey, proxy=proxy
+        )
 
     @pytest.mark.parametrize("proxy_type", BaseTest.proxyTypes)
-    def test_params_context(self, proxy_type: str):
-        with MtCaptcha(api_key=self.API_KEY, websiteURL=websiteURL, websiteKey=websiteKey, proxy=proxy) as instance:
+    @pytest.mark.parametrize("captcha_type", captcha_types)
+    def test_params_context(self, proxy_type: str, captcha_type: str):
+        with MtCaptcha(
+            api_key=self.API_KEY, captcha_type=captcha_type, websiteURL=websiteURL, websiteKey=websiteKey, proxy=proxy
+        ) as instance:
             pass
 
     """
     Failed tests
     """
 
-    def test_proxy_err(self):
+    @pytest.mark.parametrize("captcha_type", captcha_types)
+    def test_proxy_err(self, captcha_type: str):
         resp = MtCaptcha(
-            api_key=self.API_KEY, websiteURL=websiteURL, websiteKey=websiteKey, proxy=proxy
+            api_key=self.API_KEY, captcha_type=captcha_type, websiteURL=websiteURL, websiteKey=websiteKey, proxy=proxy
         ).captcha_handler()
         assert isinstance(resp, CaptchaResponseSer)
         assert resp.status == ResponseStatusEnm.Processing
         assert resp.errorId == 1
         assert resp.errorCode == "ERROR_INVALID_TASK_DATA"
         assert resp.solution is None
 
-    async def test_aio_proxy_err(self):
+    @pytest.mark.parametrize("captcha_type", captcha_types)
+    async def test_aio_proxy_err(self, captcha_type: str):
         resp = await MtCaptcha(
-            api_key=self.API_KEY, websiteURL=websiteURL, websiteKey=websiteKey, proxy=proxy
+            api_key=self.API_KEY, captcha_type=captcha_type, websiteURL=websiteURL, websiteKey=websiteKey, proxy=proxy
         ).aio_captcha_handler()
         assert isinstance(resp, CaptchaResponseSer)
         assert resp.status == ResponseStatusEnm.Processing