154 lines
5.3 KiB
Python
154 lines
5.3 KiB
Python
import inspect
|
|
|
|
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 isinstance(customcaptcha, CustomCaptchaTask.CustomCaptchaTask)
|
|
|
|
response = customcaptcha.captcha_handler(imageUrl=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_aiocustomcaptcha(self):
|
|
customcaptcha = CustomCaptchaTask.aioCustomCaptchaTask(
|
|
anticaptcha_key=self.anticaptcha_key_fail, assignment=self.CUSTOM_TASK
|
|
)
|
|
# check response type
|
|
assert isinstance(customcaptcha, CustomCaptchaTask.CustomCaptchaTask)
|
|
|
|
response = await customcaptcha.captcha_handler(imageUrl=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_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"]
|