diff --git a/locust/runners.py b/locust/runners.py index 0c38fc3825..50e2f18844 100644 --- a/locust/runners.py +++ b/locust/runners.py @@ -71,12 +71,20 @@ def __init__(self, environment): self.target_user_count = None # set up event listeners for recording requests - def on_request(request_type, name, response_time, response_length, exception, context, **kwargs): + def on_request_success(request_type, name, response_time, response_length, **_kwargs): self.stats.log_request(request_type, name, response_time, response_length) - if exception: - self.stats.log_error(request_type, name, exception) - self.environment.events.request.add_listener(on_request) + def on_request_failure(request_type, name, response_time, response_length, exception, **_kwargs): + self.stats.log_request(request_type, name, response_time, response_length) + self.stats.log_error(request_type, name, exception) + + # temporarily set log level to ignore warnings to suppress deprication message + loglevel = logging.getLogger().level + logging.getLogger().setLevel(logging.ERROR) + self.environment.events.request_success.add_listener(on_request_success) + self.environment.events.request_failure.add_listener(on_request_failure) + logging.getLogger().setLevel(loglevel) + self.connection_broken = False # register listener that resets stats when spawning is complete