locust 2.29.2.dev32__py3-none-any.whl → 2.29.2.dev42__py3-none-any.whl

locust/_version.py

@@ -3,6 +3,7 @@
 TYPE_CHECKING = False
 if TYPE_CHECKING:
     from typing import Tuple, Union
+
     VERSION_TUPLE = Tuple[Union[int, str], ...]
 else:
     VERSION_TUPLE = object
@@ -12,5 +13,8 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '2.29.2.dev32'
-__version_tuple__ = version_tuple = (2, 29, 2, 'dev32')
+
+__version__ = "2.29.2.dev42"
+version = __version__
+__version_tuple__ = (2, 29, 2, "dev42")
+version_tuple = __version_tuple__

locust/contrib/fasthttp.py

@@ -483,7 +483,7 @@ class FastResponse(CompatResponse):
 
     _response: HTTPSocketPoolResponse | None = None
 
-    encoding: str | None = None
+    encoding: str | float | None = None
     """In some cases setting the encoding explicitly is needed. If so, do it before calling .text"""
 
     request: FastRequest | None = None

locust/dispatch.py

@@ -90,13 +90,13 @@ class UsersDispatcher(Iterator):
         assert len(user_classes) > 0
         assert len(set(self._user_classes)) == len(self._user_classes)
 
-        self._target_user_count: int = None
+        self._target_user_count: int = 0
 
-        self._spawn_rate: float = None
+        self._spawn_rate: float = 0.0
 
-        self._user_count_per_dispatch_iteration: int = None
+        self._user_count_per_dispatch_iteration: int = 0
 
-        self._wait_between_dispatch: float = None
+        self._wait_between_dispatch: float = 0.0
 
         self._initial_users_on_workers = {
             worker_node.id: {user_class.__name__: 0 for user_class in self._user_classes}
@@ -107,7 +107,8 @@ class UsersDispatcher(Iterator):
 
         self._current_user_count = self.get_current_user_count()
 
-        self._dispatcher_generator: Generator[dict[str, dict[str, int]], None, None] = None
+        self._dispatcher_generator: Generator[dict[str, dict[str, int]], None, None] = None  # type: ignore
+        # a generator is assigned (in new_dispatch()) to _dispatcher_generator before it's used
 
         self._user_generator = self._user_gen()
 
@@ -131,7 +132,7 @@ class UsersDispatcher(Iterator):
         return sum(map(sum, map(dict.values, self._users_on_workers.values())))
 
     @property
-    def dispatch_in_progress(self):
+    def dispatch_in_progress(self) -> bool:
         return self._dispatch_in_progress
 
     @property

locust/stats.py

@@ -9,25 +9,12 @@ import signal
 import time
 from abc import abstractmethod
 from collections import OrderedDict, defaultdict, namedtuple
-from collections import (
-    OrderedDict as OrderedDictType,
-)
 from collections.abc import Iterable
 from copy import copy
 from html import escape
 from itertools import chain
-from tempfile import NamedTemporaryFile
 from types import FrameType
-from typing import (
-    TYPE_CHECKING,
-    Any,
-    Callable,
-    NoReturn,
-    Protocol,
-    TypedDict,
-    TypeVar,
-    cast,
-)
+from typing import TYPE_CHECKING, Any, Callable, NoReturn, Protocol, TypedDict, TypeVar, cast
 
 import gevent
 
@@ -319,12 +306,12 @@ class StatsEntry:
         A {response_time => count} dict that holds the response time distribution of all
         the requests.
 
-        The keys (the response time in ms) are rounded to store 1, 2, ... 9, 10, 20. .. 90,
-        100, 200 .. 900, 1000, 2000 ... 9000, in order to save memory.
+        The keys (the response time in ms) are rounded to store 1, 2, ... 98, 99, 100, 110, 120, ... 980, 990, 1000,
+        1100, 1200, ... 9800, 9900, 10_000, 11_000, 12_000 ... in order to save memory.
 
         This dict is used to calculate the median and percentile response times.
         """
-        self.response_times_cache: OrderedDictType[int, CachedResponseTimes] | None = None
+        self.response_times_cache: OrderedDict[int, CachedResponseTimes] | None = None
         """
         If use_response_times_cache is set to True, this will be a {timestamp => CachedResponseTimes()}
         OrderedDict that holds a copy of the response_times dict for each of the last 20 seconds.

{locust-2.29.2.dev32.dist-info → locust-2.29.2.dev42.dist-info}/METADATA

@@ -1,15 +1,15 @@
 Metadata-Version: 2.1
 Name: locust
-Version: 2.29.2.dev32
+Version: 2.29.2.dev42
 Summary: Developer-friendly load testing framework
+Home-page: https://locust.io/
 License: MIT
-Project-URL: Homepage, https://github.com/locustio/locust
-Project-URL: Documentation, https://docs.locust.io/
-Project-URL: Code, https://github.com/locustio/locust
-Project-URL: Help/Questions, https://stackoverflow.com/questions/tagged/locust
-Project-URL: Issue tracker, https://github.com/locustio/locust/issues
-Classifier: Topic :: Software Development :: Testing :: Traffic Generation
+Author: Jonatan Heyman
+Maintainer: Lars Holmberg
+Requires-Python: >=3.9
 Classifier: Development Status :: 5 - Production/Stable
+Classifier: Intended Audience :: Developers
+Classifier: Intended Audience :: System Administrators
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: OS Independent
 Classifier: Programming Language :: Python
@@ -18,29 +18,32 @@ Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
-Classifier: Intended Audience :: Developers
-Classifier: Intended Audience :: System Administrators
 Classifier: Topic :: Software Development :: Testing
 Classifier: Topic :: Software Development :: Testing :: Traffic Generation
 Classifier: Topic :: System :: Distributed Computing
-Requires-Python: >=3.9
+Requires-Dist: ConfigArgParse (>=1.5.5)
+Requires-Dist: Flask-Cors (>=3.0.10)
+Requires-Dist: Flask-Login (>=0.6.3)
+Requires-Dist: Werkzeug (>=2.0.0)
+Requires-Dist: flask (>=2.0.0)
+Requires-Dist: gevent (>=22.10.2)
+Requires-Dist: geventhttpclient (>=2.3.1)
+Requires-Dist: msgpack (>=1.0.0)
+Requires-Dist: psutil (>=5.9.1)
+Requires-Dist: pywin32 ; sys_platform == "win32"
+Requires-Dist: pyzmq (>=25.0.0)
+Requires-Dist: requests (>=2.26.0) ; python_full_version <= "3.11.0"
+Requires-Dist: requests (>=2.32.2) ; python_full_version > "3.11.0"
+Requires-Dist: tomli (>=1.1.0) ; python_version < "3.11"
+Requires-Dist: typing_extensions (>=4.6.0) ; python_version < "3.11"
+Project-URL: Documentation, https://docs.locust.io/
+Project-URL: Help/Questions, https://github.com/orgs/locustio/discussions/
+Project-URL: Issue Tracker, https://github.com/locustio/locust/issues
+Project-URL: Repository, https://github.com/locustio/locust
+Project-URL: Stack Overflow, https://stackoverflow.com/questions/tagged/locust
+Project-URL: Slack, https://locustio.slack.com/
+Project-URL: Slack/Signup, https://communityinviter.com/apps/locustio/locust
 Description-Content-Type: text/markdown
-License-File: LICENSE
-Requires-Dist: gevent >=22.10.2
-Requires-Dist: flask >=2.0.0
-Requires-Dist: Werkzeug >=2.0.0
-Requires-Dist: requests >=2.26.0
-Requires-Dist: msgpack >=1.0.0
-Requires-Dist: pyzmq >=25.0.0
-Requires-Dist: geventhttpclient >=2.3.1
-Requires-Dist: ConfigArgParse >=1.5.5
-Requires-Dist: psutil >=5.9.1
-Requires-Dist: Flask-Login >=0.6.3
-Requires-Dist: Flask-Cors >=3.0.10
-Requires-Dist: pywin32 ; platform_system == "Windows"
-Requires-Dist: tomli >=1.1.0 ; python_version < "3.11"
-Requires-Dist: typing-extensions >=4.6.0 ; python_version < "3.11"
-Requires-Dist: requests >=2.32.2 ; python_version > "3.11"
 
 # Locust
 
@@ -112,6 +115,7 @@ Locust's code base is intentionally kept small and doesn't solve everything out
 
 * Documentation: [docs.locust.io](https://docs.locust.io)
 * Support/Questions: [StackOverflow](https://stackoverflow.com/questions/tagged/locust)
+* Github Discussions: [Github Discussions](https://github.com/orgs/locustio/discussions)
 * Chat/discussion: [Slack](https://locustio.slack.com) [(signup)](https://communityinviter.com/apps/locustio/locust)
 
 ## Authors
@@ -124,3 +128,4 @@ Locust's code base is intentionally kept small and doesn't solve everything out
 ## License
 
 Open source licensed under the MIT license (see _LICENSE_ file for details).
+

locust-2.29.2.dev42.dist-info/RECORD

@@ -0,0 +1,49 @@
+locust/__init__.py,sha256=Hmw2vNf75eLQ1mQIPXAwlQrJ_XFY65MOb92fGsNCukQ,1458
+locust/__main__.py,sha256=vBQ82334kX06ImDbFlPFgiBRiLIinwNk3z8Khs6hd74,31
+locust/_version.py,sha256=lz-VFOL8BSOzklrR9XsmlGd2RsoIxkCjTCovy1iiKAM,460
+locust/argument_parser.py,sha256=sjQoJ1NTac9LdNYT7zn8RajlWqBQs8YFNv6uRExb2gg,28941
+locust/clients.py,sha256=OHPv6hBAt4gt3HI67yqyT1qrSsF8uMdCwIRu0kIsRWI,19491
+locust/contrib/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+locust/contrib/fasthttp.py,sha256=1hR2WK-3ACGVsU6Fu4Z4KucVzL1LXMLG6k54KeZdt78,28509
+locust/debug.py,sha256=We6Z9W0btkKSc7PxWmrZx-xMynvOOsKhG6jmDgQin0g,5134
+locust/dispatch.py,sha256=0rGny9P8dEq2Kg7exRWPorKQ5D1LnhHd2xG5Jq7ge7g,17107
+locust/env.py,sha256=sP-fCnZs0e2xodRemLHgTgyyUt5eezwtdA9WsCoqJkQ,12767
+locust/event.py,sha256=iXEwIYFzra-j1WRldXB9SUibydtD8q8EIKaFPGTTIjk,8729
+locust/exception.py,sha256=jGgJ32ubuf4pWdlaVOkbh2Y0LlG0_DHi-lv3ib8ppOE,1791
+locust/html.py,sha256=_n3aB3fxiYzSeE_7RqHF3iiEPjPnbQ3e2Pw9P8AVtPU,3920
+locust/input_events.py,sha256=ZIyePyAMuA_YFYWg18g_pE4kwuQV3RbEB250MzXRwjY,3314
+locust/log.py,sha256=Wrkn0Ibugh5Sqjm4hGQ2-jUsy1tNMBdTctp4FyXQI24,3457
+locust/main.py,sha256=NGjL5QqakU5aeyUzwu2Fh00xVZfC3eoBE3DtfOmRtcM,27854
+locust/py.typed,sha256=gkWLl8yD4mIZnNYYAIRM8g9VarLvWmTAFeUfEbxJLBw,65
+locust/rpc/__init__.py,sha256=nVGoHWFQxZjnhCDWjbgXIbmFbN9sizAjkhvSs9_642c,58
+locust/rpc/protocol.py,sha256=n-rb3GZQcAlldYDj4E4GuFGylYj_26GSS5U29meft5Y,1282
+locust/rpc/zmqrpc.py,sha256=AAY6w7wSFHsW34qqN28666boHFf6dTSAXPQJnAO6iUI,2707
+locust/runners.py,sha256=otH-ZxTygBRXN46Nmocs5ac8R4b0MsnKAUcHRwVnD1E,69869
+locust/shape.py,sha256=t-lwBS8LOjWcKXNL7j2U3zroIXJ1b0fazUwpRYQOKXw,1973
+locust/stats.py,sha256=KLVIdw-jh09u75JFucl7PHXNktAnPnQx-DrV_rUH5xM,45772
+locust/user/__init__.py,sha256=S2yvmI_AU9kXirtTIVqiV_Hs7yXzqXvaSgkNo9ig-fk,71
+locust/user/inspectuser.py,sha256=KgrWHyE5jhK6or58R7soLRf-_st42AaQrR72qbiXw9E,2641
+locust/user/sequential_taskset.py,sha256=SbrrGU9HV2nEWe6zQVtjymn8NgPISP7QSNoVdyoXjYg,2687
+locust/user/task.py,sha256=doCOclhPC92ePfVa1KfIceZC-YofpnM4VaKplWYNH8o,16730
+locust/user/users.py,sha256=BsKyxzLq1lmdYBXnwHNyMH_Nxfy7QRlyLnfiE8539HY,9989
+locust/user/wait_time.py,sha256=bGRKMVx4lom75sX3POYJUa1CPeME2bEAXG6CEgxSO5U,2675
+locust/util/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+locust/util/cache.py,sha256=IxbpGawl0-hoWKvCrtksxjSLf2GbBBTVns06F7mFBkM,1062
+locust/util/date.py,sha256=pqyUuqzPds-A8ai6ZYTxe5Eak1M1BR9iuQcqEcrtHwc,179
+locust/util/deprecation.py,sha256=pFipULkYFjYnuFSycJZ0s95r0CTSlvx8otGnqSu1bsQ,2028
+locust/util/exception_handler.py,sha256=jTMyBq2a0O07fRjmqGkheyaPj58tUgnbbcjoesKGPws,797
+locust/util/load_locustfile.py,sha256=hn70KcIG8jHmZyuKv2pcEmwgWtOEu24Efeji1KRYNUE,2964
+locust/util/rounding.py,sha256=5haxR8mKhATqag6WvPby-MSRRgIw5Ob6thbyvMYZM7o,92
+locust/util/timespan.py,sha256=Y0LtnhUq2Mq19p04u0XtBlYQ_-S2cRvwRdgru8W9WhA,986
+locust/web.py,sha256=rN1NVeZ9LKSEeDwvpRbOJ0bcy8U1U4VjP-7vK7ejlwM,27367
+locust/webui/dist/assets/favicon.ico,sha256=IUl-rYqfpHdV38e-s0bkmFIeLS-n3Ug0DQxk-h202hI,8348
+locust/webui/dist/assets/index-e9ad42b4.js,sha256=aXKF6Qug5b9n5e160A0PFvM0qgJQgVwtwQhLwCsek-4,1647570
+locust/webui/dist/assets/logo.png,sha256=EIVPqr6wE_yqguHaqFHIsH0ZACLSrvNWyYO7PbyIj4w,19299
+locust/webui/dist/auth.html,sha256=NE_pYWbFjnBEkq2tG5mS_9YoL7JHX4Gkx9zhQoxlSC8,501
+locust/webui/dist/index.html,sha256=S78UvAUZbQ-sH0wBxqFKrT2ZSfRxUFGx5xwQY6FaVMk,507
+locust/webui/dist/report.html,sha256=sOdZZVgZbqgu86BBCSQf3uQUYXgmgSnXF32JpnyAII8,513
+locust-2.29.2.dev42.dist-info/LICENSE,sha256=78XGpIn3fHVBfaxlPNUfjVufSN7QsdhpJMRJHv2AFpo,1095
+locust-2.29.2.dev42.dist-info/METADATA,sha256=zxWkyHFAuy0txwl12tVAxmjgvpmXHSgWdVeCOUk7nDk,7674
+locust-2.29.2.dev42.dist-info/WHEEL,sha256=HBsDV7Hj4OTiS1GX6ua7iQXUQTB9UHftbBxr7Q8Xm9c,110
+locust-2.29.2.dev42.dist-info/entry_points.txt,sha256=RhIxlLwU_Ae_WjimS5COUDLagdCh6IOMyLtgaQxNmlM,43
+locust-2.29.2.dev42.dist-info/RECORD,,

locust-2.29.2.dev42.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: poetry-core 1.9.0
+Root-Is-Purelib: false
+Tag: cp312-cp312-manylinux_2_35_x86_64

locust-2.29.2.dev42.dist-info/entry_points.txt

@@ -0,0 +1,3 @@
+[console_scripts]
+locust=locust.main:main
+

locust/test/__init__.py

@@ -1,15 +0,0 @@
-try:
-    import resource
-
-    # work around occasional "zmq.error.ZMQError: Too many open files"
-    # this is done in main.py when running locust proper so we need to do it here as well
-    resource.setrlimit(
-        resource.RLIMIT_NOFILE,
-        (
-            10000,
-            resource.RLIM_INFINITY,
-        ),
-    )
-    changed_rlimit = True
-except Exception:
-    changed_rlimit = False

locust/test/fake_module1_for_env_test.py

@@ -1,7 +0,0 @@
-"""Module for locust.test.test_env.TestEnvironment.test_user_classes_with_same_name_is_error"""
-
-from locust import User
-
-
-class MyUserWithSameName(User):
-    pass

locust/test/fake_module2_for_env_test.py

@@ -1,7 +0,0 @@
-"""Module for locust.test.test_env.TestEnvironment.test_user_classes_with_same_name_is_error"""
-
-from locust import User
-
-
-class MyUserWithSameName(User):
-    pass

locust/test/mock_locustfile.py

@@ -1,56 +0,0 @@
-import os
-import random
-import time
-from contextlib import contextmanager
-
-MOCK_LOCUSTFILE_CONTENT = '''
-"""This is a mock locust file for unit testing"""
-
-from locust import HttpUser, TaskSet, task, between, LoadTestShape
-
-
-def index(l):
-    l.client.get("/")
-
-def stats(l):
-    l.client.get("/stats/requests")
-
-
-class UserTasks(TaskSet):
-    # one can specify tasks like this
-    tasks = [index, stats]
-
-
-class UserSubclass(HttpUser):
-    host = "http://127.0.0.1:8089"
-    wait_time = between(2, 5)
-    tasks = [UserTasks]
-
-
-class NotUserSubclass():
-    host = "http://localhost:8000"
-
-'''
-
-
-class MockedLocustfile:
-    __slots__ = ["filename", "directory", "file_path"]
-
-
-@contextmanager
-def mock_locustfile(filename_prefix="mock_locustfile", content=MOCK_LOCUSTFILE_CONTENT, dir=None):
-    mocked = MockedLocustfile()
-    mocked.directory = dir or os.path.dirname(os.path.abspath(__file__))
-    mocked.filename = "%s_%s_%i.py" % (
-        filename_prefix,
-        str(time.time()).replace(".", "_"),
-        random.randint(0, 100000),
-    )
-    mocked.file_path = os.path.join(mocked.directory, mocked.filename)
-    with open(mocked.file_path, "w") as file:
-        file.write(content)
-
-    try:
-        yield mocked
-    finally:
-        os.remove(mocked.file_path)

locust/test/mock_logging.py

@@ -1,28 +0,0 @@
-from __future__ import annotations
-
-import logging
-from types import TracebackType
-from typing import Union
-
-LogMessage = list[Union[str, dict[str, TracebackType]]]
-
-
-class MockedLoggingHandler(logging.Handler):
-    debug: list[LogMessage] = []
-    warning: list[LogMessage] = []
-    info: list[LogMessage] = []
-    error: list[LogMessage] = []
-    critical: list[LogMessage] = []
-
-    def emit(self, record):
-        if record.exc_info:
-            value = {"message": record.getMessage(), "exc_info": record.exc_info}
-        else:
-            value = record.getMessage()
-        getattr(self.__class__, record.levelname.lower()).append(value)
-
-    @classmethod
-    def reset(cls):
-        for attr in dir(cls):
-            if isinstance(getattr(cls, attr), list):
-                setattr(cls, attr, [])

locust/test/test_debugging.py

@@ -1,39 +0,0 @@
-from locust import debug, task
-from locust.test.testcases import LocustTestCase
-from locust.user.task import LOCUST_STATE_STOPPING
-from locust.user.users import HttpUser
-
-import os
-from threading import Timer
-from unittest import mock
-
-
-class DebugTestCase(LocustTestCase):
-    def setUp(self):
-        super().setUp()
-        debug._env = None
-
-
-class TestDebugging(DebugTestCase):
-    @mock.patch.dict(os.environ, {"LOCUST_HOST": "http://localhost"})
-    def test_run_single_user_pass_host_to_user_classes(self):
-        """
-        HttpUser should receive host from environment variable
-        """
-
-        class MyUser1(HttpUser):
-            @task
-            def my_task(self):
-                pass
-
-        def _stop_user():
-            if user := getattr(debug._env, "single_user_instance", None):
-                user._state = LOCUST_STATE_STOPPING
-
-        t = Timer(1, _stop_user)
-        t.start()
-
-        debug.run_single_user(
-            MyUser1,
-            loglevel=None,  # another log setup might mess with other tests...
-        )