package org.apache.maven.surefire.testng;

 /*

  * 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.ResourceBundle;

 import org.apache.maven.surefire.report.CategorizedReportEntry;

 import org.apache.maven.surefire.report.PojoStackTraceWriter;

 import org.apache.maven.surefire.report.ReportEntry;

 import org.apache.maven.surefire.report.RunListener;

 import org.apache.maven.surefire.report.SimpleReportEntry;

 import org.testng.ISuite;

 import org.testng.ISuiteListener;

 import org.testng.ITestContext;

 import org.testng.ITestListener;

 import org.testng.ITestResult;

 import org.testng.TestNG;

 /**

  * Listens for and provides and adaptor layer so that

  * TestNG tests can report their status to the current

  * {@link org.apache.maven.surefire.report.RunListener}.

  *

  * @author jkuhnert

  * @noinspection ThrowableResultOfMethodCallIgnored

  */

 public class TestNGReporter

     implements ITestListener, ISuiteListener

 {

     public static final String SUREFIRE_BUNDLE_NAME = "org.apache.maven.surefire.surefire";

     private final ResourceBundle bundle = ResourceBundle.getBundle( SUREFIRE_BUNDLE_NAME );

     /**

      * core Surefire reporting

      */

     private final RunListener reporter;

     /**

      * Constructs a new instance that will listen to

      * test updates from a {@link TestNG} class instance.

      * <p/>

      * <p/>It is assumed that the requisite {@link TestNG#addListener(ITestListener)}

      * method call has already associated with this instance <i>before</i> the test

      * suite is run.

      *

      * @param reportManager Instance to report suite status to

      */

     public TestNGReporter( RunListener reportManager )

     {

         this.reporter = reportManager;

         if ( reportManager == null )

         {

             throw new IllegalArgumentException( "ReportManager passed in was null." );

         }

     }

     public void onTestStart( ITestResult result )

     {

         String rawString = bundle.getString( "testStarting" );

         String group = groupString( result.getMethod().getGroups(), result.getTestClass().getName() );

         ReportEntry report =

             new CategorizedReportEntry( getSource( result ), getUserFriendlyTestName( result ), group );

         reporter.testStarting( report );

     }

     private String getSource( ITestResult result )

     {

         return result.getTestClass().getName();

     }

     public void onTestSuccess( ITestResult result )

     {

         ReportEntry report = new SimpleReportEntry( getSource( result ), getUserFriendlyTestName( result ) );

         reporter.testSucceeded( report );

     }

     public void onTestFailure( ITestResult result )

     {

         ReportEntry report = SimpleReportEntry.withException( getSource( result ), getUserFriendlyTestName( result ),

                                                               new PojoStackTraceWriter(

                                                                   result.getTestClass().getRealClass().getName(),

                                                                   result.getMethod().getMethodName(),

                                                                   result.getThrowable() ) );

         reporter.testFailed( report );

     }

     private static String getUserFriendlyTestName( ITestResult result )

     {

         // This is consistent with the JUnit output

         return result.getName() + "(" + result.getTestClass().getName() + ")";

     }

     public void onTestSkipped( ITestResult result )

     {

         ReportEntry report = new SimpleReportEntry( getSource( result ), getUserFriendlyTestName( result ) );

         reporter.testSkipped( report );

     }

     public void onTestFailedButWithinSuccessPercentage( ITestResult result )

     {

         ReportEntry report = SimpleReportEntry.withException( getSource( result ), getUserFriendlyTestName( result ),

                                                               new PojoStackTraceWriter(

                                                                   result.getTestClass().getRealClass().getName(),

                                                                   result.getMethod().getMethodName(),

                                                                   result.getThrowable() ) );

         reporter.testError( report );

     }

     public void onStart( ITestContext context )

     {

     }

     public void onFinish( ITestContext context )

     {

     }

     public void onStart( ISuite suite )

     {

     }

     public void onFinish( ISuite suite )

     {

     }

     /**

      * Creates a string out of the list of testng groups in the

      * form of <pre>"group1,group2,group3"</pre>.

      *

      * @param groups       The groups being run

      * @param defaultValue The default to use if no groups

      * @return a string describing the groups

      */

     private static String groupString( String[] groups, String defaultValue )

     {

         String retVal;

         if ( groups != null && groups.length > 0 )

         {

             StringBuilder str = new StringBuilder();

             for ( int i = 0; i < groups.length; i++ )

             {

                 str.append( groups[i] );

                 if ( i + 1 < groups.length )

                 {

                     str.append( "," );

                 }

             }

             retVal = str.toString();

         }

         else

         {

             retVal = defaultValue;

         }

         return retVal;

     }

     public void onConfigurationFailure( ITestResult result )

     {

         onTestFailure( result );

     }

     public void onConfigurationSkip( ITestResult result )

     {

         onTestSkipped( result );

     }

     public void onConfigurationSuccess( ITestResult result )

     {

         // DGF Don't record configuration successes as separate tests

         //onTestSuccess( result );

     }

 }