def query_json(json_content, query, delimiter='.'): """ Do an xpath-like query with json_content. @param (json_content) json_content json_content = { "ids": [1, 2, 3, 4], "person": { "name": { "first_name": "Leo", "last_name": "Lee", }, "age": 29, "cities": ["Guangzhou", "Shenzhen"] } } @param (str) query "person.name.first_name" => "Leo" "person.cities.0" => "Guangzhou" @return queried result """ if json_content == "": raise exception.ResponseError("response content is empty!") try: for key in query.split(delimiter): if isinstance(json_content, list): json_content = json_content[int(key)] elif isinstance(json_content, (dict, CaseInsensitiveDict)): json_content = json_content[key] else: raise exception.ParseResponseError( "response content is in text format! failed to query key {}!".format(key)) except (KeyError, ValueError, IndexError): raise exception.ParseResponseError("failed to query json when extracting response!") return json_content
def query_json(json_content, query, delimiter='.'): """ Do an xpath-like query with json_content. @param (json_content) json_content json_content = { "ids": [1, 2, 3, 4], "person": { "name": { "first_name": "Leo", "last_name": "Lee", }, "age": 29, "cities": ["Guangzhou", "Shenzhen"] } } @param (str) query "person.name.first_name" => "Leo" "person.cities.0" => "Guangzhou" @return queried result """ stripped_query = query.strip(delimiter) if not stripped_query: return None try: for key in stripped_query.split(delimiter): if isinstance(json_content, list): key = int(key) json_content = json_content[key] except (KeyError, ValueError, IndexError): raise exception.ParseResponseError("failed to query json when extracting response!") return json_content
def extract_field(self, field, delimiter='.'): """ extract field from requests.Response @param (str) field of requests.Response object, and may be joined by delimiter "status_code" "content" "headers.content-type" "content.person.name.first_name" """ try: field += "." # string.split(sep=None, maxsplit=-1) -> list of strings # e.g. "content.person.name" => ["content", "person.name"] top_query, sub_query = field.split(delimiter, 1) if top_query in ["body", "content", "text"]: json_content = self.parsed_body() else: json_content = getattr(self.resp_obj, top_query) if sub_query: # e.g. key: resp_headers_content_type, sub_query = "content-type" return utils.query_json(json_content, sub_query) else: # e.g. key: resp_status_code, resp_content return json_content except AttributeError: raise exception.ParseResponseError("failed to extract bind variable in response!")
def validate(self, validators, variables_mapping): """ Bind named validators to value within the context. @param (list) validators [ {"check": "status_code", "comparator": "eq", "expected": 201}, {"check": "resp_body_success", "comparator": "eq", "expected": True} ] @param (dict) variables_mapping { "resp_body_success": True } @return (list) content differences [ { "check": "status_code", "comparator": "eq", "expected": 201, "value": 200 } ] """ for validator_dict in validators: check_item = validator_dict.get("check") if not check_item: raise exception.ParamsError("invalid check item in testcase validators!") if "expected" not in validator_dict: raise exception.ParamsError("expected item missed in testcase validators!") expected = validator_dict.get("expected") comparator = validator_dict.get("comparator", "eq") if check_item in variables_mapping: validator_dict["actual_value"] = variables_mapping[check_item] else: try: validator_dict["actual_value"] = self.extract_field(check_item) except exception.ParseResponseError: raise exception.ParseResponseError("failed to extract check item in response!") utils.match_expected( validator_dict["actual_value"], expected, comparator, check_item ) return True