locust 2.40.4.dev3__tar.gz → 2.40.5.dev5__tar.gz

{locust-2.40.4.dev3 → locust-2.40.5.dev5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: locust
-Version: 2.40.4.dev3
+Version: 2.40.5.dev5
 Summary: Developer-friendly load testing framework
 Project-URL: homepage, https://locust.io/
 Project-URL: repository, https://github.com/locustio/locust

{locust-2.40.4.dev3 → locust-2.40.5.dev5}/locust/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '2.40.4.dev3'
-__version_tuple__ = version_tuple = (2, 40, 4, 'dev3')
+__version__ = version = '2.40.5.dev5'
+__version_tuple__ = version_tuple = (2, 40, 5, 'dev5')
 
 __commit_id__ = commit_id = None

{locust-2.40.4.dev3 → locust-2.40.5.dev5}/locust/contrib/fasthttp.py

@@ -199,7 +199,7 @@ class FastHttpSession:
         allow_redirects: bool = True,
         context: dict = {},
         **kwargs,
-    ) -> ResponseContextManager:
+    ) -> ResponseContextManager:  # technically it can also return FastResponse
         """
         Send an HTTP request
 
@@ -226,6 +226,8 @@ class FastHttpSession:
             Another side effect of setting stream to True is that the time for downloading the response
             content will not be accounted for in the request time that is reported by Locust.
         :param allow_redirects: (optional) Set to True by default.
+        :return: A :py:class:`FastResponse <locust.contrib.fasthttp.FastResponse>` object if catch_response is False, and
+            :py:class:`ResponseContextManager <locust.contrib.fasthttp.ResponseContextManager>` if True.
         """
         # prepend url with hostname unless it's already an absolute URL
         built_url = self._build_url(url)
@@ -283,12 +285,12 @@ class FastHttpSession:
             except HTTPParseError as e:
                 request_meta["response_time"] = (time.perf_counter() - start_perf_counter) * 1000
                 request_meta["exception"] = e
-                rcm = ResponseContextManager(response, self.request_event, request_meta, catch_response)
-                if not catch_response:
-                    rcm.__exit__(None, None, None)
-                return rcm
+                if catch_response:
+                    return ResponseContextManager(response, self.request_event, request_meta, catch_response)
+                else:
+                    self.request_event.fire(**request_meta)
+                    return response  # type: ignore[return-value]
 
-        rcm = ResponseContextManager(response, self.request_event, request_meta, catch_response)
         # Record the consumed time
         # Note: This is intentionally placed after we record the content_size above, since
         # we'll then trigger fetching of the body (unless stream=True)
@@ -299,9 +301,12 @@ class FastHttpSession:
         except FAILURE_EXCEPTIONS as e:
             request_meta["exception"] = e  # type: ignore[assignment] # mypy, why are you so dumb..
 
-        if not catch_response:  # if not using with-block, report the request immediately
-            rcm.__exit__(None, None, None)
-        return rcm
+        if catch_response:
+            return ResponseContextManager(response, self.request_event, request_meta, catch_response)
+        else:
+            # if not using with-block, report the request immediately
+            self.request_event.fire(**request_meta)
+            return response  # type: ignore[return-value]
 
     def delete(self, url: str, **kwargs: Unpack[RESTKwargs]) -> ResponseContextManager:
         """Sends a DELETE request"""

{locust-2.40.4.dev3 → locust-2.40.5.dev5}/locust/util/load_locustfile.py

@@ -105,7 +105,7 @@ def load_locustfile_pytest(path) -> dict[str, type[User]]:
         {},
         [
             "-q",  # suppress pytest loggings about "test session starts" and "collected 0 items" etc
-            "-s",  # dont capture stdin (locust uses it for opening a browser and for input events)
+            "-s",  # dont capture stdin (locust uses it for opening a browser and other input events)
             path,
         ],
     )