Merge pull request #1671 from stanislawskwark/patch-1

/swarm adjusted for tests with shape class
locustio · Jan 4, 2021 · 01087c7 · 01087c7
2 parents 11de532 + ceffa15
commit 01087c7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/locust/web.py b/locust/web.py
@@ -134,8 +134,6 @@ def index():
         @self.auth_required_if_enabled
         def swarm():
             assert request.method == "POST"
-            user_count = int(request.form["user_count"])
-            spawn_rate = float(request.form["spawn_rate"])
 
             if request.form.get("host"):
                 # Replace < > to guard against XSS
@@ -146,6 +144,8 @@ def swarm():
                 return jsonify(
                     {"success": True, "message": "Swarming started using shape class", "host": environment.host}
                 )
+            user_count = int(request.form["user_count"])
+            spawn_rate = float(request.form["spawn_rate"])
 
             environment.runner.start(user_count, spawn_rate)
             return jsonify({"success": True, "message": "Swarming started", "host": environment.host})