Mockito Answer 实现动态返回值技巧

本文揭秘了 Mockito 中实现模拟方法动态返回值的关键技巧——用 `thenAnswer()` 替代 `thenReturn()`，通过 Lambda 表达式实现延迟求值，使 mock 方法每次调用时都能实时读取被测对象的最新状态（如运行中被修改的属性值），彻底解决静态快照导致的测试失真问题；同时附带实用提示：灵活利用 `invocation` 参数支持复杂场景、兼顾可读性与线程安全，让单元测试真正精准还原真实交互逻辑。

本文介绍如何在 Mockito 中让模拟方法返回随测试状态实时变化的值，解决 thenReturn() 仅捕获调用时快照值的问题，核心方案是改用 thenAnswer() 配合 Lambda 表达式实现延迟求值。

本文介绍如何在 Mockito 中让模拟方法返回随测试状态实时变化的值，解决 `thenReturn()` 仅捕获调用时快照值的问题，核心方案是改用 `thenAnswer()` 配合 Lambda 表达式实现延迟求值。

在使用 Mockito 编写单元测试时，一个常见误区是误以为 thenReturn() 能“监听”被测对象状态的变化。例如，当模拟方法依赖于某个对象的动态属性（如 ob.getParam()），而该属性在被测方法执行过程中被修改，直接使用 thenReturn(ob.getParam()) 会导致 Mockito 在 stub 定义时就立即求值——此时返回的是初始值（如 5），而非后续更新后的值。

根本原因在于：thenReturn() 接收的是一个静态值，而非表达式；它不参与运行时上下文，无法感知后续状态变更。

✅ 正确解法是使用 thenAnswer() —— 它允许你传入一个 Answer 实例（常用 Lambda 表达式），使返回值在每次 mock 方法被实际调用时才动态计算：

@Test
void testMyMethodUpdatesParamAndUsesLatestValue() {
    // 准备测试对象
    @Mock OtherClass otherClassMock;
    MyClass ob = new MyClass(otherClassMock, /* other deps */);

    // 设置初始参数
    ob.setParam(5);

    // ✅ 关键：使用 thenAnswer 延迟求值，每次调用 otherMethod() 时重新获取当前 param 值
    when(otherClassMock.otherMethod()).thenAnswer(invocation -> ob.getParam());

    // 执行被测方法（内部会修改 param 并调用 otherMethod）
    ob.myMethod();

    // 此时 otherMethod() 返回的是 myMethod() 内部更新后的 param 值，而非初始的 5
}

? 小贴士：

• invocation 参数提供了对原始调用的完整访问（如 invocation.getArguments()、invocation.getMethod()），适用于更复杂的场景（如根据入参返回不同值）；
• 若需复用逻辑，可定义具名 Answer 类或静态方法，提升可读性与可维护性；
• 注意线程安全：若测试涉及多线程并发修改 ob.getParam()，需确保 getParam() 是线程安全的，否则可能引发竞态问题；
• thenAnswer() 的性能开销极小，与 thenReturn() 几乎无差别，无需顾虑性能损耗。

总结：当 mock 方法的返回值需依赖被测对象的运行时状态时，请始终优先考虑 thenAnswer()。它将 stub 行为从“静态赋值”升级为“动态响应”，让测试真正反映代码的真实交互逻辑。

以上就是《Mockito Answer 实现动态返回值技巧》的详细内容，更多关于的资料请关注golang学习网公众号！