Add tests for image-captcha

tests/test_ImageToText.py

@@ -0,0 +1,180 @@
+import inspect
+
+import pytest
+
+from python3_anticaptcha import ImageToTextTask
+
+from main import MainAntiCaptcha
+
+
+class TestAntiCaptcha(MainAntiCaptcha):
+    WRONG_SAVE_FORMAT = "qwerty"
+    """
+    Params check
+    """
+
+    def test_customcatpcha_params(self):
+        default_init_params = [
+            "self",
+            "anticaptcha_key",
+            "sleep_time",
+            "save_format",
+            "language",
+            "callbackUrl",
+        ]
+        default_handler_params = [
+            "self",
+            "captcha_link",
+            "captcha_file",
+            "captcha_base64",
+        ]
+        # get customcaptcha init and captcha_handler params
+        aioinit_params = inspect.getfullargspec(
+            ImageToTextTask.aioImageToTextTask.__init__
+        )
+        aiohandler_params = inspect.getfullargspec(
+            ImageToTextTask.aioImageToTextTask.captcha_handler
+        )
+
+        # get customcaptcha init and captcha_handler params
+        init_params = inspect.getfullargspec(ImageToTextTask.ImageToTextTask.__init__)
+        handler_params = inspect.getfullargspec(
+            ImageToTextTask.ImageToTextTask.captcha_handler
+        )
+        # check aio module params
+        assert default_init_params == aioinit_params[0]
+        assert default_handler_params == aiohandler_params[0]
+        # check sync module params
+        assert default_init_params == init_params[0]
+        assert default_handler_params == handler_params[0]
+
+    """
+    Response checking
+    """
+
+    def test_response_imagecaptcha(self):
+        imagecaptcha = ImageToTextTask.ImageToTextTask(
+            anticaptcha_key=self.anticaptcha_key_fail
+        )
+        # check response type
+        assert isinstance(imagecaptcha, ImageToTextTask.ImageToTextTask)
+
+        response = imagecaptcha.captcha_handler(captcha_link=self.image_url)
+        # check response type
+        assert isinstance(response, dict)
+        # check all dict keys
+        assert ["errorId", "errorCode", "errorDescription"] == list(response.keys())
+
+    @pytest.mark.asyncio
+    async def test_response_aioimagecaptcha(self):
+        imagecaptcha = ImageToTextTask.aioImageToTextTask(
+            anticaptcha_key=self.anticaptcha_key_fail
+        )
+        # check response type
+        assert isinstance(imagecaptcha, ImageToTextTask.ImageToTextTask)
+
+        response = await imagecaptcha.captcha_handler(captcha_link=self.image_url)
+
+        # check response type
+        assert isinstance(response, dict)
+        # check all dict keys
+        assert ["errorId", "errorCode", "errorDescription"] == list(response.keys())
+
+    """
+    Fail tests
+    """
+
+    def test_fail_imagecaptcha_value(self):
+        with pytest.raises(ValueError):
+            assert ImageToTextTask.ImageToTextTask(
+                anticaptcha_key=self.anticaptcha_key_fail,
+                save_format=self.WRONG_SAVE_FORMAT,
+            )
+
+    def test_fail_imagecaptcha_const(self):
+        imagecaptcha = ImageToTextTask.ImageToTextTask(
+            anticaptcha_key=self.anticaptcha_key_fail,
+            save_format=ImageToTextTask.SAVE_FORMATS[0],
+        )
+
+        response = imagecaptcha.captcha_handler(captcha_link=self.image_url)
+
+        assert 1 == response["errorId"]
+
+    def test_fail_imagecaptcha_const_context(self):
+        with ImageToTextTask.ImageToTextTask(
+            anticaptcha_key=self.anticaptcha_key_fail,
+            save_format=ImageToTextTask.SAVE_FORMATS[0],
+        ) as imagecaptcha:
+
+            response = imagecaptcha.captcha_handler(captcha_link=self.image_url)
+
+            assert 1 == response["errorId"]
+
+    def test_fail_imagecaptcha_temp(self):
+        imagecaptcha = ImageToTextTask.ImageToTextTask(
+            anticaptcha_key=self.anticaptcha_key_fail,
+            save_format=ImageToTextTask.SAVE_FORMATS[1],
+        )
+
+        response = imagecaptcha.captcha_handler(captcha_link=self.image_url)
+
+        assert 1 == response["errorId"]
+
+    def test_fail_imagecaptcha_temp_context(self):
+        with ImageToTextTask.ImageToTextTask(
+            anticaptcha_key=self.anticaptcha_key_fail,
+            save_format=ImageToTextTask.SAVE_FORMATS[1],
+        ) as imagecaptcha:
+
+            response = imagecaptcha.captcha_handler(captcha_link=self.image_url)
+
+            assert 1 == response["errorId"]
+
+    @pytest.mark.asyncio
+    async def test_fail_aioimagecaptcha_value(self):
+        with pytest.raises(ValueError):
+            assert ImageToTextTask.ImageToTextTask(
+                anticaptcha_key=self.anticaptcha_key_fail,
+                save_format=self.WRONG_SAVE_FORMAT,
+            )
+
+    @pytest.mark.asyncio
+    async def test_fail_aioimagecaptcha_temp(self):
+        imagecaptcha = ImageToTextTask.aioImageToTextTask(
+            anticaptcha_key=self.anticaptcha_key_fail,
+            save_format=ImageToTextTask.SAVE_FORMATS[1],
+        )
+        response = await imagecaptcha.captcha_handler(captcha_link=self.image_url)
+        assert 1 == response["errorId"]
+
+    @pytest.mark.asyncio
+    async def test_fail_aioimagecaptcha_temp_context(self):
+        with ImageToTextTask.aioImageToTextTask(
+            anticaptcha_key=self.anticaptcha_key_fail,
+            save_format=ImageToTextTask.SAVE_FORMATS[1],
+        ) as imagecaptcha:
+            response = await imagecaptcha.captcha_handler(captcha_link=self.image_url)
+            assert 1 == response["errorId"]
+
+    @pytest.mark.asyncio
+    async def test_fail_aioimagecaptcha_const(self):
+        imagecaptcha = ImageToTextTask.aioImageToTextTask(
+            anticaptcha_key=self.anticaptcha_key_fail,
+            save_format=ImageToTextTask.SAVE_FORMATS[0],
+        )
+        response = await imagecaptcha.captcha_handler(captcha_link=self.image_url)
+        assert 1 == response["errorId"]
+
+    @pytest.mark.asyncio
+    async def test_fail_aioimagecaptcha_const_context(self):
+        with ImageToTextTask.aioImageToTextTask(
+            anticaptcha_key=self.anticaptcha_key_fail,
+            save_format=ImageToTextTask.SAVE_FORMATS[0],
+        ) as imagecaptcha:
+            response = await imagecaptcha.captcha_handler(captcha_link=self.image_url)
+            assert 1 == response["errorId"]
+
+    """
+    True tests
+    """