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