tests: upgrade code to support locust 1.0+

* More at https://docs.locust.io/en/latest/changelog.html#changelog-1-0
reanahub · Nov 24, 2020 · d4a1201 · d4a1201
1 parent c20c85c
commit d4a1201
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/benchmark/locustfile.py b/tests/benchmark/locustfile.py
@@ -24,7 +24,7 @@
 
 import os
 
-from locust import HttpLocust, TaskSet, between, task
+from locust import HttpUser, TaskSet, between, task
 
 TOKEN = os.environ.get("REANA_ACCESS_TOKEN")
 
@@ -80,8 +80,8 @@ def get_worflow_logs(self):
         )
 
 
-class WebsiteUser(HttpLocust):
+class WebsiteUser(HttpUser):
     """Locust instance. Represent the user that attacks the system."""
 
-    task_set = WorkflowsTaskSet
+    tasks = [WorkflowsTaskSet]
     wait_time = between(5, 10)