Rooba
/
python3-anticaptcha
mirror of https://github.com/AndreiDrang/python3-anticaptcha.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Move tests to separate folder

This commit is contained in:
Andrei 2019-09-06 00:07:36 +03:00
parent 9a0337342d
commit 4c91f6d081
4 changed files with 345 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
tests/main.py Normal file
View File

@ -0,0 +1,11 @@
import os
class MainAntiCaptcha(object):
WRONG_QUEUE_ID = WRONG_TASK_ID = -1
def setup_class(self):
self.anticaptcha_key_fail = os.getenv("anticaptcha_key")[:5]
self.anticaptcha_key_true = os.getenv("anticaptcha_key")
self.server_ip = "85.255.8.26"
self.image_url = "https://raw.githubusercontent.com/AndreiDrang/python3-anticaptcha/master/anticaptcha_examples/088636.png"

16
tests/test_Callback.py Normal file
View File

@ -0,0 +1,16 @@
import requests
from main import MainAntiCaptcha
class TestAntiCaptcha(MainAntiCaptcha):
def test_callback_server(self):
# test server alive
response = requests.get(f"http://{self.server_ip}:8001/ping")
assert response.status_code == 200
# try register new queue
response = requests.post(
f"http://{self.server_ip}:8001/register_key",
json={"key": "fwefefefopewofkewopfkop", "vhost": "anticaptcha_vhost"},
)
assert response.status_code == 200

154
tests/test_CustomCaptcha.py Normal file
View File

@ -0,0 +1,154 @@
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"]

164
tests/test_CustomResultHandler.py Normal file
View File

@ -0,0 +1,164 @@
import inspect
import asyncio
import pytest
from python3_anticaptcha import CustomResultHandler
from main import MainAntiCaptcha
class TestAntiCaptcha(MainAntiCaptcha):
"""
Params check
"""
def test_result_handler_params(self):
default_init_params = ["self", "anticaptcha_key", "sleep_time"]
default_handler_params = ["self", "task_id"]
# get customcaptcha init and task_handler params
aioinit_params = inspect.getfullargspec(
CustomResultHandler.aioCustomResultHandler.__init__
)
aiohandler_params = inspect.getfullargspec(
CustomResultHandler.aioCustomResultHandler.task_handler
)
# get customcaptcha init and task_handler params
init_params = inspect.getfullargspec(
CustomResultHandler.CustomResultHandler.__init__
)
handler_params = inspect.getfullargspec(
CustomResultHandler.CustomResultHandler.task_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_result_handler(self):
# prepare client
custom_result = CustomResultHandler.CustomResultHandler(
anticaptcha_key=self.anticaptcha_key_true
)
# get balance
response = custom_result.task_handler(task_id=self.WRONG_TASK_ID)
# check response type
assert type(response) is dict
# check all dict keys
assert ["errorId", "errorCode", "errorDescription", "taskId"] == list(
response.keys()
)
@pytest.mark.asyncio
async def test_response_aioresult_handler(self):
# prepare client
custom_result = CustomResultHandler.aioCustomResultHandler(
anticaptcha_key=self.anticaptcha_key_true
)
# get balance
response = await custom_result.task_handler(task_id=self.WRONG_TASK_ID)
# check response type
assert type(response) is dict
# check all dict keys
assert ["errorId", "errorCode", "errorDescription", "taskId"] == list(
response.keys()
)
"""
Fail tests
"""
def test_fail_result_handler(self):
# prepare client
custom_result = CustomResultHandler.CustomResultHandler(
anticaptcha_key=self.anticaptcha_key_fail
)
# get balance
response = custom_result.task_handler(task_id=self.WRONG_TASK_ID)
# check error code
assert 1 == response["errorId"]
def test_fail_result_handler_context(self):
# prepare client
with CustomResultHandler.CustomResultHandler(
anticaptcha_key=self.anticaptcha_key_fail
) as custom_result:
# get balance
response = custom_result.task_handler(task_id=self.WRONG_TASK_ID)
# check error code
assert 1 == response["errorId"]
@pytest.mark.asyncio
async def test_fail_aioresult_handler(self):
# prepare client
custom_result = CustomResultHandler.aioCustomResultHandler(
anticaptcha_key=self.anticaptcha_key_fail
)
# get balance
response = await custom_result.task_handler(task_id=self.WRONG_TASK_ID)
# check error code
assert 1 == response["errorId"]
@pytest.mark.asyncio
async def test_fail_aioresult_handler_context(self):
# prepare client
with CustomResultHandler.aioCustomResultHandler(
anticaptcha_key=self.anticaptcha_key_fail
) as custom_result:
# get balance
response = await custom_result.task_handler(task_id=self.WRONG_TASK_ID)
# check error code
assert 1 == response["errorId"]
"""
True tests
"""
def test_true_result_handler(self):
# prepare client
custom_result = CustomResultHandler.CustomResultHandler(
anticaptcha_key=self.anticaptcha_key_true
)
# get balance
response = custom_result.task_handler(task_id=self.WRONG_TASK_ID)
# check error code
assert 16 == response["errorId"]
def test_true_result_handler_context(self):
# prepare client
with CustomResultHandler.CustomResultHandler(
anticaptcha_key=self.anticaptcha_key_true
) as custom_result:
# get balance
response = custom_result.task_handler(task_id=self.WRONG_TASK_ID)
# check error code
assert 16 == response["errorId"]
@pytest.mark.asyncio
async def test_true_aioresult_handler(self):
# prepare client
custom_result = CustomResultHandler.aioCustomResultHandler(
anticaptcha_key=self.anticaptcha_key_true
)
# get balance
response = await custom_result.task_handler(task_id=self.WRONG_TASK_ID)
# check error code
assert 16 == response["errorId"]
@pytest.mark.asyncio
async def test_true_aioresult_handler_context(self):
# prepare client
with CustomResultHandler.aioCustomResultHandler(
anticaptcha_key=self.anticaptcha_key_true
) as custom_result:
# get balance
response = await custom_result.task_handler(task_id=self.WRONG_TASK_ID)
# check error code
assert 16 == response["errorId"]