Skip to content

Commit 68780b5

Browse files
committed
Improve error message on missing user_count or spawn_rate in /swarm payload. Adresses #3051
1 parent fab2227 commit 68780b5

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

locust/web.py

+6
Original file line numberDiff line numberDiff line change
@@ -253,6 +253,8 @@ def swarm() -> Response:
253253

254254
parsed_options_dict = vars(environment.parsed_options) if environment.parsed_options else {}
255255
run_time = None
256+
user_count = None
257+
spawn_rate = None
256258
for key, value in request.form.items():
257259
if key == "user_count": # if we just renamed this field to "users" we wouldn't need this
258260
user_count = int(value)
@@ -303,6 +305,10 @@ def swarm() -> Response:
303305
self._swarm_greenlet = None
304306

305307
if environment.runner is not None:
308+
if user_count is None or spawn_rate is None:
309+
err_msg = "Missing user_count or spawn_rate from /swarm request"
310+
logger.error(err_msg)
311+
return jsonify({"success": False, "message": err_msg, "host": environment.host})
306312
self._swarm_greenlet = gevent.spawn(environment.runner.start, user_count, spawn_rate)
307313
self._swarm_greenlet.link_exception(greenlet_exception_handler)
308314
response_data = {

0 commit comments

Comments
 (0)