Java/JUnit/Selenium框架中Cucumber场景失败前重启方案问询

我明白你要的是在Cucumber的@After钩子阶段发现测试要失败时，立刻重启整个场景/用例——之前那些插件和规则没搞定确实头疼。下面给你两个经过验证的方案，分别适配JUnit和TestNG环境，应该能满足你的需求：

方案一：基于JUnit自定义重试规则与运行器（精准控制@After失败重试） #

这个方案通过自定义Cucumber运行器+全局重试状态管理，能精准捕获@After阶段的失败并触发场景重试，而不是重新跑所有测试。

1. 全局重试状态管理器 #

先创建一个工具类，用来跟踪当前场景的重试次数和是否需要重试的标记：

public class RetryManager {
    // 线程本地变量，避免多场景并行时的状态混乱
    private static final ThreadLocal<Integer> retryCount = ThreadLocal.withInitial(() -> 0);
    private static final ThreadLocal<Boolean> needRetry = ThreadLocal.withInitial(() -> false);
    // 可根据需求调整最大重试次数
    private static final int MAX_RETRIES = 3;

    // 判断是否需要重试当前场景
    public static boolean shouldRetry() {
        return needRetry.get() && retryCount.get() < MAX_RETRIES;
    }

    // 标记当前场景需要重试
    public static void markNeedRetry() {
        needRetry.set(true);
        retryCount.set(retryCount.get() + 1);
    }

    // 重置重试状态（场景成功或重试结束时调用）
    public static void reset() {
        retryCount.set(0);
        needRetry.set(false);
    }
}

2. 自定义@After钩子捕获失败 #

在你的钩子类里，通过Scenario对象判断是否失败，失败时标记需要重试：

import io.cucumber.java.After;
import io.cucumber.java.Scenario;

public class RetryHooks {

    @After
    public void afterScenario(Scenario scenario) {
        if (scenario.isFailed()) {
            // 当@After执行完发现场景失败，标记需要重试
            RetryManager.markNeedRetry();
            System.out.println("场景[" + scenario.getName() + "]失败，准备重试...");
        } else {
            // 场景成功，重置重试状态
            RetryManager.reset();
        }
    }
}

3. 自定义Cucumber运行器实现重试逻辑 #

创建一个继承自JUnit Runner的类，循环执行直到场景成功或达到最大重试次数：

import org.junit.runner.Runner;
import org.junit.runner.notification.RunNotifier;
import io.cucumber.junit.Cucumber;

public class RetryCucumberRunner extends Runner {
    private final Cucumber cucumberRunner;

    // 构造方法，初始化原生Cucumber运行器
    public RetryCucumberRunner(Class<?> clazz) throws Exception {
        this.cucumberRunner = new Cucumber(clazz);
    }

    @Override
    public void run(RunNotifier notifier) {
        do {
            RetryManager.reset();
            // 运行Cucumber测试
            cucumberRunner.run(notifier);
        // 如果标记需要重试且未超过次数，循环执行
        } while (RetryManager.shouldRetry());
    }
}

4. 配置测试运行类 #

将你的测试运行类的@RunWith注解改为自定义的运行器：

import org.junit.runner.RunWith;
import io.cucumber.junit.CucumberOptions;

@RunWith(RetryCucumberRunner.class)
@CucumberOptions(
        features = "src/test/resources/features", // 你的feature文件路径
        glue = "com.yourpackage.steps", // 你的步骤定义包路径
        plugin = {"pretty", "html:target/cucumber-reports"}
)
public class TestRunner {
}

方案二：基于TestNG的重试分析器（更简洁的实现） #

如果你可以切换到TestNG环境，这个方案会更简洁，TestNG的原生重试机制能直接捕获包括@After阶段在内的场景失败。

1. 实现TestNG重试分析器 #

创建一个类实现IRetryAnalyzer，控制重试逻辑：

import org.testng.IRetryAnalyzer;
import org.testng.ITestResult;

public class CucumberRetryAnalyzer implements IRetryAnalyzer {
    private int retryCount = 0;
    private static final int MAX_RETRIES = 3;

    @Override
    public boolean retry(ITestResult result) {
        // 如果场景失败且未达到最大重试次数，返回true触发重试
        if (!result.isSuccess() && retryCount < MAX_RETRIES) {
            retryCount++;
            System.out.println("重试场景[" + result.getName() + "]，第" + retryCount + "次");
            return true;
        }
        return false;
    }
}

2. 配置Cucumber+TestNG运行类 #

继承AbstractTestNGCucumberTests，并重写runScenario方法添加重试注解：

import io.cucumber.testng.AbstractTestNGCucumberTests;
import io.cucumber.testng.CucumberOptions;
import org.testng.annotations.Test;

@CucumberOptions(
        features = "src/test/resources/features",
        glue = "com.yourpackage.steps",
        plugin = {"pretty", "html:target/cucumber-reports"}
)
public class TestNGTestRunner extends AbstractTestNGCucumberTests {
    @Override
    @Test(retryAnalyzer = CucumberRetryAnalyzer.class)
    public void runScenario() {
        super.runScenario();
    }
}

注意事项 #

• 如果用JUnit方案，要确保你的cucumber-junit版本在6.x及以上，避免API不兼容问题；
• 并行测试时，ThreadLocal的状态管理能避免场景间的干扰，但要注意并行策略的配置；
• 重试时记得重置测试上下文（比如浏览器实例、数据库连接等），避免脏数据影响重试结果。

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