Junit單元測試多線程的問題

今天下午很快完成了一個接口的監控功能，而後屁顛屁顛地用Junit開始單元測試。而後我就開始陷入崩潰的邊緣...

監控結束後須要將監控結果以郵件的形式發送給運營的小夥伴維護，前面測試仍是很順利，到了開多線程發郵件時就不行了，

程序也不報錯，也接收不到郵件。而後改代碼再測試，再冥思一下子，再改再測試，仍是無果，最後選擇度娘一下，結論是：

Junit單元測試不支持多線程

而後，整我的都很差了...浪費了我好多時間，就是由於這個！！!

雖然知道告終果，可是筆者仍是須要親自驗證一下。

/**
 * @Title: TestDoWork.java
 * @Describe:
 * @author: Mr.Yanphet
 * @Email: mr_yanphet@163.com
 * @date: 2016年8月15日 下午5:50:03
 * @version: 1.0
 */
public class TestDoWork {

    class DoWork implements Runnable {

        @Override
        public void run() {
            for (int i = 0; i < 10000; i++) {
                long milliSecond = System.currentTimeMillis();
                System.out.println("i=" + i + "，milliSecond=" + milliSecond);// 輸出循環次數和當前的系統時間
            }
        }

    }

    @Test
    public void test() {
        DoWork dw = new DoWork();
        Thread t = new Thread(dw);
        t.start();
    }

}

輸出結果以下（筆者省略了部分輸出）：

.....
i=751，milliSecond=1471257586416
i=752，milliSecond=1471257586416
i=753，milliSecond=1471257586416
i=754，milliSecond=1471257586416
i=755，milliSecond=1471257586416
i=756，milliSecond=1471257586416
i=757，milliSecond=1471257586416
i=758，milliSecond=1471257586416

從結果能夠看到，循環到了759次後就沒再輸出了，說明子線程還沒結束任務，整個程序就被強迫結束了。

既然知道了現象，那麼爲何會出現這樣的現象呢，貼出部分Junit4 TestRunner源碼就知道了

public static final int SUCCESS_EXIT = 0;
public static final int FAILURE_EXIT = 1;
public static final int EXCEPTION_EXIT = 2;

public static void main(String args[]) {
    TestRunner aTestRunner = new TestRunner();
    try {
        TestResult r = aTestRunner.start(args);
        if (!r.wasSuccessful())
            System.exit(FAILURE_EXIT);
        System.exit(SUCCESS_EXIT);
    } catch (Exception e) {
        System.err.println(e.getMessage());
        System.exit(EXCEPTION_EXIT);
    } 
}

再貼上TestResult部分源碼，以供參考

protected  List<TestFailure>    fFailures
protected  List<TestFailure>    fErrors

public synchronized boolean wasSuccessful() {
    return failureCount() == 0 && errorCount() == 0;
}

public synchronized int errorCount() {
    return fErrors.size();
}

public synchronized int failureCount() {
    return fFailures.size();
}

在TestRunner中能夠看出，若是是單線程，當測試主線程執行結束後，無論子線程是否結束，都會回調TestResult的wasSuccessful方法，

而後判斷結果是成功仍是失敗，最後調用相應的System.exit()方法。你們都知道這個方法是用來結束當前正在運行中的java虛擬機，jvm都自身難保了，因此子線程也就對不住你咧...

解決辦法：

1 簡單粗暴地讓主線程休眠一段時間，而後讓子線程可以運行結束。可是這個方法的弊端是，你不知道子線程的運行時間，因此須要看臉＝_＝

　　Thread.sleep();

2 使用CountDownLatch工具類，讓主線程阻塞，直到子線程運行結束或者阻塞超時，這個方法要比第一個方法好點。

　　countDownLatch.await(5, TimeUnit.MINUTES);

至於還有其餘方法，筆者看到不少大神本身寫的Junit支持多線程，有興趣的讀者自行度娘...