Mockito MockBean在Spring Batch测试中返回累积结果问题

问题现象 #

通过@MockBean注入UtilisateurSuiviRepository后，在不同测试方法中为同一方法的不同参数设置不同返回值时，第二个测试方法会返回前一个测试的结果（两个元素的列表），而非预期的单独结果。尝试用@Before结合MockitoAnnotations.initMocks(this)重置mock无效，手动创建mock并赋值则导致stub不生效。

根本原因 #

1. @MockBean的mock由Spring容器管理：MockitoAnnotations.initMocks(this)仅负责初始化用@Mock注解的字段，无法重置Spring创建并管理的@MockBean实例，因此之前的stub状态会残留。
2. 手动创建mock无法替换容器实例：直接给测试类字段赋值新的mock对象，只是改变了测试类内的引用，Spring容器中被业务逻辑依赖的mock实例并未被替换，所以手动设置的stub不会生效。

解决方案 #

方案1：用Mockito.reset()重置Spring管理的mock #

在每个测试方法执行前，调用Mockito.reset()清除mock的所有stub和交互记录，确保每个测试的stub都是独立的。

代码示例：

@Before
public void setUp(){
    Mockito.reset(utilisateurSuiviRepository);
}

方案2：使用@DirtiesContext重建Spring上下文（谨慎使用） #

标记测试类，让Spring在每个测试方法执行后重建上下文，彻底重置所有bean（包括mock）。但此方法会显著增加测试耗时，仅在必要时使用。

代码示例：

@DirtiesContext(classMode = DirtiesContext.ClassMode.AFTER_EACH_TEST_METHOD)
@RunWith(SpringRunner.class)
@MyCustomBatchUnitTest
@TestPropertySource(properties = {"spring.batch.job.names=myJob"})
public class JobTest {
    // 测试代码
}

方案3：在测试方法中明确覆盖stub（临时方案） #

如果不想重置mock，可在第二个测试方法中直接覆盖对应参数的stub，但这种方式不够优雅，且易出错：

// 第二个测试方法内
when(utilisateurSuiviRepository
        .findUtilisateursInactifsPourPremiereRelance(16))
        .thenReturn(new ArrayList<Object[]>(){{
            add(new Object[]{5L,null});
        }});

关键注意事项 #

• 当使用@MockBean时，必须通过Spring提供的机制或Mockito的reset()方法来管理mock状态，不要混用@Mock的初始化方式。
• 优先选择方案1，它在保证测试独立性的同时，不会增加额外的上下文重建开销。

内容的提问来源于stack exchange，提问作者mr.Penguin