封装一个unittest多线程执行测试用例的装饰器

这是一个基于Python的unittest框架的多线程执行测试用例的装饰器实现：

import threading
import unittest

def multi_threaded(num_threads):
    """
    A decorator that allows a unittest.TestCase class to be executed
    using multiple threads.
    """
    def decorator(cls):
        # Create a new subclass of the original TestCase class
        class MultiThreadedTestCase(cls):
            def __init__(self, methodName='runTest', thread_id=0):
                super().__init__(methodName)
                self.thread_id = thread_id

            def run(self, result=None):
                # Override the run() method to set the thread id
                # before calling the superclass run() method.
                self._result = result
                self._thread_id = self.thread_id
                super().run(result)

        # Define a new test suite that runs each test case in a separate thread
        def suite():
            loader = unittest.TestLoader()
            suite = unittest.TestSuite()
            for method_name in loader.getTestCaseNames(cls):
                for i in range(num_threads):
                    test_case = MultiThreadedTestCase(method_name, i)
                    suite.addTest(test_case)
            return suite

        # Override the default run() method to run the test suite in multiple threads
        def run(self, result=None):
            threads = []
            for test_case in suite():
                thread = threading.Thread(target=test_case.run, args=(result,))
                threads.append(thread)
                thread.start()

            # Wait for all threads to finish
            for thread in threads:
                thread.join()

        cls.run = run
        return cls

    return decorator

使用示例：

@multi_threaded(num_threads=2)
class MyTestCase(unittest.TestCase):
    def test_addition(self):
        self.assertEqual(1 + 1, 2)

if __name__ == '__main__':
    unittest.main()

在上面的示例中，MyTestCase 类被装饰为多线程执行测试用例，num_threads 参数指定了线程数。test_addition 方法将在两个线程中执行，每个线程都会运行一次该方法。