使用Mockito的JUnit测试用户输入

我对Mocking Framework 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。

提前致谢！