python自动化断言封装

简述：一般把断言方法集合封装为一个类，独立作为一个模块使用，输入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_result)
这个是开放给外部的方法，传入的参数为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)

尾声：
没啥想说的。。。。。。。。。（对了，有问题可以在评论区即时指正）