用Mockito模拟具有相同参数的同一方法的多次调用

使用 Mockito 应对带有相同参数的同一方法的多次调用

简介

在单元测试中，模拟外部依赖项和服务对于隔离我们的代码并验证其行为至关重要。Mockito 作为 Java 模拟框架，提供了对具有相同参数的同一方法进行多次调用的灵活控制，使我们能够测试非确定性响应并验证结果的稳定性。

问题

我们希望在不同调用中模拟一个方法返回不同的对象，例如 ExecutorCompletionService 的 take() 方法，该方法返回已完成任务的 Future。

解决方法

Mockito 的 thenReturn() 方法允许我们为特定方法调用设置不同的返回值。对于具有相同参数的同一方法的多次调用，我们可以通过使用链式语法指定多个返回值：

when(mockObject.method(argument)).thenReturn(value1).thenReturn(value2).thenReturn(value3);

这将导致方法在第一次调用时返回 value1，在第二次调用时返回 value2，依此类推。

应用场景

此技术对于测试非确定性响应非常有用，例如使用 ExecutorCompletionService 的情况下。无论完成任务的顺序如何，我们都可以验证我们的代码对返回结果保持一致。

优点

• 轻松模拟具有相同参数的同一方法的多次调用
• 验证非确定性响应和确保代码对返回顺序不敏感
• 提高测试的可读性和可维护性

实现

使用 Mockito 模拟具有相同参数的同一方法的多次调用非常简单：

1. 创建被模拟对象的模拟。
2. 使用 thenReturn() 方法指定每个调用的返回值。
3. 在测试代码中注入模拟对象并验证结果。

示例

考虑以下代码示例：

// 创建模拟对象
ExecutorCompletionService<T> mockCompletionService = mock(ExecutorCompletionService.class);

// 设置返回值
when(mockCompletionService.take())
    .thenReturn(new FutureTask<>(new T1()))
    .thenReturn(new FutureTask<>(new T2()))
    .thenReturn(new FutureTask<>(new T3()));

// 测试代码
// ...

常见问题解答

• 我可以设置无限数量的返回值吗？

  • 是的，thenReturn() 方法允许您指定任意数量的返回值。

• 我可以为不同的参数指定不同的返回值吗？

  • 是的，您可以通过使用 when(mockObject.method(argument1, argument2)) 等来指定不同的参数。

• 我可以模拟抛出异常吗？

  • 是的，您可以使用 thenThrow() 方法模拟方法抛出异常。

• 如何验证模拟方法的调用顺序？

  • 您可以使用 inOrder 方法来验证方法的调用顺序。

• 如何重置模拟对象？

  • 您可以使用 reset() 方法来重置模拟对象，清除所有先前设置的预期和交互。

总结

通过使用 Mockito 的 thenReturn() 方法，我们可以轻松地模拟具有相同参数的同一方法的多次调用。这对于测试非确定性响应和验证代码对返回顺序不敏感非常有用，从而提高了测试的可靠性和有效性。