mirror of
https://github.com/librecaptcha/lc-core.git
synced 2025-01-13 15:03:20 -05:00
67 lines
2.5 KiB
Python
67 lines
2.5 KiB
Python
from locust import task, between, SequentialTaskSet
|
|
from locust.contrib.fasthttp import FastHttpUser
|
|
from locust import events
|
|
import json
|
|
import logging
|
|
import subprocess
|
|
|
|
@events.quitting.add_listener
|
|
def _(environment, **kw):
|
|
if environment.stats.total.fail_ratio > 0.02:
|
|
logging.error("Test failed due to failure ratio > 2%")
|
|
environment.process_exit_code = 1
|
|
elif environment.stats.total.avg_response_time > 300:
|
|
logging.error("Test failed due to average response time ratio > 300 ms")
|
|
environment.process_exit_code = 1
|
|
elif environment.stats.total.get_response_time_percentile(0.95) > 800:
|
|
logging.error("Test failed due to 95th percentile response time > 800 ms")
|
|
environment.process_exit_code = 1
|
|
else:
|
|
environment.process_exit_code = 0
|
|
|
|
class QuickStartUser(SequentialTaskSet):
|
|
wait_time = between(0.1,0.2)
|
|
|
|
@task
|
|
def captcha(self):
|
|
captcha_params = {"level":"debug","media":"image/png","input_type":"text"}
|
|
|
|
with self.client.post(path="/v1/captcha", json=captcha_params, name="/captcha", catch_response = True) as resp:
|
|
if resp.status_code != 200:
|
|
resp.failure("Status was not 200: " + resp.text)
|
|
captchaJson = resp.json()
|
|
uuid = captchaJson.get("id")
|
|
if not uuid:
|
|
resp.failure("uuid not returned on /captcha endpoint: " + resp.text)
|
|
|
|
with self.client.get(path="/v1/media?id=%s" % uuid, name="/media", stream=True, catch_response = True) as resp:
|
|
if resp.status_code != 200:
|
|
resp.failure("Status was not 200: " + resp.text)
|
|
|
|
media = resp.content
|
|
|
|
ocrAnswer = self.solve(uuid, media)
|
|
|
|
answerBody = {"answer": ocrAnswer,"id": uuid}
|
|
with self.client.post(path='/v1/answer', json=answerBody, name="/answer", catch_response=True) as resp:
|
|
if resp.status_code != 200:
|
|
resp.failure("Status was not 200: " + resp.text)
|
|
else:
|
|
if resp.json().get("result") != "True":
|
|
resp.failure("Answer was not accepted")
|
|
|
|
def solve(self, uuid, media):
|
|
mediaFileName = "tests/test-%s.png" % uuid
|
|
with open(mediaFileName, "wb") as f:
|
|
f.write(media)
|
|
ocrResult = subprocess.Popen("gocr %s" % mediaFileName, shell=True, stdout=subprocess.PIPE)
|
|
ocrAnswer = ocrResult.stdout.readlines()[0].strip().decode()
|
|
return ocrAnswer
|
|
|
|
|
|
|
|
class User(FastHttpUser):
|
|
wait_time = between(0.1,0.2)
|
|
tasks = [QuickStartUser]
|
|
host = "http://localhost:8888"
|