selenium从入门到应用 - 8，selenium+testNG实现多线程的并发测试

本系列所有代码 https://github.com/zhangting85/simpleWebtest
本文将介绍一个Java+TestNG+Maven+Selenium的web自动化测试脚本环境下selenium和testNG的多线程并发测试的实现，并提供全部代码。

 

这里我的多线程的执行测试用例，目的首先是为了提高测试执行速度，然后是为做分布式测试提供技术基础。

 

testNG自带了多线程机制，那么为什么要我们还要自己写一些代码呢？

因为selenium webdriver本身不是线程安全的。比如说，我直接在一个testNG的case里新建两个webdriver，再做操作，他们就会冲突。然后其中一个会死掉。

如果我们直接用testNG的多线程，也会遇到类似的问题。

这里，我用thread local来解决线程安全问题：

 

在DriverManager类中，

ThreadLocal的泛型变量的声明如下：

/**
* 每个DriverManager只管理一个driver，所以他是static的，但是我引入ThreadLocal来处理多线程
* shares the same web driver and use thread local to handle the multi-thread
*/
public static ThreadLocal<WebDriver> ThreadDriver=new ThreadLocal<WebDriver>() ;

之后是执行case时，要获取一个可用的WebDriver对象来执行测试，此时调用DriverManger.getDriver（）方法来取得。

当当前进程未绑定driver时，创建一个driver，如果已绑定driver，则直接返回。所谓给当前进程绑定一个driver，使用的是ThreadDriver.set(driver)。

/**
* 如果当前进程没有绑定driver，创建一个然后绑定上，如果已经有了就直接返回
* create a driver for this thread if not exist. or return it directly
*/
public static WebDriver getDriver(){
WebDriver driver= DriverManager.ThreadDriver.get();
if (driver==null){
if (browserType.equals("firefox")){
driver = new EventFiringWebDriver(new FirefoxDriver()).register(new LogEventListener());
ThreadDriver.set(driver);
//找东西前等三秒wait 3 second for every find by
DriverManager.getDriver().manage().timeouts().implicitlyWait(3, TimeUnit.SECONDS);
}
//有需求的同学自己在这里添加IE等浏览器的支持
//you can add ie/chrome or other driver here
}
return driver;
}

 

这样添加完之后，我们的webdriver已经变成了线程安全的。

之后就可以执行并发的测试，这里用了testng的dataprovider的多线程功能：

    /**
* JD首页上搜索一个商品 主要介绍定位某个商品名称的N种写法
* 这里用了DataProvider
* （不会用DataProvier不要说自己懂testNG）
*/
@Test(dataProvider="product_to_search")
public void searchProduct(String keyword) throws InterruptedException {
log.info("keyword="+keyword);
String actual_procut_name= new JDHomepage().init().searchHeader.search(keyword).getProductNameByIndexMethodTwo(1);
log.info("actual_procut_name="+actual_procut_name);
assert(actual_procut_name.contains(keyword));
log.info("--------------------------------");
}

/**
* 用了DataProvider并且是并行的。但是这个是运行不通过的。因为WebDriver不是线程安全的。
* 稍后我会再修复这个。。。暂时先这样吧。
*
* 2014-4-20：更新了DriverManager，现在这个框架里的WebDriver是线程安全的了，所以这个脚本就可以并行执行4个测试数据了。
* 详见DriverManager类
*
*/
@DataProvider(name="product_to_search",parallel = true)
public Iterator<Object[]> createData1() {
//一个Array
String[] astringarray={"巧克力","饼","糕","蛋"};
//转成一个String的List
List<String> lines=Arrays.asList(astringarray);

//一个Object的空的List
List<Object[]> data = new ArrayList<Object[]>();

//往空List里放东西，然后返回
for (String line :lines )
{
data.add(new Object[]{line});//加了一个FirefoxDriver进去哦
}
return data.iterator();

}

 

当dataprovider设置了parallel参数为true时，测试即会并发执行。这里的真实并发数受到CPU内核数限制。

具体代码详见我的github

 

另外，使用了这些代码将webdriver和现成绑定，并使用testng提供的多线程功能后，这个测试框架下的测试用例变得可以简单得移植到selenium grid上，实现分布式测试。只需要再改动一下启动driver的代码。

下期再见。