如何在BDD Behave框架中跟踪步骤执行超时？

要实现**无需修改步骤函数、全局监控每个步骤执行时长（超15分钟即失败）**的需求，可以通过Behave的before_step钩子结合线程监控的方式来实现，核心思路是用线程包裹步骤执行逻辑，主线程等待超时时间，超时则终止并标记步骤失败。

实现步骤 #

1. 编写环境控制文件（environment.py） #

在项目根目录的environment.py中添加以下代码，全局替换步骤执行逻辑，加入超时监控：

import threading
import traceback
from behave import *

class StepTimeoutMonitor:
    def __init__(self, timeout_seconds):
        self.timeout = timeout_seconds
        self.step_thread = None
        self.step_result = None
        self.step_exception = None
        self.traceback_info = None

    def _execute_step(self, step_func, context, args, kwargs):
        """在子线程中执行原始步骤函数"""
        try:
            self.step_result = step_func(context, *args, **kwargs)
        except Exception as e:
            self.step_exception = e
            self.traceback_info = traceback.format_exc()

    def run_with_timeout(self, step_func, context, args, kwargs):
        """启动线程并等待超时"""
        self.step_thread = threading.Thread(
            target=self._execute_step,
            args=(step_func, context, args, kwargs),
            daemon=True
        )
        self.step_thread.start()
        # 等待指定超时时间
        self.step_thread.join(self.timeout)
        
        if self.step_thread.is_alive():
            # 步骤超时，抛出异常标记失败
            raise TimeoutError(f"步骤执行超时，已超过{self.timeout}秒（15分钟）")
        if self.step_exception:
            # 步骤自身抛出异常，重新抛出保留原始错误信息
            print(self.traceback_info)
            raise self.step_exception

def before_step(context, step):
    # 设置超时时间为15分钟（900秒）
    TIMEOUT_SECONDS = 900
    # 保存原始步骤执行函数
    original_step_func = step.func
    # 替换为带超时监控的包装函数
    def wrapped_step_execution(*args, **kwargs):
        monitor = StepTimeoutMonitor(TIMEOUT_SECONDS)
        monitor.run_with_timeout(original_step_func, context, args, kwargs)
    step.func = wrapped_step_execution

方案说明 #

• 全局生效：通过before_step钩子自动为所有步骤添加超时监控，无需修改任何步骤实现代码
• 超时检测：将步骤执行放到子线程中，主线程等待900秒，若子线程仍在运行则抛出TimeoutError，Behave会自动标记该步骤为失败
• 保留原始错误：如果步骤自身抛出异常，会重新抛出并打印堆栈信息，不影响原有错误排查

注意事项 #

• 由于Python线程无法强制终止，超时后子线程会以守护线程身份随主线程结束而终止，若步骤涉及资源占用（如数据库连接、文件句柄），建议在步骤函数中添加资源释放逻辑，或在after_step钩子中补充清理操作
• 该方案对IO阻塞型步骤的超时检测效果最佳；若为CPU密集型任务，可结合signal模块（仅Linux/macOS支持）实现更强制的超时终止

内容的提问来源于stack exchange，提问作者Goutham A G