locust 2.32.0__py3-none-any.whl → 2.32.1.dev13__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.dev13"
 version = __version__
-__version_tuple__ = (2, 32, 0)
+__version_tuple__ = (2, 32, 1, "dev13")
 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]