GithubBackup/httprunner
1
0
Fork 0
mirror of https://github.com/httprunner/httprunner.git synced 2026-05-15 20:38:44 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #1113 from lihuacai168/master

Browse Source 
feat: support variable and function in response extract expression
This commit is contained in:
debugtalk
2022-03-22 17:17:23 +08:00
committed by GitHub
parent d614f260f6 0cfd3fbea6
commit e78cd0376f
3 changed files with 27 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
httprunner/response.py
View File

@@ -167,12 +167,21 @@ class ResponseObject(object):
return check_value
def extract(self, extractors: Dict[Text, Text]) -> Dict[Text, Any]:
def extract(self,
extractors: Dict[Text, Text],
variables_mapping: VariablesMapping = None,
functions_mapping: FunctionsMapping = None,
) -> Dict[Text, Any]:
if not extractors:
return {}
extract_mapping = {}
for key, field in extractors.items():
if '$' in field:
# field contains variable or function
field = parse_data(
field, variables_mapping, functions_mapping
)
field_value = self._search_jmespath(field)
extract_mapping[key] = field_value

2
httprunner/runner.py
View File

@@ -195,7 +195,7 @@ class HttpRunner(object):
# extract
extractors = step.extract
extract_mapping = resp_obj.extract(extractors)
extract_mapping = resp_obj.extract(extractors, step.variables, self.__project_meta.functions)
step_data.export_vars = extract_mapping
variables_mapping = step.variables