mocking - mock 时如何跳出循环

Question

我有一段代码是这样的:

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

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

Answer 1

您可以在训练 mock 时提供多个 return，最后一个可以为 null:

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