File tree 1 file changed +6
-0
lines changed
1 file changed +6
-0
lines changed Original file line number Diff line number Diff line change @@ -253,6 +253,8 @@ def swarm() -> Response:
253
253
254
254
parsed_options_dict = vars (environment .parsed_options ) if environment .parsed_options else {}
255
255
run_time = None
256
+ user_count = None
257
+ spawn_rate = None
256
258
for key , value in request .form .items ():
257
259
if key == "user_count" : # if we just renamed this field to "users" we wouldn't need this
258
260
user_count = int (value )
@@ -303,6 +305,10 @@ def swarm() -> Response:
303
305
self ._swarm_greenlet = None
304
306
305
307
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 })
306
312
self ._swarm_greenlet = gevent .spawn (environment .runner .start , user_count , spawn_rate )
307
313
self ._swarm_greenlet .link_exception (greenlet_exception_handler )
308
314
response_data = {
You can’t perform that action at this time.
0 commit comments