diff --git a/tests/locustfile.py b/tests/locustfile.py index d3a72b0..54774e0 100644 --- a/tests/locustfile.py +++ b/tests/locustfile.py @@ -7,31 +7,32 @@ class QuickStartUser(SequentialTaskSet): @task def captcha(self): - captcha_params = {"level":"some","media":"some","input_type":"some"} + # 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.C-------------------\n\n") + 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 /captcha endpoint: ") + print("\nError on /media endpoint: ") print(resp) print(resp.text) - print("----------------END.C-------------------\n\n") + print("----------------END.MEDIA-------------------\n\n") resp = self.client.post(path='/v1/answer', json=answerBody, name="/answer") if resp.status_code != 200: - print("\nError on /captcha endpoint: ") + print("\nError on /answer endpoint: ") print(resp) print(resp.text) - print("----------------END.C-------------------\n\n") + print("----------------END.ANSWER-------------------\n\n") class User(FastHttpUser):