locust 2.47.1.dev39__tar.gz → 2.47.1.dev43__tar.gz

{locust-2.47.1.dev39 → locust-2.47.1.dev43}/PKG-INFO

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

{locust-2.47.1.dev39 → locust-2.47.1.dev43}/locust/_version.py

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

{locust-2.47.1.dev39 → locust-2.47.1.dev43}/locust/contrib/fasthttp.py

@@ -12,6 +12,7 @@ import socket
 import time
 import zlib
 from base64 import b64encode
+from codecs import getincrementaldecoder
 from contextlib import contextmanager
 from http.cookiejar import CookieJar
 from ssl import SSLError
@@ -335,17 +336,16 @@ class FastHttpSession:
         response = self.request(method, url, stream=True, **kwargs)
         response.raise_for_status()
 
+        decoder = getincrementaldecoder("utf-8")(errors="replace")
         buffer = ""
-        for chunk in response.iter_content(chunk_size=1024, decode_content=True):
-            #  Ensure that chunk is a string.
-            if isinstance(chunk, bytes):
-                chunk = chunk.decode("utf-8", errors="replace")
-
-            buffer += chunk
+        for chunk in response.iter_content(chunk_size=1024, decode_content=False):
+            buffer += decoder.decode(chunk)
             while "\n" in buffer:
                 line, buffer = buffer.split("\n", 1)
                 yield line
 
+        # Flush a final incomplete character using the same replacement policy.
+        buffer += decoder.decode(b"", final=True)
         # Return to the last line that may be incomplete.
         if buffer:
             yield buffer

{locust-2.47.1.dev39 → locust-2.47.1.dev43}/locust/user/inspectuser.py

@@ -4,6 +4,7 @@ import inspect
 from collections import defaultdict
 from json import dumps
 
+from ..dispatch import _kl_generator
 from .task import TaskSet
 from .users import User
 
@@ -32,11 +33,15 @@ def _calc_distribution(user_classes, num_users):
     num_users = num_users or (total_weight if not fixed_count else 1)
     weighted_count = num_users - fixed_count
     weighted_count = max(0, weighted_count)
-    user_classes_count = {}
-
-    for u in user_classes:
-        count = u.fixed_count or (u.weight / total_weight) * weighted_count
-        user_classes_count[u.__name__] = round(count)
+    user_classes_count = {u.__name__: u.fixed_count for u in user_classes}
+
+    # Pick weighted users the way the dispatcher does, so the counts match a real run.
+    weighted_users = _kl_generator((u.__name__, u.weight) for u in user_classes if not u.fixed_count)
+    for _ in range(int(weighted_count)):
+        name = next(weighted_users)
+        if name is None:
+            break
+        user_classes_count[name] += 1
 
     return user_classes_count