基於Python的接口自動化unittest測試框架和ddt數據驅動詳解
引言
在編寫接口自動化用例時，我們一般針對一個接口建立一個.py文件，一條接口測試用例封裝為一個函數（方法），但是在批量執行的過程中，如果其中一條出錯，後面的用例就無法執行，還有在運行大量的接口測試用例時測試數據如何管理和加載。針對測試用例加載以及執行控制，python語言提供瞭unittest單元測試框架，將測試用例編寫在unittest框架下，使用該框架可以單個或者批量加載互不影響的用例執行及更靈活的執行控制，對於更好的進行測試數據的管理和加載，這裡我們引入數據驅動的模塊：ddt，測試數據和測試腳本的分離，通過ddt數據驅動來加載測試數據到測試用例腳本中，通常在接口自動化測試中會將unittest和ddt結合起來使用，從而實現測試用例腳本和測試數據的載入來完成測試的執行。下面來看看unittest框架和ddt這兩個模塊具體的應用。
一、unittest測試框架
unittest單元測試框架是python語言的一套標準模塊，封裝提供瞭諸多操作測試用例和用例加載、測試前置和場景恢復以及測試結果輸出等一系列類和方法。
1.unittest框架中最核心四個組件概念：
（1）TestCase：測試用例類，編寫測試用例腳本時需要繼承該類，從而具有該類的屬性和方法，一個TestCase實例就是一個測試用例，其中測試用例方法都以test開頭。
（2）TestSuite：測試集，也就是測試用例的集合，用來組織用例。
（3）testrunner：用來執行測試用例，並返回測試用例的執行結果，可以用圖形或者文本將測試結果形象地展現出來，HTMLTestRunner用來生成圖形化的報告，TextTestRunner用來生成簡單的文本測試結果。
（4）testfixure：測試夾件，主要用於測試用例的前置初始化和執行後的銷毀。
2.testcase—-測試用例
- 新建一個的.py測試用例文件必須是test開頭，如test_login.py，主要後續用於識別測試用例文件編寫測試用例的類，必須繼承unittest.TestCase，做為測試類
- 測試類中用例的方法名稱必須以test開頭，用於識別測試用例數
- 測試類中的用例執行順序，按照以test開頭的方法後的Ascill碼順序執行(0~9,A~Z,a~z)
3.testfixure—-測試夾件
- 也叫測試夾具，主要是用例前置的初始化以及執行後的銷毀
- 測試夾件提供兩種方法，一種是類級別的：setup(）和teardown()，一種是方法級別的：setUpClass()和tearDownClass()
- 類級別的測試夾件，每一條測試用例執行之前與之後都要運行一次setup(）和teardown()；方法級別的測試夾件，所有測試用例執行之前到執行完成隻運行一次setUpClass()和tearDownClass()
下面通過簡單的代碼示例看看TestCase與TestFixure的使用
（1）使用setup(）和teardown()，創建test_666.py文件編輯如下代碼：
import unittest
class test_unittest(unittest.TestCase):
def setUp(self):
print("測試環境初始化，開始執行setup")
def tearDown(self):
print("測試執行完成，運行teardown")
print("------------------------------")
def test_a(self):
print("開始執行test_a用例")
def test_A(self):
print("開始執行test_A用例")
def test_1(self):
print("開始執行test_1用例")
def notest_1(self):
print("不執行notest_1用例")
if __name__ == "__main__":
unittest.main()
執行後，輸出如下：
在代碼中我們編寫瞭4個def用例方法，隻執行瞭3個def，因為最後一個def不是test開頭。可以看到每執行一個def用例，setup(）和teardown()都會執行一次，其中按照執行順序：test_1最先執行，test_A其後，test_a最後執行
（2）使用setUpClass()和tearDownClass()
對於setUpClass()和tearDownClass()我們隻需將上面代碼，稍微修改即可
import unittest
class test_unittest(unittest.TestCase):
@classmethod
def setUpClass(cls):
print("測試環境初始化，開始執行setup")
@classmethod
def tearDownClass(cls):
print("測試執行完成，運行teardown")
print("------------------------------")
def test_a(self):
print("開始執行test_a用例")
def test_A(self):
print("開始執行test_A用例")
def test_1(self):
print("開始執行test_1用例")
if __name__ == "__main__":
unittest.main()
運行效果如下：
可以看到所有用例都執行完後，setUpClass()和tearDownClass()隻運行瞭一次。因此在編寫一個測試腳本時，裡面寫瞭多個測試用例，
這時我們希望的是所有用例執行完成後再銷毀環境，這時使用setUpClass()和tearDownClass()就比較好瞭。
3.TestSuite—-測試集
unittest框架下提供瞭unittest.TestSuite()和unittest.TestLoader()類，這兩個類下封裝瞭加載用例的方法，用於加載測試用例到測試集中
（1）unittest.TestSuite()提供單個用例加載方法
addTest()：單個用例加載，當然也可以將多個用例的方法名放入列表中添加到addTest()中，加載多條測試用例
（2）unittest.TestLoader()提供批量加載或發現用例的方法
loadTestsFromTestCase(測試類名)：添加一個測試類
loadTestsFromModule(模塊名)：添加一個模塊
discover(測試用例的所在目錄)：指定目錄去加載，會自動尋找這個目錄下所有符合命名規則的測試用例
4.testrunner—-測試運行
testrunner就是用來執行測試用例的，並且可以生成相應的測試報告。測試報告有兩種展示形式，一種是text文本，一種是html格式。
html格式的就是HTMLTestRunner瞭，HTMLTestRunner是Python標準庫的unittest框架的一個擴展，它可以生成一個直觀清晰的HTML測試報告。使用的前提就是要下載HTMLTestRunner.py，下載完後放在python的安裝目錄下的scripts目錄下即可。
通過代碼示例看看testsuite和testrunner這兩個組件的使用，上面的test_666.py用例文件我們已經寫好瞭3條用例瞭，現在我們來加載這些用例
新建run_case.py文件，該文件和test_666.py文件放置在同一個包文件：test下，run_case.py文件編輯如下代碼運行：
import unittest
from test.test_666 import test_unittest
# 單個用例加載
suite = unittest.TestSuite()
case1 = test_unittest('test_1')
case2 = test_unittest('test_a')
suite.addTest(case1)
suite.addTest(case2)
print(suite)
print("------------------")
# 批量用例加載
case_path = r"E:\api_test\test"
# 按文件路徑加載，註意該文件為包文件即文件下有__init__.py
all_case = unittest.defaultTestLoader.discover(case_path,pattern="test_666*.py",top_level_dir=None)
all_case1 = unittest.defaultTestLoader.loadTestsFromTestCase(test_unittest) # 按類名稱加載
print(all_case)
print("------------------")
print(all_case1)
輸出結果如下：
E:\api_test\Scripts\python.exe E:/api_test/test_bak/run_case.py
<unittest.suite.TestSuite tests=[<test.test_666.test_unittest testMethod=test_1>, <test.test_666.test_unittest testMethod=test_a>]>
——————
<unittest.suite.TestSuite tests=[<unittest.suite.TestSuite tests=[<unittest.suite.TestSuite tests=[<test_666.test_unittest testMethod=test_1>, <test_666.test_unittest testMethod=test_A>, <test_666.test_unittest testMethod=test_a>]>]>]>
——————
<unittest.suite.TestSuite tests=[<test.test_666.test_unittest testMethod=test_1>, <test.test_666.test_unittest testMethod=test_A>, <test.test_666.test_unittest testMethod=test_a>]>
Process finished with exit code 0
通過unittest框架下提供的加載用例的諸多方法，我們就可以單個或者批量加載用例，後續可以將加載的用例集引入到HTMLTestRunner.py模塊生成可視化的測試報告
5.assert—-測試斷言
無論是什麼樣的測試用例，最後都需要有用例執行後的驗證，在接口自動化測試中我們執行完接口用例也需要驗證斷言用例執行是否滿足我們的預期。unittest提供瞭豐富的斷言方法，常見的斷言如下表：
二、ddt數據驅動
- @ddt：類的裝飾器，繼承的是TestCase類
- @data()：@data裝飾符可以把參數當成測試數據，參數可以是單個值、列表、元祖、字典這些類型，用於輸入測試數據
- @unpack：分解數據標志，主要是把元祖和列表解析成多個參數
- @file_data()：輸入文件，如json或者yaml類型文件
（1）輸入簡單的參數：單個值、列表、元祖、字典
import unittest
from ddt import data,unpack,ddt
@ddt
class myddt(unittest.TestCase):
@data("123") # 單個值
def test1(self,testdata1):
print(testdata1)
print("------------------")
@data([1,2,3],[4,5,6]) # 列表
def test2(self,testdata2):
print(testdata2)
print("------------------")
"""
@data((1, 2, 3)) # 元組
def test2(self, testdata3):
print(testdata3)
print("------------------")
@data({'zhangshan':1,'wangwu':2,'lisi':3}) # 字典
def test2(self, testdata4):
print(testdata4)
print("------------------")
"""
if __name__ == '__main__':
unittest.main()
（2）使用@unpack對復雜數據結構，如元組、列表數據進行分解
代碼示例：
import unittest
from ddt import data,unpack,ddt
@ddt
class myddt(unittest.TestCase):
@data([1,2],[3,4]) # 列表
@unpack
def test2(self, testdata1,testdata2):
print("拆解的第一個參數：",testdata1)
print("拆解的第二個參數：", testdata2)
print("------------------")
if __name__ == '__main__':
unittest.main()
運行後輸出如下：
（3）使用@file_data()輸入文件格式測試數據
編輯一個data.json的文件，代碼示例：
import unittest from ddt import file_data,ddt @ddt class myddt(unittest.TestCase): @file_data(r"E:\api_test\test\data.json") def test1(self, *value): print(value) if __name__ == '__main__': unittest.main()
通過ddt和unittest框架的結合就可以實現測試用例腳本編寫、測試執行控制以及測試數據的批量加載，從而完成不同接口測試用例的批量執行和覆蓋測試不同測試場景。
到此這篇關於基於Python的接口自動化unittest測試框架和ddt數據驅動詳解的文章就介紹到這瞭,更多相關Python的接口自動化ddt數據驅動內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet！
推薦閱讀：
- python unittest單元測試的步驟分析
- python中的unittest框架實例詳解
- Pytest執行unittest TestSuite(測試套件)的實現方法
- python 如何在測試中使用 Mock
- Python接口自動化系列之unittest結合ddt的使用教程詳解