org.scijava.test.TestUtils Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of scijava-common Show documentation
Show all versions of scijava-common Show documentation
SciJava Common is a shared library for SciJava software. It provides a plugin framework, with an extensible mechanism for service discovery, backed by its own annotation processor, so that plugins can be loaded dynamically. It is used by downstream projects in the SciJava ecosystem, such as ImageJ and SCIFIO.
/*
* #%L
* SciJava Common shared library for SciJava software.
* %%
* Copyright (C) 2009 - 2014 Board of Regents of the University of
* Wisconsin-Madison, Broad Institute of MIT and Harvard, and Max Planck
* Institute of Molecular Cell Biology and Genetics.
* %%
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are met:
*
* 1. Redistributions of source code must retain the above copyright notice,
* this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright notice,
* this list of conditions and the following disclaimer in the documentation
* and/or other materials provided with the distribution.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDERS OR CONTRIBUTORS BE
* LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
* POSSIBILITY OF SUCH DAMAGE.
* #L%
*/
package org.scijava.test;
import java.io.File;
import java.io.IOException;
import java.net.URL;
import java.util.AbstractMap;
import java.util.Map;
import org.scijava.util.ClassUtils;
import org.scijava.util.FileUtils;
/**
* A bunch of helpful functions for unit tests.
*
* @author Johannes Schindelin
*/
public class TestUtils {
/**
* Makes a temporary directory for use with unit tests.
*
* When the unit test runs in a Maven context, the temporary directory will be
* created in the target/ directory corresponding to the calling class
* instead of /tmp/.
*
*
* @param prefix the prefix for the directory's name
* @return the reference to the newly-created temporary directory
* @throws IOException
*/
public static File createTemporaryDirectory(final String prefix) throws IOException {
final Map.Entry, String> calling = getCallingCodeLocation(null);
return createTemporaryDirectory(prefix, calling.getKey(), calling.getValue());
}
/**
* Makes a temporary directory for use with unit tests.
*
* When the unit test runs in a Maven context, the temporary directory will be
* created in the corresponding target/ directory instead of
* /tmp/.
*
*
* @param prefix the prefix for the directory's name
* @param forClass the class for context (to determine whether there's a
* target/ directory)
* @return the reference to the newly-created temporary directory
* @throws IOException
*/
public static File createTemporaryDirectory(final String prefix,
final Class forClass) throws IOException
{
return createTemporaryDirectory(prefix, forClass, "" + temporaryDirectoryCounter++);
}
private static int temporaryDirectoryCounter = 1;
/**
* Makes a temporary directory for use with unit tests.
*
* When the unit test runs in a Maven context, the temporary directory will be
* created in the corresponding target/ directory instead of
* /tmp/.
*
*
* @param prefix the prefix for the directory's name
* @param forClass the class for context (to determine whether there's a
* target/ directory)
* @param suffix the suffix for the directory's name
* @return the reference to the newly-created temporary directory
* @throws IOException
*/
public static File createTemporaryDirectory(final String prefix,
final Class forClass, final String suffix) throws IOException
{
final URL directory = ClassUtils.getLocation(forClass);
if (directory == null) {
throw new IllegalArgumentException("No location for class " + forClass);
}
if (!"file".equals(directory.getProtocol())) {
throw new IllegalArgumentException("Invalid directory: " + directory);
}
final String path = directory.getPath();
if (path == null) throw new IllegalArgumentException("Directory has null path");
final File baseDirectory;
if (path.endsWith("/target/test-classes/")) {
baseDirectory = new File(path).getParentFile();
} else {
baseDirectory = new File(path);
}
File file = new File(baseDirectory, prefix + suffix);
if (file.isDirectory()) {
if (!FileUtils.deleteRecursively(file)) {
// Oh, how I *love* Windows. Love, love, love.
for (int i = -1; file.isDirectory(); i--) {
file = new File(baseDirectory, prefix + i + suffix);
}
}
}
else if (file.exists() && !file.delete()) {
throw new IOException("Could not remove " + file);
}
if (!file.mkdir()) throw new IOException("Could not make directory " + file);
return file;
}
/**
* Returns the class of the caller (excluding the specified class).
*
* Sometimes it is convenient to determine the caller's context, e.g. to
* determine whether running in a maven-surefire-plugin context (in which case
* the location of the caller's class would end in
* target/test-classes/).
*
*
* @param excluding the class to exclude (or null)
* @return the class of the caller
*/
public static Class getCallingClass(final Class excluding) {
return getCallingCodeLocation(excluding).getKey();
}
/**
* Returns the class and the method/line number of the caller (excluding the specified class).
*
* Sometimes it is convenient to determine the caller's context, e.g. to
* determine whether running in a maven-surefire-plugin context (in which case
* the location of the caller's class would end in
* target/test-classes/).
*
*
* @param excluding the class to exclude (or null)
* @return the class of the caller and the method and line number
*/
public static Map.Entry, String> getCallingCodeLocation(final Class excluding) {
final String thisClassName = TestUtils.class.getName();
final String thisClassName2 = excluding == null ? null : excluding.getName();
final Thread currentThread = Thread.currentThread();
for (final StackTraceElement element : currentThread.getStackTrace()) {
final String thatClassName = element.getClassName();
if (thatClassName == null || thatClassName.equals(thisClassName) ||
thatClassName.equals(thisClassName2) ||
thatClassName.endsWith("TestUtils") ||
thatClassName.startsWith("java.lang.")) {
continue;
}
final ClassLoader loader = currentThread.getContextClassLoader();
final Class clazz;
try {
clazz = loader.loadClass(element.getClassName());
}
catch (ClassNotFoundException e) {
throw new UnsupportedOperationException("Could not load " +
element.getClassName() + " with the current context class loader (" +
loader + ")!");
}
final String suffix = element.getMethodName() + "-L" + element.getLineNumber();
return new AbstractMap.SimpleEntry, String>(clazz, suffix);
}
throw new UnsupportedOperationException("No calling class outside " + thisClassName + " found!");
}
}