简述:一般把断言方法集合封装为一个类,独立作为一个模块使用,输入response和期望数据,然后返回断言结果
一、断言返回数据
这里我将不同断言结果的返回数据分别做为一字典类型的实例属性
1、断言通过数据(pass_result)
self.pass_result = { 'code': 0, 'response_code': self.response_data.status_code, 'response_reason': self.response_data.reason, 'response_headers': self.response_data.headers, 'response_body': self.response_data.text, 'message': '测试用例执行通过', 'check_result': True }
2、断言失败数据(fail_result)
self.fail_result = { 'code': 2, 'response_code': self.response_data.status_code, 'response_reason': self.response_data.reason, 'response_headers': self.response_data.headers, 'response_body': self.response_data.text, 'message': '测试用例断言失败,测试用例执行不通过', 'check_result': False }
二、私有断言方法
1、跳过断言函数
有些用例不需要进行断言,并且又不想影响结构,就需要用到该函数,该函数固定返回self.pass_result
def __none_check(self): logger.info("断言类型[none]-->不进行断言操作,本次断言操作通过") return self.pass_result
2、断言json类型响应数据是否包含期望的键
源码如下
def __key_check(self, check_data): """ 检查json类型的响应body中是否包含一个或多个key :param check_data: 检查数据字符串 :return:都包含则返回True; 有一个不包含就返回False """ key_list = check_data.split(',') tmp_result = [] for check_key in key_list: if check_key in self.response_data.json().keys(): tmp_result.append(True) else: tmp_result.append(False) if False in tmp_result: error_message = '断言类型[json_key]-->实际结果:%s ;期望结果:%s 不相符,断言失败'%(self.response_data.text, check_data) logger.error(error_message) self.fail_result["message"] = error_message return self.fail_result else: pass_message = '断言类型[json_key]-->实际结果:%s ;期望结果:%s 相符,断言通过'%(self.response_data.text, check_data) logger.info(pass_message) self.pass_result["message"] = pass_message return self.pass_result
思路:
(1)本函数是先使用split(‘,’)对检查数据(该数据为str类型)进行分隔,返回期望key列表(key_list)
(2)再使用for循环key_list 使每个key对json类型响应数据进行判断是否存在,存在则在tmp_result列表添加True,反之则添加False
(3)最后判断tmp_result列表是否存在False,存在则return self.fail_result,反之则return self.pass_result
3、断言json类型响应数据是否包含期望的键值对
源码如下
def __key_value_check(self, check_data): """ 检查json类型的响应body中是否包含一个或多个键值对 :param check_data:检查数据字符串 :return:都包含则返回True; 有一个不包含就返回False """ key_value_dict = json.loads(check_data) tmp_result = [] for key_value in key_value_dict.items(): if key_value in self.response_data.json().items(): tmp_result.append(True) else: tmp_result.append(False) if False in tmp_result: error_message = '断言类型[json_key_value]-->实际结果:%s ;期望结果:%s 不相符,断言失败' % (self.response_data.text, check_data) logger.error(error_message) self.fail_result["message"] = error_message return self.fail_result else: pass_message = '断言类型[json_key_value]-->实际结果:%s ;期望结果:%s 相符,断言通过' % (self.response_data.text, check_data) logger.info(pass_message) self.pass_result["message"] = pass_message return self.pass_result
思路:
(1)本函数是先使用json.loads(也可用eval内置函数)见检查数据(该数据为str类型)进行转换,返回期望数据key_value_dict字典
(2)再使用for循环key_value_dict.items() 使每个键值对对json类型响应数据进行判断是否存在,存在则在tmp_result列表添加True,反之则添加False
(3)最后判断tmp_result列表是否存在False,存在则return self.fail_result,反之则return self.pass_result
4、补充
对键和值的判断还可以用在判断响应头上,实际代码类似就不多赘述
5、使用正则表达式对响应正文进行提取,断言是否有表达式提取到的数据
源码如下
def __body_regexp_check(self, check_data): if re.findall(check_data, self.response_data.text): pass_message = '断言类型[body_regexp]-->实际结果:%s ;期望结果:%s 相符,断言通过' % (self.response_data.text, check_data) logger.info(pass_message) self.pass_result["message"] = pass_message return self.pass_result else: error_message = '断言类型[body_regexp]-->实际结果:%s ;期望结果:%s 不相符,断言失败' % (self.response_data.text, check_data) logger.error(error_message) self.fail_result["message"] = error_message return self.fail_result
思路:
(1)这个的模块很简单,就是使用re模块的findall方法传入用与检查的正则表达式和str类型的response数据,findall会return列表数据,该列表类型数据每个单元为符合表达式的str类型数据
(2)最后判断列表是否存在数据,存在则return self.pass_result,反之则return self.fail_result
(3)升级思路:这个函数还可以进一步判断,通过正则表达式返回的字符串1是否与期望值一致,
只需对输入的check_data数据统一一个规则,比如“《|》”符号前面的是正则表达式,后面的是期望值,这个符号随你怎么定(当然最好是那种不常出现的字符组合),然后用split分割出来,然后在去判断
6、对响应status_code进行断言
源码如下
def __response_code_check(self, check_data): if self.response_data.status_code == int(check_data): pass_message = '断言类型[response_code]-->实际结果:%s ;期望结果:%s 相符,断言通过' logger.info(pass_message) self.pass_result["message"] = pass_message return self.pass_result else: error_message = '断言类型[response_code]-->实际结果:%s ;期望结果:%s 不相符,断言失败' logger.error(error_message) self.fail_result["message"] = error_message return self.fail_result
思路:
这个比较简单一看会,不多赘述(至于为啥还写在这里,单纯是因为作者强迫症,就只差这个函数没有写进来)
三、断言方法封装
看前面的第2部分名字“私有断言方法”就明白这些不是开放给外部使用的,所以这里需要一个开放给外部的函数用来使用这些断言方法,这里我分为两个部分
1、实例属性:fuction
源码如下:
self.function = { 'none': self.__none_check, 'json_key': self.__key_check, 'json_key_value': self.__key_value_check, 'body_regexp': self.__body_regexp_check, 'header_key': self.__header_key_check, 'header_key_value': self.__header_key_value_check, 'response_code': self.__response_code_check }
这个属性是一个字典类型的数据,数据中key是断言方法名称,而value是前面的第2部分的私有断言方法对象(注意⚠️ :传递函数对象时,函数名末尾不能有括号“()” ,加上括号就是调用函数返回该函数return的值了)
2、实例函数:run_check
源码如下:
def run_check(self, check_type, excepted_result): logger.info('根据断言类型[%s]进行断言,检查是否满足期望结果:%s'%(check_type, excepted_result)) if check_type == 'none' or excepted_result == "": return self.function['none']() return self.function[check_type](excepted_resul> t)
这个是开放给外部的方法,传入的参数为check_type(断言类型)和excpted_result(断言数据),然后除了none的情况判断一下
if check_type == 'none' or excepted_result == "":
return self.function'none'
其他的情况直接返回第一部分的 function[check_type](exceptd_result)
return self.function[check_type](excepted_result)