From ffffdb849be863d5a7b1bed8f14bd63ff29af7e2 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Dennis=20Krone=CC=81?= Date: Mon, 17 May 2021 14:31:08 +0200 Subject: [PATCH 1/2] Register stats from request_success and request_failure Old update was causing breaking issues. --- locust/runners.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/locust/runners.py b/locust/runners.py index 0c38fc3825..e2b09d8b5a 100644 --- a/locust/runners.py +++ b/locust/runners.py @@ -71,12 +71,15 @@ 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) + + self.environment.events.request_success.add_listener(on_request_success) + self.environment.events.request_failure.add_listener(on_request_failure) self.connection_broken = False # register listener that resets stats when spawning is complete From 9856e5e7d464d478a73fede616c27f6f0f746b97 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Dennis=20Krone=CC=81?= Date: Mon, 17 May 2021 16:21:07 +0200 Subject: [PATCH 2/2] Suppress deprication message when adding request handler in runner --- locust/runners.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/locust/runners.py b/locust/runners.py index e2b09d8b5a..50e2f18844 100644 --- a/locust/runners.py +++ b/locust/runners.py @@ -78,8 +78,13 @@ def on_request_failure(request_type, name, response_time, response_length, excep 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