数据驱动—ddt

Posted 2023-03-28

参考技术A ddt是python的第三方库，安装非常简单：

pip install ddt

ddt官方文档地址： https://pypi.org/project/ddt/

1、ddt模块包含类的装饰器ddt和两个方法装饰器data

1.ddt.ddt：装饰类，也就是继承TestCase的类。

2.ddt.data：装饰测试方法，参数是一系列的值。

3.ddt.file_data：装饰测试方法，参数是文件名。文件可以是json或者yaml类型。

注意，如果文件是以“.yml”或者".yaml"结尾，ddt会作为yaml类型处理，其他文件都会作为json文件处理。

如果文件是列表，列表的值会作为测试用例参数，同时，会作为测试用例方法名后缀显示。

如果文件是字典，字典的key会作为测试用例方法的后缀显示，字典的value会作为测试用例参数。

1.ddt.unpack：传递的是复杂的数据结构时使用，比如使用列表或者元组，添加unpack后，ddt会自动把元组或者列表对应到多个参数上。

2、测试用例方法名生成规则

使用ddt后，会产生一个新的测试用例方法名，即定义的测试用例方法名，比如：def test_one()，这里就是test_one。

data：如果传递过来的数据存在__ name__ 属性，则这里就是该数据的__ name__ 值。如果未定义__ name__ 属性，ddt会尽量将传递过来的值转化为python标识符，作为data显示。

1、使用data传参数给测试用例

from ddt import ddt,data,file_data,unpack

import unittest

@ddt

class MyTestDdt(unittest.TestCase):

    def setUp(self):

        print('start')

    @data(2,4)#使用data传参数给测试用例

    def test_one(self,value):

        print(f'the @data number is :value')

    def tearDown(self):

        print('end')

if __name__ == '__main__':

    unittest.main(verbosity=2)

结果：

start

test_one_1_2 (__main__.MyTestDdt) ... ok

the @data number is :2

end

start

the @data number is :4

test_one_2_4 (__main__.MyTestDdt) ... ok

end

----------------------------------------------------------------------

Ran 2 tests in 0.000s

OK

2、使用unpack分解list或者tuple

from ddt import ddt,data,file_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(f'the @data number is :value1,value2')

    def tearDown(self):

        print('end')

if __name__ == '__main__':

    unittest.main(verbosity=2)

结果：

start

the @data number is :(1, 2)

end

test_one_1__1__2_ (__main__.MyTestDdt) ... ok

start

the @data number is :(4, 5)

test_one_2__4__5_ (__main__.MyTestDdt) ... ok

end

----------------------------------------------------------------------

Ran 2 tests in 0.002s

OK

3、用unpack分解字典

from ddt import ddt,data,file_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(f'the @data number is :value1,value2')

    def tearDown(self):

        print('end')

if __name__ == '__main__':

    unittest.main(verbosity=2)

结果：

test_one_1 (__main__.MyTestDdt) ... ok

start

test_one_2 (__main__.MyTestDdt) ... ok

the @data number is :(1, 2)

----------------------------------------------------------------------

end

Ran 2 tests in 0.000s

start

the @data number is :(3, 4)

OK

end

4、自动化测试实战

这里我们登录新浪邮箱，并利用数据驱动的方式带参进行测试

import unittest

from ddt import ddt,data,unpack

from selenium import webdriver

def testdata():

    '''测试数据'''

    return [('', '', '请输入邮箱名'), ('', '123456', '请输入邮箱名'),

            ('123456', '', '您输入的邮箱名格式不正确')]

@ddt

class MyMailLogin(unittest.TestCase):

    def setUp(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()) #这里加*后会将返回数据分为一个个的元组

    @unpack

    def test_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)

    def tearDown(self):

        self.driver.quit()

if __name__ == '__main__':

    unittest.main(verbosity=2)

结果：

test_login_1___________请输入邮箱名__ (__main__.MyMailLogin)

测试邮箱登录 ... ok

test_login_2_______123456____请输入邮箱名__ (__main__.MyMailLogin)

测试邮箱登录 ... ok

test_login_3___123456________您输入的邮箱名格式不正确__ (__main__.MyMailLogin)

测试邮箱登录 ... ok

需要注意@data( * testdata()) 这里加*后会将返回数据分为一个个的元组，否则unpack无法分解数据，就会报错，列如：

def testdata():

    return (('', '', '请输入邮箱名'), ('', '123456', '请输入邮箱名'),

            ('123456', '', '您输入的邮箱名格式不正确'))

print(*testdata()) #三个元组

print(testdata()) #一个元组

结果：

('', '', '请输入邮箱名') ('', '123456', '请输入邮箱名') ('123456', '', '您输入的邮箱名格式不正确') (('', '', '请输入邮箱名'), ('', '123456', '请输入邮箱名'), ('123456', '', '您输入的邮箱名格式不正确'))

笔者创建了一个测试交流群，如果对软件测试、接口测试、自动化测试、面试经验交流感兴趣可以加测试交流群：829792258，还会有同行一起技术交流

Python ddt数据驱动

import unittest
from time import sleep

import ddt
# 导入ddt模块
from selenium import webdriver


@ddt.ddt
# 声明使用ddt
class MyTestCase(unittest.TestCase):

    def setUp(self):
        self.driver = webdriver.Firefox()
        self.base_url = "https://www.baidu.com/"
        self.driver.get(self.base_url)
        self.driver.maximize_window()
        sleep(2)

    @ddt.data(["中国", "中国_百度百科"],
              ["美国", "美国_百度百科"],
              ["英国", "英国_百度百科"])
    # 测试数据

    @ddt.unpack
    # 对测试数据进行解包并传递
    def test_something(self, searchTerm, searchResult):
        driver = self.driver
        driver.find_element_by_xpath(".//*[@id=‘kw‘]").send_keys(searchTerm)
        driver.find_element_by_xpath(".//*[@id=‘su‘]").click()
        sleep(2)
        responseText = driver.find_element_by_xpath(".//*[@id=‘1‘]/h3/a").text
        self.assertEqual(responseText, searchResult)

    def tearDown(self):
        self.driver.close()
        self.driver.quit()

if __name__ == ‘__main__‘:
    unittest.main()