前言
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
