unittest错误重跑与测试用例跳过机制

错误重跑与测试用例跳过机制

在 unittest 框架中，错误重跑机制和测试用例跳过机制是两个重要的功能，它们分别用于处理不稳定测试和选择性执行测试。以下是对这两个机制的详细介绍：

6. 错误重跑机制

错误重跑机制允许在测试用例失败时自动重试，有助于过滤因环境波动导致的偶发失败（Flaky Tests）。unittest 原生不支持重跑，但可以通过以下方式实现：

6.1 自定义重试装饰器

通过装饰器为单个测试方法添加重试逻辑：

import unittest import functools def retry(tries=3, exceptions=(AssertionError,)):     """重试装饰器，用于测试方法"""     def decorator(func):         @functools.wraps(func)         def wrapper(self, *args, **kwargs):             attempt = 0             while attempt < tries:                 try:                     return func(self, *args, **kwargs)                 except exceptions as e:                     attempt += 1                     if attempt == tries:                         raise                     print(f"测试失败，重试 ({attempt}/{tries}): {e}")         return wrapper     return decorator class TestExample(unittest.TestCase):     @retry(tries=3)  # 失败时重试3次     def test_flaky(self):         import random         self.assertTrue(random.choice([True, False]))  # 模拟不稳定测试 if __name__ == '__main__':     unittest.main()

6.2 自定义 TestCase 基类

通过继承 unittest.TestCase 并重写 run() 方法，为所有测试用例添加重试逻辑：

import unittest class RetryTestCase(unittest.TestCase):     max_retries = 3  # 全局重试次数     def run(self, result=None):         retry_count = 0         success = False                  while not success and retry_count <= self.max_retries:             # 重置结果对象             if result:                 result._mirrorOutput = False  # 避免重复输出                              # 执行测试             super().run(result)                          # 检查是否需要重试             if result.failures or result.errors:                 retry_count += 1                 print(f"测试失败，重试 ({retry_count}/{self.max_retries})")                 # 清空当前失败记录                 result.failures = []                 result.errors = []             else:                 success = True class TestMath(RetryTestCase):     def test_flaky(self):         import random         self.assertEqual(random.randint(1, 10) % 2, 0)  # 50% 概率失败

6.3 使用第三方库（如 XTestRunner）

XTestRunner 提供了更完善的重跑功能，支持生成包含重跑记录的 HTML 报告：

import unittest from XTestRunner import HTMLTestRunner class TestExample(unittest.TestCase):     def test_flaky(self):         import random         self.assertTrue(random.choice([True, False])) if __name__ == '__main__':     suite = unittest.TestSuite()     suite.addTest(TestExample("test_flaky"))          with open('report.html', 'wb') as f:         runner = HTMLTestRunner(             stream=f,             title="测试报告",             rerun=3  # 失败重试3次         )         runner.run(suite)

7. Case 跳过机制

unittest 提供了内置的跳过机制，允许根据条件或版本选择性执行测试用例。

7.1 无条件跳过

使用 @unittest.skip(reason) 装饰器直接跳过测试：

import unittest class TestFeatures(unittest.TestCase):     @unittest.skip("该功能尚未实现")     def test_feature_x(self):         pass  # 此测试会被跳过          def test_feature_y(self):         self.assertTrue(True)

7.2 条件性跳过

使用 @unittest.skipIf(condition, reason) 或 @unittest.skipUnless(condition, reason)：

import unittest import sys class TestPlatform(unittest.TestCase):     @unittest.skipIf(sys.platform != 'linux', "仅在 Linux 上运行")     def test_linux_feature(self):         pass          @unittest.skipUnless(sys.version_info >= (3, 8), "需要 Python 3.8+")     def test_python38_feature(self):         pass

7.3 动态跳过

在测试方法中使用 self.skipTest(reason) 动态跳过：

import unittest import requests class TestAPI(unittest.TestCase):     def test_external_api(self):         try:             response = requests.get("https://api.example.com")         except requests.ConnectionError:             self.skipTest("无法连接到外部 API")                  self.assertEqual(response.status_code, 200)

7.4 跳过整个测试类

在类定义前添加装饰器：

import unittest @unittest.skip("模块重构中，暂不执行测试") class TestLegacyCode(unittest.TestCase):     def test_case1(self):         pass  # 整个类的测试都会被跳过

总结

• 错误重跑机制：用于处理不稳定测试，通过自定义装饰器、基类或第三方库实现。
• Case 跳过机制：用于选择性执行测试，通过 @skip、@skipIf、@skipUnless 装饰器或 self.skipTest() 方法实现。

这两种机制结合使用可以显著提高测试效率，减少无效测试的干扰，让测试聚焦于真正需要验证的功能。