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的代码。

下期再见。