3

我正在编写一个测试,我想捕获测试方法在 STDOUT 上发送的消息。这是我的代码:

@Test
public void testAction() throws IllegalArgumentException, NoSuchMethodException, IllegalAccessException, InvocationTargetException,
        CmdLineException, IOException {
    PrintStream originalSTDOUT = System.out;
    ByteArrayOutputStream baos = new ByteArrayOutputStream();
    PrintStream ps = new PrintStream(baos);

    try {
        System.setOut(ps);

        // Call to the method that will print text to STDOUT...

        ps.flush();
        String batchLog = baos.toString("UTF-8");
        assertTrue("Invalid exception text !", batchLog.contains("my expected text..."));
    } finally {
        System.setOut(originalSTDOUT);//Restore original STDOUT
        originalSTDOUT.write(baos.toByteArray());// Write back to STDOUT, even if I comment this line, outputs go to STDOUT... 

        ps.close();
    }
}

不幸的是,batchLog尽管我仍然可以将预期的文本读取到 STDOUT,但它始终是空的。

将文本打印到 STDOUT 的方法实际上捕获了一个异常。然后它将它传递给Logger如下所示:

log.warn("发生错误:", e);

e我在测试中调用的方法中引发的异常在哪里。

使用Logger此附加程序打印其消息:org.apache.log4j.ConsoleAppender. 此附加程序没有应用特殊配置。

我错过了什么?

PS:这个SO 答案很有趣,但它对我不起作用,因为ConsoleAppender它已经在StandardOutputStreamLog 规则可以起作用之前创建。

Java 6
Junit 4.10

4

3 回答 3

8

我终于解决了我的问题!

我的第一次尝试是错误的,因为我重定向 STDOUT 为时已晚,正如@Joni 在他的评论中所说:ConsoleAppender已经创建。

由于异常消息由 a 处理Logger,因此我决定添加 aWriterAppender支持的 aStringWriter以获取消息。

这是我的工作解决方案:

@Test
public void testAction() throws IllegalArgumentException, NoSuchMethodException, IllegalAccessException, InvocationTargetException,
        CmdLineException, IOException {

        // Setup WriterAppender
        Writer w = new StringWriter();
        Layout l = new PatternLayout("%m%n");

        WriterAppender wa = new WriterAppender(l, w);
        wa.setEncoding("UTF-8");
        wa.setThreshold(Level.ALL);
        wa.activateOptions();// WriterAppender does nothing here, but I like defensive code...

        // Add it to logger
        Logger log = Logger.getLogger(ExceptionHandler.class);// ExceptionHandler is the class that contains this code : `log.warn("An error has occured:", e);'
        log.addAppender(wa);

        try {
             // Call to the method that will print text to STDOUT...

             String batchLog = w.toString();
             assertTrue("Invalid exception text !", batchLog.contains("my expected text..."));
        } finally {
             // Cleanup everything...
             log.removeAppender(wa);
             wa.close();
        }
}
于 2013-08-12T14:19:09.700 回答
3

稍微偏离主题,但如果有些人(比如我,当我第一次发现这个线程时)可能对通过 SLF4J 捕获日志输出感兴趣,commons-testing的 JUnit@Rule可能会有所帮助:

public class FooTest {
    @Rule
    public final ExpectedLogs logs = new ExpectedLogs() {{
        captureFor(Foo.class, LogLevel.WARN);
    }};

    @Test
    public void barShouldLogWarning() {
        assertThat(logs.isEmpty(), is(true)); // Nothing captured yet.

        // Logic using the class you are capturing logs for:
        Foo foo = new Foo();
        assertThat(foo.bar(), is(not(nullValue())));

        // Assert content of the captured logs:
        assertThat(logs.isEmpty(), is(false));
        assertThat(logs.contains("Your warning message here"), is(true));
    }
}

免责声明

  • 我开发了这个库,因为我找不到任何适合我自己需要的解决方案。
  • log4j目前只有,log4j2和的绑定logback可用,但我很乐意添加更多。
于 2014-12-11T08:32:15.383 回答
0

尝试在 PrintStream 上启用 autoflushing true:

PrintStream ps = new PrintStream(baos,true);
于 2013-08-12T11:31:22.583 回答