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

Question

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

由于 System.in 是 InputStream 类型，而 System.out 是 PrintStream 类型，我用这个原型(prototype)在 func 中编写了我的代码:

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，而后者又能够返回一个字符串。