Pytest執行unittest TestSuite(測試套件)的方法
阿新 • • 發佈:2021-08-31
前言
TestSuite一直是unittest的靈活與精髓之處,在繁多的測試用例中,可以任意挑選和組合各種用例集,比如smoke用例集、
level1用例集、
webtest用例集、
bug迴歸用例集
等等,當然這些TestSuite需要我們提前定義好,並把用例載入進去。
Pytest採取的是完全不同的用例組織和執行方式。用例的執行主要基於名稱匹配;組織則基於用例目錄,用例命名格式及用例mark標籤, 這種方式省去了麻煩的提前定義TestSuite及載入用例的過程,執行時通過路徑/用例名格式/不同的標籤組合來動態匹配出要執行的用例,使用更加靈活。
然而,從原有的unittest框架轉向pytest懷抱時仍不得不面臨這樣一個問題:我原先定義好的TestSuite怎麼執行?實現方法
主要思路:
①迭代遍歷TestSuite中的所有case得到每個case的路徑test_demo.TestDemo.test_a
②將case路徑轉化為Pytest支援的執行格式test_demo.py::TestDemo::test_a並組成一個case名稱列表供Pytest呼叫。
示例用例: test_demo.py:
import unittest class TestDemo(unittest.TestCase): def test_a(self): print("a") def test_b(self): print("b")
示例測試套件:demo.py:
import unittest import pytest from test_demo import TestDemo suite = unittest.TestSuite() suite.addTests([TestDemo('test_a'), TestDemo('test_b')]) # 因為suite中可能會存在巢狀, 所以我們要迭代取出其中所有的用例: def collect(suite): cases = [] # 用於存放Pytest支援的用例路徑字串 def _collect(tests): # 遞迴,如果下級元素還是TestSuite則繼續往下找if isinstance(tests, unittest.TestSuite): [_collect(i) for i in tests if tests.countTestCases() != 0] else: _path = tests.id().split(".") # case.id()可以獲取用例路徑(字串) _path[0] += ".py" cases.append("::".join(_path)) # 如果下級元素是TestCase,則新增到TestSuite中 _collect(suite) return cases if __name__ == '__main__': cases = collect(suite) pytest.main([*cases, "-v"]) # pytest.main(cases) # 不加額外引數的化可直接執行cases
作者:韓志超
連結:https://www.jianshu.com/p/6a05ccd3ca94
來源:簡書 去期待陌生,去擁抱驚喜。