config: Make run id header optional

Author: shahar99s

httprunner/config.py

@@ -94,6 +94,7 @@ def __init__(self, name: Text) -> None:
         self.__base_url: Text = ""
         self.__variables: VariablesMapping = {}
         self.__config = TConfig(name=name, path=caller_frame.filename)
+        self.__add_request_id = True
 
     @property
     def name(self) -> Text:
@@ -115,6 +116,10 @@ def verify(self, verify: bool) -> "Config":
         self.__config.verify = verify
         return self
 
+    def add_request_id(self, add_request_id: bool) -> "Config":
+        self.__config.add_request_id = add_request_id
+        return self
+
     def export(self, *export_var_name: Text) -> "Config":
         self.__config.export.extend(export_var_name)
         self.__config.export = list(set(self.__config.export))

httprunner/models.py

@@ -118,6 +118,7 @@ class TConfig(BaseModel):
     name: Name
     verify: Verify = False
     base_url: BaseUrl = ""
+    add_request_id: bool = True
     # Text: prepare variables in debugtalk.py, ${gen_variables()}
     variables: Union[VariablesMapping, Text] = {}
     parameters: Union[VariablesMapping, Text] = {}

httprunner/step_request.py

@@ -102,9 +102,10 @@ def run_step_request(runner: HttpRunner, step: TStep) -> StepResult:
     request_headers = {
         key: request_headers[key] for key in request_headers if not key.startswith(":")
     }
-    request_headers[
-        "HRUN-Request-ID"
-    ] = f"HRUN-{runner.case_id}-{str(int(time.time() * 1000))[-6:]}"
+    if runner.config.add_request_id:
+        request_headers[
+            "HRUN-Request-ID"
+        ] = f"HRUN-{runner.case_id}-{str(int(time.time() * 1000))[-6:]}"
     parsed_request_dict["headers"] = request_headers
 
     step_variables["request"] = parsed_request_dict

httprunner/step_sql_request.py

@@ -85,10 +85,11 @@ def run_step_sql_request(runner: HttpRunner, step: TStep) -> StepResult:
             f"?charset=utf8mb4"
         )
 
-    # parsed_request_dict["headers"].setdefault(
-    #     "HRUN-Request-ID",
-    #     f"HRUN-{self.__case_id}-{str(int(time.time() * 1000))[-6:]}",
-    # )
+    # if runner.config.add_request_id:
+    #   parsed_request_dict["headers"].setdefault(
+    #       "HRUN-Request-ID",
+    #       f"HRUN-{self.__case_id}-{str(int(time.time() * 1000))[-6:]}",
+    #   )
 
     # setup hooks
     if step.setup_hooks:

httprunner/step_thrift_request.py

@@ -95,10 +95,11 @@ def run_step_thrift_request(runner: HttpRunner, step: TStep) -> StepResult:
     )
     parsed_request_dict["thrift_client"] = parsed_request_dict["thrift_client"]
 
-    # parsed_request_dict["headers"].setdefault(
-    #     "HRUN-Request-ID",
-    #     f"HRUN-{self.__case_id}-{str(int(time.time() * 1000))[-6:]}",
-    # )
+    # if runner.config.add_request_id:
+    #   parsed_request_dict["headers"].setdefault(
+    #       "HRUN-Request-ID",
+    #       f"HRUN-{self.__case_id}-{str(int(time.time() * 1000))[-6:]}",
+    #   )
     step_variables["thrift_request"] = parsed_request_dict
 
     psm = parsed_request_dict["psm"]