多类测试套件运行时Mock偶尔未生效问题排查求助

我在测试一个重度依赖Project Reactor的服务，测试中会Mock负责API调用的组件返回值。单个测试文件单独运行时全部通过，但一次性运行所有测试文件时，部分测试失败，错误显示Mock未生效——注入的组件要么返回null，要么调用了真实组件的实现，且日志中没有Mock失败的相关信息。

使用的依赖版本：

• spring-boot-test:2.7.8
• org.mockito:mockito-core:4.5.1、org.mockito:mockito-junit-jupiter:4.5.1
• org.junit.jupiter:junit-jupiter:5.8.2
• spring-boot-starter-webflux:2.7.8（依赖Reactor相关组件）
• 构建工具：Gradle 7.8

代码示例 #

interface API {
  Flux<Bird> getBirds();
}

@Component
class BirdWatcher {
  API api;
  BirdWatcher(API api) {
    this.api = api;
  }
  
  Flux<Bird> getUncommonBirds() {
    return api.getBirds()           // 测试中Mock了该方法，但部分运行场景下返回null或调用真实实现
      .filter(Bird::isUncommon);
  }
}

@SpringBootTest
class BirdWatcherTests {
  @Autowired
  BirdWatcher birdWatcher;
  @MockBean
  API api;
  
  @Test
  void findsUncommonBirds() {
    // 准备测试数据
    Bird birdCommon = new Bird("Sparrow", "common");
    Bird birdUncommon = new Bird("Parrot", "uncommon");
    Mockito.when(api.getBirds()).thenReturn(Flux.just(birdCommon, birdUncommon));
    
    // 执行方法
    Flux<Bird> uncommonBirds = birdWatcher.getUncommonBirds();
    
    // 断言结果
    assertThat(uncommonBirds.collectList().block().size(), equalTo(1));
  }
}

原因分析 #

1. Spring Boot Test上下文缓存复用：Spring Boot Test会缓存测试上下文，当多个测试类共享相同配置时，会复用同一个上下文。如果前一个测试对@MockBean设置了存根行为，后续测试如果没有重置Mock，就会继承之前的存根；若某个测试未定义存根，Mock会返回null（Mockito默认未存根方法返回null）。
2. Mockito Mock状态未自动重置：@MockBean创建的Mock实例默认不会在测试方法间自动重置状态，批量运行时测试执行顺序可能导致Mock行为交叉污染。
3. Reactor异步特性与阻塞测试的冲突：使用block()方法阻塞获取结果，可能引发线程调度问题，在Gradle默认并行运行测试的场景下，上下文初始化或Mock注入的线程安全问题会被放大。

解决方法 #

• 重置Mock状态：在每个测试方法执行前重置Mock，避免状态残留，通过@BeforeEach实现：

  @BeforeEach
  void resetMock() {
    Mockito.reset(api);
  }
  

• 禁用上下文缓存（针对性使用）：对存在问题的测试类添加@DirtiesContext注解，强制测试结束后销毁上下文，避免被后续测试复用。注意该注解会增加测试执行时间，建议仅在必要时使用：

  @SpringBootTest
  @DirtiesContext
  class BirdWatcherTests {
    // 测试代码
  }
  

• 使用Reactor专用测试工具：替换block()为StepVerifier，这是Reactor官方推荐的测试方式，避免阻塞带来的线程问题，同时能更精准验证异步流：

  @Test
  void findsUncommonBirds() {
    Bird birdCommon = new Bird("Sparrow", "common");
    Bird birdUncommon = new Bird("Parrot", "uncommon");
    Mockito.when(api.getBirds()).thenReturn(Flux.just(birdCommon, birdUncommon));
    
    StepVerifier.create(birdWatcher.getUncommonBirds())
      .expectNext(birdUncommon)
      .verifyComplete();
  }
  

• 统一测试上下文配置：确保所有测试类使用一致的配置（如@ActiveProfiles、@TestConfiguration等），避免Spring创建多个上下文，减少并发初始化风险。
• 禁用测试并行执行：如果是Gradle并行测试导致的线程安全问题，可在build.gradle中禁用并行：

  test {
    maxParallelForks = 1
  }
  

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