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 org.apache.maven.surefire.api.report.ReportEntry;
  import org.apache.maven.surefire.api.report.RunMode;
  import org.apache.maven.surefire.api.report.StackTraceWriter;
  import org.apache.maven.surefire.api.report.TestSetReportEntry;
  
  import javax.annotation.Nonnull;
  import java.util.Collections;
  import java.util.Map;
  
  import static java.util.Collections.unmodifiableMap;
  import static org.apache.maven.plugin.surefire.report.ReporterUtils.formatElapsedTime;
  import static org.apache.maven.surefire.api.util.internal.StringUtils.NL;
  import static org.apache.maven.surefire.shared.utils.StringUtils.isBlank;
  
  /**
   * @author Kristian Rosenvold
   */
  public class WrappedReportEntry
      implements TestSetReportEntry
  {
      private final ReportEntry original;
  
      private final ReportEntryType reportEntryType;
  
      private final Integer elapsed;
  
      private final Utf8RecodingDeferredFileOutputStream stdout;
  
      private final Utf8RecodingDeferredFileOutputStream stdErr;
  
      private final Map<String, String> systemProperties;
  
      public WrappedReportEntry( ReportEntry original, ReportEntryType reportEntryType, Integer estimatedElapsed,
                                 Utf8RecodingDeferredFileOutputStream stdout,
                                 Utf8RecodingDeferredFileOutputStream stdErr,
                                 Map<String, String> systemProperties )
      {
          this.original = original;
          this.reportEntryType = reportEntryType;
          this.elapsed = estimatedElapsed;
          this.stdout = stdout;
          this.stdErr = stdErr;
          this.systemProperties = unmodifiableMap( systemProperties );
      }
  
      public WrappedReportEntry( ReportEntry original, ReportEntryType reportEntryType, Integer estimatedElapsed,
                                 Utf8RecodingDeferredFileOutputStream stdout,
                                 Utf8RecodingDeferredFileOutputStream stdErr )
      {
          this( original, reportEntryType, estimatedElapsed, stdout, stdErr, Collections.<String, String>emptyMap() );
      }
  
      @Override
      public Integer getElapsed()
      {
          return elapsed;
      }
  
      @Override
      public int getElapsed( int fallback )
      {
          return elapsed == null ? fallback : elapsed;
      }
  
      public ReportEntryType getReportEntryType()
      {
          return reportEntryType;
      }
  
      public Utf8RecodingDeferredFileOutputStream getStdout()
      {
          return stdout;
      }
  
      public Utf8RecodingDeferredFileOutputStream getStdErr()
      {
          return stdErr;
      }
 
     @Override
     public String getSourceName()
     {
         return original.getSourceName();
     }
 
     @Override
     public String getSourceText()
     {
         return original.getSourceText();
     }
 
     @Override
     public String getName()
     {
         return original.getName();
     }
 
     @Override
     public String getNameText()
     {
         return original.getNameText();
     }
 
     public String getClassMethodName()
     {
         return original.getSourceName() + "." + original.getName();
     }
 
     @Override
     public String getGroup()
     {
         return original.getGroup();
     }
 
     @Override
     public StackTraceWriter getStackTraceWriter()
     {
         return original.getStackTraceWriter();
     }
 
     @Override
     public String getMessage()
     {
         return original.getMessage();
     }
 
     public String getStackTrace( boolean trimStackTrace )
     {
         StackTraceWriter w = original.getStackTraceWriter();
         return w == null ? null : ( trimStackTrace ? w.writeTrimmedTraceToString() : w.writeTraceToString() );
     }
 
     public String elapsedTimeAsString()
     {
         return formatElapsedTime( getElapsed() );
     }
 
     String getReportSourceName()
     {
         String sourceName = getSourceName();
         String sourceText = getSourceText();
         return isBlank( sourceText ) ? sourceName : sourceText;
     }
 
     String getReportSourceName( String suffix )
     {
         return isBlank( suffix ) ? getReportSourceName() : getReportSourceName() + "(" + suffix + ")";
     }
 
     String getSourceName( String suffix )
     {
         return isBlank( suffix ) ? getSourceName() : getSourceName() + "(" + suffix + ")";
     }
 
     String getReportName()
     {
         String name = getName();
         String nameText = getNameText();
         return isBlank( nameText ) ? name : nameText;
     }
 
     public String getOutput( boolean trimStackTrace )
     {
         String outputLine = getElapsedTimeSummary() + "  <<< " + getReportEntryType().name() + "!";
         String trimmedStackTrace = getStackTrace( trimStackTrace );
         return trimmedStackTrace == null ? outputLine : outputLine + NL + trimmedStackTrace;
     }
 
     public String getElapsedTimeVerbose()
     {
         return "Time elapsed: " + elapsedTimeAsString() + " s";
     }
 
     public String getElapsedTimeSummary()
     {
         String description = getName() == null ? getSourceName() : getClassMethodName();
         return description + "  " + getElapsedTimeVerbose();
     }
 
     public boolean isErrorOrFailure()
     {
         ReportEntryType thisType = getReportEntryType();
         return ReportEntryType.FAILURE == thisType || ReportEntryType.ERROR == thisType;
     }
 
     public boolean isSkipped()
     {
         return ReportEntryType.SKIPPED == getReportEntryType();
     }
 
     public boolean isSucceeded()
     {
         return ReportEntryType.SUCCESS == getReportEntryType();
     }
 
     @Override
     public String getNameWithGroup()
     {
         return original.getNameWithGroup();
     }
 
     @Override
     public String getReportNameWithGroup()
     {
         String reportNameWithGroup = original.getReportNameWithGroup();
 
         if ( isBlank ( reportNameWithGroup ) )
         {
             return getNameWithGroup();
         }
 
         return reportNameWithGroup;
     }
 
     @Nonnull
     @Override
     public RunMode getRunMode()
     {
         return original.getRunMode();
     }
 
     @Override
     public Long getTestRunId()
     {
         return original.getTestRunId();
     }
 
     @Override
     public Map<String, String> getSystemProperties()
     {
         return systemProperties;
     }
 }