package org.jooby.test;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;
import org.junit.runner.Runner;
import org.junit.runners.Suite;
import org.junit.runners.model.InitializationError;
import com.google.common.base.Predicate;
import com.google.common.base.Predicates;
/**
* JUnit suite for Jooby. Internal use only.
*
* @author edgar
*/
public class JoobySuite extends Suite {
private List<Runner> runners;
static {
System.setProperty("io.netty.leakDetectionLevel", "advanced");
}
public JoobySuite(final Class<?> klass) throws InitializationError {
super(klass, Collections.emptyList());
runners = runners(klass);
}
@SuppressWarnings("rawtypes")
private List<Runner> runners(final Class<?> klass) throws InitializationError {
List<Runner> runners = new ArrayList<>();
Predicate<Class> filter = Predicates.alwaysTrue();
OnServer onserver = klass.getAnnotation(OnServer.class);
if (onserver != null) {
List<Class<?>> server = Arrays.asList(onserver.value());
filter = server::contains;
}
String[] servers = {"org.jooby.undertow.Undertow", "org.jooby.jetty.Jetty",
"org.jooby.netty.Netty" };
for (String server : servers) {
try {
Class<?> serverClass = getClass().getClassLoader().loadClass(server);
if (filter.apply(serverClass)) {
runners.add(new JoobyRunner(getTestClass().getJavaClass(), serverClass));
}
} catch (ClassNotFoundException ex) {
// do nothing
}
}
return runners;
}
@Override
protected List<Runner> getChildren() {
return runners;
}
}