Unittest 多进程+装饰器+额外传参给测试类

最新推荐文章于 2023-04-01 03:17:06 发布

qq_27525737

最新推荐文章于 2023-04-01 03:17:06 发布

阅读量666

点赞数

本文链接：https://blog.csdn.net/qq_27525737/article/details/108561368

版权

本文介绍了如何在Unittest中实现测试类的参数化，通过多进程执行测试，并利用装饰器来格式化日志输出。首先，展示了如何使用unittest_utils.py为测试类添加额外参数，然后讲解了run.py中多进程执行测试的实现，最后展示了logger.py和test_demo.py的代码示例及其执行结果。

摘要由CSDN通过智能技术生成

一、给Unittest测试类添加参数

调用unittest.main()时无法传入额外的参数给我们的测试类时，可以使用如下工具给我们的测试类添加额外的参数：

unittest_utils.py代码：

import unittest


class ParametrizedTestCase(unittest.TestCase):
    """ TestCase classes that want to be parametrized should
        inherit from this class.
    """

    def __init__(self, method_name='runTest', param=None):
        super(ParametrizedTestCase, self).__init__(method_name)
        self.param = param

    @staticmethod
    def parametrize(test_case_class, param=None):
        """ Create a suite containing all tests taken from the given
            subclass, passing them the parameter 'param'.
        """
        test_loader = unittest.TestLoader()
        test_names = test_loader.getTestCaseNames(test_case_class)
        suite = unittest.TestSuite()
        for name in test_names:
            suite