| package org.apache.maven.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.SafeThrowable; |
| import org.apache.maven.surefire.api.report.StackTraceWriter; |
| import org.apache.maven.surefire.api.util.internal.StringUtils; |
| |
| import java.io.PrintWriter; |
| import java.io.StringWriter; |
| import java.util.Objects; |
| |
| /** |
| * Write the trace out for a POJO test. |
| * |
| * @author <a href="mailto:brett@apache.org">Brett Porter</a> |
| */ |
| public class PojoStackTraceWriter |
| implements StackTraceWriter |
| { |
| private final Throwable t; |
| |
| private final String testClass; |
| |
| private final String testMethod; |
| |
| public PojoStackTraceWriter( String testClass, String testMethod, Throwable t ) |
| { |
| this.testClass = testClass; |
| this.testMethod = testMethod; |
| this.t = t; |
| } |
| |
| @Override |
| public String writeTraceToString() |
| { |
| if ( t != null ) |
| { |
| StringWriter w = new StringWriter(); |
| try ( PrintWriter stackTrace = new PrintWriter( w ) ) |
| { |
| t.printStackTrace( stackTrace ); |
| } |
| StringBuffer builder = w.getBuffer(); |
| if ( isMultiLineExceptionMessage( t ) ) |
| { |
| // SUREFIRE-986 |
| String exc = t.getClass().getName() + ": "; |
| if ( StringUtils.startsWith( builder, exc ) ) |
| { |
| builder.insert( exc.length(), '\n' ); |
| } |
| } |
| return builder.toString(); |
| } |
| return ""; |
| } |
| |
| @Override |
| public String smartTrimmedStackTrace() |
| { |
| return t == null ? "" : new SmartStackTraceParser( testClass, t, testMethod ).getString(); |
| } |
| |
| @Override |
| public String writeTrimmedTraceToString() |
| { |
| return t == null ? "" : SmartStackTraceParser.stackTraceWithFocusOnClassAsString( t, testClass ); |
| } |
| |
| @Override |
| public SafeThrowable getThrowable() |
| { |
| return t == null ? null : new SafeThrowable( t ); |
| } |
| |
| private static boolean isMultiLineExceptionMessage( Throwable t ) |
| { |
| String msg = t.getLocalizedMessage(); |
| if ( msg != null ) |
| { |
| int countNewLines = 0; |
| for ( int i = 0, length = msg.length(); i < length; i++ ) |
| { |
| if ( msg.charAt( i ) == '\n' ) |
| { |
| if ( ++countNewLines == 2 ) |
| { |
| break; |
| } |
| } |
| } |
| return countNewLines > 1 || countNewLines == 1 && !msg.trim().endsWith( "\n" ); |
| } |
| return false; |
| } |
| |
| @Override |
| public boolean equals( Object o ) |
| { |
| if ( this == o ) |
| { |
| return true; |
| } |
| if ( o == null || getClass() != o.getClass() ) |
| { |
| return false; |
| } |
| PojoStackTraceWriter that = ( PojoStackTraceWriter ) o; |
| return Objects.equals( t, that.t ) |
| && Objects.equals( testClass, that.testClass ) |
| && Objects.equals( testMethod, that.testMethod ); |
| } |
| |
| @Override |
| public int hashCode() |
| { |
| return Objects.hash( t, testClass, testMethod ); |
| } |
| } |