Created
February 1, 2011 20:41
-
-
Save bjpbakker/806616 to your computer and use it in GitHub Desktop.
JUnit test suite with all tests; see http://blog.iprofs.nl/2011/02/01/junit-suite-with-all-tests/
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| package nl.iprofs.blogs.junit4features.runner; | |
| import java.io.File; | |
| import java.lang.reflect.Method; | |
| import java.lang.reflect.Modifier; | |
| import java.util.LinkedList; | |
| import java.util.List; | |
| import org.junit.Test; | |
| import org.junit.runners.Suite; | |
| import org.junit.runners.model.InitializationError; | |
| /** | |
| * <p>An extension of JUnit's {@link Suite} test runner that gathers the test | |
| * classes from the classpath instead of having to specify them in | |
| * {@link SuiteClasses}.</p> | |
| * | |
| * <p> | |
| * A test suite including all tests on the classpath looks like this: | |
| * <pre> | |
| * @RunWith(ClasspathSuite.class) | |
| * public class AllTestsSuite {} | |
| * </pre> | |
| * </p> | |
| * | |
| * @author Bart Bakker <bbakker at iPROFS dot nl> | |
| */ | |
| public class ClasspathSuite extends Suite { | |
| private static final String CLASSPATH = System.getProperty("java.class.path"); | |
| private static final String CLASS_EXT = ".class"; | |
| public ClasspathSuite(Class<?> setupClass) throws InitializationError { | |
| super(setupClass, getTestClasses()); | |
| } | |
| private static Class<?>[] getTestClasses() throws InitializationError { | |
| List<Class<?>> testClasses = new LinkedList<Class<?>>(); | |
| for (String each : CLASSPATH.split(File.pathSeparator)) { | |
| File root = new File(each); | |
| // We're really only interested in classpath directories as we'd | |
| // like to run tests in this project. If other tests are included | |
| // on the classpath in libraries we'd like not to know of them. | |
| if (root.isDirectory()) { | |
| gatherTestClasses(root, root, testClasses); | |
| } | |
| } | |
| return testClasses.toArray(new Class<?>[testClasses.size()]); | |
| } | |
| private static void gatherTestClasses(File root, File cur, List<Class<?>> classes) throws InitializationError { | |
| for (File each : cur.listFiles()) { | |
| if (each.isDirectory()) { | |
| gatherTestClasses(root, each, classes); | |
| } else if (each.getName().endsWith(CLASS_EXT)) { | |
| try { | |
| Class<?> clazz = Class.forName(getClassName(each.getPath(), root.getPath())); | |
| if (isTestClass(clazz)) { | |
| classes.add(clazz); | |
| } | |
| } catch (ClassNotFoundException e) { | |
| throw new InitializationError(e); | |
| } | |
| } | |
| } | |
| } | |
| private static String getClassName(String fileName, String classPathRoot) { | |
| String className = fileName.substring(classPathRoot.length()); | |
| if (className.charAt(0) == File.separatorChar) { | |
| className = className.substring(1); | |
| } | |
| return className.substring(0, className.length() - CLASS_EXT.length()).replace(File.separatorChar, '.'); | |
| } | |
| private static boolean isTestClass(Class<?> clazz) { | |
| if (isAbstractClass(clazz)) { | |
| return false; | |
| } | |
| for (Method method : clazz.getDeclaredMethods()) { | |
| if (isTest(method)) { | |
| return true; | |
| } | |
| } | |
| return false; | |
| } | |
| private static boolean isAbstractClass(Class<?> clazz) { | |
| return (clazz.getModifiers() & Modifier.ABSTRACT) != 0; | |
| } | |
| private static boolean isTest(Method method) { | |
| return method.getAnnotation(Test.class) != null; | |
| } | |
| } |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment