locust 2.32.0__py3-none-any.whl → 2.32.1__py3-none-any.whl

locust/_version.py

@@ -14,7 +14,7 @@ __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
 
-__version__ = "2.32.0"
+__version__ = "2.32.1"
 version = __version__
-__version_tuple__ = (2, 32, 0)
+__version_tuple__ = (2, 32, 1)
 version_tuple = __version_tuple__

locust/contrib/fasthttp.py

@@ -587,6 +587,14 @@ class ErrorResponse:
         raise self.error
 
 
+class LocustBadStatusCode(ConnectionError):
+    def __repr__(self):
+        repr_str = super().__repr__()
+        if self.kw_text:
+            return repr_str.replace(repr(self.url) + ", ", "")
+        return repr_str
+
+
 class LocustUserAgent(UserAgent):
     response_type = FastResponse
     request_type = FastRequest
@@ -606,6 +614,11 @@ class LocustUserAgent(UserAgent):
         )
         return self.response_type(resp, request=request, sent_request=resp._sent_request)
 
+    def _verify_status(self, status_code, url=None):
+        """Hook for subclassing"""
+        if status_code not in self.valid_response_codes:
+            raise LocustBadStatusCode(url, code=status_code)
+
 
 class ResponseContextManager(FastResponse):
     """

locust/debug.py

@@ -150,7 +150,7 @@ def run_single_user(
         _env.events.init.fire(environment=_env, runner=None, web_ui=None)
         # uncaught events will be suppressed, so check if that happened
         if locust.log.unhandled_greenlet_exception:
-            raise Exception("Unhandled exception in init")
+            os._exit(1)
 
     # do the things that the Runner usually does
     _env.user_classes = [user_class]

locust/web.py

@@ -24,6 +24,7 @@ from flask import (
     request,
     send_file,
     send_from_directory,
+    session,
     url_for,
 )
 from flask_cors import CORS
@@ -504,6 +505,8 @@ class WebUI:
             if not self.web_login:
                 return redirect(url_for("index"))
 
+            self.auth_args["error"] = session.get("auth_error", None)
+
             return render_template_from(
                 "auth.html",
                 auth_args=self.auth_args,
@@ -573,6 +576,7 @@ class WebUI:
         def wrapper(*args, **kwargs):
             if self.web_login:
                 try:
+                    session["auth_error"] = None
                     return login_required(view_func)(*args, **kwargs)
                 except Exception as e:
                     return f"Locust auth exception: {e} See https://docs.locust.io/en/stable/extending-locust.html#adding-authentication-to-the-web-ui for configuring authentication."