package org.apache.maven.surefire.junit4;

/*
 * 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.suite.AbstractDirectoryTestSuite;
import org.apache.maven.surefire.testset.SurefireTestSet;
import org.apache.maven.surefire.testset.TestSetFailedException;

import java.io.File;
import java.lang.annotation.Annotation;
import java.lang.reflect.Method;
import java.util.ArrayList;

/**
 * Test suite for JUnit4 based on a directory of Java test classes. This is
 * capable of running both JUnit3 and JUnit4 test classes (I think).
 *
 * @author Karl M. Davis
 */
public class JUnit4DirectoryTestSuite
    extends AbstractDirectoryTestSuite
{
    /**
     * Constructor.
     */
    public JUnit4DirectoryTestSuite( File basedir, ArrayList includes, ArrayList excludes )
    {
        super( basedir, includes, excludes );
    }

    /**
     * This method will be called for each class to be run as a test. It returns
     * a surefire test set that will later be executed.
     *
     * @see org.apache.maven.surefire.suite.AbstractDirectoryTestSuite#createTestSet(java.lang.Class,
     *java.lang.ClassLoader)
     */
    protected SurefireTestSet createTestSet( Class testClass, ClassLoader classLoader )
        throws TestSetFailedException
    {
		// this is JUnit4: check to see if there are any methods with a @Test annotations
		boolean lTestFound = false;
		for (Method lMethod : testClass.getDeclaredMethods())
		{
			for (Annotation lAnnotation : lMethod.getAnnotations())
			{
				if (org.junit.Test.class.isAssignableFrom( lAnnotation.annotationType() ))
				{
					lTestFound = true;
					break;
				}
			}
		}
		// if not test found, simply return null
		if (lTestFound == false) return null;
		
		// create the testset
        return new JUnit4TestSet( testClass );
    }

}