java - 如何将输入发送到 Junit 中的 System.console.readLine(...)？

Question

我有一个要测试的主要方法。

我只是想知道如何通过 console.readLine(...) 和 console.readLine(...) 对我的 Junit 测试的期望 - 无需重构 main(...) - 我正在使用JMockit 是否可以在这里使用 - 即模拟 System.console()？。

class MyClass {
    public static void main(String[] args) {
        Console console = System.console();

        String username = console.readLine("Enter your username: ");
        char[] newPassword = console.readPassword("Enter your new password: ");

        ...
    }
}

class MyJunitTest {
    @Test
    public void test() {
        MyClass.main(null);
        // here I'd just like to pass the username and password to the console?
    }
}

Answer 1

我使用 JMockit 来模拟 Console 类:

  @Test
  public void test(
      @Mocked final System systemMock,
      @Mocked final Console consoleMock) {
    new NonStrictExpectations() {
      {
        System.console();
        result = consoleMock;

        consoleMock.readLine(anyString);
        result = "aUsername";

        consoleMock.readPassword(anyString);
        result = "aPassword";
      }
    };

    MyClass.main(null);
  }