java - JUnit 使用 Mockito 测试用户输入

Question

我在使用模拟框架 Mockito 时遇到问题。我想测试用户在控制台中键入的输入。如果没有 Mocking，它工作得很好，但有了它，我不知道如何从函数中获取返回值，该函数获取控制台中的输入。

没有 Mockito:

读取输入法

public int readUserInput() {
    BufferedReader console = new BufferedReader(new InputStreamReader(
            System.in));
    return Integer.parseInt(console.readLine());
    ...
}

测试方法:

public class TestClass {
    Controller contr = new Controller();
    @Test
    public void testUserInput() {
        System.setIn(new ByteArrayInputStream("1".getBytes()));
        assertEquals(1, contr.readUserInput());
        System.setIn(System.in);
    }
}

但是现在如何使用 Mockito 编写测试。我尝试了几种方法，但没有一个有效。我一直得到 0，而不是我在 ByteArrayInputStream 中写入的数字。

提前致谢!

Answer 1

您需要模拟 BufferedReader 的 readLine() API 来返回您的值。即

 Mockito.when(mockedBufferedReader.readLine())
                .thenReturn("1");

现在，当您从测试中调用 readInput() 时，将调用您的模拟缓冲读取器的 API。希望这能回答您的问题。