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的泛型变量的声明如下：

1 /**

2 * 每个DriverManager只管理一个driver，所以他是static的，但是我引入ThreadLocal来处理多线程

3 * shares the same web driver and use thread local to handle the multi-thread

4 */

5 public static ThreadLocal<WebDriver> ThreadDriver=new ThreadLocal<WebDriver>() ;

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

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

 1 /**

 2 * 如果当前进程没有绑定driver，创建一个然后绑定上，如果已经有了就直接返回

 3 * create a driver for this thread if not exist. or return it directly

 4 */

 5 public static WebDriver getDriver(){

 6 WebDriver driver= DriverManager.ThreadDriver.get();

 7 if (driver==null){

 8 if (browserType.equals("firefox")){

 9 driver = new EventFiringWebDriver(new FirefoxDriver()).register(new LogEventListener());

10 ThreadDriver.set(driver);

11 //找东西前等三秒wait 3 second for every find by

12 DriverManager.getDriver().manage().timeouts().implicitlyWait(3, TimeUnit.SECONDS);

13 }

14 //有需求的同学自己在这里添加IE等浏览器的支持

15 //you can add ie/chrome or other driver here

16 }

17 return driver;

18 }

 

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

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

 1     /**

 2 * JD首页上搜索一个商品 主要介绍定位某个商品名称的N种写法

 3 * 这里用了DataProvider

 4 * （不会用DataProvier不要说自己懂testNG）

 5 */

 6 @Test(dataProvider="product_to_search")

 7 public void searchProduct(String keyword) throws InterruptedException {

 8 log.info("keyword="+keyword);

 9 String actual_procut_name= new JDHomepage().init().searchHeader.search(keyword).getProductNameByIndexMethodTwo(1);

10 log.info("actual_procut_name="+actual_procut_name);

11 assert(actual_procut_name.contains(keyword));

12 log.info("--------------------------------");

13 }

14 

15 /**

16 * 用了DataProvider并且是并行的。但是这个是运行不通过的。因为WebDriver不是线程安全的。

17 * 稍后我会再修复这个。。。暂时先这样吧。

18 *

19 * 2014-4-20：更新了DriverManager，现在这个框架里的WebDriver是线程安全的了，所以这个脚本就可以并行执行4个测试数据了。

20 * 详见DriverManager类

21 *

22 */

23 @DataProvider(name="product_to_search",parallel = true)

24 public Iterator<Object[]> createData1() {

25 //一个Array

26 String[] astringarray={"巧克力","饼","糕","蛋"};

27 //转成一个String的List

28 List<String> lines=Arrays.asList(astringarray);

29 

30 //一个Object的空的List

31 List<Object[]> data = new ArrayList<Object[]>();

32 

33 //往空List里放东西，然后返回

34 for (String line :lines )

35 {

36 data.add(new Object[]{line});//加了一个FirefoxDriver进去哦

37 }

38 return data.iterator();

39 

40 }

View Code

 

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

具体代码详见我的github

 

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

下期再见。