测试框架Unitest之参数化 - ddt 数据驱动模块详解（下）

zxdznb

三、ddt读取文件

2.ddt读取excel文件
思路：先从excel文件中读取数据，然后再用ddt加载已读取的数据


第一步：读取excel

1. # get_excel.py
   
2. 
   
3. from openpyxl import load_workbook
   
4. 
   
5. class ExcelData():
   
6. 
   
7.     def __init__(self, file="config.xlsx"):
   
8.         '''
   
9.         初始化Excel对象
   
10.         '''
    
11.         self.file = file
    
12.         self.wb = load_workbook(self.file)
    
13. 
    
14.     def get_row_value(self, row, sheet_name="Sheet1"):
    
15.         '''
    
16.         获取Excel中某一行的数据
    
17.         '''
    
18.         sh = self.wb[sheet_name]
    
19.         max_col = sh.max_column
    
20.         row_value = []
    
21.         for col in range(1, max_col+1):
    
22.             value = sh.cell(row, col).value
    
23.             row_value.append(value)
    
24.         return row_value
    
25. 
    
26.     def get_all_row(self, sheet_name="Sheet1"):
    
27.         '''
    
28.         获取Excel中所有行的数据，并存放在列表中
    
29.         '''
    
30.         sh = self.wb[sheet_name]
    
31.         max_row = sh.max_row
    
32.         row_value = []
    
33.         for row in range(2, max_row+1):
    
34.             value = self.get_row_value(row)
    
35.             row_value.append(value)
    
36.         return row_value
    
37. 
    
38. if __name__ == "__main__":
    
39.     excel = ExcelData()
    
40.     testdata = excel.get_all_row()
    
41.     print(testdata)

复制代码

第二步：使用ddt使用读取好的数据

1. # get_ddt.py
   
2. 
   
3. import requests
   
4. import unittest
   
5. from ddt import ddt, data, unpack, file_data
   
6. from get_excel import ExcelData
   
7. 
   
8. @ddt
   
9. class SignTest(unittest.TestCase):
   
10. 
    
11.     # 从get_excel.py中读取测试数据
    
12.     excel = ExcelData()
    
13.     testdata = excel.get_all_row()
    
14. 
    
15.     @data(*testdata)
    
16.     def test_sign(self, datas):
    
17.         # 由于从excel中读取到的数据为列表形式，所以采用下标来提取各参数
    
18.         ID = datas[0]
    
19.         model = datas[1]
    
20.         title = datas[2]
    
21.         method = datas[3]
    
22.         url = datas[4]
    
23.         username = datas[5]
    
24.         pwd = datas[6]
    
25.         cpwd = datas[7]
    
26.         check = datas[8]
    
27.         body = {
    
28.             "username": username,
    
29.             "pwd": pwd,
    
30.             "cpwd": cpwd
    
31.         }
    
32.         self.sign_test(ID,model,title,url,method,body,check)
    
33. 
    
34.     def sign_test(self,ID,model,title,url,method,body,check):
    
35.         print("用例ID:", ID)
    
36.         print("模块:", model)
    
37.         print("用例标题:", title)
    
38.         response = requests.request(url=url, method=method, data=body).text
    
39.         try:
    
40.             # 通过断言，比较实际结果是否与预期结果一致
    
41.             # 由于从excel中读取到的check为str类型，所以response不用转换为dict，直接断言比较是否相等
    
42.             assert check == response
    
43.             print("测试通过")
    
44.         except Exception as e:
    
45.             print("测试失败")
    
46.             raise e
    
47. 
    
48. if __name__ == "__main__":
    
49.     unittest.main()

复制代码

结果：

1. 运行结果：
   
2. 
   
3. 用例ID: 001
   
4. 模块: 注册模块
   
5. 用例标题: 正确的用户名和密码，注册成功
   
6. .测试通过
   
7. 用例ID: 002
   
8. 模块: 注册模块
   
9. 用例标题: 用户名长度小于6位，注册失败
   
10. .测试通过
    
11. OK
    
12. ----------------------------------------------------------------------
    
13. Ran 2 tests in 0.190s
    
14. 
    
15. Process finished with exit code 0

复制代码

转载：python ddt模块的使用 - www.pu - 博客园 (cnblogs.com)