并行运行Cucumber时如何避免钩子重复并管理WebDriver实例

我尝试用maven-surefire-plugin并行运行Cucumber特性文件，pom.xml里的并行配置如下：

<plugin>
    <groupId>org.apache.maven.plugins</groupId>
    <artifactId>maven-surefire-plugin</artifactId>
    <version>3.0.0</version>
    <configuration>
        <parallel>methods</parallel>
        <useUnlimitedThreads>true</useUnlimitedThreads>
    </configuration>
</plugin>

目前每个步骤定义类都通过TestSetUp创建专属WebDriver实例，代码如下：

public class TestSetUp {

    private static WebDriver driver;

    public TestSetUp() {
        driver = createChromeDriver();
    }

    private WebDriver createWebDriver() {
        driver = new ChromeDriver();
        return driver;
    }
}

步骤定义类示例：

public class Test_A {

    private WebDriver driver;

    @Before
    public void beforeScenario() {
        TestSetUp setUp = new TestSetUp();
        driver = setUp.driver;

    }

    @When("The user navigates to the site")
    public void openSite() {
        driver.get("https://www.examplesite.com");
    }

    @And("The user executes task on the site")
    public void executeTask() {
        // 网页操作逻辑
    }

    @After
    public void tearDown() {
        driver.close();
    }
}

现在需要并行运行多个特性文件（对应多个步骤定义类），想找个更优方案：既能避免@Before和@After钩子代码重复，又能为每个步骤定义类分配独立的WebDriver实例？

1. 用ThreadLocal解决WebDriver线程安全问题 #

你当前TestSetUp里的static WebDriver会导致多线程并行时实例共享，必须换成ThreadLocal存储每个线程专属的WebDriver：

public class WebDriverManager {
    // ThreadLocal保证每个线程有独立的WebDriver实例
    private static final ThreadLocal<WebDriver> driverThreadLocal = new ThreadLocal<>();

    // 获取当前线程的WebDriver，不存在则创建
    public static WebDriver getDriver() {
        if (driverThreadLocal.get() == null) {
            driverThreadLocal.set(createChromeDriver());
        }
        return driverThreadLocal.get();
    }

    private static WebDriver createChromeDriver() {
        // 可添加自定义配置，比如无头模式、窗口大小
        ChromeOptions options = new ChromeOptions();
        options.addArguments("--start-maximized");
        return new ChromeDriver(options);
    }

    // 关闭WebDriver并清理ThreadLocal中的实例
    public static void quitDriver() {
        WebDriver driver = driverThreadLocal.get();
        if (driver != null) {
            driver.quit();
            driverThreadLocal.remove();
        }
    }
}

2. 全局钩子类消除重复代码 #

创建一个全局Cucumber钩子类，统一处理WebDriver的初始化和销毁，所有步骤定义类不再需要重复写@Before/@After：

import io.cucumber.java.After;
import io.cucumber.java.Before;

public class GlobalHooks {

    @Before
    public void setUp() {
        // 触发当前线程的WebDriver初始化
        WebDriverManager.getDriver();
    }

    @After
    public void tearDown() {
        // 关闭当前线程的WebDriver
        WebDriverManager.quitDriver();
    }
}

3. 简化步骤定义类 #

现在步骤定义类无需再写钩子方法，直接从ThreadLocal获取WebDriver即可：

public class Test_A {

    private WebDriver driver;

    public Test_A() {
        // 获取当前线程专属的WebDriver实例
        this.driver = WebDriverManager.getDriver();
    }

    @When("The user navigates to the site")
    public void openSite() {
        driver.get("https://www.examplesite.com");
    }

    @And("The user executes task on the site")
    public void executeTask() {
        // 网页操作逻辑
    }
}

4. 优化Surefire并行配置（可选） #

当前<parallel>methods</parallel>是并行跑单个测试方法，如果你要并行跑特性文件/场景，更合适的配置是：

<plugin>
    <groupId>org.apache.maven.plugins</groupId>
    <artifactId>maven-surefire-plugin</artifactId>
    <version>3.0.0</version>
    <configuration>
        <!-- 按特性文件并行，更贴合Cucumber的测试结构 -->
        <parallel>features</parallel>
        <!-- 限制线程数，避免浏览器实例过多耗尽资源，2C表示2倍CPU核心数 -->
        <forkCount>2C</forkCount>
        <reuseForks>true</reuseForks>
    </configuration>
</plugin>

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