/*
 * Copyright 2010 the original author or authors.
 *
 * Licensed 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.
 */

package org.gradle.api.internal.tasks.testing.junit;

import junit.framework.JUnit4TestAdapter;
import junit.framework.Test;
import junit.framework.TestListener;
import junit.framework.TestResult;
import org.gradle.api.internal.tasks.testing.*;
import org.gradle.util.IdGenerator;
import org.gradle.util.TimeProvider;
import org.junit.runner.Describable;
import org.junit.runner.Description;

import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;

public class JUnitTestClassExecuter {
    private final ClassLoader applicationClassLoader;
    private final TestListener listener;
    private final TestResultProcessor resultProcessor;
    private final IdGenerator<?> idGenerator;
    private final TimeProvider timeProvider;

    public JUnitTestClassExecuter(ClassLoader applicationClassLoader, TestListener listener, TestResultProcessor resultProcessor, IdGenerator<?> idGenerator, TimeProvider timeProvider) {
        this.applicationClassLoader = applicationClassLoader;
        this.listener = listener;
        this.resultProcessor = resultProcessor;
        this.idGenerator = idGenerator;
        this.timeProvider = timeProvider;
    }

    public void execute(String testClassName) {
        TestDescriptorInternal testInternal = new DefaultTestClassDescriptor(idGenerator.generateId(), testClassName);
        resultProcessor.started(testInternal, new TestStartEvent(timeProvider.getCurrentTime()));

        Test adapter = createTest(testClassName);
        TestResult result = new TestResult();
        result.addListener(listener);
        adapter.run(result);

        resultProcessor.completed(testInternal.getId(), new TestCompleteEvent(timeProvider.getCurrentTime()));
    }

    private Test createTest(String testClassName) {
        Class<?> testClass;
        try {
            testClass = Class.forName(testClassName, true, applicationClassLoader);
        } catch (Throwable e) {
            return new BrokenTest(
                    Description.createSuiteDescription(String.format("initializationError(%s)", testClassName)), e);
        }

        // Look for a static suite method
        try {
            Method suiteMethod = testClass.getMethod("suite", new Class[0]);
            return (Test) suiteMethod.invoke(null);
        } catch (NoSuchMethodException e) {
            // Ignore
        } catch (InvocationTargetException e) {
            return new BrokenTest(Description.createTestDescription(testClass, "suite"), e.getCause());
        } catch (IllegalAccessException e) {
            return new BrokenTest(Description.createTestDescription(testClass, "suite"), e);
        }

        return new JUnit4TestAdapter(testClass);
    }

    private static class BrokenTest implements Test, Describable {
        private final Throwable failure;
        private final Description description;

        public BrokenTest(Description description, Throwable failure) {
            this.failure = failure;
            this.description = description;
        }

        public Description getDescription() {
            return description;
        }

        public int countTestCases() {
            return 1;
        }

        public void run(TestResult result) {
            result.startTest(this);
            result.addError(this, failure);
            result.endTest(this);
        }
    }
}

@@ -2519,203 +2519,13 @@
 r);%0A
+%0A
     
-    %7D catch (Throwable e) %7B%0A            return new BrokenTest(%0A                    Description.createSuiteDescription(String.format(%22initializationError(%25s)%22, testClassName)), e);%0A        %7D%0A%0A
     
@@ -2562,38 +2562,46 @@
  method%0A        
+ 
+   
 try %7B%0A
+    
             Meth
@@ -2661,16 +2661,20 @@
 ss%5B0%5D);%0A
+    
         
@@ -2717,32 +2717,36 @@
 (null);%0A        
+    
 %7D catch (NoSuchM
@@ -2777,16 +2777,20 @@
         
+    
 // Ignor
@@ -2791,32 +2791,36 @@
  Ignore%0A        
+  
+  
 %7D catch (Invocat
@@ -2835,32 +2835,36 @@
 tException e) %7B%0A
+    
             retu
@@ -2963,39 +2963,40 @@
     
-%7D catch (IllegalAccessException
+    %7D%0A        %7D catch (Throwable
  e) 
@@ -3023,32 +3023,53 @@
  new BrokenTest(
+%0A                    
 Description.crea
@@ -3062,36 +3062,37 @@
 scription.create
-Test
+Suite
 Description(test
@@ -3087,34 +3087,71 @@
 ription(
-testClass, %22suite%22
+String.format(%22initializationError(%25s)%22, testClassName)
 ), e);%0A 

Handle case where attempting to locate a test class's suite() method results in a error, such as NoClassDefFoundError.