Python中如何在单测运行超过x秒时终止coverage run进程？

coverage run本身没有提供单测试用例超时终止的参数，它的作用是包裹后续的pytest执行过程、统计代码覆盖率，因此你只需要给pytest加上超时控制逻辑即可，两种常用实现方案如下：

方案1：使用pytest-timeout插件（推荐） #

• 先安装插件：
  pip install pytest-timeout
• 修改原执行命令，新增--timeout=X参数，将X替换为你设置的单测试最大运行时长（单位：秒），例如设置10秒超时的命令如下：

coverage run --source . -m pytest --timeout=10 --junitxml=unit_test_results.xml --disable-pytest-warnings --ignore=<some_file_path>

• 附加配置：如果需要超时后直接终止整个测试进程而非仅终止超时用例，可额外添加参数--timeout_method=thread。

方案2：通过pytest钩子实现（无需额外安装依赖） #

如果你的运行环境不允许安装第三方插件，可以通过pytest的内置钩子实现超时控制：

• 在项目根目录新建conftest.py文件，添加如下代码，将MAX_TEST_DURATION的值修改为你需要的超时秒数：

import pytest
from threading import Timer

MAX_TEST_DURATION = 10 # 自定义单测试最大运行时长，单位秒

def pytest_runtest_call(item):
    def timeout_handler():
        pytest.fail(f"测试用例 {item.name} 运行时长超过阈值，强制终止")
    
    timer = Timer(MAX_TEST_DURATION, timeout_handler)
    timer.start()
    try:
        item.runtest()
    finally:
        timer.cancel()

• 无需修改原执行命令，直接运行原来的coverage run命令即可生效。

两种方案触发超时终止时，coverage进程都会同步退出，已经生成的覆盖率数据不会丢失。

内容的提问来源于stack exchange，提问作者Abinash Biswal