JUnit test for System.out.println()

2022-08-31 04:34:58

我需要为一个设计不佳的旧应用程序编写JUnit测试,并且正在向标准输出编写大量错误消息。当该方法行为正确时,它将返回 XML 响应:getResponse(String request)

@BeforeClass
public static void setUpClass() throws Exception {
    Properties queries = loadPropertiesFile("requests.properties");
    Properties responses = loadPropertiesFile("responses.properties");
    instance = new ResponseGenerator(queries, responses);
}

@Test
public void testGetResponse() {
    String request = "<some>request</some>";
    String expResult = "<some>response</some>";
    String result = instance.getResponse(request);
    assertEquals(expResult, result);
}

但是,当它得到格式错误的XML或不理解请求时,它会返回并将一些东西写入标准输出。null

有没有办法在 JUnit 中断言控制台输出?要捕获以下情况:

System.out.println("match found: " + strExpr);
System.out.println("xml not well formed: " + e.getMessage());

答案 1

使用ByteArrayOutputStream和System.setXXX很简单:

private final ByteArrayOutputStream outContent = new ByteArrayOutputStream();
private final ByteArrayOutputStream errContent = new ByteArrayOutputStream();
private final PrintStream originalOut = System.out;
private final PrintStream originalErr = System.err;

@Before
public void setUpStreams() {
    System.setOut(new PrintStream(outContent));
    System.setErr(new PrintStream(errContent));
}

@After
public void restoreStreams() {
    System.setOut(originalOut);
    System.setErr(originalErr);
}

示例测试用例:

@Test
public void out() {
    System.out.print("hello");
    assertEquals("hello", outContent.toString());
}

@Test
public void err() {
    System.err.print("hello again");
    assertEquals("hello again", errContent.toString());
}

我用这段代码来测试命令行选项(断言-version输出版本字符串等)

编辑:这个答案的先前版本在测试后调用;这就是评论者引用的NullPointerExceptions的原因。System.setOut(null)


答案 2

我知道这是一个旧线程,但是有一个很好的库可以做到这一点:来自文档的系统规则
示例:

public void MyTest {
    @Rule
    public final SystemOutRule systemOutRule = new SystemOutRule().enableLog();

    @Test
    public void overrideProperty() {
        System.out.print("hello world");
        assertEquals("hello world", systemOutRule.getLog());
    }
}

它还允许您捕获命令行工具需要测试的其他内容。System.exit(-1)


推荐