| /* |
| * 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 |
| * |
| * https://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.apache.ant.antunit.junit4; |
| |
| import java.lang.reflect.InvocationTargetException; |
| import java.lang.reflect.Method; |
| import java.lang.reflect.Modifier; |
| import java.util.Collection; |
| import java.util.Collections; |
| import java.util.Comparator; |
| import java.util.Enumeration; |
| import java.util.HashMap; |
| import java.util.Iterator; |
| import java.util.LinkedList; |
| import java.util.List; |
| import java.util.Map; |
| |
| import junit.framework.Test; |
| import junit.framework.TestCase; |
| |
| import org.apache.ant.antunit.AntUnitExecutionNotifier; |
| import org.apache.ant.antunit.AssertionFailedException; |
| import org.apache.ant.antunit.junit3.AntUnitSuite; |
| import org.apache.ant.antunit.junit3.AntUnitTestCase; |
| import org.junit.internal.runners.InitializationError; |
| import org.junit.runner.Description; |
| import org.junit.runner.Runner; |
| import org.junit.runner.manipulation.Filter; |
| import org.junit.runner.manipulation.Filterable; |
| import org.junit.runner.manipulation.NoTestsRemainException; |
| import org.junit.runner.manipulation.Sortable; |
| import org.junit.runner.manipulation.Sorter; |
| import org.junit.runner.notification.Failure; |
| import org.junit.runner.notification.RunNotifier; |
| |
| /** |
| * JUnit4 Runner to put in a RunWith annotation of the AntUnitSuite when using a |
| * JUnit4 runner. Using this runner is not mandatory because junit4 is able to |
| * run junit3 test. However, the test may be faster with this Runner (with the |
| * default junit4 adapter, the suiteSetUp and suiteTearDown will be executed |
| * around every test target). Also, more features are available when this runner |
| * is used (filtering & sorting) |
| */ |
| @SuppressWarnings("deprecation") |
| public class AntUnitSuiteRunner extends Runner implements Filterable, Sortable { |
| |
| private final AntUnitSuite junit3Suite; |
| private final Map<String, Description> targetDescriptions = new HashMap<String, Description>(); |
| private final List<String> targetsOrder = new LinkedList<String>(); |
| |
| private AntUnitSuiteRunner(AntUnitSuite suite, Class<?> junitTestClass) throws InitializationError { |
| junit3Suite = suite; |
| if (suite.hasAntInitError()) { |
| throw new InitializationError(suite.getAntInitialisationException()); |
| } |
| for (Enumeration<Test> tests = suite.tests(); tests.hasMoreElements();) { |
| TestCase nextTc = (TestCase) tests.nextElement(); |
| //TODO Handle the possibility for the user to define suite of AntUnit scripts |
| AntUnitTestCase tc = (AntUnitTestCase) nextTc; |
| Description tc_desc = Description.createTestDescription(junitTestClass, tc.getName()); |
| targetDescriptions.put(tc.getTarget(), tc_desc); |
| targetsOrder.add(tc.getTarget()); |
| } |
| } |
| |
| public AntUnitSuiteRunner(Class<?> testCaseClass) throws InitializationError { |
| this(getJUnit3AntSuite(testCaseClass), testCaseClass); |
| } |
| |
| private static AntUnitSuite getJUnit3AntSuite(Class<?> testCaseClass) |
| throws InitializationError { |
| try { |
| Method suiteMethod = testCaseClass.getMethod("suite"); |
| if (!Modifier.isStatic(suiteMethod.getModifiers())) { |
| throw new InitializationError("suite method must be static"); |
| } |
| Object suite = suiteMethod.invoke(null); |
| if (suite == null) { |
| throw new InitializationError("suite method can not return null"); |
| } |
| if (!(suite instanceof AntUnitSuite)) { |
| throw new InitializationError("suite method must return an AntUnitSuite"); |
| } |
| return (AntUnitSuite) suite; |
| } catch (NoSuchMethodException e) { |
| throw new InitializationError(e); |
| } catch (IllegalAccessException e) { |
| throw new InitializationError(e); |
| } catch (InvocationTargetException e) { |
| throw new InitializationError(e); |
| } |
| } |
| |
| /** |
| * Filterable implementation |
| */ |
| public void filter(Filter filter) throws NoTestsRemainException { |
| for (Iterator<Map.Entry<String, Description>> iter = |
| targetDescriptions.entrySet().iterator(); iter.hasNext();) { |
| Map.Entry<String, Description> mapEntry = iter.next(); |
| if (!filter.shouldRun(mapEntry.getValue())) { |
| iter.remove(); |
| targetsOrder.remove(mapEntry.getKey()); |
| } |
| } |
| } |
| |
| /** |
| * Sortable implementation |
| */ |
| public void sort(final Sorter sorter) { |
| Collections.sort(targetsOrder, new Comparator<String>() { |
| public int compare(String target1, String target2) { |
| Description d2 = targetDescriptions.get(target2); |
| Description d1 = targetDescriptions.get(target1); |
| return sorter.compare(d1, d2); |
| } |
| }); |
| } |
| |
| /** |
| * Runner implementation |
| */ |
| public Description getDescription() { |
| Description r = Description.createSuiteDescription(junit3Suite.getName()); |
| |
| Collection<Description> childDesc = targetDescriptions.values(); |
| for (Description desc : childDesc) { |
| r.addChild(desc); |
| } |
| return r; |
| } |
| |
| /** |
| * Runner implementation |
| */ |
| public void run(final RunNotifier junitNotifier) { |
| List<String> targetList = new LinkedList<String>(targetDescriptions.keySet()); |
| |
| AntUnitExecutionNotifier antUnitNotifier = new AntUnitExecutionNotifier() { |
| public void fireStartTest(String targetName) { |
| junitNotifier.fireTestStarted(getDescription(targetName)); |
| } |
| public void fireEndTest(String targetName) { |
| junitNotifier.fireTestFinished(getDescription(targetName)); |
| } |
| public void fireError(String targetName, Throwable t) { |
| Failure failure = new Failure(getDescription(targetName), t); |
| junitNotifier.fireTestFailure(failure); |
| } |
| public void fireFail(String targetName, AssertionFailedException ae) { |
| Failure failure = new Failure(getDescription(targetName), ae); |
| junitNotifier.fireTestFailure(failure); |
| } |
| private Description getDescription(String targetName) { |
| return targetDescriptions.get(targetName); |
| } |
| }; |
| |
| junit3Suite.runInContainer(targetList, antUnitNotifier); |
| } |
| |
| } |