From 7255015b163dac5a9931d86adc44d69f94fbdcb9 Mon Sep 17 00:00:00 2001
From: AndresJRamirez <prof.andresjimenez@gmail.com>
Date: Tue, 27 Oct 2020 16:43:49 +0100
Subject: [PATCH] Update: locustfile to Locust v1.3.1

Adapt the classes and attributes to work with Locust v1.3.1
---
 loadtest/locustfile.py | 36 ++++++++++++++++++++----------------
 1 file changed, 20 insertions(+), 16 deletions(-)

diff --git a/loadtest/locustfile.py b/loadtest/locustfile.py
index 1e7ceb119f..85cae8ee99 100644
--- a/loadtest/locustfile.py
+++ b/loadtest/locustfile.py
@@ -3,11 +3,11 @@
 from random import choice
 
 from locust import (
-    HttpLocust,
-    TaskSequence,
+    HttpUser,
+    SequentialTaskSet,
     TaskSet,
-    seq_task,
     task,
+    between
 )
 
 
@@ -22,17 +22,14 @@ def index(self):
         self.client.get("/visualizer/{0}/".format(VOTING))
 
 
-class DefVoters(TaskSequence):
+class DefVoters(SequentialTaskSet):
 
     def on_start(self):
         with open('voters.json') as f:
             self.voters = json.loads(f.read())
         self.voter = choice(list(self.voters.items()))
 
-    def on_quit(self):
-        self.voter = None
-
-    @seq_task(1)
+    @task
     def login(self):
         username, pwd = self.voter
         self.token = self.client.post("/authentication/login/", {
@@ -40,11 +37,12 @@ def login(self):
             "password": pwd,
         }).json()
 
-    @seq_task(2)
+    @task
     def getuser(self):
-        self.user = self.client.post("/authentication/getuser/", self.token).json()
+        self.usr= self.client.post("/authentication/getuser/", self.token).json()
+        print( str(self.user))
 
-    @seq_task(3)
+    @task
     def voting(self):
         headers = {
             'Authorization': 'Token ' + self.token.get('token'),
@@ -56,16 +54,22 @@ def voting(self):
                 "a": "12",
                 "b": "64"
             },
-            "voter": self.user.get('id'),
+            "voter": self.usr.get('id'),
             "voting": VOTING
         }), headers=headers)
 
 
-class Visualizer(HttpLocust):
+    def on_quit(self):
+        self.voter = None
+
+class Visualizer(HttpUser):
     host = HOST
-    task_set = DefVisualizer
+    tasks = [DefVisualizer]
+    wait_time = between(3,5)
+
 
 
-class Voters(HttpLocust):
+class Voters(HttpUser):
     host = HOST
-    task_set = DefVoters
+    tasks = [DefVoters]
+    wait_time= between(3,5)