java - 如何模拟 hasNext 并获得高返回

Question

这个问题已经有答案了:

Using Mockito with multiple calls to the same method with the same arguments (14 个回答)

已关闭 4 年前。

对于我的测试，我需要模拟我的cursos 的hasNext() 方法。然而，为了充分测试我的代码，我需要 250 次迭代才能发送到批量请求。因此我最后需要 250 x true 和 1 x false。

我创建了一个 boolean 数组，其中包含 250 个 true 和 1 个 false

我得到了什么

@Mock
private Cursor<Record> cursor;

public void myTest(){
  when(cursor.hasNext()).thenReturn(true, false);
}

但现在我需要 250 个光标条件所以我创建了一个 boolean 数组，但显然它无法编译

final boolean[] cursorsResponses = fillCursors();
when(cursor.hasNext()).thenReturn(cursorsResponses);

Answer 1

所以在你的情况下:

when(cursor.hasNext()).thenAnswer(new Answer() {
   private int count = 0;

   public Object answer(InvocationOnMock invocation) {
        return (count++ < 250);
   }
});