locust 2.24.2.dev11__tar.gz → 2.24.2.dev17__tar.gz

{locust-2.24.2.dev11 → locust-2.24.2.dev17}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: locust
-Version: 2.24.2.dev11
+Version: 2.24.2.dev17
 Summary: Developer friendly load testing framework
 License: MIT
 Project-URL: Homepage, https://github.com/locustio/locust

{locust-2.24.2.dev11 → locust-2.24.2.dev17}/docs/custom-load-shape.rst

@@ -101,7 +101,7 @@ This shape would create create in the first 10 seconds 10 User of ``UserA``. In
 Reusing command line parameters in custom shapes
 ------------------------------------------------
 
-By default, using a custom shape will disable default run paramaters (in both the CLI and the Web UI):
+By default, using a custom shape will disable default run parameters (in both the CLI and the Web UI):
 - `--run-time` (providing this one with a custom shape will make locust to bail out)
 - `--spawn-rate`
 - `--users`

{locust-2.24.2.dev11 → locust-2.24.2.dev17}/locust/_version.py

@@ -12,5 +12,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '2.24.2.dev11'
-__version_tuple__ = version_tuple = (2, 24, 2, 'dev11')
+__version__ = version = '2.24.2.dev17'
+__version_tuple__ = version_tuple = (2, 24, 2, 'dev17')

{locust-2.24.2.dev11 → locust-2.24.2.dev17}/locust/env.py

@@ -198,6 +198,9 @@ class Environment:
         return self.web_ui
 
     def update_user_class(self, user_settings):
+        if isinstance(self.runner, MasterRunner):
+            self.runner.send_message("update_user_class", user_settings)
+
         user_class_name = user_settings.get("user_class_name")
         user_class = self.available_user_classes[user_class_name]
         user_tasks = self.available_user_tasks[user_class_name]

{locust-2.24.2.dev11 → locust-2.24.2.dev17}/locust/main.py

@@ -394,7 +394,7 @@ See https://github.com/locustio/locust/wiki/Installation#increasing-maximum-numb
 
                     environment.update_user_class(user_config)
             except Exception as e:
-                logger.error(f"The --config-users arugment must be in valid JSON string or file: {e}")
+                logger.error(f"The --config-users argument must be in valid JSON string or file: {e}")
                 sys.exit(-1)
 
     if (

{locust-2.24.2.dev11 → locust-2.24.2.dev17}/locust/runners.py

@@ -1395,6 +1395,8 @@ class WorkerRunner(DistributedRunner):
                 self.reset_connection()
             elif msg.type == "heartbeat":
                 self.last_heartbeat_timestamp = time.time()
+            elif msg.type == "update_user_class":
+                self.environment.update_user_class(msg.data)
             elif msg.type in self.custom_messages:
                 logger.debug("Received %s message from master" % msg.type)
                 listener, concurrent = self.custom_messages[msg.type]

{locust-2.24.2.dev11 → locust-2.24.2.dev17}/locust/test/test_env.py

@@ -241,3 +241,43 @@ class TestEnvironment(LocustTestCase):
             environment.available_user_classes["User1"].json(),
             {"host": "http://localhost", "tasks": ["my_task_2"], "fixed_count": 0, "weight": 1},
         )
+
+    def test_distributed_update_user_class(self):
+        class MyUser1(User):
+            @task
+            def my_task(self):
+                pass
+
+            @task
+            def my_task_2(self):
+                pass
+
+        class MyUser2(User):
+            @task
+            def my_task(self):
+                pass
+
+        master_env = Environment(
+            user_classes=[MyUser1, MyUser2],
+            available_user_classes={"User1": MyUser1, "User2": MyUser2},
+            available_user_tasks={"User1": MyUser1.tasks, "User2": MyUser2.tasks},
+        )
+        master = master_env.create_master_runner("*", 0)
+
+        worker_env = Environment(
+            user_classes=[MyUser1, MyUser2],
+            available_user_classes={"User1": MyUser1, "User2": MyUser2},
+            available_user_tasks={"User1": MyUser1.tasks, "User2": MyUser2.tasks},
+        )
+        worker = worker_env.create_worker_runner("127.0.0.1", master.server.port)
+
+        master_env.update_user_class({"user_class_name": "User1", "host": "http://localhost", "tasks": ["my_task_2"]})
+
+        self.assertEqual(
+            master_env.available_user_classes["User1"].json(),
+            {"host": "http://localhost", "tasks": ["my_task_2"], "fixed_count": 0, "weight": 1},
+        )
+        self.assertEqual(
+            worker_env.available_user_classes["User1"].json(),
+            {"host": "http://localhost", "tasks": ["my_task_2"], "fixed_count": 0, "weight": 1},
+        )

{locust-2.24.2.dev11 → locust-2.24.2.dev17}/locust/test/test_web.py

@@ -1351,3 +1351,32 @@ class TestModernWebUI(LocustTestCase, _HeaderCheckMixin):
         self.assertTrue(d("#root"))
         self.assertIn('"locustfile": "locust.py"', str(d))
         self.assertIn('"host": "http://localhost"', str(d))
+
+    def test_update_user(self):
+        class MyUser1(User):
+            @task
+            def my_task(self):
+                pass
+
+            @task
+            def my_task_2(self):
+                pass
+
+        class MyUser2(User):
+            @task
+            def my_task(self):
+                pass
+
+        self.environment.user_classes = [MyUser1, MyUser2]
+        self.environment.available_user_classes = {"User1": MyUser1, "User2": MyUser2}
+        self.environment.available_user_tasks = {"User1": MyUser1.tasks, "User2": MyUser2.tasks}
+
+        requests.post(
+            "http://127.0.0.1:%i/user" % self.web_port,
+            json={"user_class_name": "User1", "host": "http://localhost", "tasks": ["my_task_2"]},
+        )
+
+        self.assertEqual(
+            self.environment.available_user_classes["User1"].json(),
+            {"host": "http://localhost", "tasks": ["my_task_2"], "fixed_count": 0, "weight": 1},
+        )

{locust-2.24.2.dev11 → locust-2.24.2.dev17}/locust/webui/src/hooks/useSortByField.ts

@@ -55,7 +55,7 @@ export default function useSortByField<Row>(
 
       if (sortField === currentSortField.current) {
         if (shouldReverse) {
-          // reset to intial state on 3rd click
+          // reset to initial state on 3rd click
           setShouldReverse(false);
           currentSortField.current = undefined;
           sortStats(defaultSortKey);

{locust-2.24.2.dev11 → locust-2.24.2.dev17}/locust.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: locust
-Version: 2.24.2.dev11
+Version: 2.24.2.dev17
 Summary: Developer friendly load testing framework
 License: MIT
 Project-URL: Homepage, https://github.com/locustio/locust