如何在behave的after_all中获取失败测试用例并实现重试？

要实现失败用例的重试，核心思路是先在每个场景跑完后收集失败用例的关键标识，再在after_all钩子中读取这些信息并触发重试。具体步骤如下：

1. 收集失败场景信息 #

在environment.py的after_scenario钩子中，捕获失败的场景并记录可靠的唯一标识（比如feature文件路径+场景行号，比场景名称更精准），存储到context.config.userdata这个全局可访问的存储区：

def after_scenario(context, scenario):
    if scenario.status == "failed":
        # 用feature文件路径+场景行号作为唯一标识，避免场景名称重复的问题
        failed_identifier = f"{scenario.feature.filename}:{scenario.line}"
        # 初始化存储失败用例的列表
        if "failed_cases" not in context.config.userdata:
            context.config.userdata["failed_cases"] = []
        context.config.userdata["failed_cases"].append(failed_identifier)

2. 在after_all中触发重试 #

在after_all钩子中读取收集到的失败用例，通过subprocess调用behave命令，指定只运行这些失败的场景：

import subprocess

def after_all(context):
    failed_cases = context.config.userdata.get("failed_cases", [])
    if not failed_cases:
        print("没有失败的测试用例，无需重试")
        return

    print(f"准备重试 {len(failed_cases)} 个失败用例：{failed_cases}")
    # 构造behave重试命令
    cmd = ["behave"]
    for case in failed_cases:
        feature_path, line_num = case.split(":", 1)
        # 通过--include指定feature文件，--line指定场景行号
        cmd.extend(["--include", feature_path, "--line", line_num])
    
    # 执行重试命令，捕获输出结果
    retry_result = subprocess.run(cmd, capture_output=True, text=True)
    print("\n=== 重试输出 ===")
    print(retry_result.stdout)
    if retry_result.stderr:
        print("\n=== 重试错误输出 ===")
        print(retry_result.stderr)

注意事项 #

• 优先用场景行号：场景名称可能重复，而行号是唯一的，能精准定位到失败场景
• 路径适配：如果feature文件不在当前工作目录，需要在subprocess.run中指定cwd参数，确保命令能找到目标文件
• 多次重试扩展：如果需要多次重试，可以在after_all中加入循环逻辑，每次重试后重新收集失败用例，判断是否需要继续重试

内容的提问来源于stack exchange，提问作者Andrei Starunov