Pytest执行unittest TestSuite(测试套件)的方法

前言

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
来源:简书
上一篇:unittest系统(六)如何在一个测试类多个测试用例执行中只初始化和清理一次?


下一篇:python基础知识——测试代码