数据驱动测试框架设计——11操作header

思路：1.config创建header.json储存header

2.Base下封装一个handle.py来获取header

# -*-coding:utf-8-*-
from Base.handle_value import *

class HundleHeader:
    def get_header(self):
        return getValue.readjson("header.json")

getHeader = HundleHeader()

3.给用例加一列header操作；

run_main加一个headers附一个初始值None，如果是yes 则获取header.json中储存的header值，否则headers为默认的None

# -*-coding:utf-8-*-
from Base.handle_excel import *
from Base.base_request import *
from Base.handle_result import *
from Base.handle_cookie import *
from Base.handle_header import *
'''创建RunMain类实现运行用例的功能'''
class RunMain:

    def run_case(self):

        ''' 获取行数'''
        rows = excel_data.get_rows()
        '''除去表头需要遍历rows-1次'''
        for i in range(rows-1):
            '''给一个默认值'''
            cookies = None
            get_cookie = None
            headers = None
            '''i是从0开始，所以i+2，就和用例可以对应上，获取第i+2用例数据'''
            data = excel_data.get_rows_value(i+2)
            is_run = data[2]
            '''判断如果第二列数据为yes则执行'''
            if is_run == "yes":
                method = data[5]
                url = data[4]
                cookie_method = data[8]
                header_method = data[7]
                if cookie_method == "yes":
                    cookies = handlecookie.get_cookie_value("app")
                if cookie_method == "write":
                    get_cookie = {"is_cookie":"web"}
                    '''操作header的情况'''
                if header_method == "yes":
                    headers = getHeader.get_header()
                '''用封装的request发送请求'''
                res = request.run_main(method=method,url=url,get_cookie = get_cookie,cookies = cookies,headers = headers)
                '''调试打印'''
                print(res)

                '''获取实际result对应的code'''
                code = res["result"]

                '''获取实际的msg的值'''
                msg = res["msg"]
                '''获取断言方法'''
                assertion_method = data[9]
                '''获取预期结果'''
                excepect_result = data[10]
                '''如果断言方法是mer'''
                if assertion_method == "mer":
                    '''传入实际code获取config对应的msg值'''
                    config_msg = data_result.get_result(url,code)
                    '''进行断言'''
                    if msg == config_msg:
                        excel_data.excel_write_data(i + 2,12,"case通过")
                    else:
                        excel_data.excel_write_data(i + 2, 12, "case失败")
                        '''如果断言方法是result'''
                if assertion_method == "result":
                    '''这块需要注意返回结果获取的result是整型，而我们Excel中获取
                    的是字符串，所以这边需要把格式统一再做比对'''
                    if str(res["result"]) == excepect_result:
                        excel_data.excel_write_data(i + 2, 12, "case通过")
                    else:
                        excel_data.excel_write_data(i + 2, 12, "case失败")

if __name__ == '__main__':
    run = RunMain()
    run.run_case()