TestNG方法并行执行时如何为每个测试生成唯一WebDriver对象？

我在使用TestNG以方法并行模式执行Selenium测试时，发现每个测试方法拿到的WebDriver对象并非唯一，导致测试失败。相关代码如下：

BaseTest类 #

public class BaseTest {

public WebDriver driver;

@BeforeMethod
public void setup() {
driver = new FireFoxDriver();
}

@AfterMethod
public void teardown() {
    driver.quit();
}
}

测试类（继承自BaseTest） #

public class Tests extends BaseTest{

@Test
public void first() {
    System.identityHashCode(driver);
}

@Test
public void second() {
    System.identityHashCode(driver);
}
}

TestNG配置文件 #

<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE suite SYSTEM "http://testng.org/testng-1.0.dtd">
<suite name="Testing Suite Selenium" parallel="methods" thread-count="5">
<test name="regression">
       <classes>
            <class name="Tests"/>
        </classes>
    </test>
</suite >

解决方法：使用ThreadLocal隔离线程级别的WebDriver实例 #

问题根源是：TestNG以parallel="methods"模式运行时，多个测试方法会在不同线程中执行，但普通实例变量driver无法保证每个线程拥有独立的实例。改用ThreadLocal<WebDriver>可以为每个线程存储专属的WebDriver对象，确保线程间互不干扰。

修改后的BaseTest类

public class BaseTest {
    // 用ThreadLocal存储WebDriver，每个线程对应独立实例
    private ThreadLocal<WebDriver> driverThreadLocal = new ThreadLocal<>();

    @BeforeMethod
    public void setup() {
        FirefoxDriver driver = new FirefoxDriver();
        // 将当前线程的driver存入ThreadLocal
        driverThreadLocal.set(driver);
    }

    // 提供获取当前线程driver的方法
    protected WebDriver getDriver() {
        return driverThreadLocal.get();
    }

    @AfterMethod
    public void teardown() {
        // 获取当前线程的driver并关闭
        WebDriver driver = driverThreadLocal.get();
        if (driver != null) {
            driver.quit();
            // 移除ThreadLocal中的实例，避免内存泄漏
            driverThreadLocal.remove();
        }
    }
}

修改后的测试类

public class Tests extends BaseTest{

@Test
public void first() {
    System.identityHashCode(getDriver());
}

@Test
public void second() {
    System.identityHashCode(getDriver());
}
}

关键说明 #

• ThreadLocal会为每个访问它的线程创建独立的变量副本，确保每个测试方法（对应一个线程）拿到的都是专属的WebDriver实例。
• 在@AfterMethod中调用remove()是为了清理ThreadLocal中的实例，避免线程池复用线程时引发内存泄漏。
• TestNG配置文件无需修改，保持parallel="methods"和thread-count配置即可。

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