如何对使用 Scanner 和 InputStream 的方法进行单元测试？

Question

代码如下：

public int foo(InputStream in) {
    int counter = 0;
    Scanner scanner = new Scanner(in);
    while(scanner.hasNextLine()) {
        counter++;
        scanner.nextLine();
    }
    return counter;
}

通常我会将 FileInputStream 传递给此方法，但是我希望能够在不访问物理文件的情况下对其进行测试。

我应该模拟 File 对象吗？如何实现

@Test
public void shouldReturnThree(){

}

Answer 1

您可以像这样将测试字符串作为 InputStream 传递给方法（：

InputStream stream = new ByteArrayInputStream(exampleString.getBytes());

（无耻从this answer盗取）

例如，这段代码将打印 3：

String str = "\n\n\n";
InputStream stream = new ByteArrayInputStream(str.getBytes());
System.out.println(foo(stream));