python3-anticaptcha/tests/test_ImageToText.py

import inspect

import pytest

from python3_anticaptcha import ImageToTextTask

from tests.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 await 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
    """