mirror of
https://github.com/librecaptcha/lc-core.git
synced 2025-01-12 22:43:20 -05:00
42 lines
1.5 KiB
Python
42 lines
1.5 KiB
Python
from locust import task, between, SequentialTaskSet
|
|
from locust.contrib.fasthttp import FastHttpUser
|
|
import json
|
|
|
|
class QuickStartUser(SequentialTaskSet):
|
|
wait_time = between(0.1,1)
|
|
|
|
@task
|
|
def captcha(self):
|
|
# TODO: Iterate over parameters for a more comprehensive test
|
|
captcha_params = {"level":"easy","media":"image/png","input_type":"text"}
|
|
|
|
resp = self.client.post(path="/v1/captcha", json=captcha_params, name="/captcha")
|
|
if resp.status_code != 200:
|
|
print("\nError on /captcha endpoint: ")
|
|
print(resp)
|
|
print(resp.text)
|
|
print("----------------END.CAPTCHA-------------------\n\n")
|
|
|
|
uuid = json.loads(resp.text).get("id")
|
|
answerBody = {"answer": "qwer123","id": uuid}
|
|
|
|
resp = self.client.get(path="/v1/media?id=%s" % uuid, name="/media")
|
|
if resp.status_code != 200:
|
|
print("\nError on /media endpoint: ")
|
|
print(resp)
|
|
print(resp.text)
|
|
print("----------------END.MEDIA-------------------\n\n")
|
|
|
|
resp = self.client.post(path='/v1/answer', json=answerBody, name="/answer")
|
|
if resp.status_code != 200:
|
|
print("\nError on /answer endpoint: ")
|
|
print(resp)
|
|
print(resp.text)
|
|
print("----------------END.ANSWER-------------------\n\n")
|
|
|
|
|
|
class User(FastHttpUser):
|
|
wait_time = between(0.1,1)
|
|
tasks = [QuickStartUser]
|
|
host = "http://localhost:8888"
|