1 package org.apache.maven.surefire.report;
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22 import java.io.ByteArrayOutputStream;
23 import java.io.IOException;
24 import java.io.PrintStream;
25
26
27
28
29
30 public class ConsoleOutputCapture
31 {
32
33 private static final PrintStream oldOut = System.out;
34
35 private static final PrintStream oldErr = System.err;
36
37 public ConsoleOutputCapture( ConsoleOutputReceiver target )
38 {
39 System.setOut( new ForwardingPrintStream( true, target ) );
40
41 System.setErr( new ForwardingPrintStream( false, target ) );
42 }
43
44 public void restoreStreams()
45 {
46 System.setOut( oldOut );
47 System.setErr( oldErr );
48 }
49
50 static class ForwardingPrintStream
51 extends PrintStream
52 {
53 private final boolean isStdout;
54
55 private final ConsoleOutputReceiver target;
56
57 ForwardingPrintStream( boolean stdout, ConsoleOutputReceiver target )
58 {
59 super( new ByteArrayOutputStream() );
60 isStdout = stdout;
61 this.target = target;
62 }
63
64 public void write( byte[] buf, int off, int len )
65 {
66 target.writeTestOutput( buf, off, len, isStdout );
67 }
68
69 public void write( byte[] b )
70 throws IOException
71 {
72 target.writeTestOutput( b, 0, b.length, isStdout );
73 }
74
75 public void close()
76 {
77 }
78
79 public void flush()
80 {
81 }
82 }
83
84 }