locust 2.46.7.dev19__tar.gz → 2.46.7.dev21__tar.gz

{locust-2.46.7.dev19 → locust-2.46.7.dev21}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: locust
-Version: 2.46.7.dev19
+Version: 2.46.7.dev21
 Summary: Developer-friendly load testing framework
 Project-URL: Homepage, https://locust.cloud/
 Project-URL: Repository, https://github.com/locustio/locust

{locust-2.46.7.dev19 → locust-2.46.7.dev21}/locust/_version.py

@@ -18,7 +18,7 @@ version_tuple: tuple[int | str, ...]
 commit_id: str | None
 __commit_id__: str | None
 
-__version__ = version = '2.46.7.dev19'
-__version_tuple__ = version_tuple = (2, 46, 7, 'dev19')
+__version__ = version = '2.46.7.dev21'
+__version_tuple__ = version_tuple = (2, 46, 7, 'dev21')
 
 __commit_id__ = commit_id = None

{locust-2.46.7.dev19 → locust-2.46.7.dev21}/locust/contrib/fasthttp.py

@@ -12,6 +12,7 @@ import socket
 import time
 import zlib
 from base64 import b64encode
+from contextlib import contextmanager
 from http.cookiejar import CookieJar
 from ssl import SSLError
 from typing import TYPE_CHECKING
@@ -111,6 +112,11 @@ class FastHttpSession:
         self.request_event = request_event
         self.cookiejar = CookieJar()
         self.user = user
+
+        # Requests can be grouped under a common name by setting request_name, or by using the rename_request context manager
+        # This is an alternative to passing in the "name" parameter to the requests function
+        self.request_name: str | None = None
+
         if not ssl_context_factory:
             if insecure:
                 ssl_context_factory = insecure_ssl_context_factory
@@ -153,6 +159,16 @@ class FastHttpSession:
         else:
             return f"{self.base_url}{path}"
 
+    @contextmanager
+    def rename_request(self, name: str) -> Generator[None]:
+        """Group requests using the "with" keyword"""
+
+        self.request_name = name
+        try:
+            yield
+        finally:
+            self.request_name = None
+
     def _send_request_safe_mode(self, method: str, url: str, **kwargs) -> FastResponse:
         """
         Send an HTTP request, and catch any exception that might occur due to either
@@ -221,6 +237,10 @@ class FastHttpSession:
         :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.
         """
+        # if group name has been set and no name parameter has been passed in; set the name parameter to group_name
+        if self.request_name and not name:
+            name = self.request_name
+
         # prepend url with hostname unless it's already an absolute URL
         built_url = self._build_url(url)