从0开始python学习-48.pytest框架之断言

目录

1. 响应进行断言

1.1 在yaml用例中写入断言内容

1.2 封装断言方法

1.3 在执行流程中加入断言判断内容

2. 数据库数据断言

2.1 在yaml用例中写入断言内容

2.2 连接数据库并封装执行sql的方法

2.3 封装后校验方法是否可执行

2.4 使用之前封装的断言方法，加入数据库的case

---

1. 响应进行断言

1.1 在yaml用例中写入断言内容

validate:equals: # 相等断言断言状态码为200: [200,status_code] # [预期结果，实际结果]contains: # 包含断言断言响应结果包含access_token: [access_token,json]

1.2 封装断言方法

class AssertUtil:def assert_all_case(self,res,assert_type,value):# 深拷贝响应结果resnew_res = copy.deepcopy(res)# 把json()方法变更为json属性，用于反射使用,如果无法获取到new_res.json()抛出异常try:new_res.json = new_res.json()except Exception:new_res.json = {"msg":'response not json data'}# 循环判断断言for msg,data in value.items(): # 这里的value值就是断言的value值yq,sj = data[0],data[1] # 将预期值和实际值取出来# 过反射获取属性的值try:sj_value = getattr(new_res,sj) # 将获取到的响应结果反射给sjexcept Exception:sj_value = sj# 判断断言match assert_type:case "equals":assert yq == sj_value,msgcase "contains":assert yq in sj_value,msg

1.3 在执行流程中加入断言判断内容

def stand_case_flow(caseinfo):# 读取yaml文件中的数据case_obj = verify_yaml(caseinfo)# 请求之后得到响应后判断是否需要断言，如果validate不为None，则需要断言if case_obj.validate: # 这里可以将yaml中的validate内容取出来for assert_type,value in ExtractUtil().use_extract_value(case_obj.validate).items():AssertUtil().assert_all_case(res,assert_type,value)else:print("该用例没有涉及断言内容")

结果示例-断言失败

结果示例-断言成功

 

2. 数据库数据断言

2.1 在yaml用例中写入断言内容

validate:db_contains: # 数据库包含断言断言返回结果包含sql查询结果: ["SELECT email FROM pw_user WHERE uid = 1",text]

2.2 连接数据库并封装执行sql的方法

class AssertUtil:# 连接数据库def conn_mysql(self):self.conn = pymysql.connect(user="user",password="pwd",host="127.0.0.1",database="sd",port=3306)return self.conn# 执行sql语句def execute_sql(self,sql):# 创建数据库链接conn = self.conn_mysql()# 创建游标cs = conn.cursor()# 执行SQLcs.execute(sql)# 取值value = cs.fetchone()# 关闭资源cs.close()conn.close()# 返回值return value

2.3 封装后校验方法是否可执行

if __name__ == "__main__":a = AssertUtil().execute_sql("SELECT email FROM pw_user WHERE uid = 1")print(a)print(type(a))

执行结果

2.4 使用之前封装的断言方法，加入数据库的case

def assert_all_case(self,res,assert_type,value):# 深拷贝响应结果resnew_res = copy.deepcopy(res)# 把json()方法变更为json属性，用于反射使用,如果无法获取到new_res.json()抛出异常try:new_res.json = new_res.json()except Exception:new_res.json = {"msg":'response not json data'}# 循环判断断言for msg,data in value.items():yq,sj = data[0],data[1] # 将预期值和实际值取出来# 过反射获取属性的值try:sj_value = getattr(new_res,sj)except Exception:sj_value = sj# 判断断言match assert_type:case "equals":assert yq == sj_value,msgcase "contains":assert yq in sj_value,msgcase "db_contains":yq_value = self.execute_sql(yq)  # 将预期值的sql写入执行sql的方法中，获取结果assert yq_value[0] in sj_value, msg  # 注意这里的yq_value获取到的是tuple，需要通过下标取值str

结果示例-断言失败