如何在Mockito中为同一方法设置不同的行为？

Question

我有循环：

while (set.next()) {
    // something to do...
}

set反对它是模拟的：

final ResultSet set = mock(ResultSet.class);
when(set.next()).thenReturn(true);

我有无限循环。

如何仅在第一次通话时为true定义set.next()，为第二次通话定义false？谢谢。