测试从stdin读取并写入stdout的Java程序

Question

我正在为Java编程竞赛编写一些代码。程序的输入是使用stdin给出的，输出是在stdout上给出的。你们如何测试可在stdin /
stdout上运行的程序？这就是我的想法：

由于System.in的类型为InputStream，而System.out的类型为PrintStream，因此我使用以下原型在函数中编写了代码：

void printAverage(InputStream in, PrintStream out)

现在，我想使用junit对此进行测试。我想使用字符串伪造System.in并以字符串形式接收输出。

@Test
void testPrintAverage() {

    String input="10 20 30";
    String expectedOutput="20";

    InputStream in = getInputStreamFromString(input);
    PrintStream out = getPrintStreamForString();

    printAverage(in, out);

    assertEquals(expectedOutput, out.toString());
}

实现getInputStreamFromString（）和getPrintStreamForString（）的“正确”方法是什么？

我使这个变得比需要的复杂吗？

Answer 1

请尝试以下操作：

String string = "aaa";
InputStream stringStream = new java.io.ByteArrayInputStream(string.getBytes())

stringStream 是将从输入字符串中读取字符的流。

OutputStream outputStream = new java.io.ByteArrayOutputStream();
PrintStream printStream = new PrintStream(outputStream);
// .. writes to printWriter and flush() at the end.
String result = outputStream.toString()

printStream是PrintStream将写入的，而outputStream后者又将能够返回字符串。