使用线程在 TestNG 中使用多个类进行测试执行

Question

我试图在我的 TestNG 项目中运行多个类，但其中一个类总是失败。

AbstractTest.java （sampleTest1 和 sampleTest2 类只包含一对一的 @Test 和 get() 方法）

public class AbstractTest {

  private List<WebDriver> webDriverPool = Collections.synchronizedList(new ArrayList<WebDriver>());
  private ThreadLocal<WebDriver> driverThread;

  @BeforeTest
  public void beforeTest()
  {
    this.driverThread = new ThreadLocal<WebDriver>() {
      @Override
      protected WebDriver initialValue() {
        final WebDriver webDriver = new FirefoxDriver();
        webDriverPool.add(webDriver);
        return webDriver;
      }
    };
  }

  @AfterTest
  public void afterTest() {
    for (WebDriver driver : this.webDriverPool) {
      driver.quit();
    }
  }

  public WebDriver getDriver() {
    return this.driverThread.get();
  }

}

testng.xml

<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE suite SYSTEM "http://testng.org/testng-1.0.dtd">
<suite thread-count="11" name="testSuite" parallel="tests">
    <test name="Test1" group-by-instances="true">
        <classes>
            <class name="test.sample.sampleTest1" />
            <class name="test.sample.sampleTest2" />
        </classes>
    </test> <!-- Test1 -->
</suite> <!-- testSuite -->

问题是其中一个线程看不到 WebDriver，它返回 null 并且会失败。我知道如果我将 ThreadLocal 更改为 static 它会起作用，但如果我并行运行测试会导致更多问题，我的目标是什么 :)

提前致谢！

Answer 1

使用其他使用 ThreadLocal 的方法。例如一种线程安全的单例实现可以提供帮助：

public class DriverSession
{
    private static ThreadLocal<DriverSession> instance = new ThreadLocal<DriverSession>();
    private WebDriver                  driver;

    public static DriverSession getInstance()
    {
        if (null == instance.get())
        {
            instance.set(new Proxy1());
        }
        return instance.get();
    }

    protected DriverSession()
    {
        driver = new FirefoxDriver();
    }

    public WebDriver getDriver()
    {
        return this.driver;
    }
}

我就是这样做的。因此，为了访问当前的驱动程序实例，请使用：

DriverSession.getInstance().getDriver().get("http://google.com");