ddt數據驅動常見的用法【多測師_王sir】

一、背景
一般進行接口測試時，每個接口的傳參都不止一種情況，一般會考慮正向、逆向等多種組合，所以在測試一個接口時，通
常會編寫多條case，而這些除了傳參不同外，並沒有什么區別。這個時候就可以利用ddt來管理測試數據，提高代碼復用率。

二、ddt安裝
ddt是python的第三方庫，安裝非常簡單：（基於Python3.7版本）
pip3 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傳參數給測試用例

import unittest
from ddt import ddt,data,unpack

@ddt
class MyTestDdt(unittest.TestCase):
    def setUp(self):
        print('start')

    @data(66,88)#使用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)

運行后結果為：
C:\Python37\python3.exe C:/project/dcs6/decorator/裝飾器.py
test_one_1_66 (__main__.MyTestDdt) ... ok
start
test_one_2_88 (__main__.MyTestDdt) ... ok

the @data number is :66
----------------------------------------------------------------------
end
Ran 2 tests in 0.000s
start

the @data number is :88
OK
end

 
          
              可以看出data分別傳參數66和88給測試用例，然后測試用例執行了兩遍。另外，注意此時打印出的測試用例名稱已經不同， 
           
          
              比如：test_one_1_66和test_one_2_88，最后的66和88，分別是傳遞進去的參數。 
           
          
  2、使用unpack分解list或者tuple 
         

import unittest
from ddt import ddt,data,unpack
@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)

運行結果如下：

  C:\Python37\python3.exe C:/project/dcs6/decorator/裝飾器.py
  test_one_1__1__2_ (__main__.MyTestDdt) ... ok
  start
  test_one_2__4__5_ (__main__.MyTestDdt) ... ok
  the @data number is :(1, 2)

  end
  ----------------------------------------------------------------------
  start
  Ran 2 tests in 0.000s
  the @data number is :(4, 5)

  end
  OK

  可以看出data分別傳元組參數（1，2）和（4，5）給測試用例，然后測試用例執行了兩遍。另外，注意此時打印出的

  測試用例名稱已經不同，比如：test_one_1__ 1__ 2_和test_one_2__ 4__5_ ，最后的1，2和4，5，分別是傳遞進去的參數。

3、用unpack分解字典
import unittest
from ddt import ddt,data,unpack
@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)

運行結果如下：
C:\Python37\python3.exe C:/project/dcs6/decorator/裝飾器.py
start
test_one_1 (__main__.MyTestDdt) ... ok
the @data number is :(1, 2)
end
start
test_one_2 (__main__.MyTestDdt) ... ok
the @data number is :(3, 4)

end
----------------------------------------------------------------------
Ran 2 tests in 0.000s

OK

4、自動化測試實戰
這里以登錄多測師cms后台管理系統，並利用數據驅動的方式帶參進行測試
#coding=utf-8
"""
===========================
Author:多測師_王sir
Time:2020-09-08 13:51
Wechat:xiaoshubass
website:www.duoceshi.cn
===========================
"""

import unittest
from ddt import ddt,data,unpack
from selenium import webdriver

def testdata():
    '''測試數據'''
    return [('admin', '123456'), ('duoceshi', '123456'),
            ('duoceshi1', '123456')]

@ddt
class Cms_Login(unittest.TestCase):
    def setUp(self):
        self.url = 'http://cms.duoceshi.cn/cms/manage/login.do'
        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):
        '''測試登錄多測師cms后台管理系統'''
        self.driver.find_element_by_id('userAccount').send_keys(username)
        self.driver.find_element_by_id('loginPwd').send_keys(password)
        self.driver.find_element_by_id('loginBtn').click()

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

if __name__ == '__main__':
    unittest.main(verbosity=2)

運行結果如下：
C:\Python37\python3.exe C:/project/dcs6/decorator/裝飾器.py
test_login_1___admin____123456__ (__main__.Cms_Login)
測試登錄多測師cms后台管理系統 ... ok
test_login_2___duoceshi____123456__ (__main__.Cms_Login)
測試登錄多測師cms后台管理系統 ... ok
test_login_3___duoceshi1____123456__ (__main__.Cms_Login)
測試登錄多測師cms后台管理系統 ... ok

----------------------------------------------------------------------
Ran 3 tests in 32.658s

OK