如何在Java Selenium页面对象工厂模式中实现ThreadLocal<WebDriver>以支持并行测试

1. 改造基础Page类 #

将原有静态WebDriver替换为ThreadLocal，确保每个线程持有独立的WebDriver实例：

import org.openqa.selenium.WebDriver;
import org.openqa.selenium.support.PageFactory;

public class Page {
    // 用ThreadLocal存储每个线程的WebDriver实例
    private static final ThreadLocal<WebDriver> driverThreadLocal = new ThreadLocal<>();

    // 获取当前线程的WebDriver
    public static WebDriver getDriver() {
        return driverThreadLocal.get();
    }

    // 设置当前线程的WebDriver
    public static void setDriver(WebDriver driver) {
        driverThreadLocal.set(driver);
    }

    // 移除当前线程的WebDriver（避免内存泄漏）
    public static void removeDriver() {
        driverThreadLocal.remove();
    }

    public Page() {
        // 从ThreadLocal获取当前线程的driver初始化页面对象
        PageFactory.initElements(getDriver(), this);
    }
}

2. 改造页面对象类（LandingPage） #

无需再通过构造方法传递driver，直接继承改造后的Page类即可：

import org.openqa.selenium.WebElement;
import org.openqa.selenium.support.FindBy;

public class LandingPage extends Page {

    @FindBy(xpath = "this is xpath")
    private WebElement pageWebElement;

    public LandingPage() {
        super(); // 调用父类无参构造，自动从ThreadLocal取driver初始化
    }

    public void doSomething() {
        pageWebElement.click();
    }
}

3. 改造步骤定义类（LandingPageStepDefs） #

在测试生命周期的初始化/销毁方法中，完成WebDriver的创建、绑定与清理，同时调整页面对象初始化逻辑：

import io.cucumber.java.After;
import io.cucumber.java.Before;
import org.openqa.selenium.chrome.ChromeDriver;

public class LandingPageStepDefs {
    private LandingPage landingPage;

    // 每个线程测试前初始化WebDriver并绑定到ThreadLocal
    @Before
    public void setup() {
        // 根据需求创建对应浏览器的WebDriver实例（示例为Chrome）
        WebDriver driver = new ChromeDriver();
        Page.setDriver(driver);
        // 初始化页面对象
        landingPage = new LandingPage();
    }

    // 每个线程测试后销毁WebDriver并清理ThreadLocal
    @After
    public void teardown() {
        WebDriver driver = Page.getDriver();
        if (driver != null) {
            driver.quit();
            Page.removeDriver();
        }
    }

    @And("Customer does something")
    public void customerDoesSomething() {
        landingPage.doSomething();
    }
}

4. 启用并行测试配置 #

若使用Cucumber #

在junit-platform.properties中添加并行配置：

# 开启并行执行
cucumber.execution.parallel.enabled=true
# 设置并行线程数（根据机器配置调整）
cucumber.execution.parallel.config.strategy=fixed
cucumber.execution.parallel.config.fixed.parallelism=3

若使用TestNG #

在testng.xml中配置并行：

<suite name="Parallel Tests" parallel="tests" thread-count="3">
    <test name="Test 1">
        <classes>
            <class name="your.package.TestClass"/>
        </classes>
    </test>
    <!-- 可添加更多测试类 -->
</suite>

核心注意事项 #

• 线程隔离：ThreadLocal保证每个线程的WebDriver实例完全独立，不会出现跨线程干扰
• 内存泄漏预防：测试结束后必须调用removeDriver()，避免ThreadLocal持有过期实例导致内存泄漏
• 框架配合：必须开启对应测试框架的并行执行开关，否则ThreadLocal的线程隔离优势无法发挥

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