python3-anticaptcha/tests/test_CustomCaptcha.py

import inspect
import asyncio

import pytest

from python3_anticaptcha import CustomCaptchaTask

from main import MainAntiCaptcha


class TestAntiCaptcha(MainAntiCaptcha):
    CUSTOM_TASK = "2+2=?"
    """
    Params check
    """

    def test_customcatpcha_params(self):
        default_init_params = [
            "self",
            "anticaptcha_key",
            "sleep_time",
            "assignment",
            "forms",
            "callbackUrl",
        ]
        default_handler_params = ["self", "imageUrl"]
        # get customcaptcha init and captcha_handler params
        aioinit_params = inspect.getfullargspec(
            CustomCaptchaTask.aioCustomCaptchaTask.__init__
        )
        aiohandler_params = inspect.getfullargspec(
            CustomCaptchaTask.aioCustomCaptchaTask.captcha_handler
        )

        # get customcaptcha init and captcha_handler params
        init_params = inspect.getfullargspec(
            CustomCaptchaTask.CustomCaptchaTask.__init__
        )
        handler_params = inspect.getfullargspec(
            CustomCaptchaTask.CustomCaptchaTask.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_customcaptcha(self):
        customcaptcha = CustomCaptchaTask.CustomCaptchaTask(
            anticaptcha_key=self.anticaptcha_key_fail, assignment=self.CUSTOM_TASK
        )
        # check response type
        assert type(customcaptcha) is CustomCaptchaTask.CustomCaptchaTask

        response = customcaptcha.captcha_handler(imageUrl=self.image_url)
        # check response type
        assert type(response) is dict
        # check all dict keys
        assert ["errorId", "errorCode", "errorDescription"] == list(response.keys())

    @pytest.mark.asyncio
    async def test_response_aiocustomcaptcha(self):
        customcaptcha = CustomCaptchaTask.aioCustomCaptchaTask(
            anticaptcha_key=self.anticaptcha_key_fail, assignment=self.CUSTOM_TASK
        )
        # check response type
        assert type(customcaptcha) is CustomCaptchaTask.CustomCaptchaTask

        response = await customcaptcha.captcha_handler(imageUrl=self.image_url)
        # check response type
        assert type(response) is dict
        # check all dict keys
        assert ["errorId", "errorCode", "errorDescription"] == list(response.keys())

    """
    Fail tests
    """

    def test_fail_customcaptcha(self):
        customcaptcha = CustomCaptchaTask.CustomCaptchaTask(
            anticaptcha_key=self.anticaptcha_key_fail, assignment=self.CUSTOM_TASK
        )

        response = customcaptcha.captcha_handler(imageUrl=self.image_url)

        assert 1 == response["errorId"]

    def test_fail_customcaptcha_context(self):
        with CustomCaptchaTask.CustomCaptchaTask(
            anticaptcha_key=self.anticaptcha_key_fail, assignment=self.CUSTOM_TASK
        ) as customcaptcha:

            response = customcaptcha.captcha_handler(imageUrl=self.image_url)

            assert 1 == response["errorId"]

    @pytest.mark.asyncio
    async def test_fail_aiocustomcaptcha(self):
        customcaptcha = CustomCaptchaTask.aioCustomCaptchaTask(
            anticaptcha_key=self.anticaptcha_key_fail, assignment=self.CUSTOM_TASK
        )
        response = await customcaptcha.captcha_handler(imageUrl=self.image_url)
        assert 1 == response["errorId"]

    @pytest.mark.asyncio
    async def test_fail_aiocustomcaptcha_context(self):
        with CustomCaptchaTask.aioCustomCaptchaTask(
            anticaptcha_key=self.anticaptcha_key_fail, assignment=self.CUSTOM_TASK
        ) as customcaptcha:
            response = await customcaptcha.captcha_handler(imageUrl=self.image_url)
            assert 1 == response["errorId"]

    """
    True tests
    """

    def test_true_customcaptcha(self):
        customcaptcha = CustomCaptchaTask.CustomCaptchaTask(
            anticaptcha_key=self.anticaptcha_key_true, assignment=self.CUSTOM_TASK
        )

        response = customcaptcha.captcha_handler(imageUrl=self.image_url)

        assert 0 == response["errorId"]

    def test_true_customcaptcha_context(self):
        with CustomCaptchaTask.CustomCaptchaTask(
            anticaptcha_key=self.anticaptcha_key_true, assignment=self.CUSTOM_TASK
        ) as customcaptcha:

            response = customcaptcha.captcha_handler(imageUrl=self.image_url)

            assert 0 == response["errorId"]

    @pytest.mark.asyncio
    async def test_true_aiocustomcaptcha(self):
        customcaptcha = CustomCaptchaTask.aioCustomCaptchaTask(
            anticaptcha_key=self.anticaptcha_key_true, assignment=self.CUSTOM_TASK
        )
        response = await customcaptcha.captcha_handler(imageUrl=self.image_url)
        assert 0 == response["errorId"]

    @pytest.mark.asyncio
    async def test_true_aiocustomcaptcha_context(self):
        with CustomCaptchaTask.aioCustomCaptchaTask(
            anticaptcha_key=self.anticaptcha_key_true, assignment=self.CUSTOM_TASK
        ) as customcaptcha:
            response = await customcaptcha.captcha_handler(imageUrl=self.image_url)
            assert 0 == response["errorId"]