java - 如何确保 mock 的对象仅在 mock 中被调用一次

我有一个while循环如下

while (nodeIterator.hasNext())

我已经模拟了hasNext返回true的方法，以便我可以在while循环中测试代码，但是现在的问题是每次它返回true时，这个循环就永远不会结束。无论如何，请告诉我是否可以确保只调用一次此方法，否则，如何在第一次执行后返回false？

最佳答案

我得到了答案，我们可以通过以下方式做到这一点

when(nodeIterator.hasNext()).thenReturn(true).thenReturn(false);

这称为方法 stub 。同样，如果您想调用它两次，然后又想返回false，那么请执行以下操作

when(nodeIterator.hasNext()).thenReturn(true).thenReturn(true).thenReturn(false);