feat: response extract expresion support variable and function

This commit is contained in:
lihuacai
2021-09-01 15:29:23 +08:00
parent 5467326f8e
commit 0bad623780
3 changed files with 27 additions and 3 deletions

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