PyPI - locust - Versions diffs - 2.47.1.dev16__tar.gz → 2.47.1.dev21__tar.gz - Mend

@@ -20,10 +20,12 @@ if typing.TYPE_CHECKING:
     from paho.mqtt.subscribeoptions import SubscribeOptions
-# A SUBACK response for MQTT can only contain 0x00, 0x01, 0x02, or 0x80. 0x80
-# indicates a failure to subscribe.
+# A SUBACK response for MQTT 3.1.1 can only contain 0x00, 0x01, 0x02, or 0x80.
+# 0x80 indicates a failure to subscribe. In MQTT 5, any reason code of 0x80 or
+# greater indicates a failure.
 #
 # http://docs.oasis-open.org/mqtt/mqtt/v3.1.1/os/mqtt-v3.1.1-os.html#_Figure_3.26_-
+# https://docs.oasis-open.org/mqtt/mqtt/v5.0/os/mqtt-v5.0-os.html#_Toc3901031
 SUBACK_FAILURE = 0x80
 REQUEST_TYPE = "MQTT"
@@ -189,7 +191,7 @@ class MqttClient(mqtt.Client):
                 },
             )
         else:
-            if SUBACK_FAILURE in granted_qos:
+            if any(code >= SUBACK_FAILURE for code in granted_qos):
                 self.environment.events.request.fire(
                     request_type=REQUEST_TYPE,
                     name=self._generate_event_name("subscribe", qos, topic),

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

@@ -18,7 +18,7 @@ version_tuple: tuple[int | str, ...]
 commit_id: str | None
 __commit_id__: str | None
-__version__ = version = '2.47.1.dev16'
-__version_tuple__ = version_tuple = (2, 47, 1, 'dev16')
+__version__ = version = '2.47.1.dev21'
+__version_tuple__ = version_tuple = (2, 47, 1, 'dev21')
 __commit_id__ = commit_id = None

@@ -122,11 +122,12 @@ class HttpSession(requests.Session):
     def rename_request(self, name: str) -> Generator[None]:
         """Group requests using the "with" keyword"""
+        previous_name = self.request_name
         self.request_name = name
         try:
             yield
         finally:
-            self.request_name = None
+            self.request_name = previous_name
     def request(  # type: ignore[override]
         self,

@@ -164,11 +164,12 @@ class FastHttpSession:
     def rename_request(self, name: str) -> Generator[None]:
         """Group requests using the "with" keyword"""
+        previous_name = self.request_name
         self.request_name = name
         try:
             yield
         finally:
-            self.request_name = None
+            self.request_name = previous_name
     def _send_request_safe_mode(self, method: str, url: str, **kwargs) -> FastResponse:
         """

@@ -54,11 +54,12 @@ class OpenAIClient(OpenAI):
     def rename_request(self, name: str) -> Generator[None]:
         """Group requests using the "with" keyword"""
+        previous_name = self.request_name
         self.request_name = name
         try:
             yield
         finally:
-            self.request_name = None
+            self.request_name = previous_name
 class OpenAIUser(User):

locust 2.47.1.dev16__tar.gz → 2.47.1.dev21__tar.gz