package org.apache.maven.surefire.junitcore;

 /*

  * 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.common.junit4.JUnit4RunListener;

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

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

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

 import org.apache.maven.surefire.testset.TestSetFailedException;

 import org.junit.runner.Description;

 import org.junit.runner.Result;

 import org.junit.runner.notification.Failure;

 /**

  * A class to be used when there is no JUnit parallelism (methods or/and class). This allow to workaround JUnit

  * limitation a la Junit4 provider. Specifically, we can redirect properly the output even if we don't have class

  * demarcation in JUnit. It works when if there is a JVM instance per test run, i.e. with forkMode=always or perthread.

  */

 public class NonConcurrentRunListener

     extends JUnit4RunListener

     implements ConsoleOutputReceiver

 {

     private java.lang.Class<?> currentTestClass;

     private Description lastFinishedDescription;

     public NonConcurrentRunListener( RunListener reporter )

         throws TestSetFailedException

     {

         super( reporter );

     }

     public synchronized void writeTestOutput( byte[] buf, int off, int len, boolean stdout )

     {

         // We can write immediately: no parallelism and a single class.

         ( (ConsoleOutputReceiver) reporter ).writeTestOutput( buf, off, len, stdout );

     }

     protected SimpleReportEntry createReportEntry( Description description )

     {

         return new SimpleReportEntry( description.getClassName(), description.getDisplayName()/*,

                                        (int) ( System.currentTimeMillis() - startTime ) */);

     }

     protected SimpleReportEntry createReportEntryForTestSet( Description description )

     {

         return new SimpleReportEntry( description.getClassName(), description.getClassName() /*,

                                        (int) ( System.currentTimeMillis() - startTime ) */);

     }

     @Override

     public void testStarted( Description description )

         throws Exception

     {

         finishLastTestSetIfNeccessary( description );

         super.testStarted( description );

     }

     private void finishLastTestSetIfNeccessary( Description description )

     {

         if ( !description.getTestClass().equals( currentTestClass ) )

         {

             currentTestClass = description.getTestClass();

             if ( lastFinishedDescription != null )

             {

                 reporter.testSetCompleted( createReportEntryForTestSet( lastFinishedDescription ) );

                 lastFinishedDescription = null;

             }

             reporter.testSetStarting( createReportEntryForTestSet( description ) );

         }

     }

     @Override

     public void testFinished( Description description )

         throws Exception

     {

         super.testFinished( description );

         this.lastFinishedDescription = description;

     }

     @Override

     public void testIgnored( Description description )

         throws Exception

     {

         finishLastTestSetIfNeccessary( description );

         super.testIgnored( description );

         this.lastFinishedDescription = description;

     }

     @Override

     public void testFailure( Failure failure )

         throws Exception

     {

         super.testFailure( failure );

         this.lastFinishedDescription = failure.getDescription();

     }

     @Override

     public void testAssumptionFailure( Failure failure )

     {

         super.testAssumptionFailure( failure );

         this.lastFinishedDescription = failure.getDescription();

     }

     @Override

     public void testRunStarted( Description description )

         throws Exception

     {

     }

     @Override

     public void testRunFinished( Result result )

         throws Exception

     {

         if ( lastFinishedDescription != null )

         {

             reporter.testSetCompleted( createReportEntryForTestSet( lastFinishedDescription ) );

             lastFinishedDescription = null;

         }

     }

 }