JUnit单元测试不支持多线程

最新推荐文章于 2022-09-15 14:16:06 发布

Amazing_deron

最新推荐文章于 2022-09-15 14:16:06 发布

阅读量266

点赞数

CC 4.0 BY-SA版权

分类专栏： Java 文章标签： JUnit

本文链接：https://blog.youkuaiyun.com/Deronn/article/details/90268856

Java 专栏收录该内容

38 篇文章

订阅专栏

在进行JUnit单元测试时遇到多线程问题，主线程结束导致子线程未完成任务。分析Junit4 TestRunner源码发现，测试主线程结束后，不管子线程是否结束，都会强制结束程序。为解决此问题，提出了两种方案：1) 主线程休眠，但时间不确定；2) 使用CountDownLatch工具类，使主线程等待子线程完成或超时。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在写一个多线程启动时，使用单元测试时子线程莫名其妙结束。

网上找了一些资料发现问题所在：

1.测试发现问题的demo


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);
    } 
}

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);