package org.apache.maven.plugin.surefire.report;
   
   /*
    * Licensed to the Apache Software Foundation (ASF) under one
    * or more contributor license agreements.  See the NOTICE file
    * distributed with this work for additional information
    * regarding copyright ownership.  The ASF licenses this file
    * to you under the Apache License, Version 2.0 (the
    * "License"); you may not use this file except in compliance
   * with the License.  You may obtain a copy of the License at
   *
   *     http://www.apache.org/licenses/LICENSE-2.0
   *
   * Unless required by applicable law or agreed to in writing,
   * software distributed under the License is distributed on an
   * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
   * KIND, either express or implied.  See the License for the
   * specific language governing permissions and limitations
   * under the License.
   */
  
  import java.util.ArrayList;
  import java.util.Collections;
  import java.util.List;
  import org.apache.maven.plugin.surefire.runorder.StatisticsReporter;
  import org.apache.maven.surefire.report.ConsoleLogger;
  import org.apache.maven.surefire.report.ConsoleOutputReceiver;
  import org.apache.maven.surefire.report.ReportEntry;
  import org.apache.maven.surefire.report.RunListener;
  import org.apache.maven.surefire.report.RunStatistics;
  import org.apache.maven.surefire.util.internal.ByteBuffer;
  
  /**
   * Reports data for a single test set.
   * <p/>
   * 
   * @author Kristian Rosenvold
   */
  public class TestSetRunListener
      implements RunListener, ConsoleOutputReceiver, ConsoleLogger
  {
      private final RunStatistics globalStatistics;
  
      private final TestSetStats detailsForThis;
  
      private final List<ByteBuffer> testStdOut = Collections.synchronizedList( new ArrayList<ByteBuffer>() );
  
      private final List<ByteBuffer> testStdErr = Collections.synchronizedList( new ArrayList<ByteBuffer>() );
  
      private final TestcycleConsoleOutputReceiver consoleOutputReceiver;
  
      private final boolean briefOrPlainFormat;
  
      private final StatelessXmlReporter simpleXMLReporter;
  
      private final ConsoleReporter consoleReporter;
  
      private final FileReporter fileReporter;
  
      private final StatisticsReporter statisticsReporter;
  
      public TestSetRunListener( ConsoleReporter consoleReporter, FileReporter fileReporter,
                                 StatelessXmlReporter simpleXMLReporter,
                                 TestcycleConsoleOutputReceiver consoleOutputReceiver,
                                 StatisticsReporter statisticsReporter, RunStatistics globalStats,
                                 boolean trimStackTrace, boolean isPlainFormat, boolean briefOrPlainFormat )
      {
          this.consoleReporter = consoleReporter;
          this.fileReporter = fileReporter;
          this.statisticsReporter = statisticsReporter;
          this.simpleXMLReporter = simpleXMLReporter;
          this.consoleOutputReceiver = consoleOutputReceiver;
          this.briefOrPlainFormat = briefOrPlainFormat;
          this.detailsForThis = new TestSetStats( trimStackTrace, isPlainFormat );
          this.globalStatistics = globalStats;
      }
  
      public void info( String message )
      {
          if ( consoleReporter != null )
          {
              consoleReporter.writeMessage( message );
          }
      }
  
      public void writeTestOutput( byte[] buf, int off, int len, boolean stdout )
      {
          ByteBuffer byteBuffer = new ByteBuffer( buf, off, len );
          if ( stdout )
          {
              testStdOut.add( byteBuffer );
          }
          else
          {
              testStdErr.add( byteBuffer );
          }
          consoleOutputReceiver.writeTestOutput( buf, off, len, stdout );
      }
  
     public void testSetStarting( ReportEntry report )
     {
         detailsForThis.testSetStart();
         if ( consoleReporter != null )
         {
             consoleReporter.testSetStarting( report );
         }
         consoleOutputReceiver.testSetStarting( report );
     }
 
     public void clearCapture()
     {
         testStdErr.clear();
         testStdOut.clear();
     }
 
     public void testSetCompleted( ReportEntry report )
     {
         WrappedReportEntry wrap = wrapTestSet( report );
         List<String> testResults = briefOrPlainFormat ? detailsForThis.getTestResults() : null;
         if ( consoleReporter != null )
         {
             consoleReporter.testSetCompleted( wrap, detailsForThis, testResults );
         }
         consoleOutputReceiver.testSetCompleted( wrap );
         if ( fileReporter != null )
         {
             fileReporter.testSetCompleted( wrap, detailsForThis, testResults );
         }
         if ( simpleXMLReporter != null )
         {
             simpleXMLReporter.testSetCompleted( wrap, detailsForThis );
         }
         if ( statisticsReporter != null )
         {
             statisticsReporter.testSetCompleted();
         }
         if ( consoleReporter != null )
         {
             consoleReporter.reset();
         }
 
         globalStatistics.add( detailsForThis );
         detailsForThis.reset();
 
     }
 
     // ----------------------------------------------------------------------
     // Test
     // ----------------------------------------------------------------------
 
     public void testStarting( ReportEntry report )
     {
         detailsForThis.testStart();
 
     }
 
     public void testSucceeded( ReportEntry reportEntry )
     {
         WrappedReportEntry wrapped = wrap( reportEntry, ReportEntryType.success );
         detailsForThis.testSucceeded( wrapped );
         if ( statisticsReporter != null )
         {
             statisticsReporter.testSucceeded( reportEntry );
         }
         clearCapture();
     }
 
     public void testError( ReportEntry reportEntry )
     {
         WrappedReportEntry wrapped = wrap( reportEntry, ReportEntryType.error );
         detailsForThis.testError( wrapped );
         if ( statisticsReporter != null )
         {
             statisticsReporter.testError( reportEntry );
         }
         globalStatistics.addErrorSource( reportEntry.getStackTraceWriter() );
         clearCapture();
     }
 
     public void testFailed( ReportEntry reportEntry )
     {
         WrappedReportEntry wrapped = wrap( reportEntry, ReportEntryType.failure );
         detailsForThis.testFailure( wrapped );
         if ( statisticsReporter != null )
         {
             statisticsReporter.testFailed( reportEntry );
         }
         globalStatistics.addFailureSource( reportEntry.getStackTraceWriter() );
         clearCapture();
     }
 
     // ----------------------------------------------------------------------
     // Counters
     // ----------------------------------------------------------------------
 
     public void testSkipped( ReportEntry reportEntry )
     {
         WrappedReportEntry wrapped = wrap( reportEntry, ReportEntryType.skipped );
         detailsForThis.testSkipped( wrapped );
         if ( statisticsReporter != null )
         {
             statisticsReporter.testSkipped( reportEntry );
         }
         clearCapture();
     }
 
     public void testAssumptionFailure( ReportEntry report )
     {
         testSkipped( report );
     }
 
     public String getAsString( List<ByteBuffer> byteBufferList )
     {
         StringBuilder stringBuffer = new StringBuilder();
         // To avoid getting a java.util.ConcurrentModificationException while iterating (see SUREFIRE-879) we need to
         // iterate over a copy or the elements array. Since the passed in byteBufferList is always wrapped with
         // Collections.synchronizedList( ) we are guaranteed toArray() is going to be atomic, so we are safe.
         for ( Object byteBuffer : byteBufferList.toArray() )
         {
             stringBuffer.append( byteBuffer.toString() );
         }
         return stringBuffer.toString();
     }
 
     private WrappedReportEntry wrap( ReportEntry other, ReportEntryType reportEntryType )
     {
         final int estimatedElapsed;
         if ( reportEntryType != ReportEntryType.skipped )
         {
             if ( other.getElapsed() != null )
             {
                 estimatedElapsed = other.getElapsed();
             }
             else
             {
                 estimatedElapsed = detailsForThis.getElapsedSinceLastStart();
             }
         }
         else
         {
             estimatedElapsed = 0;
         }
 
         return new WrappedReportEntry( other, reportEntryType, estimatedElapsed, getAsString( testStdOut ),
                                        getAsString( testStdErr ) );
     }
 
     private WrappedReportEntry wrapTestSet( ReportEntry other )
     {
         return new WrappedReportEntry( other, null, other.getElapsed() != null 
                         ? other.getElapsed()
                         : detailsForThis.getElapsedSinceTestSetStart(), getAsString( testStdOut ),
                                        getAsString( testStdErr ) );
     }
 
     public void close()
     {
         if ( consoleOutputReceiver != null )
         {
             consoleOutputReceiver.close();
         }
     }
 }