locust 2.29.2.dev34__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.dev34'
-__version_tuple__ = version_tuple = (2, 29, 2, 'dev34')
+
+__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-2.29.2.dev34.dist-info → locust-2.29.2.dev42.dist-info}/METADATA

@@ -1,15 +1,15 @@
 Metadata-Version: 2.1
 Name: locust
-Version: 2.29.2.dev34
+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.dev34.dist-info → locust-2.29.2.dev42.dist-info}/RECORD

@@ -1,10 +1,12 @@
 locust/__init__.py,sha256=Hmw2vNf75eLQ1mQIPXAwlQrJ_XFY65MOb92fGsNCukQ,1458
 locust/__main__.py,sha256=vBQ82334kX06ImDbFlPFgiBRiLIinwNk3z8Khs6hd74,31
-locust/_version.py,sha256=JBB9SnH1JVeprCyiBeHwUccW_DTDmJF3v1M32Sj6GuA,428
+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=vYh0QEDFgJ3hY0HgSk-EiNO7IP9ffzXF_Et8wB9JvsI,16995
+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
@@ -13,44 +15,12 @@ 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/runners.py,sha256=otH-ZxTygBRXN46Nmocs5ac8R4b0MsnKAUcHRwVnD1E,69869
-locust/shape.py,sha256=t-lwBS8LOjWcKXNL7j2U3zroIXJ1b0fazUwpRYQOKXw,1973
-locust/stats.py,sha256=KLVIdw-jh09u75JFucl7PHXNktAnPnQx-DrV_rUH5xM,45772
-locust/web.py,sha256=rN1NVeZ9LKSEeDwvpRbOJ0bcy8U1U4VjP-7vK7ejlwM,27367
-locust/contrib/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-locust/contrib/fasthttp.py,sha256=tcCeXkDou-WlQwdXhj0Wt29sS9gDrqyA2aA9BMqTzeY,28501
 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/test/__init__.py,sha256=CaVC4yA4DnCO8EY3LbedPHFg86a9Lqlpe92JuiX3THw,396
-locust/test/fake_module1_for_env_test.py,sha256=dzGYWCr1SSkd8Yyo68paUNrCNW7YY_QgjRb7sM37gG0,164
-locust/test/fake_module2_for_env_test.py,sha256=dzGYWCr1SSkd8Yyo68paUNrCNW7YY_QgjRb7sM37gG0,164
-locust/test/mock_locustfile.py,sha256=4xgoAYlhvdIBjGsLFFN0abpTNM7k12iSkrfTPUQhAMQ,1271
-locust/test/mock_logging.py,sha256=qapKrKhTdlVc8foJB2Hxjn7SB6soaLeAj3VF4A6kZtw,806
-locust/test/test_debugging.py,sha256=omQ0w5_Xh1xuTBzkd3VavEIircwtlmoOEHcMInY67vU,1053
-locust/test/test_dispatch.py,sha256=RjoncanN4FFt-aiTl4G8XRoc81n6fwfO8CacbjzpvP8,168856
-locust/test/test_env.py,sha256=l0fLl9nubdgzxwFNajmBkJvQc5cO5rOTE4p12lbCbs0,8919
-locust/test/test_fasthttp.py,sha256=cuWsS9t875Mrik0xTXlsfHde90F6_cgKWKJ3_Dr0ldw,30723
-locust/test/test_http.py,sha256=VQCVY0inLC0RS-V3E9WHL3vBLGokZjQt0zKSrTNlQmM,12536
-locust/test/test_interruptable_task.py,sha256=LZKSV-aJNnwfvAxguz6SckBEuGEnfGimoIgVfJ2wQTA,1377
-locust/test/test_load_locustfile.py,sha256=v-muHoM-CYu8t7DXm4AQtFP2q8RYfnTTUBqj7uVqhig,8494
-locust/test/test_locust_class.py,sha256=oGhhOX848jHRQnIfFlhLlW-kHGYLyYsfDX8hM07Ro7g,25506
-locust/test/test_log.py,sha256=YPY6vgTAy1KaNU2qoVvQrTH5x_mzRrljEHrkSBy3yxs,7553
-locust/test/test_main.py,sha256=7OuH2-7noD_rbeoKJD9hIZsylSugu7ze3XFIrU1u0HI,85016
-locust/test/test_old_wait_api.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-locust/test/test_parser.py,sha256=-2VO5Dopg-VoWvIgXrmr7GN40cqrnjUoctBHmVlyewg,17826
-locust/test/test_runners.py,sha256=jvyv5JCD4sJ_8nmB8ZzqGINLkepIAKyIgwdVTzMtG7I,169468
-locust/test/test_sequential_taskset.py,sha256=4RMJp6_xEIrLh8wjNzhIQjKmBvoqgzCNz57Zz6kWz-k,3900
-locust/test/test_stats.py,sha256=F51VkL3k3y4OhYBlRyV6vWzisenSAOmSWKy2IPVrnWM,33929
-locust/test/test_tags.py,sha256=mzhGLPMizSnSItTHLHizYvloxDfuIDAOgelwInyrf28,13138
-locust/test/test_taskratio.py,sha256=SQ-sBqeFm2GhkfCD_57-fPzQrk1ilSw3DRb0_nwyxAI,2742
-locust/test/test_users.py,sha256=lp6yAKGK9_MIs9F7s1Vc3561P4oRavhpeVo2y9w3SUU,2135
-locust/test/test_util.py,sha256=DmFTgNSWWx8zrsx9_ZGO6MsySmBV1H_GzNIVzzyapCM,1229
-locust/test/test_wait_time.py,sha256=3evSEp6amMWFrzmSYs71MCeIsu7Rtarldb_HnwgSrU0,2353
-locust/test/test_web.py,sha256=MxLdLIiOFdS6zoRoPJqviFPKwymLsaQRzVzbvr3qpEc,46789
-locust/test/test_zmqrpc.py,sha256=kONaZ11hwnneLwaVn7lIDVV7KHpEP2nkxuKhfb9ba3o,2173
-locust/test/testcases.py,sha256=ZaPYNxSSChAs0nts_13mCGY7WFW8AjXQZdPOvwAK0TY,6961
-locust/test/util.py,sha256=98HXLClkycNTxLiuy1d3W_tM6dBU9bA-p5ZXMfncaWE,2754
+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
@@ -65,15 +35,15 @@ locust/util/exception_handler.py,sha256=jTMyBq2a0O07fRjmqGkheyaPj58tUgnbbcjoesKG
 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/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/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-2.29.2.dev34.dist-info/LICENSE,sha256=78XGpIn3fHVBfaxlPNUfjVufSN7QsdhpJMRJHv2AFpo,1095
-locust-2.29.2.dev34.dist-info/METADATA,sha256=JxBhCMRud-_bXYMsQ-Bbo6Gs9yXAEKQCdfo40CmNP1Q,7390
-locust-2.29.2.dev34.dist-info/WHEEL,sha256=y4mX-SOX4fYIkonsAGA5N0Oy-8_gI4FXw5HNI1xqvWg,91
-locust-2.29.2.dev34.dist-info/entry_points.txt,sha256=RAdt8Ku-56m7bFjmdj-MBhbF6h4NX7tVODR9QNnOg0E,44
-locust-2.29.2.dev34.dist-info/top_level.txt,sha256=XSsjgPA8Ggf9TqKVbkwSqZFuPlZ085X13M9orDycE20,7
-locust-2.29.2.dev34.dist-info/RECORD,,
+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...
-        )