【python做接口测试的学习记录day8——pytest自动化测试框架之热加载和断言封装】

一、热加载:

就是在代码执行过程当中动态的调用Python中的方法达到或得动态参数的目的

在根目录下创建debug_talk.py,这里我们写入了获得随机数的方法,后面使用值直接调用即可

import random
import time
from common.yaml_util import YamlUtil

class DebugTalk:

    #获得随机数
    def get_randon_number(self,min,max):
        return random.randint(int(min),int(max))

    #读取extract.yaml文件中的值
    def read_extract_data(self,key):
        return YamlUtil().read_yaml(key)

另外根目录创建test.py,这个是获取随机时间和本地时间的一个方法,因为我后面的接口输入数据中包含一个参数是时间

import random
import time
from common.yaml_util import YamlUtil


class Test:
    #获取随机时间
    def get_random_time(self):
        # return str(int(time.time()))[1:6]     #获取随机时间,得到的是时间戳
        return time.strftime('%H:%M:%S', time.localtime(time.time()))  #获取本地时间,并将格式转换成时分秒

    #读取extract.yaml文件中的值
    def read_extract_data(self,key):
        return YamlUtil().read_yaml(key)

接下来就是request.py文件的修改:

import jsonpath
import requests
import json
from common.yaml_util import YamlUtil
from builtins import str
import re
from debug_talk import DebugTalk
from test import  Test


class RequestUtil:

    def __init__(self,two_node,obj):
        self.base_url=YamlUtil().read_config('base',two_node)
        self.obj=obj


    #替换值的方法
    # #(替换url,params,data,json,headers)
    # #(string,int,float,list,dict)
    def replace_value(self, data):
        if data:
            # 保存数据类型
            data_type = type(data)
            # 判断数据类型转换成str
            if isinstance(data, dict) or isinstance(data, list):
                str_data = json.dumps(data)
            else:
                str_data = str(data)
            for cs in range(1, str_data.count('${') + 1):
                # 替换
                if "${" in str_data and "}" in str_data:
                    start_index = str_data.index("${")
                    end_index = str_data.index("}", start_index)
                    old_value = str_data[start_index:end_index + 1]
                    print("old_value:"+old_value)
                    #反射:通过类的对象和方法字符串调用方法
                    func_name=old_value[2:old_value.index('(')]
                    args_value1=old_value[old_value.index('(')+1:old_value.index(')')]
                    new_value=""
                    if args_value1 !="" :
                        args_value2 = args_value1.split(',')
                        new_value=getattr(self.obj,func_name)(*args_value2)
                    else:
                        new_value = getattr(self.obj, func_name)()
                    str_data = str_data.replace(old_value, str(new_value))
            # 还原数据类型
            if isinstance(data, dict) or isinstance(data, list):
                data = json.loads(str_data)
            else:
                data = data_type(str_data)
        return data


    #规范yaml测试用例
    def standard_yaml(self,caseinfo):
        caseinfo_keys= caseinfo.keys()
        #判断一级关键字是否包含:name,request,validate
        if "name" in caseinfo_keys and "request" in caseinfo_keys and "validate" in caseinfo_keys:
            #判断request下面是否包含:method、url
            request_keys=caseinfo["request"].keys()
            if "method" in  request_keys and "url" in request_keys:
                print("yaml基本架构检查通过")
                method = caseinfo['request'].pop("method") #pop() 函数用于移除列表中的一个元素,并且返回该元素的值。
                url= caseinfo['request'].pop("url")
                res = self.send_request(method,url,**caseinfo['request']) #caseinfo需要解包加**
                return_text=res.text
                return_code = res.status_code
                return_json=""
                try:
                    return_json = res.json()
                except Exception as e:
                    print("extract返回的结果不是JSON格式")

                # 提取值并写入extract.yaml文件
                if "extract" in caseinfo.keys():
                    for key, value in caseinfo["extract"].items():
                        if "(.*?)" in value or "(.+?)" in value:  # 正则表达式
                            zz_value = re.search(value, return_text)
                            if zz_value:
                                extract_value = {key: zz_value.group(1)}
                                YamlUtil().write_yaml(extract_value)
                        else:  # jsonpath
                            js_value = jsonpath.jsonpath(return_json, value)
                            if js_value:
                                extract_value = {key: js_value[0]}
                                YamlUtil().write_yaml(extract_value)
            else:
                print("在request下必须包含method,url")
        else:
            print("一级关键字必须包含name,request,validate")



    sess= requests.session()

    # 统一请求封装

    def send_request(self,method,url,**kwargs):
        method=str(method).lower()  #转换小写
        #基础路径的拼接和替换
        url= self.base_url + self.replace_value(url)
        print(url)
        #参数替换
        for key,value in kwargs.items():
            if key in ['params','data','json','headers']:
                kwargs[key]=self.replace_value(value)
            elif key == "files":
                for file_key, file_path in value.items():
                    value[file_key] = open(file_path, 'rb')
        res = RequestUtil.sess.request(method, url, **kwargs)
        print(res.text)
        return res

参数为本地时间的yaml用例文件:

-
  name: 获取城市列表
  request:
    method: post
    url: /jlcloud/simulation/${read_extract_data(data)}/ranAsPlan
    json: {"time": "${get_random_time()}"}
    headers:
      'Content-Type': 'application/json'
      'X-Token': ${read_extract_data(token)}
  validate: None

这里仅使用了随机时间,如果接口包含随机数,可类比随机时间的方法

 最后需要在用例上需要创建初始化对象,用到哪个就写哪个,我这里只用到了Test()

import requests
import json
from common.request_util import RequestUtil
from common.yaml_util import YamlUtil
from test import Test

class TestRequest:
    pass
    @pytest.mark.parametrize("caseinfo",YamlUtil().read_testcase('get_token.yaml'))
    def test_login(self,caseinfo):
        res = RequestUtil("base_test_url",Test()).standard_yaml(caseinfo)

    #
    @pytest.mark.parametrize("caseinfo", YamlUtil().read_testcase('userinfo.yaml'))
    def test_userinfo(self, caseinfo):
        res = RequestUtil("base_test_url",Test()).standard_yaml(caseinfo)
    #     print(res.text)
    #
    @pytest.mark.parametrize("caseinfo", YamlUtil().read_testcase('city_type.yaml'))
    def test_city(self, caseinfo):
        res = RequestUtil("base_test_url",Test()).standard_yaml(caseinfo)

    @pytest.mark.parametrize("caseinfo", YamlUtil().read_testcase('simulation.yaml'))
    def test_simulation(self, caseinfo):
        res = RequestUtil("base_test_url",Test()).standard_yaml(caseinfo)

    @pytest.mark.parametrize("caseinfo", YamlUtil().read_testcase('ranAsPlan.yaml'))
    def test_ranAsPlan(self, caseinfo):
        res = RequestUtil("base_test_url",Test()).standard_yaml(caseinfo)

二、断言

断言一般有两种,一种是通过返回值的状态码是否等于200,一种是业务的判断,业务判断可以使用两种方式,一种是想等,一种是包含

1、状态断言:返回的状态码是否等于200

2、业务断言:相等的断言、包含的断言

在request.py中进行断言方法的封装:

import jsonpath
import requests
import json
from common.yaml_util import YamlUtil
from builtins import str
import re
from debug_talk import DebugTalk
from test import  Test


class RequestUtil:

    def __init__(self,two_node,obj):
        self.base_url=YamlUtil().read_config('base',two_node)
        self.obj=obj


    #替换值的方法
    # #(替换url,params,data,json,headers)
    # #(string,int,float,list,dict)
    def replace_value(self, data):
        if data:
            # 保存数据类型
            data_type = type(data)
            # 判断数据类型转换成str
            if isinstance(data, dict) or isinstance(data, list):
                str_data = json.dumps(data)
            else:
                str_data = str(data)
            for cs in range(1, str_data.count('${') + 1):
                # 替换
                if "${" in str_data and "}" in str_data:
                    start_index = str_data.index("${")
                    end_index = str_data.index("}", start_index)
                    old_value = str_data[start_index:end_index + 1]
                    print("old_value:"+old_value)
                    #反射:通过类的对象和方法字符串调用方法
                    func_name=old_value[2:old_value.index('(')]
                    args_value1=old_value[old_value.index('(')+1:old_value.index(')')]
                    new_value=""
                    if args_value1 !="" :
                        args_value2 = args_value1.split(',')
                        new_value=getattr(self.obj,func_name)(*args_value2)
                    else:
                        new_value = getattr(self.obj, func_name)()
                    str_data = str_data.replace(old_value, str(new_value))
            # 还原数据类型
            if isinstance(data, dict) or isinstance(data, list):
                data = json.loads(str_data)
            else:
                data = data_type(str_data)
        return data


    #规范yaml测试用例
    def standard_yaml(self,caseinfo):
        caseinfo_keys= caseinfo.keys()
        #判断一级关键字是否包含:name,request,validate
        if "name" in caseinfo_keys and "request" in caseinfo_keys and "validate" in caseinfo_keys:
            #判断request下面是否包含:method、url
            request_keys=caseinfo["request"].keys()
            if "method" in  request_keys and "url" in request_keys:
                print("yaml基本架构检查通过")
                method = caseinfo['request'].pop("method") #pop() 函数用于移除列表中的一个元素,并且返回该元素的值。
                url= caseinfo['request'].pop("url")
                res = self.send_request(method,url,**caseinfo['request']) #caseinfo需要解包加**
                return_text=res.text
                return_code = res.status_code
                return_json=""
                try:
                    return_json = res.json()
                except Exception as e:
                    print("extract返回的结果不是JSON格式")

                # 提取值并写入extract.yaml文件
                if "extract" in caseinfo.keys():
                    for key, value in caseinfo["extract"].items():
                        if "(.*?)" in value or "(.+?)" in value:  # 正则表达式
                            zz_value = re.search(value, return_text)
                            if zz_value:
                                extract_value = {key: zz_value.group(1)}
                                YamlUtil().write_yaml(extract_value)
                        else:  # jsonpath
                            js_value = jsonpath.jsonpath(return_json, value)
                            if js_value:
                                extract_value = {key: js_value[0]}
                                YamlUtil().write_yaml(extract_value)
                #断言:
                self.assert_result(caseinfo['validate'],return_json,return_code)
            else:
                print("在request下必须包含method,url")
        else:
            print("一级关键字必须包含name,request,validate")



    sess= requests.session()

    # 统一请求封装

    def send_request(self,method,url,**kwargs):
        method=str(method).lower()  #转换小写
        #基础路径的拼接和替换
        url= self.base_url + self.replace_value(url)
        print(url)
        #参数替换
        for key,value in kwargs.items():
            if key in ['params','data','json','headers']:
                kwargs[key]=self.replace_value(value)
            elif key == "files":
                for file_key, file_path in value.items():
                    value[file_key] = open(file_path, 'rb')
        res = RequestUtil.sess.request(method, url, **kwargs)
        print(res.text)
        return res
    #断言
    def assert_result(self,yq_result,sj_result,return_code):
        all_flag = 0
        for yq in yq_result:
            for key,value in yq.items():
                print(key,value)
                if key=="equals":
                    flag=self.equals_assert(value,return_code,sj_result)
                    all_flag =all_flag + flag
                elif key == 'contains':
                    flag=self.contains_assert(value,sj_result)
                    all_flag = all_flag + flag
                else:
                    print("框架暂不支持此段断言方式")
        assert all_flag==0

    # 相等断言
    def equals_assert(self,value,return_code,sj_result):
        flag=0
        for assert_key,assert_value in value.items():
            print(assert_key,assert_value)
            if assert_key=="status_code":  #状态断言
                assert_value==return_code
                if assert_value!=return_code:
                    flag=flag+1
                    print("断言失败,返回的状态码不等于%s"%assert_value)
            else:
                lists=jsonpath.jsonpath(sj_result,'$..%s'%assert_key)
                if lists:
                    if assert_value not in lists:
                        flag=flag+1
                        print("断言失败:"+assert_key+"不等于"+str(assert_value))
                else:
                    flag = flag + 1
                    print("断言失败:返回的结果不存在:"+assert_key)
        return flag
    # 包含断言
    def contains_assert(self,value,sj_result):
        flag=0
        if value not in str(sj_result):
            flag = flag + 1
            print("断言失败:返回的结果中不包含:"+value)
        return flag


然后在yaml文件中,输入断言的内容,例如我这个登录接口断言了返回值code是否等于200,以及是否包含data

【python做接口测试的学习记录day8——pytest自动化测试框架之热加载和断言封装】_第1张图片

完成后,让我们run起来,结果可以看到:

【python做接口测试的学习记录day8——pytest自动化测试框架之热加载和断言封装】_第2张图片

你可能感兴趣的:(pytest,功能测试,测试工具,python)