Mockito 中使用参数化测试动态模拟不同方法调用-java教程-PHP中文网

Mockito 中使用参数化测试动态模拟不同方法调用

本文介绍如何在 junit 5 参数化测试中，结合 mockito 动态模拟多个不同签名的 repository 方法调用，避免重复 stub，提升测试可维护性与表达力。

本文介绍如何在 junit 5 参数化测试中，结合 mockito 动态模拟多个不同签名的 repository 方法调用，避免重复 stub，提升测试可维护性与表达力。

在使用 @ParameterizedTest 进行多场景覆盖时，一个常见痛点是：每个测试用例需 stub 不同的 Repository 方法（如 findComponentByKey_1()、findComponentByKey_2()），但 when(...).thenReturn(...) 的入参必须是具体的方法调用表达式，无法直接传入“方法引用”或“待执行逻辑”。原始代码试图将 methodCall 作为 Object 类型传入，这在 Mockito 中不可行——when() 接收的是 实际被调用的 mock 方法表达式，而非泛型占位符。

✅ 正确解法是：用 Supplier> 封装待 stub 的方法调用逻辑，并在 when() 中通过 .get() 触发该 Supplier，从而让 Mockito 捕获真实的方法调用链。这种方式既保持类型安全，又支持延迟执行，完美适配参数化场景。

以下是优化后的完整示例（已修正原始代码中的关键问题）：

@ExtendWith(MockitoExtension.class)
public class MyServiceTest {

    @Mock
    private MyRepository myRepository; // ❗移除 static —— Mockito 不支持静态字段注入

    private MyService myService;

    @BeforeEach
    void setUp() {
        myService = new MyService(myRepository);
    }

    @ParameterizedTest
    @MethodSource("args")
    void findAllTest(Map<string string> allParameters, Supplier<list>> methodCall) {
        Element element = Element.builder().identification("id").build();

        // ✅ 关键：通过 Supplier.get() 触发实际方法调用，使 Mockito 能正确记录 stub 行为
        when(methodCall.get()).thenReturn(List.of(element));

        var expectedResponse = myService.execute(allParameters); // 注意方法名拼写修正（excute → execute）

        // ✅ 断言逻辑（示例）
        assertThat(expectedResponse).isNotNull();
        // ... 其他断言
    }

    private static Stream<arguments> args() {
        // ✅ 每个 Supplier 封装一个具体的 mock 方法调用（注意：参数需符合实际签名）
        Supplier<list>> findByKey1 = () -> myRepository.findComponentByKey_1("value_1");
        Supplier<list>> findByKey2 = () -> myRepository.findComponentByKey_2("value_2");

        return Stream.of(
            Arguments.of(Map.of("key_1", "value_1"), findByKey1),
            Arguments.of(Map.of("key_2", "value_2"), findByKey2)
        );
    }
}</list></list></arguments></list></string>

⚠️ 重要注意事项：

@Mock 字段不能是 static：MockitoExtension 仅对实例字段生效；静态 mock 会导致 NullPointerException 或 stub 失效。
Supplier 中的参数必须真实有效：例如 findComponentByKey_1("value_1") 的参数需与被测方法中实际传入的值一致（尤其当方法内含 ArgumentMatchers 时，需同步调整）。
避免在 @MethodSource 方法中直接使用 myRepository 实例：因 args() 是静态方法，无法访问实例字段。正确做法是——在 args() 内部 先创建临时 mock（如 MyRepository repo = mock(MyRepository.class)），再构造 Supplier；或更推荐：将 stub 提前放在 @BeforeEach 中，并在 args() 中仅传递方法标识符（如枚举/字符串），由测试主体决定调用哪个 stub。
若方法返回值依赖输入参数，建议改用 thenAnswer() 配合 InvocationOnMock 实现动态响应，而非固定 thenReturn()。