一、背景
一般进行接口测试时,每个接口的传参都不止一种情况,一般会考虑正向、逆向等多种组合,所以在测试一个接口时,通常会编写多条case,而这些除了传参不同外,并没有什么区别。这个时候就可以利用ddt来管理测试数据,提高代码复用率。
二、ddt安装
ddt是python的第三方库,安装非常简单:
pip install ddt
ddt官方文档地址:https://pypi.org/project/ddt/
三、ddt模块组成
1、ddt模块包含类的装饰器ddt和两个方法装饰器data
ddt.ddt:装饰类,也就是继承TestCase的类。
ddt.data:装饰测试方法,参数是一系列的值。
ddt.file_data:装饰测试方法,参数是文件名。文件可以是json或者yaml类型。
注意,如果文件是以“.yml”或者".yaml"结尾,ddt会作为yaml类型处理,其他文件都会作为json文件处理。
如果文件是列表,列表的值会作为测试用例参数,同时,会作为测试用例方法名后缀显示。
如果文件是字典,字典的key会作为测试用例方法的后缀显示,字典的value会作为测试用例参数。
ddt.unpack:传递的是复杂的数据结构时使用,比如使用列表或者元组,添加unpack后,ddt会自动把元组或者列表对应到多个参数上。
2、测试用例方法名生成规则
使用ddt后,会产生一个新的测试用例方法名,即定义的测试用例方法名,比如:def test_one(),这里就是test_one。
data:如果传递过来的数据存在__ name__ 属性,则这里就是该数据的__ name__ 值。如果未定义__ name__ 属性,ddt会尽量将传递过来的值转化为python标识符,作为data显示。比如:(1,3)就转为1_3。需要注意的是,如果数据是字典,则这里就是字典的key。
四、示例
1、使用data传参数给测试用例
fromddtimportddt,data,file_data,unpackimportunittest@ddtclassMyTestDdt(unittest.TestCase):defsetUp(self):print('start')@data(2,4)#使用data传参数给测试用例deftest_one(self,value):print(f'the @data number is :{value}')deftearDown(self):print('end')if__name__=='__main__':unittest.main(verbosity=2)
结果:
starttest_one_1_2(__main__.MyTestDdt)...okthe@datanumberis:2endstartthe@datanumberis:4test_one_2_4(__main__.MyTestDdt)...okend----------------------------------------------------------------------Ran2testsin0.000sOK
可以看出data分别传参数2和4给测试用例,然后测试用例执行了两遍。另外,注意此时打印出的测试用例名称已经不同,比如:test_one_1_2和test_one_2_4,最后的2和4,分别是传递进去的参数。
2、使用unpack分解list或者tuple
fromddtimportddt,data,file_data,unpackimportunittest@ddtclassMyTestDdt(unittest.TestCase):defsetUp(self):print('start')@data((1,2),(4,5))#元组@unpack#分解deftest_one(self,value1,value2):print(f'the @data number is :{value1,value2}')deftearDown(self):print('end')if__name__=='__main__':unittest.main(verbosity=2)
结果:
startthe@datanumberis:(1,2)endtest_one_1__1__2_(__main__.MyTestDdt)...okstartthe@datanumberis:(4,5)test_one_2__4__5_(__main__.MyTestDdt)...okend----------------------------------------------------------------------Ran2testsin0.002sOK
可以看出data分别传元组参数(1,2)和(4,5)给测试用例,然后测试用例执行了两遍。另外,注意此时打印出的测试用例名称已经不同,比如:test_one_1__ 1__ 2_和test_one_2__ 4__5_ ,最后的1,2和4,5,分别是传递进去的参数。
3、用unpack分解字典
fromddtimportddt,data,file_data,unpackimportunittest@ddtclassMyTestDdt(unittest.TestCase):defsetUp(self):print('start')@data({'value1':1,'value2':2},{'value1':3,'value2':4})#字典@unpackdeftest_one(self,value1,value2):print(f'the @data number is :{value1,value2}')deftearDown(self):print('end')if__name__=='__main__':unittest.main(verbosity=2)
结果:
test_one_1(__main__.MyTestDdt)...okstarttest_one_2(__main__.MyTestDdt)...okthe@datanumberis:(1,2)----------------------------------------------------------------------endRan2testsin0.000sstartthe@datanumberis:(3,4)OKend
4、自动化测试实战
这里我们登录新浪邮箱,并利用数据驱动的方式带参进行测试
importunittestfromddtimportddt,data,unpackfromseleniumimportwebdriverdeftestdata():'''测试数据'''return[('','','请输入邮箱名'),('','123456','请输入邮箱名'),('123456','','您输入的邮箱名格式不正确')]@ddtclassMyMailLogin(unittest.TestCase):defsetUp(self):self.url='https://mail.sina.com.cn/'self.driver=webdriver.Chrome()self.driver.maximize_window()self.driver.get(self.url)self.driver.implicitly_wait(30)@data(*testdata())#这里加*后会将返回数据分为一个个的元组@unpackdeftest_login(self,username,password,result):'''测试邮箱登录'''self.driver.find_element_by_id('freename').clear()self.driver.find_element_by_id('freename').send_keys(username)self.driver.find_element_by_id('freepassword').clear()self.driver.find_element_by_id('freepassword').send_keys(password)self.driver.find_element_by_link_text('登录').click()freeError=self.driver.find_element_by_xpath('//div[@class="freeError"]//span').text? ? ? ? self.assertEqual(freeError,result)deftearDown(self):self.driver.quit()if__name__=='__main__':unittest.main(verbosity=2)
结果:
test_login_1___________请输入邮箱名__(__main__.MyMailLogin)测试邮箱登录 ... oktest_login_2_______123456____请输入邮箱名__(__main__.MyMailLogin)测试邮箱登录 ... oktest_login_3___123456________您输入的邮箱名格式不正确__(__main__.MyMailLogin)测试邮箱登录 ... ok
需要注意@data( * testdata()) 这里加*后会将返回数据分为一个个的元组,否则unpack无法分解数据,就会报错。比如:
deftestdata():return(('','','请输入邮箱名'),('','123456','请输入邮箱名'),('123456','','您输入的邮箱名格式不正确'))print(*testdata())#三个元组print(testdata())#一个元组
结果:
('', '', '请输入邮箱名') ('', '123456', '请输入邮箱名') ('123456', '', '您输入的邮箱名格式不正确')
(('', '', '请输入邮箱名'), ('', '123456', '请输入邮箱名'), ('123456', '', '您输入的邮箱名格式不正确'))