GithubBackup/httprunner
1
0
Fork 0
mirror of https://github.com/httprunner/httprunner.git synced 2026-08-28 19:47:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: add default header HRUN-Request-ID for each testcase

Browse Source
This commit is contained in:
debugtalk
2020-03-13 19:36:47 +08:00
parent 04ef15e778
commit 7cea1e7c34
1 changed files with 10 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
httprunner/runner.py
+10 -2
View File
@@ -1,5 +1,4 @@
# encoding: utf-8 import uuid
from enum import Enum from enum import Enum
from unittest.case import SkipTest from unittest.case import SkipTest
@@ -73,6 +72,10 @@ class Runner(object):
self.export = config.get("export") or config.get("output", []) self.export = config.get("export") or config.get("output", [])
config_variables = config.get("variables", {}) config_variables = config.get("variables", {})
self.hrun_request_id = str(uuid.uuid4())
if "HRUN-Request-ID" not in config_variables:
config_variables["HRUN-Request-ID"] = self.hrun_request_id
# testcase setup hooks # testcase setup hooks
testcase_setup_hooks = config.get("setup_hooks", []) testcase_setup_hooks = config.get("setup_hooks", [])
# testcase teardown hooks # testcase teardown hooks
@@ -222,6 +225,11 @@ class Runner(object):
base_url = self.session_context.eval_content(test_dict.get("base_url", "")) base_url = self.session_context.eval_content(test_dict.get("base_url", ""))
parsed_url = utils.build_url(base_url, url) parsed_url = utils.build_url(base_url, url)
request_headers = parsed_test_request.setdefault("headers", {})
if "HRUN-Request-ID" not in request_headers:
parsed_test_request["headers"]["HRUN-Request-ID"] = \
self.session_context.session_variables_mapping["HRUN-Request-ID"]
try: try:
method = parsed_test_request.pop('method') method = parsed_test_request.pop('method')
parsed_test_request.setdefault("verify", self.verify) parsed_test_request.setdefault("verify", self.verify)