如何在多线程环境中用Mockito Mock工具类静态方法返回值？

Mockito的mockStatic默认采用线程局部作用域，只有创建MockedStatic实例的线程能识别到静态方法的mock实现。你的业务逻辑通过线程池在子线程中调用静态方法，子线程不属于创建mock的测试线程，因此无法触发mock逻辑，会调用真实方法。

方案1：重构代码（推荐） #

静态硬依赖是测试难维护的根源，通过依赖注入封装静态方法逻辑，从根本上解决问题：

1. 定义接口封装配置获取逻辑：

public interface ConfigProvider {
    SomeConfig getConfigForId(String id);
}

2. 实现接口，复用原有静态方法逻辑：

public class DefaultConfigProvider implements ConfigProvider {
    @Override
    public SomeConfig getConfigForId(String id) {
        return UtilClass.getConfigForId(id);
    }
}

3. 修改SomeService，通过构造方法注入接口：

public class SomeService{
    private final ConfigProvider configProvider;

    @Inject
    public SomeService(ConfigProvider configProvider) {
        this.configProvider = configProvider;
    }

    public void doSth(List<Msg> messages){
        List<Callable<Void>> tasks = new ArrayList<>();
        
        for (Msg msg : messages){
            tasks.add(()->{
                String id = msg.getId();
                SomeConfig config = configProvider.getConfigForId(id);
                // ... 原有业务逻辑
            });
        }
        
        // 线程池执行任务逻辑
        // ...
    }
}

4. 测试时直接Mock接口，Guice绑定实例：

Injector injector = Guice.createInjector(new AbstractModule() {
    @Override
    protected void configure() {
        ConfigProvider mockProvider = Mockito.mock(ConfigProvider.class);
        Mockito.when(mockProvider.getConfigForId(Mockito.any())).thenReturn(someConfig);
        bind(ConfigProvider.class).toInstance(mockProvider);
    }
});
SomeService someService = injector.getInstance(SomeService.class);

someService.doSth(testMessages);

这种方式完全解耦了静态方法依赖，测试更灵活，也符合面向对象设计原则。

方案2：使用Mockito全局作用域Mock（需版本≥4.5.0） #

从Mockito 4.5.0开始，mockStatic支持指定Scope.GLOBAL，让所有线程都能识别到静态mock：

try (MockedStatic<UtilClass> dummy = Mockito.mockStatic(UtilClass.class, 
        MockedStatic.withSettings().inScope(MockedStatic.Scope.GLOBAL))) {
  
    dummy.when(() -> UtilClass.getConfigForId(Mockito.any()))
          .thenReturn(someConfig);
  
    someService.doSth(testMessages);
    // 必须等待所有子线程任务执行完毕，否则try块结束后mock会被销毁
    // 示例：如果线程池是类成员，添加方法获取后等待：
    // someService.getExecutor().shutdown();
    // someService.getExecutor().awaitTermination(10, TimeUnit.SECONDS);
}

注意：全局作用域的mock可能会影响其他测试用例，需确保测试之间的隔离性。

方案3：同步测试线程与子线程（旧版本Mockito兼容） #

如果无法升级Mockito版本，可通过CountDownLatch让测试线程等待子线程在mock作用域内完成任务：

1. 修改SomeService的方法，支持传入CountDownLatch：

public void doSth(List<Msg> messages, CountDownLatch latch){
    List<Callable<Void>> tasks = new ArrayList<>();
    
    for (Msg msg : messages){
        tasks.add(()->{
            try {
                String id = msg.getId();
                SomeConfig config = UtilClass.getConfigForId(id);
                // ... 原有业务逻辑
            } finally {
                latch.countDown();
            }
            return null;
        });
    }
    
    // 线程池执行任务逻辑
    // ...
}

2. 测试代码中等待子线程完成：

Injector injector = Guice.createInjector(new BasicModule());
SomeService someService = injector.getInstance(SomeService.class);

List<Msg> testMessages = Arrays.asList(new Msg("id1"), new Msg("id2"));
CountDownLatch latch = new CountDownLatch(testMessages.size());

try (MockedStatic<UtilClass> dummy = Mockito.mockStatic(UtilClass.class)) {
  
    dummy.when(() -> UtilClass.getConfigForId(Mockito.any()))
          .thenReturn(someConfig);
  
    someService.doSth(testMessages, latch);
    // 等待所有子线程执行完成，超时时间可根据实际调整
    latch.await(10, TimeUnit.SECONDS);
}

内容的提问来源于stack exchange，提问作者waynewingorc