com.seventheye.robolectric.sqlite.util.PackageUtils Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of robolectric-sqlite Show documentation
Library to use robolectric with Sqlite instead of H2
The newest version!
package com.seventheye.robolectric.sqlite.util;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.net.URL;
import java.util.ArrayList;
import java.util.Enumeration;
import java.util.List;
import java.util.jar.JarEntry;
import java.util.jar.JarInputStream;

public class PackageUtils {

/**
 * Find the Jar of a given class, and pull all of the test classes in the
 * given packageName
 * @param referenceClass
 * @param packageName
 * @return
 */
	public static Class[] findTestClasses(Class referenceClass,
			String packageName) {
		
		Class[] classList;
		
		try {
			//Pull classes from target folder of sourcecode (used by eclipse)
			classList = getClasses(packageName);
		} catch (ClassNotFoundException e) {
			throw new RuntimeException(e);
		} catch (IOException e) {
			throw new RuntimeException(e);
		}

		if (classList.length == 0) {
			//pull classes from JAR file (used by maven)
			classList = getClassNamesInPackage(referenceClass, packageName);
		}
		System.out.println("Robolectric test classes found: " + classList.length);
		return classList;
	}

	/**
	 * Find the Jar of a given class, and pull all of the classes in the given
	 * packageName
	 * 
	 * @param referenceClass
	 * @param packageName
	 * @return
	 */
	private static Class[] getClassNamesInPackage(Class referenceClass,
			String packageName) {

		URL jarLocation = referenceClass.getProtectionDomain().getCodeSource()
				.getLocation();

		List> classes = new ArrayList>();
		packageName = packageName.replaceAll("\\.", "/");

		try {
			JarInputStream jarFile = new JarInputStream(new FileInputStream(
					jarLocation.getPath()));
			JarEntry jarEntry;

			while (true) {
				jarEntry = jarFile.getNextJarEntry();
				if (jarEntry == null) {
					break;
				}
				if ((jarEntry.getName().startsWith(packageName)) && (jarEntry.getName().endsWith("Test.class"))) {
					String className = jarEntry.getName()
							.replaceAll("/", "\\.").replace(".class", "");

					classes.add(PackageUtils.class.getClassLoader().loadClass(className));
				}
			}
		} catch (Exception e) {
			e.printStackTrace();
		}
		return classes.toArray(new Class[classes.size()]);
	}

	/**
	 * Scans all classes accessible from the context class loader which belong
	 * to the given package and subpackages.
	 * 
	 * @param packageName
	 *            The base package
	 * @return The classes
	 * @throws ClassNotFoundException
	 * @throws IOException
	 */
	private static Class[] getClasses(String packageName)
			throws ClassNotFoundException, IOException {
		ClassLoader classLoader = Thread.currentThread()
				.getContextClassLoader();
		assert classLoader != null;
		String path = packageName.replace('.', '/');
		Enumeration resources = classLoader.getResources(path);
		List dirs = new ArrayList();
		while (resources.hasMoreElements()) {
			URL resource = resources.nextElement();
			dirs.add(new File(resource.getFile()));
		}
		List> classes = new ArrayList>();
		for (File directory : dirs) {
			classes.addAll(findClasses(directory, packageName));
		}
		return classes.toArray(new Class[classes.size()]);
	}

	/**
	 * Recursive method used to find all classes in a given directory and
	 * subdirs.
	 * 
	 * @param directory
	 *            The base directory
	 * @param packageName
	 *            The package name for classes found inside the base directory
	 * @return The classes
	 * @throws ClassNotFoundException
	 */
	private static List> findClasses(File directory, String packageName)
			throws ClassNotFoundException {
		List> classes = new ArrayList>();
		if (!directory.exists()) {
			return classes;
		}
		File[] files = directory.listFiles();
		for (File file : files) {
			if (file.isDirectory()) {
				assert !file.getName().contains(".");
				classes.addAll(findClasses(file,
						packageName + "." + file.getName()));
			} else if (file.getName().endsWith("Test.class")) {
				String classname =packageName
				+ '.'
				+ file.getName().substring(0,
						file.getName().length() - 6);

				classes.add(PackageUtils.class.getClassLoader().loadClass(classname));
			}
		}
		return classes;
	}

}