locust 2.25.0__tar.gz → 2.25.1.dev17__tar.gz

{locust-2.25.0 → locust-2.25.1.dev17}/.github/workflows/tests.yml

@@ -24,7 +24,6 @@ jobs:
           - { name: "3.11", python: "3.11", os: ubuntu-latest, tox: py311 }
           - { name: "3.10", python: "3.10", os: ubuntu-latest, tox: py310 }
           - { name: "3.9", python: "3.9", os: ubuntu-latest, tox: py39 }
-          - { name: "3.8", python: "3.8", os: ubuntu-latest, tox: py38 }
 
     steps:
       - uses: actions/checkout@v3

locust-2.25.1.dev17/.pre-commit-config.yaml

@@ -0,0 +1,10 @@
+repos:
+  - repo: https://github.com/astral-sh/ruff-pre-commit
+    # Ruff version.
+    rev: v0.3.7
+    hooks:
+      # Run the linter.
+      - id: ruff
+        args: [--fix]
+      # Run the formatter.
+      - id: ruff-format

{locust-2.25.0 → locust-2.25.1.dev17}/CHANGELOG.md

@@ -1,6 +1,33 @@
 # Detailed changelog
 The most important changes can also be found in [the documentation](https://docs.locust.io/en/latest/changelog.html).
 
+## [2.25.0](https://github.com/locustio/locust/tree/2.25.0) (2024-04-14)
+
+[Full Changelog](https://github.com/locustio/locust/compare/2.24.1...2.25.0)
+
+**Fixed bugs:**
+
+- Wrong distribution of virtual users by user classes [\#2662](https://github.com/locustio/locust/issues/2662)
+- Monkey patching error [\#2661](https://github.com/locustio/locust/issues/2661)
+- cannot specify task of UserClass when locust run on distributed mode [\#2657](https://github.com/locustio/locust/issues/2657)
+- \[ModernUI\] HTML entities are incorrectly escaped in generated report [\#2635](https://github.com/locustio/locust/issues/2635)
+- Bug / Feature request: Time intensive `custom_messages` functions trigger heartbeat timeout [\#2608](https://github.com/locustio/locust/issues/2608)
+
+**Closed issues:**
+
+- Add an UI extension exemple that display proxy cache hit/miss [\#2644](https://github.com/locustio/locust/issues/2644)
+
+**Merged pull requests:**
+
+- Log deprecation warning for --legacy-ui [\#2670](https://github.com/locustio/locust/pull/2670) ([cyberw](https://github.com/cyberw))
+- Add an example that displays cache stats using the new UI. [\#2669](https://github.com/locustio/locust/pull/2669) ([obriat](https://github.com/obriat))
+- Update User Classes in Distributed Mode [\#2666](https://github.com/locustio/locust/pull/2666) ([andrewbaldwin44](https://github.com/andrewbaldwin44))
+- fix\(dispatch\): UserClasses weight distribution with gcd [\#2663](https://github.com/locustio/locust/pull/2663) ([llirrikk](https://github.com/llirrikk))
+- chore: fix some comments [\#2659](https://github.com/locustio/locust/pull/2659) ([JohnEndson](https://github.com/JohnEndson))
+- Fix issue with concurrent custom messages and docs [\#2654](https://github.com/locustio/locust/pull/2654) ([cyberw](https://github.com/cyberw))
+- Add functionality to run listener functions for `custom_messages` concurrently [\#2650](https://github.com/locustio/locust/pull/2650) ([samuelspagl](https://github.com/samuelspagl))
+- \[Modern UI\] Escape HTML Entities correctly in Exceptions Table [\#2648](https://github.com/locustio/locust/pull/2648) ([andrewbaldwin44](https://github.com/andrewbaldwin44))
+
 ## [2.24.1](https://github.com/locustio/locust/tree/2.24.1) (2024-03-20)
 
 [Full Changelog](https://github.com/locustio/locust/compare/2.24.0...2.24.1)
@@ -857,8 +884,8 @@ The most important changes can also be found in [the documentation](https://docs
 **Merged pull requests:**
 
 - Fire requests through Environment.events [\#1982](https://github.com/locustio/locust/pull/1982) ([BonelessPi](https://github.com/BonelessPi))
-- Fix docs with underscore postfix for hyperlink? [\#1979](https://github.com/locustio/locust/pull/1979) ([jeroenhendricksen](https://github.com/jeroenhendricksen))
-- Repair broken hyperlinks in documentation [\#1978](https://github.com/locustio/locust/pull/1978) ([jeroenhendricksen](https://github.com/jeroenhendricksen))
+- Fix docs with underscore postfix for hyperlink? [\#1979](https://github.com/locustio/locust/pull/1979) ([jeroenhe](https://github.com/jeroenhe))
+- Repair broken hyperlinks in documentation [\#1978](https://github.com/locustio/locust/pull/1978) ([jeroenhe](https://github.com/jeroenhe))
 
 ## [2.6.0](https://github.com/locustio/locust/tree/2.6.0) (2022-01-23)
 

{locust-2.25.0/locust.egg-info → locust-2.25.1.dev17}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: locust
-Version: 2.25.0
+Version: 2.25.1.dev17
 Summary: Developer friendly load testing framework
 License: MIT
 Project-URL: Homepage, https://github.com/locustio/locust
@@ -14,7 +14,6 @@ Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: OS Independent
 Classifier: Programming Language :: Python
 Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.8
 Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
@@ -24,7 +23,7 @@ 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.8
+Requires-Python: >=3.9
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: gevent>=22.10.2
@@ -33,7 +32,7 @@ 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.0.11
+Requires-Dist: geventhttpclient>=2.2.1
 Requires-Dist: ConfigArgParse>=1.5.5
 Requires-Dist: tomli>=1.1.0; python_version < "3.11"
 Requires-Dist: psutil>=5.9.1
@@ -45,7 +44,7 @@ Requires-Dist: pywin32; platform_system == "Windows"
 # Locust
 
 [![PyPI](https://img.shields.io/pypi/v/locust.svg)](https://pypi.org/project/locust/)
-[![PyPI](https://img.shields.io/pypi/pyversions/locust.svg)](https://pypi.org/project/locust/)
+![Python Version from PEP 621 TOML](https://img.shields.io/python/required-version-toml?tomlFilePath=https%3A%2F%2Fraw.githubusercontent.com%2Flocustio%2Flocust%2Fmaster%2Fpyproject.toml&link=https%3A%2F%2Fpypi.org%2Fproject%2Flocust)
 [![Downloads](https://pepy.tech/badge/locust/week)](https://pepy.tech/project/locust)
 [![Build Status](https://github.com/locustio/locust/workflows/Tests/badge.svg)](https://github.com/locustio/locust/actions?query=workflow%3ATests)
 [![GitHub contributors](https://img.shields.io/github/contributors/locustio/locust.svg)](https://github.com/locustio/locust/graphs/contributors)

{locust-2.25.0 → locust-2.25.1.dev17}/README.md

@@ -1,7 +1,7 @@
 # Locust
 
 [![PyPI](https://img.shields.io/pypi/v/locust.svg)](https://pypi.org/project/locust/)
-[![PyPI](https://img.shields.io/pypi/pyversions/locust.svg)](https://pypi.org/project/locust/)
+![Python Version from PEP 621 TOML](https://img.shields.io/python/required-version-toml?tomlFilePath=https%3A%2F%2Fraw.githubusercontent.com%2Flocustio%2Flocust%2Fmaster%2Fpyproject.toml&link=https%3A%2F%2Fpypi.org%2Fproject%2Flocust)
 [![Downloads](https://pepy.tech/badge/locust/week)](https://pepy.tech/project/locust)
 [![Build Status](https://github.com/locustio/locust/workflows/Tests/badge.svg)](https://github.com/locustio/locust/actions?query=workflow%3ATests)
 [![GitHub contributors](https://img.shields.io/github/contributors/locustio/locust.svg)](https://github.com/locustio/locust/graphs/contributors)

{locust-2.25.0 → locust-2.25.1.dev17}/docs/developing-locust.rst

@@ -20,7 +20,9 @@ Now the ``locust`` command will run *your* code with no need for reinstalling af
 
 To contribute your changes, push to a branch in your repo and then `open a PR on github <https://github.com/locustio/locust/compare>`_. 
 
-Before you open a pull request, make sure all the checks work. And if you are adding a feature, make sure it is documented (in ``docs/*.rst``).
+If you install `pre-commit <https://pre-commit.com/>`_ linting and format checks/fixes will be automatically performed saving you a round trip to GitHub actions.
+
+Before you open a pull request, make sure all the tests work. And if you are adding a feature, make sure it is documented (in ``docs/*.rst``).
 
 Testing your changes
 ====================

{locust-2.25.0 → locust-2.25.1.dev17}/docs/installation.rst

@@ -3,7 +3,7 @@
 Installation
 ============
 
-0. `Install Python <https://docs.python-guide.org/starting/installation/>`_ (3.8 or later)
+0. `Install Python <https://docs.python-guide.org/starting/installation/>`_ (3.9 or later)
 
 1. Install the package (check `the wiki <https://github.com/locustio/locust/wiki/Installation>`_ if the installation fails)
 

{locust-2.25.0 → locust-2.25.1.dev17}/examples/rest.py

@@ -2,8 +2,8 @@ from locust import FastHttpUser, run_single_user, task
 from locust.contrib.fasthttp import RestResponseContextManager
 from locust.user.wait_time import constant
 
+from collections.abc import Generator
 from contextlib import contextmanager
-from typing import Generator
 
 
 class MyUser(FastHttpUser):
@@ -40,12 +40,10 @@ class MyUser(FastHttpUser):
             # use a trailing comma to append the response text to the custom message
             assert resp.js["data"]["foo"] == 2, "my custom error message with response text,"
 
-        # this only works in python 3.8 and up, so it is commented out:
-        # if sys.version_info >= (3, 8):
-        #     with self.rest("", "/post", json={"foo": 1}) as resp:
-        #         # assign and assert in one line
-        #         assert (foo := resp.js["foo"])
-        #         print(f"the number {foo} is awesome")
+        with self.rest("", "/post", json={"foo": 1}) as resp:
+            # assign and assert in one line
+            assert (foo := resp.js["foo"])
+            print(f"the number {foo} is awesome")
 
         # rest() catches most exceptions, so any programming mistakes you make automatically marks the request as a failure
         # and stores the callstack in the failure message

{locust-2.25.0 → locust-2.25.1.dev17}/locust/_version.py

@@ -12,5 +12,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '2.25.0'
-__version_tuple__ = version_tuple = (2, 25, 0)
+__version__ = version = '2.25.1.dev17'
+__version_tuple__ = version_tuple = (2, 25, 1, 'dev17')

{locust-2.25.0 → locust-2.25.1.dev17}/locust/argument_parser.py

@@ -80,8 +80,7 @@ class LocustTomlConfigParser(configargparse.TomlConfigParser):
         result = OrderedDict()
 
         for section in self.sections:
-            data = configargparse.get_toml_section(config, section)
-            if data:
+            if data := configargparse.get_toml_section(config, section):
                 for key, value in data.items():
                     if isinstance(value, list):
                         result[key] = value

{locust-2.25.0 → locust-2.25.1.dev17}/locust/clients.py

@@ -2,8 +2,8 @@ from __future__ import annotations
 
 import re
 import time
+from collections.abc import Generator
 from contextlib import contextmanager
-from typing import Generator
 from urllib.parse import urlparse, urlunparse
 
 import requests

{locust-2.25.0 → locust-2.25.1.dev17}/locust/contrib/fasthttp.py

@@ -12,11 +12,12 @@ import socket
 import time
 import traceback
 from base64 import b64encode
+from collections.abc import Generator
 from contextlib import contextmanager
 from http.cookiejar import CookieJar
 from json.decoder import JSONDecodeError
 from ssl import SSLError
-from typing import Any, Callable, Generator, cast
+from typing import Any, Callable, cast
 from urllib.parse import urlparse, urlunparse
 
 import gevent
@@ -306,11 +307,11 @@ class FastHttpUser(User):
     connection_timeout: float = 60.0
     """Parameter passed to FastHttpSession"""
 
-    max_redirects: int = 5
-    """Parameter passed to FastHttpSession. Default 5, meaning 4 redirects."""
+    max_redirects: int = 30
+    """Parameter passed to FastHttpSession."""
 
-    max_retries: int = 1
-    """Parameter passed to FastHttpSession. Default 1, meaning zero retries."""
+    max_retries: int = 0
+    """Parameter passed to FastHttpSession."""
 
     insecure: bool = True
     """Parameter passed to FastHttpSession. Default True, meaning no SSL verification."""
@@ -404,8 +405,7 @@ class FastHttpUser(User):
             except Exception as e:
                 error_lines = []
                 for l in traceback.format_exc().split("\n"):
-                    m = self._callstack_regex.match(l)
-                    if m:
+                    if m := self._callstack_regex.match(l):
                         filename = re.sub(r"/(home|Users/\w*)/", "~/", m.group(1))
                         error_lines.append(filename + ":" + m.group(2) + m.group(3))
                     short_resp = resp.text[:200] if resp.text else resp.text

{locust-2.25.0 → locust-2.25.1.dev17}/locust/dispatch.py

@@ -1,14 +1,13 @@
 from __future__ import annotations
 
 import contextlib
-import functools
 import itertools
 import math
-import sys
 import time
 from collections import defaultdict
+from collections.abc import Generator, Iterator
 from operator import attrgetter
-from typing import TYPE_CHECKING, Generator, Iterator
+from typing import TYPE_CHECKING
 
 import gevent
 from roundrobin import smooth
@@ -18,20 +17,6 @@ if TYPE_CHECKING:
     from locust.runners import WorkerNode
 
 
-def compatible_math_gcd(*args: int) -> int:
-    """
-    This function is a workaround for the fact that `math.gcd` in:
-        - 3.5 <= Python < 3.9   doesn't accept more than two arguments.
-        - 3.9 <= Python         can accept more than two arguments.
-    See more at https://docs.python.org/3.9/library/math.html#math.gcd
-    """
-    if (3, 5) <= sys.version_info < (3, 9):
-        return functools.reduce(math.gcd, args)
-    elif sys.version_info >= (3, 9):
-        return math.gcd(*args)
-    raise NotImplementedError("This function is only implemented for Python from 3.5")
-
-
 # To profile line-by-line, uncomment the code below (i.e. `import line_profiler ...`) and
 # place `@profile` on the functions/methods you wish to profile. Then, in the unit test you are
 # running, use `from locust.dispatch import profile; profile.print_stats()` at the end of the unit test.
@@ -406,7 +391,7 @@ class UsersDispatcher(Iterator):
             max_order_of_magnitude = _get_order_of_magnitude(min(abs(u[1]) for u in users))
             weights = tuple(int(u[1] * max_order_of_magnitude) for u in users)
 
-            greatest_common_divisor = compatible_math_gcd(*weights)
+            greatest_common_divisor = math.gcd(*weights)
             normalized_values = [
                 (
                     user[0].__name__,

{locust-2.25.0 → locust-2.25.1.dev17}/locust/event.py

@@ -3,8 +3,9 @@ from __future__ import annotations
 import logging
 import time
 import traceback
+from collections.abc import Generator
 from contextlib import contextmanager
-from typing import Any, Generator
+from typing import Any
 
 from . import log
 from .exception import InterruptTaskSet, RescheduleTask, RescheduleTaskImmediately, StopUser

{locust-2.25.0 → locust-2.25.1.dev17}/locust/html.py

@@ -41,8 +41,7 @@ def get_html_report(
     start_ts = stats.start_time
     start_time = datetime.datetime.utcfromtimestamp(start_ts).strftime("%Y-%m-%d %H:%M:%S")
 
-    end_ts = stats.last_request_timestamp
-    if end_ts:
+    if end_ts := stats.last_request_timestamp:
         end_time = datetime.datetime.utcfromtimestamp(end_ts).strftime("%Y-%m-%d %H:%M:%S")
     else:
         end_time = start_time

{locust-2.25.0 → locust-2.25.1.dev17}/locust/input_events.py

@@ -104,8 +104,7 @@ def input_listener(key_to_func_map: dict[str, Callable]):
 
         try:
             while True:
-                input = poller.poll()
-                if input:
+                if input := poller.poll():
                     for key in key_to_func_map:
                         if input == key:
                             key_to_func_map[key]()

{locust-2.25.0 → locust-2.25.1.dev17}/locust/main.py

@@ -216,8 +216,7 @@ def main():
         gc.collect()  # avoid freezing garbage
         gc.freeze()  # move all objects to perm gen so ref counts dont get updated
         for _ in range(options.processes):
-            child_pid = gevent.fork()
-            if child_pid:
+            if child_pid := gevent.fork():
                 children.append(child_pid)
                 logging.debug(f"Started child worker with pid #{child_pid}")
             else:
@@ -250,12 +249,8 @@ def main():
                 # nothing more to do, just wait for the children to exit
                 for child_pid in children:
                     _, child_status = os.waitpid(child_pid, 0)
-                    try:
-                        if sys.version_info >= (3, 9):
-                            child_exit_code = os.waitstatus_to_exitcode(child_status)
-                            exit_code = max(exit_code, child_exit_code)
-                    except AttributeError:
-                        pass  # dammit python 3.8...
+                    child_exit_code = os.waitstatus_to_exitcode(child_status)
+                    exit_code = max(exit_code, child_exit_code)
                 sys.exit(exit_code)
             else:
                 options.master = True
@@ -270,12 +265,8 @@ def main():
                             try:
                                 _, child_status = os.waitpid(child_pid, os.WNOHANG)
                                 children.remove(child_pid)
-                                try:
-                                    if sys.version_info >= (3, 9):
-                                        child_exit_code = os.waitstatus_to_exitcode(child_status)
-                                        exit_code = max(exit_code, child_exit_code)
-                                except AttributeError:
-                                    pass  # dammit python 3.8...
+                                child_exit_code = os.waitstatus_to_exitcode(child_status)
+                                exit_code = max(exit_code, child_exit_code)
                             except OSError as e:
                                 if e.errno == errno.EINTR:
                                     time.sleep(0.1)
@@ -291,12 +282,8 @@ def main():
                             pass  # never mind, process was already dead
                     for child_pid in children:
                         _, child_status = os.waitpid(child_pid, 0)
-                        try:
-                            if sys.version_info >= (3, 9):
-                                child_exit_code = os.waitstatus_to_exitcode(child_status)
-                                exit_code = max(exit_code, child_exit_code)
-                        except AttributeError:
-                            pass  # dammit python 3.8...
+                        child_exit_code = os.waitstatus_to_exitcode(child_status)
+                        exit_code = max(exit_code, child_exit_code)
                     if exit_code > 1:
                         logging.error(f"Bad response code from worker children: {exit_code}")
                     # ensure master doesnt finish until output from workers has arrived
@@ -327,8 +314,7 @@ def main():
 
     # make sure specified User exists
     if options.user_classes:
-        missing = set(options.user_classes) - set(user_classes.keys())
-        if missing:
+        if missing := set(options.user_classes) - set(user_classes.keys()):
             logger.error(f"Unknown User(s): {', '.join(missing)}\n")
             sys.exit(1)
         else:
@@ -356,9 +342,6 @@ It's not high enough for load testing, and the OS didn't allow locust to increas
 See https://github.com/locustio/locust/wiki/Installation#increasing-maximum-number-of-open-files-limit for more info."""
             )
 
-    if sys.version_info < (3, 9):
-        logger.warning("Python 3.8 support is deprecated and will be removed soon")
-
     # create locust Environment
     locustfile_path = None if not locustfile else os.path.basename(locustfile)
 

{locust-2.25.0 → locust-2.25.1.dev17}/locust/runners.py

@@ -14,7 +14,7 @@ import time
 import traceback
 from abc import abstractmethod
 from collections import defaultdict
-from collections.abc import MutableMapping
+from collections.abc import Iterator, MutableMapping, ValuesView
 from operator import (
     itemgetter,
     methodcaller,
@@ -24,10 +24,8 @@ from typing import (
     TYPE_CHECKING,
     Any,
     Callable,
-    Iterator,
     NoReturn,
     TypedDict,
-    ValuesView,
     cast,
 )
 from uuid import uuid4

{locust-2.25.0 → locust-2.25.1.dev17}/locust/stats.py

@@ -10,6 +10,10 @@ 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
@@ -19,16 +23,12 @@ from typing import (
     TYPE_CHECKING,
     Any,
     Callable,
-    Iterable,
     NoReturn,
     Protocol,
     TypedDict,
     TypeVar,
     cast,
 )
-from typing import (
-    OrderedDict as OrderedDictType,
-)
 
 import gevent
 
@@ -177,8 +177,7 @@ def diff_response_time_dicts(latest: dict[int, int], old: dict[int, int]) -> dic
     """
     new = {}
     for t in latest:
-        diff = latest[t] - old.get(t, 0)
-        if diff:
+        if diff := latest[t] - old.get(t, 0):
             new[t] = diff
     return new
 

{locust-2.25.0 → locust-2.25.1.dev17}/locust/test/mock_logging.py

@@ -5,7 +5,7 @@ import logging
 from typing import List, Union, Dict
 from types import TracebackType
 
-LogMessage = List[Union[str, Dict[str, TracebackType]]]
+LogMessage = list[Union[str, dict[str, TracebackType]]]
 
 
 class MockedLoggingHandler(logging.Handler):

{locust-2.25.0 → locust-2.25.1.dev17}/locust/test/test_debugging.py

@@ -27,8 +27,7 @@ class TestDebugging(DebugTestCase):
                 pass
 
         def _stop_user():
-            user = getattr(debug._env, "single_user_instance", None)
-            if user:
+            if user := getattr(debug._env, "single_user_instance", None):
                 user._state = LOCUST_STATE_STOPPING
 
         t = Timer(1, _stop_user)

{locust-2.25.0 → locust-2.25.1.dev17}/locust/test/test_fasthttp.py

@@ -506,7 +506,7 @@ class TestFastHttpUserClass(WebserverTestCase):
 
     def test_max_redirect_setting(self):
         class MyUser(FastHttpUser):
-            max_redirects = 1  # max_redirects and max_retries are funny names, because they are actually max attempts
+            max_redirects = 0
             host = "http://127.0.0.1:%i" % self.port
 
         l = MyUser(self.environment)

{locust-2.25.0 → locust-2.25.1.dev17}/locust/test/test_interruptable_task.py

@@ -3,12 +3,11 @@ from locust.env import Environment
 from locust.exception import StopUser
 
 from collections import defaultdict
-from typing import DefaultDict
 from unittest import TestCase
 
 
 class InterruptableTaskSet(SequentialTaskSet):
-    counter: DefaultDict[str, int] = defaultdict(int)
+    counter: defaultdict[str, int] = defaultdict(int)
 
     def on_start(self):
         super().on_start()

{locust-2.25.0 → locust-2.25.1.dev17}/locust/test/test_main.py

@@ -483,7 +483,6 @@ class StandaloneIntegrationTests(ProcessIntegrationTest):
                     self.assertIn("Shutting down (exit code 0)", stderr)
                     self.assertEqual(0, proc.returncode)
 
-    @unittest.skipIf(sys.version_info < (3, 9), reason="dies in 3.8 on GH and I cant be bothered to investigate it")
     def test_default_headless_spawn_options_with_shape(self):
         content = MOCK_LOCUSTFILE_CONTENT + textwrap.dedent(
             """
@@ -1618,9 +1617,10 @@ class SecondUser(HttpUser):
                     pass
             """
         )
-        with mock_locustfile(content=LOCUSTFILE_CONTENT) as mocked, patch_env(
-            "LOCUST_WAIT_FOR_WORKERS_REPORT_AFTER_RAMP_UP", "0.01"
-        ) as _:
+        with (
+            mock_locustfile(content=LOCUSTFILE_CONTENT) as mocked,
+            patch_env("LOCUST_WAIT_FOR_WORKERS_REPORT_AFTER_RAMP_UP", "0.01") as _,
+        ):
             proc = subprocess.Popen(
                 [
                     "locust",
@@ -2225,7 +2225,6 @@ class AnyUser(User):
 
             self.assertNotIn("Traceback", stderr)
             self.assertIn("INFO/locust.runners: sys.exit(42) called", stderr)
-            if sys.version_info >= (3, 9):
-                self.assertEqual(status_code, 42)
+            self.assertEqual(status_code, 42)
             self.assertNotIn("Traceback", master_stderr)
             self.assertIn("failed to send heartbeat, setting state to missing", master_stderr)