如何使用Mockito模拟ResultSet.next()方法

Question

我嘲笑java.sql.ResultSet这样的

ResultSet rs = mock(ResultSet.class);
when(rs.next()).thenReturn(true); // this seems wrong appraoch

测试代码是这样的

while (rs.next()) {
  // doing stuff here
}

所以,问题是,当我有模拟rs.next()到true再while循环永远不会终止.我希望while在2次迭代后终止循环.那么我怎么可以模拟rs.next()方法呢？

我也试过了

when(rs.next()).thenReturn(true, true, false); // always return false 

请帮忙!

Answer 1

你可以链接doReturn()方法调用:

doReturn(true).doReturn(true).doReturn(false).when(rs).next();

或者,如评论中所述,链式thenReturn方法调用:

when(rs.next()).thenReturn(true).thenReturn(true).thenReturn(false);

或者,如果你想更进一步,你可以使用Mockito Answer:

when(rs.next()).thenAnswer(new Answer() {
    private int iterations = 2;

    Object answer(InvocationOnMock invocation) {
        return iterations-- > 0;
    }
});

Answer 2

尝试

when(rs.next()).thenReturn(true).thenReturn(true).thenReturn(false);