前言
TestSuite一直是unittest的靈活與精髓之處, 在繁多的測試用例中, 可以任意挑選和組合各種用例集, 比如smoke用例集, level1用例集, webtest用例集, bug回歸用例集等等, 當然這些TestSuite需要我們提前定義好, 并把用例加載進去.
Pytest采取的是完全不同的用例組織和運行方式.用例的運行主要基于名稱匹配, 組織則基于用例目錄, 用例命名格式及用例mark標簽, 這種方式省去了麻煩的提前定義TestSuite及加載用例的過程, 執行時通過路徑/用例名格式/不同的標簽組合來動態匹配出要執行的用例, 使用更加靈活.
然而, 從原有的unittest框架轉向pytest懷抱時仍不得不面臨這樣一個問題:
我原先定義好的TestSuite怎么執行?
在Pytest官方文檔: 與原有的測試套件一起使用 (原文鏈接 | 翻譯) 一節中, 提到Pytest支持大多數測試套件, 然后怎么使用卻顧左右而言他, 相當于什么都沒寫.
實現方法
下面是個人研究的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
