自定义重复方法调用的 Mockito 返回值
在测试涉及非确定性响应的场景时,模拟返回的方法可能会很有用后续调用时的不同值。 Mockito 是一个流行的 Java 模拟框架,为这个难题提供了一个优雅的解决方案。
考虑提供的示例代码,其中 ExecutorCompletionService 用于对任务进行分组并收集其结果。为了有效地测试逻辑,无论任务执行顺序如何,我们需要确保方法completionService.take()在不同时间返回不同的值。
Mockito 提供了 thenReturn 方法来控制存根方法的返回值。值得注意的是,您可以在 thenReturn 的括号内指定多个值。每次调用该方法时,Mockito 都会按指定的顺序返回值。例如:
<code class="java">when(completionService.take()).thenReturn(value1, value2, value3);</code>
在此示例中,第一次调用completionService.take()将返回value1,第二次调用将返回value2,第三次调用将返回value3。一旦返回所有指定的值,Mockito 将继续返回最后一个值以供后续调用。
通过利用此技术,您可以有效地模拟非确定性响应,并确保您的测试逻辑保持稳健,无论任务完成执行的顺序。
以上是如何使用 Mockito 模拟重复方法调用的不同返回值?的详细内容。更多信息请关注PHP中文网其他相关文章!