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.Iterator;
  import java.util.Properties;
  import org.apache.maven.surefire.providerapi.AbstractProvider;
  import org.apache.maven.surefire.providerapi.ProviderParameters;
  import org.apache.maven.surefire.report.ReporterConfiguration;
  import org.apache.maven.surefire.report.ReporterException;
  import org.apache.maven.surefire.report.ReporterFactory;
  import org.apache.maven.surefire.suite.RunResult;
  import org.apache.maven.surefire.testset.TestRequest;
  import org.apache.maven.surefire.testset.TestSetFailedException;
  import org.apache.maven.surefire.util.NestedRuntimeException;
  import org.apache.maven.surefire.util.RunOrderCalculator;
  import org.apache.maven.surefire.util.ScanResult;
  import org.apache.maven.surefire.util.TestsToRun;
  
  /**
   * @author Kristian Rosenvold
   * @noinspection UnusedDeclaration
   */
  public class TestNGProvider
      extends AbstractProvider
  {
      private final Properties providerProperties;
  
      private final ReporterConfiguration reporterConfiguration;
  
      private final ClassLoader testClassLoader;
  
      private final ScanResult scanResult;
  
      private final TestRequest testRequest;
  
      private final ProviderParameters providerParameters;
  
      private TestsToRun testsToRun;
  
      private final RunOrderCalculator runOrderCalculator;
  
      public TestNGProvider( ProviderParameters booterParameters )
      {
          this.providerParameters = booterParameters;
          this.testClassLoader = booterParameters.getTestClassLoader();
          this.runOrderCalculator = booterParameters.getRunOrderCalculator();
          this.providerProperties = booterParameters.getProviderProperties();
          this.testRequest = booterParameters.getTestRequest();
          reporterConfiguration = booterParameters.getReporterConfiguration();
          this.scanResult = booterParameters.getScanResult();
      }
  
      public Boolean isRunnable()
      {
          return Boolean.TRUE;
      }
  
      public RunResult invoke( Object forkTestSet )
          throws TestSetFailedException, ReporterException
      {
  
          final ReporterFactory reporterFactory = providerParameters.getReporterFactory();
  
          if ( isTestNGXmlTestSuite( testRequest ) )
          {
              TestNGXmlTestSuite testNGXmlTestSuite = getXmlSuite();
              testNGXmlTestSuite.locateTestSets( testClassLoader );
              if ( forkTestSet != null && testRequest == null )
              {
                  testNGXmlTestSuite.execute( (String) forkTestSet, reporterFactory );
              }
              else
              {
                  testNGXmlTestSuite.execute( reporterFactory );
              }
          }
          else
          {
              if ( testsToRun == null )
              {
                  if ( forkTestSet instanceof TestsToRun )
                 {
                     testsToRun = (TestsToRun) forkTestSet;
                 }
                 else if ( forkTestSet instanceof Class )
                 {
                     testsToRun = TestsToRun.fromClass( (Class) forkTestSet );
                 }
                 else
                 {
                     testsToRun = scanClassPath();
                 }
             }
             TestNGDirectoryTestSuite suite = getDirectorySuite();
             suite.execute( testsToRun, reporterFactory );
         }
 
         return reporterFactory.close();
     }
 
     boolean isTestNGXmlTestSuite( TestRequest testSuiteDefinition )
     {
         return testSuiteDefinition.getSuiteXmlFiles() != null && testSuiteDefinition.getSuiteXmlFiles().size() > 0 &&
             testSuiteDefinition.getRequestedTest() == null;
 
     }
 
 
     private TestNGDirectoryTestSuite getDirectorySuite()
     {
         return new TestNGDirectoryTestSuite( testRequest.getTestSourceDirectory().toString(), providerProperties,
                                              reporterConfiguration.getReportsDirectory(),
                                              testRequest.getRequestedTestMethod(), runOrderCalculator, scanResult );
     }
 
     private TestNGXmlTestSuite getXmlSuite()
     {
         return new TestNGXmlTestSuite( testRequest.getSuiteXmlFiles(), testRequest.getTestSourceDirectory().toString(),
                                        providerProperties,
                                        reporterConfiguration.getReportsDirectory() );
     }
 
 
     public Iterator getSuites()
     {
         if ( isTestNGXmlTestSuite( testRequest ) )
         {
             try
             {
                 return getXmlSuite().locateTestSets( testClassLoader ).keySet().iterator();
             }
             catch ( TestSetFailedException e )
             {
                 throw new NestedRuntimeException( e );
             }
         }
         else
         {
             testsToRun = scanClassPath();
             return testsToRun.iterator();
         }
     }
 
     private TestsToRun scanClassPath()
     {
         final TestsToRun scanned = scanResult.applyFilter( null, testClassLoader );
         return runOrderCalculator.orderTestClasses( scanned );
     }
 
 
 }