Tiny documentation improvement

locustio · Feb 3, 2022 · f28aed1 · f28aed1
1 parent 024581b
commit f28aed1
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/docs/extending-locust.rst b/docs/extending-locust.rst
@@ -133,7 +133,7 @@ For example, you can monitor the fail ratio of your test and stop the run if it
 .. code-block:: python
 
     from locust import events
-    from locust.runners import STATE_STOPPING, STATE_STOPPED, STATE_CLEANUP, WorkerRunner
+    from locust.runners import STATE_STOPPING, STATE_STOPPED, STATE_CLEANUP, MasterRunner, LocalRunner
 
     def checker(environment):
         while not environment.runner.state in [STATE_STOPPING, STATE_STOPPED, STATE_CLEANUP]:
@@ -146,8 +146,8 @@ For example, you can monitor the fail ratio of your test and stop the run if it
 
     @events.init.add_listener
     def on_locust_init(environment, **_kwargs):
-        # only run this on master & standalone
-        if not isinstance(environment.runner, WorkerRunner):
+        # dont run this on workers, we only care about the aggregated numbers
+        if isinstance(environment.runner, MasterRunner) or isinstance(environment.runner, LocalRunner):
             gevent.spawn(checker, environment)
 
 .. _parametrizing-locustfiles: