add option to generate random body and specify headers

Author: bisgaard-itis

tests/performance/locustfiles/deployment_max_rps_single_endpoint.py

@@ -12,6 +12,7 @@
 import json
 from collections.abc import Callable
 
+import jsf
 from common.base_user import OsparcWebUserBase
 from locust import events, task
 from locust.argument_parser import LocustArgumentParser
@@ -36,7 +37,19 @@ def _(parser: LocustArgumentParser) -> None:
         "--body",
         type=str,
         default="",
-        help="The optional HTTP body as json string",
+        help="Optional HTTP body as json string",
+    )
+    parser.add_argument(
+        "--body-json-schema",
+        type=str,
+        default="",
+        help="Optional JSON schema for the request body. If specified, the request data will be randomly generated from this schema.",
+    )
+    parser.add_argument(
+        "--headers",
+        type=str,
+        default="",
+        help="Optional HTTP headers as json string",
     )
 
 
@@ -52,4 +65,11 @@ def get_endpoint(self) -> None:
         kwargs = {}
         if len(self.environment.parsed_options.body) > 0:
             kwargs["json"] = json.loads(self.environment.parsed_options.body)
+        if len(self.environment.parsed_options.body_json_schema) > 0:
+            faker = jsf.JSF(
+                json.loads(self.environment.parsed_options.body_json_schema)
+            )
+            kwargs["json"] = faker.generate()
+        if len(self.environment.parsed_options.headers) > 0:
+            kwargs["headers"] = json.loads(self.environment.parsed_options.headers)
         method(self.environment.parsed_options.endpoint, **kwargs)