OutputCapture

最新推荐文章于 2022-08-15 17:33:35 发布

欧阳某某某

最新推荐文章于 2022-08-15 17:33:35 发布

阅读量1.2k

点赞数

分类专栏： JUtil 文章标签： Rule JUnit 断言

本文链接：https://blog.csdn.net/wh_ouyangshuang/article/details/46740513

版权

JUtil 专栏收录该内容

0 篇文章 0 订阅

订阅专栏

OutputCapture是一个JUnit Rule，用于捕获System.out和System.err输出。只需简单的将捕获声明为一个@Rule，并使用toString()断言。

    @Rule
    public OutputCapture outputCapture = new OutputCapture();

用法也很方便

    @Test
    public void testDefaultSettings() {
        SampleApplication.main(new String[]{});
        String output = this.outputCapture.toString();
        TestCase.assertTrue("Wrong output: " + output, output.contains("Hello ouyang"));
    }

看看它的部分源码实现，捕获System.out和System.err输出。

    protected void captureOutput() {
        AnsiOutputControl.get().disableAnsiOutput();
        this.copy = new ByteArrayOutputStream();
        this.captureOut = new CaptureOutputStream(System.out, this.copy);
        this.captureErr = new CaptureOutputStream(System.err, this.copy);
        System.setOut(new PrintStream(this.captureOut));
        System.setErr(new PrintStream(this.captureErr));
    }
    @Override
    public String toString() {
        flush();
        return this.copy.toString();
    }