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

下期再見。