本文介绍使用 Mockito 的 thenAnswer() 捕获被模拟方法的实际入参，并将其透传给替代逻辑（如另一方法），实现参数驱动的动态行为模拟，适用于任意整型输入场景。
本文介绍使用 mockito 的 `thenanswer()` 捕获被模拟方法的实际入参，并将其透传给替代逻辑（如另一方法），实现参数驱动的动态行为模拟，适用于任意整型输入场景。
在单元测试中，我们常需对依赖对象的方法进行精细化模拟：不仅希望替换原方法的行为，还要求新行为能准确接收并复用原始调用时传入的实际参数。例如，当 mock.originalMethod(int) 被任意整数调用时，我们不希望预设固定值（如 when(...).thenReturn(42)），而是希望实时提取该整数，并将其作为参数传递给 mock.differentMethod(int) 执行。
这无法通过 thenReturn() 或静态 thenCallRealMethod() 实现，而必须借助 thenAnswer() —— 它提供 InvocationOnMock 上下文，允许我们在运行时访问真实调用参数。
✅ 正确做法如下：
import static org.mockito.ArgumentMatchers.anyInt;
import static org.mockito.Mockito.*;
// 假设 mock 是 YourClass 类型的 Mock 对象
when(mock.originalMethod(anyInt())).thenAnswer(invocation -> {
int passedInt = invocation.getArgument(0, int.class); // 获取第 0 个参数（int 类型）
return mock.differentMethod(passedInt); // 动态调用，复用原始参数
});? 关键要点说明：
-
anyInt()是匹配器（matcher），表示“接受任意int值”，确保该 stub 对所有整数输入生效； -
invocation.getArgument(0, int.class)安全提取实际传入的int参数（索引从 0 开始）；若参数为包装类（如Integer），可改用Integer.class并注意空值校验； -
thenAnswer()中的 lambda 在每次方法调用时执行，因此天然支持参数透传与上下文感知； - 若
differentMethod是真实方法（非 mock 方法），请确保mock已启用CALLS_REAL_METHODS或该方法已单独 stub；否则建议将differentMethod也声明为@Mock或使用spy配合doReturn(...).when(...)。
⚠️ 注意事项：
- 不要在
thenAnswer中直接调用mock.originalMethod(...)（会导致无限递归或 StackOverflow）； - 避免在
thenAnswer中执行耗时或有副作用的操作（如 I/O、线程睡眠），以免影响测试稳定性与可读性； - 若需处理多个参数，可连续调用
getArgument(1, String.class)等；也可用invocation.getArguments()获取参数数组。
通过这种方式，你实现了真正灵活、参数敏感的模拟逻辑——让测试既简洁又精准，完美契合契约驱动与行为验证的设计理念。
