package org.apache.maven.surefire.suite;
   
   /*
    * 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 org.apache.maven.shared.utils.StringUtils;
  import org.apache.maven.shared.utils.io.IOUtil;
  import org.apache.maven.shared.utils.xml.PrettyPrintXMLWriter;
  import org.apache.maven.shared.utils.xml.Xpp3Dom;
  import org.apache.maven.shared.utils.xml.Xpp3DomBuilder;
  import org.apache.maven.shared.utils.xml.Xpp3DomWriter;
  
  import java.io.BufferedInputStream;
  import java.io.ByteArrayOutputStream;
  import java.io.File;
  import java.io.FileInputStream;
  import java.io.FileNotFoundException;
  import java.io.FileWriter;
  import java.io.IOException;
  import java.io.InputStream;
  import java.io.PrintWriter;
  
  /**
   * Represents a test-run-result; this may be from a single test run or an aggregated result.
   * <p/>
   * In the case of timeout==true, the run-counts reflect the state of the test-run at the time
   * of the timeout.
   *
   * @author Kristian Rosenvold
   */
  public class RunResult
  {
      private final int completedCount;
  
      private final int errors;
  
      private final int failures;
  
      private final int skipped;
  
      private final int flakes;
  
      private final String failure;
  
      private final boolean timeout;
  
      public static final int SUCCESS = 0;
  
      private static final int FAILURE = 255;
  
      private static final int NO_TESTS = 254;
  
      public static RunResult timeout( RunResult accumulatedAtTimeout )
      {
          return errorCode( accumulatedAtTimeout, accumulatedAtTimeout.getFailure(), true );
      }
  
      public static RunResult failure( RunResult accumulatedAtTimeout, Exception cause )
      {
          return errorCode( accumulatedAtTimeout, getStackTrace( cause ), accumulatedAtTimeout.isTimeout() );
      }
  
      private static RunResult errorCode( RunResult other, String failure, boolean timeout )
      {
          return new RunResult( other.getCompletedCount(), other.getErrors(), other.getFailures(), other.getSkipped(),
                                failure, timeout );
  
      }
  
      public RunResult( int completedCount, int errors, int failures, int skipped )
      {
          this( completedCount, errors, failures, skipped, null, false );
      }
  
      public RunResult( int completedCount, int errors, int failures, int skipped, int flakes )
      {
          this( completedCount, errors, failures, skipped, flakes, null, false );
      }
  
      public RunResult( int completedCount, int errors, int failures, int skipped, String failure, boolean timeout )
      {
          this( completedCount, errors, failures, skipped, 0, failure, timeout );
      }
 
     public RunResult( int completedCount, int errors, int failures, int skipped, int flakes, String failure,
                       boolean timeout )
     {
         this.completedCount = completedCount;
         this.errors = errors;
         this.failures = failures;
         this.skipped = skipped;
         this.failure = failure;
         this.timeout = timeout;
         this.flakes = flakes;
     }
 
     private static String getStackTrace( Exception e )
     {
         if ( e == null )
         {
             return null;
         }
         ByteArrayOutputStream out = new ByteArrayOutputStream();
         PrintWriter pw = new PrintWriter( out );
         try
         {
             e.printStackTrace( pw );
         }
         finally
         {
             pw.close();
         }
         return new String( out.toByteArray() );
     }
 
     public int getCompletedCount()
     {
         return completedCount;
     }
 
     public int getErrors()
     {
         return errors;
     }
 
     public int getFlakes()
     {
         return flakes;
     }
 
     public int getFailures()
     {
         return failures;
     }
 
     public int getSkipped()
     {
         return skipped;
     }
 
     public Integer getFailsafeCode()  // Only used for compatibility reasons.
     {
         if ( completedCount == 0 )
         {
             return NO_TESTS;
         }
         if ( !isErrorFree() )
         {
             return FAILURE;
         }
         return null;
     }
 
     /* Indicates if the tests are error free */
     public boolean isErrorFree()
     {
         return getFailures() == 0 && getErrors() == 0;
     }
 
     /* Indicates test timeout or technical failure */
     public boolean isFailureOrTimeout()
     {
         return this.timeout || isFailure();
     }
 
     public boolean isFailure()
     {
         return failure != null;
     }
 
     public String getFailure()
     {
         return failure;
     }
 
     public boolean isTimeout()
     {
         return timeout;
     }
 
 
     public RunResult aggregate( RunResult other )
     {
         String failureMessage = getFailure() != null ? getFailure() : other.getFailure();
         boolean timeout = isTimeout() || other.isTimeout();
         int completed = getCompletedCount() + other.getCompletedCount();
         int fail = getFailures() + other.getFailures();
         int ign = getSkipped() + other.getSkipped();
         int err = getErrors() + other.getErrors();
         int flakes = getFlakes() + other.getFlakes();
         return new RunResult( completed, err, fail, ign, flakes, failureMessage, timeout );
     }
 
     public static RunResult noTestsRun()
     {
         return new RunResult( 0, 0, 0, 0 );
     }
 
     private Xpp3Dom create( String node, String value )
     {
         Xpp3Dom dom = new Xpp3Dom( node );
         dom.setValue( value );
         return dom;
     }
 
     private Xpp3Dom create( String node, int value )
     {
         return create( node, Integer.toString( value ) );
     }
 
     Xpp3Dom asXpp3Dom()
     {
         Xpp3Dom dom = new Xpp3Dom( "failsafe-summary" );
         Integer failsafeCode = getFailsafeCode();
         if ( failsafeCode != null )
         {
             dom.setAttribute( "result", Integer.toString( failsafeCode ) );
         }
         dom.setAttribute( "timeout", Boolean.toString( this.timeout ) );
         dom.addChild( create( "completed", this.completedCount ) );
         dom.addChild( create( "errors", this.errors ) );
         dom.addChild( create( "failures", this.failures ) );
         dom.addChild( create( "skipped", this.skipped ) );
         dom.addChild( create( "failureMessage", this.failure ) );
         return dom;
     }
 
     public static RunResult fromInputStream( InputStream inputStream, String encoding )
         throws FileNotFoundException
     {
         Xpp3Dom dom = Xpp3DomBuilder.build( inputStream, encoding );
         boolean timeout = Boolean.parseBoolean( dom.getAttribute( "timeout" ) );
         int completed = Integer.parseInt( dom.getChild( "completed" ).getValue() );
         int errors = Integer.parseInt( dom.getChild( "errors" ).getValue() );
         int failures = Integer.parseInt( dom.getChild( "failures" ).getValue() );
         int skipped = Integer.parseInt( dom.getChild( "skipped" ).getValue() );
         String failureMessage1 = dom.getChild( "failureMessage" ).getValue();
         String failureMessage = StringUtils.isEmpty( failureMessage1 ) ? null : failureMessage1;
         return new RunResult( completed, errors, failures, skipped, failureMessage, timeout );
     }
 
     public void writeSummary( File summaryFile, boolean inProgress, String encoding )
         throws IOException
     {
         if ( !summaryFile.getParentFile().isDirectory() )
         {
             //noinspection ResultOfMethodCallIgnored
             summaryFile.getParentFile().mkdirs();
         }
 
         FileInputStream fin = null;
         FileWriter writer = null;
         try
         {
             RunResult mergedSummary = this;
             if ( summaryFile.exists() && inProgress )
             {
                 fin = new FileInputStream( summaryFile );
 
                 RunResult runResult = RunResult.fromInputStream( new BufferedInputStream( fin ), encoding );
                 mergedSummary = mergedSummary.aggregate( runResult );
             }
 
             writer = new FileWriter( summaryFile );
             writer.write( "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n" );
             PrettyPrintXMLWriter prettyPrintXMLWriter = new PrettyPrintXMLWriter( writer );
             Xpp3DomWriter.write( prettyPrintXMLWriter, mergedSummary.asXpp3Dom() );
         }
         finally
         {
             IOUtil.close( fin );
             IOUtil.close( writer );
         }
     }
 
     @SuppressWarnings( "RedundantIfStatement" )
     public boolean equals( Object o )
     {
         if ( this == o )
         {
             return true;
         }
         if ( o == null || getClass() != o.getClass() )
         {
             return false;
         }
 
         RunResult runResult = (RunResult) o;
 
         if ( completedCount != runResult.completedCount )
         {
             return false;
         }
         if ( errors != runResult.errors )
         {
             return false;
         }
         if ( failures != runResult.failures )
         {
             return false;
         }
         if ( skipped != runResult.skipped )
         {
             return false;
         }
         if ( timeout != runResult.timeout )
         {
             return false;
         }
         if ( failure != null ? !failure.equals( runResult.failure ) : runResult.failure != null )
         {
             return false;
         }
 
         return true;
     }
 
     public int hashCode()
     {
         int result = completedCount;
         result = 31 * result + errors;
         result = 31 * result + failures;
         result = 31 * result + skipped;
         result = 31 * result + ( failure != null ? failure.hashCode() : 0 );
         result = 31 * result + ( timeout ? 1 : 0 );
         return result;
     }
 }