Python @ddt.file_data() 为.yml 文件实例

一，创建login.yml 文件（以登录接口为例）

1，创建 login.yml 文件，内容如下图：

打印login.yml 文件，代码及显示效果如下：

代码：

import yaml,json
f = open(r'C:xxx\login.yml',encoding='utf-8')
print(json.dumps(yaml.load(f),indent=4,ensure_ascii= False))

 

2，测试代码如下：

import ddt,unittest,requests,time
import BeautifulReport

@ddt.ddt
class My_case(unittest.TestCase):

    @ddt.file_data(r'xxx\login.yml')
    @ddt.unpack
    def test_run(self,**kwargs):
        url = kwargs.get('url')
        method = kwargs.get('method')
        header = kwargs.get('header')
        data =kwargs.get('data')
        is_json = kwargs.get('is_json', 0)
        cookie = kwargs.get('cookie', {})
        check = kwargs.get('check')
        if method == 'post':
            if is_json:
                res = requests.post(url,json = data,headers = header,cookies = cookie)
            else:
                res = requests.post(url,data = data,headers = header,cookies = cookie)
        else:
            res = requests.get(url,params= data,headers = header,cookies = cookie)
        print(res.text)
        # for k,v in check.items(): #校验方法为dict 
        #     self.assertEqual(res.json().get(k),v)
        for c in check: # 校验方法为list 
            self.assertIn(c, res.text)

if __name__ == '__main__':
    unittest.main()
    # suite = unittest.TestSuite()
    # suite.addTest(unittest.makeSuite(My_case))
    # report_html = BeautifulReport.BeautifulReport(suite)
    # # report_html.report(filename='test', description='用例描述')
    # fmt = '{date}_TestReport.html'.format(date=time.strftime('%Y%m%d%H%M%S'))
    # #生成报告的文件名格式20180329190544_TestReport.html
    # report_html.report(filename=fmt, description='用例描述')

二，创建createTicketOrder.yml 文件（以创建订单接口为例）

1，创建createTicketOrder.yml 文件，内容如下图：

2，创建idcard.py 文件，用来生成身份证号（附上百度代码）

import datetime,random
def getValidateCheckout(id17):
    '''获得校验码算法'''
    weight=[7,9,10,5,8,4,2,1,6,3,7,9,10,5,8,4,2] #十七位数字本体码权重
    validate=['1','0','X','9','8','7','6','5','4','3','2'] #mod11,对应校验码字符值

    sum=0
    for i in range(0,len(id17)):
        sum = sum + int(id17[i])*weight[i]
    mode=sum%11
    return validate[mode]

def getRandomIdNumber():
    addr = [513436]
    sex = random.randint(0,1) #产生随机可用身份证号，sex = 1表示男性，sex = 0表示女性
    addrId = addr[0]
    # addrName = addr[1]
    idNumber = str(addrId)
    #出生日期码
    start, end = "1960-01-01","2018-12-30" #生日起止日期
    days = (datetime.datetime.strptime(end, "%Y-%m-%d") - datetime.datetime.strptime(start, "%Y-%m-%d")).days + 1
    birthDays = datetime.datetime.strftime(datetime.datetime.strptime(start, "%Y-%m-%d") + datetime.timedelta(random.randint(0,days)), "%Y%m%d")
    idNumber = idNumber + str(birthDays)
    #顺序码
    for i in range(2):#产生前面的随机值
        n = random.randint(0,9)# 最后一个值可以包括
        idNumber = idNumber + str(n)
    # 性别数字码
    sexId = random.randrange(sex,10,step = 2) #性别码
    idNumber = idNumber + str(sexId)
    # 校验码
    checkOut = getValidateCheckout(idNumber)
    idNumber = idNumber + str(checkOut)
    # return idNumber,addrName,addrId,birthDays,sex,checkOut
    return idNumber
3，调用接口测试，代码如下：

import ddt,unittest,requests,time
import BeautifulReport
import datetime
from cases.idcard import getRandomIdNumber #随机生成身份证函数
def get_id_card(dic:dict): #随机生成身份证号
    if 'id_card' in dic.values():
        for k in dic:
            if dic.get(k) == 'id_card':
                dic[k] = getRandomIdNumber()
    return dic

def get_cur_time(dic: dict): #取系统当前日期
    if 'cur_time' in dic.values():
        for k in dic:
            if dic.get(k) == 'cur_time':
                dic[k] = '%s' % datetime.date.today()
    return dic

@ddt.ddt
class My_case(unittest.TestCase):
    @ddt.file_data(r'xxx\createTicketOrder.yml')
    @ddt.unpack
    def test_run(self,**kwargs):
        url = kwargs.get('url')
        method = kwargs.get('method')
        header = kwargs.get('header')
        data =kwargs.get('data')
        data = get_cur_time(data) # 将 yml 文件中的cur_time 字段替换为系统当前日期
        data = get_id_card(data) #将 yml 文件中的id_card 字段替换为调用函数生成的身份证号
        is_json = kwargs.get('is_json', 0)
        cookie = kwargs.get('cookie', {})
        check = kwargs.get('check')
        if method == 'post':
            if is_json:
                res = requests.post(url,json = data,headers = header,cookies = cookie)
            else:
                res = requests.post(url,data = data,headers = header,cookies = cookie)
        else:
            res = requests.get(url,params= data,headers = header,cookies = cookie)
        print(res.text)
        for k,v in check.items(): #校验方法为dict
            self.assertEqual(res.json().get(k),v)
        # for c in check: # 校验方法为list
        #     self.assertIn(c, res.text)

if __name__ == '__main__':
    unittest.main()
    # suite = unittest.TestSuite()
    # suite.addTest(unittest.makeSuite(My_case))
    # report_html = BeautifulReport.BeautifulReport(suite)
    # # report_html.report(filename='test', description='用例描述')
    # fmt = '{date}_TestReport.html'.format(date=time.strftime('%Y%m%d%H%M%S'))
    # #生成报告的文件名格式20180329190544_TestReport.html
    # report_html.report(filename=fmt, description='用例描述')

三，其他 
A,附上check 校验方法

 

校验方法为 

for c in check: # 校验方法为list
#     self.assertIn(c, res.text)

校验方法为

for k,v in check.items(): #校验方法为dict
    self.assertEqual(res.json().get(k),v)
B，如果 @ddt.file_data() 为.json 文件
如果value  是json 格式 需加转译符 \ 
如 入参为：
在json 文件里需写成 
C，附上creatTicketOrder.json 文件