View Javadoc
1   package org.apache.maven.surefire.report;
2   
3   /*
4    * Licensed to the Apache Software Foundation (ASF) under one
5    * or more contributor license agreements.  See the NOTICE file
6    * distributed with this work for additional information
7    * regarding copyright ownership.  The ASF licenses this file
8    * to you under the Apache License, Version 2.0 (the
9    * "License"); you may not use this file except in compliance
10   * with the License.  You may obtain a copy of the License at
11   *
12   *     http://www.apache.org/licenses/LICENSE-2.0
13   *
14   * Unless required by applicable law or agreed to in writing,
15   * software distributed under the License is distributed on an
16   * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
17   * KIND, either express or implied.  See the License for the
18   * specific language governing permissions and limitations
19   * under the License.
20   */
21  
22  import java.io.ByteArrayOutputStream;
23  import java.io.IOException;
24  import java.io.PrintStream;
25  import org.apache.maven.surefire.util.internal.ByteBuffer;
26  
27  /**
28   * Deals with system.out/err.
29   * <p/>
30   */
31  public class ConsoleOutputCapture
32  {
33      public static void startCapture( ConsoleOutputReceiver target )
34      {
35          System.setOut( new ForwardingPrintStream( true, target ) );
36  
37          System.setErr( new ForwardingPrintStream( false, target ) );
38      }
39  
40      private static class ForwardingPrintStream
41          extends PrintStream
42      {
43          private final boolean isStdout;
44  
45          private final ConsoleOutputReceiver target;
46  
47          ForwardingPrintStream( boolean stdout, ConsoleOutputReceiver target )
48          {
49              super( new ByteArrayOutputStream() );
50              isStdout = stdout;
51              this.target = target;
52          }
53  
54          public void write( byte[] buf, int off, int len )
55          {
56              // Note: At this point the supplied "buf" instance is reused, which means
57              // data must be copied out of the buffer
58              target.writeTestOutput( buf, off, len, isStdout );
59          }
60  
61          public void write( byte[] b )
62              throws IOException
63          {
64              target.writeTestOutput( b, 0, b.length, isStdout );
65          }
66  
67          public void write( int b )
68          {
69              byte[] buf = new byte[1];
70              buf[0] = (byte) b;
71              try
72              {
73                  write( buf );
74              }
75              catch ( IOException e )
76              {
77                  setError();
78              }
79          }
80  
81          static final byte[] NL = new byte[]{ (byte) '\n' };
82  
83          public void println( String s )
84          {
85              if ( s == null )
86              {
87                  s = "null"; // Shamelessly taken from super.print
88              }
89              final byte[] bytes = s.getBytes();
90              final byte[] join = ByteBuffer.join( bytes, 0, bytes.length, NL, 0, 1 );
91              target.writeTestOutput( join, 0, join.length, isStdout );
92          }
93  
94          public void close()
95          {
96          }
97  
98          public void flush()
99          {
100         }
101     }
102 
103 }