如何在嘲笑时突破循环

Question

我有一段代码如下：

while(count = inputStream.readLine()) != null) {
    //do something
}

在单元测试用例中，我正在模拟inputStream.readLine()并返回一个字符串，但在这种情况下，每次返回一个字符串时它将进入无限循环

Answer 1

您可以在训练模拟时提供多个返回，最后一个可以为null：

when(inputStream.readLine()).thenReturn("one", "two", "three", (String)null);