Junit单元测试不支持多线程测试问题全解析

一、背景

今天@段段提出了一个很好的问题，她发现单元测试时如果开多个线程，主线程运行结束就结束了，并不会等待子线程结束。

如果用main方法就没问题，技术群里展开了激烈的讨论。

本文将“复现”这种现象，并给出多种解决方案，并纠正个别文章的错误，并追到源头带大家找出问题的原因。

本文会分享几点干货，如调用栈大法、发编译大法等。

二、复现场景的源码

我们的线程类

代码语言：javascript复制

public class DemoThread extends Thread {

    private static int num = 0;

    public DemoThread() {
    }

    @Override
    public void run() {
        synchronized (DemoThread.class) {
            for (int i = 0; i < 4; i  ) {
                System.out.println(Thread.currentThread().getName()   "-->"   num  );
                try {
                    TimeUnit.MILLISECONDS.sleep(200);
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
        }
    }

    public static int getNum() {
        return num;
    }
}

测试代码

代码语言：javascript复制

public class ThreadDemoTest {

    @Test
    public void test(){

        DemoThread thread1 = new DemoThread();

        DemoThread thread2 = new DemoThread();

        thread1.start();
        thread2.start();

    }
}

运行

和预期完全不符，预期应该是线程1输出0-3，线程2输出4-7，结果就输出一个，什么情况？？

三、解决方案

所有的解决方法的核心是：在子线程没结束之前让主线程阻塞住。

3.1 使用main函数

代码语言：javascript复制

public class ThreadDemoTest {


    public static void main(String[] args) {
        DemoThread thread1 = new DemoThread();

        DemoThread thread2 = new DemoThread();

        thread1.start();
        thread2.start();
    }
}

结果

发现符合预期

3.2 CountDownLatch

我们的线程类：

代码语言：javascript复制

public class DemoThread extends Thread {
    private CountDownLatch latch;
    private static int num = 0;

    public DemoThread(CountDownLatch latch) {
        this.latch = latch;
    }

    @Override
    public void run() {
        synchronized (DemoThread.class) {
            for (int i = 0; i < 4; i  ) {
                System.out.println(Thread.currentThread().getName()   "-->"   num  );
                try {
                    TimeUnit.MILLISECONDS.sleep(200);
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
            latch.countDown();
        }
    }

    public static int getNum() {
        return num;
    }
}

我们的单元测试类

代码语言：javascript复制

public class ThreadDemoTest {

    @Test
    public void test() throws InterruptedException {
        CountDownLatch latch = new CountDownLatch(2);

        DemoThread thread1 = new DemoThread(latch);

        DemoThread thread2 = new DemoThread(latch);

        thread1.start();
        thread2.start();

        latch.await();
    }
}

结果依然符合预期

3.3 使用join函数

代码语言：javascript复制

public class ThreadDemoTest {

    @Test
    public void test() throws InterruptedException{

        DemoThread thread1 = new DemoThread();

        DemoThread thread2 = new DemoThread();

        thread1.start();
        thread2.start();

        thread1.join();
        thread2.join();
    }
}

结果依然符合预期

当然还有其他办法，殊途同归。