Question

在使用ExecutorService时，一个单元如何测试为Runnable任务生成新线程？

基本上，我的应用程序有一个静态线程池。

public static final ExecutorService executorService = Executors.newCachedThreadPool();

我想将此线程池用于我的单元测试，而不是模拟一个或注入新的线程池，因为不同的线程池可以显着改变我的应用程序的行为（固定与缓存与预定等）;我想确保用运行时线程池测试应用程序的行为。

缓存线程池似乎最适合我。问题是因为它是静态的并且线程被缓存了60秒，所以只有第一个测试实际上会在池中产生一个新线程，而后续测试会重用该线程。

单元测试代码：

public void testDoCallExecutesTaskInAnotherThread() {    
    final Client client = this.createClient();
    final ThreadPoolExecutor threadPoolExecutor = (ThreadPoolExecutor) client.getExecutorService(); // gets the static thread pool
    final int initPoolSize = threadPoolExecutor.getPoolSize();
    final Response response = client.doCall();
    Assert.assertEquals(initPoolSize + 1, threadPoolExecutor.getPoolSize());
}

关于如何使这项工作或其他方法的建议将不胜感激。

Answer 1

模拟ThreadFactory：

  ThreadFactory mock = new CustomObservableThreadFactory();
  ExecutorService executorService = Executors.newCachedThreadPool(mock);

像往常一样将ExecutorService注入正在测试的类
但是使用自定义ThreadFactory 来创建缓存的ThreadPool：每当要调用新的Thread时，都会调用ThreadFactory。然后，您可以根据需要跟踪这些实例化，例如使用监听器或计数器。

如何单元测试ExecutorService为任务生成新线程？

1 个答案: