java - 控制台输入和输出的 JUnit 测试

Question

我只有一个方法 main.如何使用 JUnit 检查 System.out.println() 并替换 Scanner 自动输入值？

附注请提供一些解决方案...

public static void main(String[] args) {
    Scanner scanner = new Scanner(System.in);
    int[] arr = new int[4];

    for (int i = 0; i < arr.length; i++) {
        arr[i] = scanner.nextInt();
    }

    for (int i = 0; i < arr.length; i++) {
        int res = 0;
        int k = 0;
        int num = arr[i];
        /*.....*/
        System.out.println(num);
    }
}

Answer 1

理想情况下，提取尴尬的依赖项，以便您可以在没有它们的情况下进行测试。将 main 更改为简单的:

public static void main(String[] args) {
  doWork(new Scanner(System.in), System.out);
}

// TODO: Rename to something meaningful
public static void doWork(Scanner input, PrintStream output) {
  // Remainder of code
}

(考虑使用 Writer 而不是 PrintStream 进行输出。)

那么您实际上并不需要对 main 进行单元测试 - 但您可以使用基于 的 Scanner 来测试 doWork StringReader，以及基于 StringWriter 的输出，提供您想要的任何输入并检查输出。