locust 2.43.5.dev10__tar.gz → 2.43.5.dev19__tar.gz

{locust-2.43.5.dev10 → locust-2.43.5.dev19}/PKG-INFO

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

{locust-2.43.5.dev10 → locust-2.43.5.dev19}/locust/_version.py

@@ -18,7 +18,7 @@ version_tuple: tuple[int | str, ...]
 commit_id: str | None
 __commit_id__: str | None
 
-__version__ = version = '2.43.5.dev10'
-__version_tuple__ = version_tuple = (2, 43, 5, 'dev10')
+__version__ = version = '2.43.5.dev19'
+__version_tuple__ = version_tuple = (2, 43, 5, 'dev19')
 
 __commit_id__ = commit_id = None

{locust-2.43.5.dev10 → locust-2.43.5.dev19}/locust/contrib/fasthttp.py

@@ -10,6 +10,7 @@ import re
 import socket
 import time
 import traceback
+import zlib
 from base64 import b64encode
 from contextlib import contextmanager
 from http.cookiejar import CookieJar
@@ -84,6 +85,7 @@ FAILURE_EXCEPTIONS = (
     SSLError,
     Timeout,
     HTTPConnectionClosed,
+    zlib.error,
 )
 
 
@@ -254,7 +256,7 @@ class FastHttpSession:
 
         if not allow_redirects:
             old_redirect_response_codes = self.client.redirect_resonse_codes
-            self.client.redirect_resonse_codes = []
+            self.client.redirect_resonse_codes = frozenset()
 
         start_perf_counter = time.perf_counter()
         # send request, and catch any exceptions
@@ -672,6 +674,7 @@ class LocustUserAgent(UserAgent):
     response_type = FastResponse
     request_type = FastRequest
     valid_response_codes = frozenset([200, 201, 202, 203, 204, 205, 206, 207, 208, 226, 301, 302, 303, 304, 307, 308])
+    redirect_resonse_codes = frozenset([301, 302, 303, 307, 308])
 
     def __init__(self, client_pool: HTTPClientPool | None = None, **kwargs):
         super().__init__(**kwargs)

{locust-2.43.5.dev10 → locust-2.43.5.dev19}/locust/stats.py

@@ -69,6 +69,8 @@ class StatsEntryDict(StatsBaseDict):
 class StatsErrorDict(StatsBaseDict):
     error: str
     occurrences: int
+    first_seen: float | None
+    last_seen: float | None
 
 
 class StatsHolder(Protocol):
@@ -705,11 +707,21 @@ class StatsEntry:
 
 
 class StatsError:
-    def __init__(self, method: str, name: str, error: Exception | str | None, occurrences: int = 0):
+    def __init__(
+        self,
+        method: str,
+        name: str,
+        error: Exception | str | None,
+        occurrences: int = 0,
+        first_seen: float | None = None,
+        last_seen: float | None = None,
+    ):
         self.method = method
         self.name = name
         self.error = error
         self.occurrences = occurrences
+        self.first_seen = first_seen
+        self.last_seen = last_seen
 
     @classmethod
     def parse_error(cls, error: Exception | str | None) -> str:
@@ -735,6 +747,10 @@ class StatsError:
 
     def occurred(self) -> None:
         self.occurrences += 1
+        now = time.time()
+        if self.first_seen is None:
+            self.first_seen = now
+        self.last_seen = now
 
     def to_name(self) -> str:
         error = self.error
@@ -767,7 +783,14 @@ class StatsError:
 
     @classmethod
     def unserialize(cls, data: StatsErrorDict) -> StatsError:
-        return cls(data["method"], data["name"], data["error"], data["occurrences"])
+        return cls(
+            data["method"],
+            data["name"],
+            data["error"],
+            data["occurrences"],
+            data.get("first_seen"),
+            data.get("last_seen"),
+        )
 
     def to_dict(self):
         return {
@@ -775,6 +798,8 @@ class StatsError:
             "name": self.name,
             "error": self.parse_error(self.error),
             "occurrences": self.occurrences,
+            "first_seen": self.first_seen,
+            "last_seen": self.last_seen,
         }
 
 
@@ -815,7 +840,16 @@ def setup_distributed_stats_event_listeners(events: Events, stats: RequestStats)
             if error_key not in stats.errors:
                 stats.errors[error_key] = StatsError.unserialize(error)
             else:
-                stats.errors[error_key].occurrences += error["occurrences"]
+                existing = stats.errors[error_key]
+                existing.occurrences += error["occurrences"]
+                if incoming_first := error.get("first_seen"):
+                    existing.first_seen = (
+                        incoming_first if existing.first_seen is None else min(existing.first_seen, incoming_first)
+                    )
+                if incoming_last := error.get("last_seen"):
+                    existing.last_seen = (
+                        incoming_last if existing.last_seen is None else max(existing.last_seen, incoming_last)
+                    )
 
         stats.total.extend(StatsEntry.unserialize(data["stats_total"]))
 
@@ -988,6 +1022,8 @@ class StatsCSV:
             "Name",
             "Error",
             "Occurrences",
+            "First Seen",
+            "Last Seen",
         ]
 
         self.exceptions_columns = [
@@ -1045,6 +1081,8 @@ class StatsCSV:
                     stats_error.name,
                     StatsError.parse_error(stats_error.error),
                     stats_error.occurrences,
+                    format_utc_timestamp(stats_error.first_seen) if stats_error.first_seen is not None else "",
+                    format_utc_timestamp(stats_error.last_seen) if stats_error.last_seen is not None else "",
                 ]
             )