JUnit:如何模拟System.in测试？

Question

我有一个Java命令行程序.我想创建JUnit测试用例以便能够模拟System.in.因为当我的程序运行时,它将进入while循环并等待来自用户的输入.我如何在JUnit中模拟它？

谢谢

Answer 1

切换技术上是可行的System.in,但一般情况下,不直接在代码中调用它会更加健壮,而是添加一个间接层,以便从应用程序的一个点控制输入源.具体如何做到这一点是一个实现细节 - 依赖注入的建议很好,但你不一定需要引入第三方框架; 例如,您可以从调用代码传递I/O上下文.

如何切换System.in:

String data = "Hello, World!\r\n";
InputStream stdin = System.in;
try {
  System.setIn(new ByteArrayInputStream(data.getBytes()));
  Scanner scanner = new Scanner(System.in);
  System.out.println(scanner.nextLine());
} finally {
  System.setIn(stdin);
}

Answer 2

根据@ McDowell的回答和另一个显示如何测试System.out 的答案,我想分享我的解决方案,为程序提供输入并测试其输出.

作为参考,我使用JUnit 4.12.

假设我们有这个程序简单地将输入复制到输出:

import java.util.Scanner;

public class SimpleProgram {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        System.out.print(scanner.next());
        scanner.close();
    }
}

要测试它,我们可以使用以下类:

import static org.junit.Assert.*;

import java.io.*;

import org.junit.*;

public class SimpleProgramTest {
    private final InputStream systemIn = System.in;
    private final PrintStream systemOut = System.out;

    private ByteArrayInputStream testIn;
    private ByteArrayOutputStream testOut;

    @Before
    public void setUpOutput() {
        testOut = new ByteArrayOutputStream();
        System.setOut(new PrintStream(testOut));
    }

    private void provideInput(String data) {
        testIn = new ByteArrayInputStream(data.getBytes());
        System.setIn(testIn);
    }

    private String getOutput() {
        return testOut.toString();
    }

    @After
    public void restoreSystemInputOutput() {
        System.setIn(systemIn);
        System.setOut(systemOut);
    }

    @Test
    public void testCase1() {
        final String testString = "Hello!";
        provideInput(testString);

        SimpleProgram.main(new String[0]);

        assertEquals(testString, getOutput());
    }
}

我不会解释太多,因为我相信代码是可读的,我引用了我的来源.

当JUnit运行时testCase1(),它将按照它们出现的顺序调用辅助方法:

1. setUpOutput(),因为@Before注释
2. provideInput(String data),来自 testCase1()
3. getOutput(),来自 testCase1()
4. restoreSystemInputOutput(),因为@After注释

我没有测试,System.err因为我不需要它,但它应该很容易实现,类似于测试System.out.

Answer 3

有几种方法可以解决这个问题.最完整的方法是在运行被测试类时传入一个InputStream,这是一个伪造的InputStream,它将模拟数据传递给您的类.如果您需要在代码中执行此操作,可以查看依赖注入框架(例如Google Guice),但简单的方法是:

 public class MyClass {
     private InputStream systemIn;

     public MyClass() {
         this(System.in);
     }

     public MyClass(InputStream in) {
         systemIn = in;
     }
 }

在测试中,您将调用获取输入流的构造函数.您甚至将该构造函数包设为私有并将测试放在同一个包中,以便其他代码通常不会考虑使用它.

Answer 4

尝试重构代码以使用依赖注入。不要让您的方法System.in直接使用，而是让方法接受一个InputStream作为参数。然后，在junit测试中，您可以通过测试InputStream实现来代替System.in。