一、ddt 用途
- 一般进行接口测试时，每个接口的传参都不止一种情况，一般会考虑正向、逆向等多种组合，所以在测试一个接口时，通常会编写多条 case，而这些除了传参不同外，并没有什么区别。这个时候就可以利用 ddt 来管理测试数据，提高代码复用率。
- ddt => data driver test 数据驱动测试
二、安装 ddt
- pip install ddt
三、ddt 模块组成
四、示例
-
ddt.ddt：装饰类，也就是继承 TestCase 的类。
-
ddt.data：装饰测试方法，参数是一系列的值
- ddt.unpack：用来拆解 data 的数据。通常 data 中包含的每一个值都会作为一个单独的参数传给测试方法，如果这些值是用元组或者列表传进来的，可以用 unpack 方法将其自动分解成多个参数
-
ddt.file_data：装饰测试方法，参数是文件名。文件可以是 json 或者 yaml 类型。
-
注意：如果文件是以 “.yml”或者".yaml" 结尾，ddt 会作为 yaml 类型处理，其他文件都会作为 json 文件处理。
-
如果文件是列表，列表的值会作为测试用例参数，同时，会作为测试用例方法名后缀显示。
- 如果文件是字典，字典的 key 会作为测试用例方法的后缀显示，字典的 value 会作为测试用例参数。
-
- 使用 data 传参数给测试用例
- 如下示例，可以看出 data 分别传参数 2 和 4 给测试用例，然后测试用例执行了两遍
from ddt import ddt, data import unittest @ddt class MyTestDdt(unittest.TestCase): def setUp(self): print('start') @data(2, 4) # 使用 data 传参数给测试用例 def test_one(self,value): print('the @data number is {}'.format(value)) def tearDown(self): print('end') if __name__ == '__main__': unittest.main() # 结果如下： start the @data number is :2 end start the @data number is :4 end
- 使用 unpack 分解 list 或者 tuple
- 可以看出 data 分别传元组参数（1，2）和（4，5）给测试用例，然后测试用例执行了两遍
from ddt import ddt, data, unpack import unittest @ddt class MyTestDdt(unittest.TestCase): def setUp(self): print('start') @data((1, 2), (4, 5)) # 元组 @unpack # 二次分解元组 def test_one(self, value1, value2): print('the @data number is {} and {}'.format(value1, value2)) def tearDown(self): print('end') if __name__ == '__main__': unittest.main() # 结果如下 start the @data number is 1 and 2 end start the @data number is 4 and 5 end
- 用 unpack 分解字典
from ddt import ddt, data, unpack import unittest @ddt class MyTestDdt(unittest.TestCase): def setUp(self): print('start') @data({'value1': 1, 'value2': 2}, {'value1': 3, 'value2': 4}) # 字典 @unpack def test_one(self, value1, value2): print('the @data number is {} and {}'.format(value1, value2)) def tearDown(self): print('end') if __name__ == '__main__': unittest.main() # 结果如下 start the @data number is 1 and 2 end start the @data number is 3 and 4 end
from ddt import ddt, data, unpack import unittest @ddt class MyTestDdt(unittest.TestCase): def setUp(self): print('start') @data(*[{'value1': 1, 'value2': 2}, {'value1': 3, 'value2': 4}]) # 这里加 * 后会将返回数据分为一个个的字典 @unpack def test_one(self, **value): print('the @data number is {} and {}'.format(value.get("value1"), value.get("value2"))) def tearDown(self): print('end') if __name__ == '__main__': unittest.main() # 结果如下 start the @data number is 1 and 2 end start the @data number is 3 and 4 end
五、cms 登录实例
""" getTestData.py 封装读取 excel 文件内容且返回 [['Evan', '123456'], ['duoceshi007', '123456'], ['Evan', 'xxxxxx']] 格式数据的方法 """
import xlrd def getdata(): book = xlrd.open_workbook("data.xls") sheet = book.sheet_by_name("Sheet1") rows = sheet.nrows data_list = [] for t in range(rows-1): row_list = [] row_data = sheet.row_values(t + 1) for i in row_data: if isinstance(i, float): d = int(i) row_list.append(str(d)) else: row_list.append(i) data_list.append(row_list) return data_list
""" lesson3_5_1.py cms 登录测试用例 """ from selenium import webdriver import time import unittest from ddt import ddt, data, unpack from Python_selenium.lesson3_2020_11_2.lesson3_5_ddt.getTestData import getdata @ddt class cmsLogin(unittest.TestCase): def setUp(self) -> None: self.dr = webdriver.Chrome() self.dr.get("http://xxx.xxxxxxxxx.xx/cms/manage/login.do") self.dr.maximize_window() self.dr.implicitly_wait(10) def tearDown(self) -> None: time.sleep(2) self.dr.quit() @data(*getdata()) @unpack def test_login(self, username, password): self.dr.find_element_by_id("userAccount").send_keys(username) self.dr.find_element_by_id("loginPwd").send_keys(password) self.dr.find_element_by_id("loginBtn").click() time.sleep(2) if username == "Evan" and password == "123456": try: nickName = self.dr.find_element_by_class_name("c-white").text except: raise AssertionError("登录失败！") else: assert username in nickName else: loginText = self.dr.find_element_by_css_selector('[for="online"]').text assert "使我保持登录状态" in loginText if __name__ == '__main__': unittest.main()