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 文件