1. 程式人生 > >教你快速寫出多執行緒Junit單元測試用例

教你快速寫出多執行緒Junit單元測試用例

寫過Junit單元測試的同學應該會有感覺,Junit本身是不支援普通的多執行緒測試的,這是因為Junit的底層實現上,是用System.exit退出用例執行的。JVM都終止了,在測試執行緒啟動的其他執行緒自然也無法執行。JunitCore程式碼如下

/**

* Run the tests contained in the classes named in the <code>args</code>.

* If all tests run successfully, exit with a status of 0. Otherwise exit with a status of 1.

* Write feedback while tests are running and write

* stack traces for all failed tests after the tests all complete.

* @param args names of classes in which to find tests to run

*/
public static void main(String... args) { runMainAndExit(new RealSystem(), args); } /** * Do not use. Testing purposes only. * @param system */ public static void runMainAndExit(JUnitSystem system, String... args) { Result result= new JUnitCore().runMain(system, args); system.exit(result.wasSuccessful
() ? 0 : 1); }

RealSystem:

public void exit(int code) {

System.exit(code);

}

所以要想編寫多執行緒Junit測試用例,就必須讓主執行緒等待所有子執行緒執行完成後再退出。想到的辦法自然是Thread中的join方法。話又說回來,這樣一個簡單而又典型的需求,難道會沒有第三方的包支援麼?通過google,筆者很快就找到了GroboUtils這個Junit多執行緒測試的開源的第三方的工具包。

Maven依賴方式:

<dependency>

<groupId>net.sourceforge.groboutils</groupId>
<artifactId>groboutils-core</artifactId> <version>5</version> </dependency>

注:需要第三方庫支援:Repository url https://oss.sonatype.org/content/repositories/opensymphony-releases

依賴好Jar包後就可以編寫多執行緒測試用例了。上手很簡單:

/**

* 多執行緒測試用例

*

* @author lihzh(One Coder)

* @date 2012-6-12 下午9:18:11

* @blog http://www.coderli.com

*/

@Test

public void MultiRequestsTest() {

// 構造一個Runner

TestRunnable runner = new TestRunnable() {

@Override

public void runTest() throws Throwable {

// 測試內容

}

};

int runnerCount = 100;

//Rnner陣列,想當於併發多少個。

TestRunnable[] trs = new TestRunnable[runnerCount];

for (int i = 0; i < runnerCount; i++) {

trs[i] = runner;

}

// 用於執行多執行緒測試用例的Runner,將前面定義的單個Runner組成的陣列傳入

MultiThreadedTestRunner mttr = new MultiThreadedTestRunner(trs);

try {

// 開發併發執行數組裡定義的內容

mttr.runTestRunnables();

} catch (Throwable e) {

e.printStackTrace();

}

}

執行一下,看看效果。怎麼樣,你的Junit也可以執行多執行緒測試用例了吧:)。