Overwrite weight by config user before environment weight validation

locust/env.py

@@ -37,6 +37,7 @@ def __init__(
         available_user_classes: dict[str, User] | None = None,
         available_shape_classes: dict[str, LoadTestShape] | None = None,
         available_user_tasks: dict[str, list[TaskSet | Callable]] | None = None,
+        override_user_weights: list[dict] | None = None,
         dispatcher_class: type[UsersDispatcher] = UsersDispatcher,
     ):
         self.runner: Runner | None = None
@@ -105,6 +106,10 @@ def __init__(
         self.worker_logs: dict[str, list[str]] = {}
         """Captured logs from all connected workers"""
 
+        if override_user_weights:
+            for user_dict in override_user_weights:
+                self.update_user_class(user_dict)
+
         self._remove_user_classes_with_weight_zero()
         self._validate_user_class_name_uniqueness()
         self._validate_shape_class_instance()

locust/main.py

@@ -73,6 +73,7 @@ def create_environment(
     available_user_classes=None,
     available_shape_classes=None,
     available_user_tasks=None,
+    override_user_weights=None,
 ):
     """
     Create an Environment instance from options
@@ -89,6 +90,7 @@ def create_environment(
         available_user_classes=available_user_classes,
         available_shape_classes=available_shape_classes,
         available_user_tasks=available_user_tasks,
+        override_user_weights=override_user_weights,
     )
 
 
@@ -366,21 +368,7 @@ def kill_workers(children):
 See https://github.com/locustio/locust/wiki/Installation#increasing-maximum-number-of-open-files-limit for more info."""
             )
 
-    # create locust Environment
-    locustfile_path = None if not locustfile else os.path.basename(locustfile)
-
-    environment = create_environment(
-        user_classes,
-        options,
-        events=locust.events,
-        shape_class=shape_class,
-        locustfile=locustfile_path,
-        parsed_locustfiles=locustfiles,
-        available_user_classes=available_user_classes,
-        available_shape_classes=available_shape_classes,
-        available_user_tasks=available_user_tasks,
-    )
-
+    user_new_weights: list[dict] = []
     if options.config_users:
         for json_user_config in options.config_users:
             try:
@@ -398,12 +386,10 @@ def ensure_user_class_name(config):
                 if isinstance(user_config, list):
                     for config in user_config:
                         ensure_user_class_name(config)
-
-                        environment.update_user_class(config)
+                        user_new_weights.append(config)
                 else:
                     ensure_user_class_name(user_config)
-
-                    environment.update_user_class(user_config)
+                    user_new_weights.append(user_config)
             except json.decoder.JSONDecodeError as e:
                 logger.error(f"The --config-users argument must be in valid JSON string or file: {e}")
                 sys.exit(-1)
@@ -416,6 +402,22 @@ def ensure_user_class_name(config):
                 logger.exception(e)
                 sys.exit(-1)
 
+    # create locust Environment
+    locustfile_path = None if not locustfile else os.path.basename(locustfile)
+
+    environment = create_environment(
+        user_classes,
+        options,
+        events=locust.events,
+        shape_class=shape_class,
+        locustfile=locustfile_path,
+        parsed_locustfiles=locustfiles,
+        available_user_classes=available_user_classes,
+        available_shape_classes=available_shape_classes,
+        available_user_tasks=available_user_tasks,
+        override_user_weights=user_new_weights,
+    )
+
     if (
         shape_class
         and not shape_class.use_common_options