本文介绍如何在 mockito 中对被覆写方法进行动态响应，使模拟对象能根据原始方法传入的实际参数（如 int 类型）实时调用另一个方法，而非静态预设返回值。
本文介绍如何在 mockito 中对被覆写方法进行动态响应，使模拟对象能根据原始方法传入的实际参数（如 int 类型）实时调用另一个方法，而非静态预设返回值。
在单元测试中，我们常需对依赖对象的方法行为进行精细控制。当希望保留原始调用参数的语义（例如：originalMethod(x) 的输入 x 应直接作为 differentMethod(x) 的输入），而非为每个可能参数单独配置 when(...).thenReturn(...) 时，必须借助 thenAnswer() 配合 InvocationOnMock 实现动态参数捕获。
核心思路是：
- 使用
anyInt()（或更通用的any()）匹配任意整数参数，确保所有调用均被拦截； - 在
thenAnswer的 Lambda 中，通过invocation.getArgument(0, int.class)安全提取第 1 个参数（索引从 0 开始），类型明确指定为int.class，避免自动装箱/拆箱异常； - 将该参数直接传递给目标方法（如
mock.differentMethod(theInt)），实现参数透传。
✅ 正确示例代码：
when(mock.originalMethod(anyInt())).thenAnswer(invocation -> {
int theInt = invocation.getArgument(0, int.class);
return mock.differentMethod(theInt);
});
⚠️ 注意事项：
-
避免使用
thenCallRealMethod()或thenAnswer(invocation -> differentMethod(invocation.getArgument(0)))直接调用未 mock 的真实方法——这可能导致副作用或测试不隔离；应确保differentMethod本身也已合理 mock 或属于测试可控范围。 - 若
originalMethod有多个参数，可通过getArgument(1, String.class)等方式按需提取；建议始终显式声明泛型类型，提升类型安全性和可读性。 -
anyInt()属于宽松匹配，若需验证具体参数值（如断言某次调用是否传入了42），应在verify()阶段配合eq(42)使用，而非在when()中硬编码。
总结：thenAnswer + invocation.getArgument() 是 Mockito 实现“参数驱动响应”的标准范式，兼顾灵活性与类型安全性，适用于参数透传、日志记录、条件返回等高级模拟场景。
