org.testatoo.config.junit.TestatooJunitRunner Maven / Gradle / Ivy
/**
* Copyright (C) 2008 Ovea
*
* Licensed 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
*
* http://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.testatoo.config.junit;
import org.aopalliance.intercept.MethodInvocation;
import org.junit.Rule;
import org.junit.internal.runners.model.ReflectiveCallable;
import org.junit.internal.runners.statements.Fail;
import org.junit.rules.MethodRule;
import org.junit.runner.notification.RunNotifier;
import org.junit.runners.BlockJUnit4ClassRunner;
import org.junit.runners.model.FrameworkMethod;
import org.junit.runners.model.InitializationError;
import org.junit.runners.model.Statement;
import org.testatoo.config.annotation.TestatooModules;
import org.testatoo.config.testatoo.Testatoo;
import java.lang.reflect.AccessibleObject;
import java.lang.reflect.Method;
public final class TestatooJunitRunner extends BlockJUnit4ClassRunner {
private Testatoo testatoo;
public TestatooJunitRunner(Class klass) throws InitializationError {
super(klass);
}
@Override
protected void runChild(final FrameworkMethod method, final RunNotifier notifier) {
testatoo.scheduleTest(getTestClass().getJavaClass(), method.getMethod(), new Runnable() {
@Override
public void run() {
TestatooJunitRunner.super.runChild(method, notifier);
}
});
}
@Override
protected Statement classBlock(RunNotifier notifier) {
final Statement statement = super.classBlock(notifier);
return new Statement() {
@Override
public void evaluate() throws Throwable {
testatoo = Testatoo.configure(getTestClass().getJavaClass().getAnnotation(TestatooModules.class));
testatoo.start();
try {
statement.evaluate();
} finally {
testatoo.stop();
}
}
};
}
@Override
protected Statement methodBlock(final FrameworkMethod method) {
final Object test;
try {
test = new ReflectiveCallable() {
@Override
protected Object runReflectiveCall() throws Throwable {
return createTest();
}
}.run();
} catch (Throwable e) {
return new Fail(e);
}
Statement statement = methodInvoker(method, test);
statement = possiblyExpectingExceptions(method, test, statement);
statement = withPotentialTimeout(method, test, statement);
statement = withBefores(method, test, statement);
statement = withAfters(method, test, statement);
statement = withRules(method, test, statement);
final Statement st = statement;
return new Statement() {
@Override
public void evaluate() throws Throwable {
try {
testatoo.on(test, method.getMethod());
st.evaluate();
} catch (Throwable throwable) {
throw new RuntimeException(throwable.getMessage(), throwable);
}
}
};
}
private Statement withRules(FrameworkMethod method, Object target,
Statement statement) {
Statement result = statement;
for (MethodRule each : getTestClass().getAnnotatedFieldValues(target,
Rule.class, MethodRule.class))
result = each.apply(result, method, target);
return result;
}
@Override
protected final Statement methodInvoker(final FrameworkMethod method, final Object test) {
return new Statement() {
@SuppressWarnings({"ThrowableResultOfMethodCallIgnored"})
@Override
public void evaluate() throws Throwable {
testatoo.executeTestMethod(new MethodInvocation() {
@Override
public Method getMethod() {
return method.getMethod();
}
@Override
public Object[] getArguments() {
return new Object[0];
}
@Override
public Object proceed() throws Throwable {
TestatooJunitRunner.super.methodInvoker(method, test).evaluate();
return null;
}
@Override
public Object getThis() {
return test;
}
@Override
public AccessibleObject getStaticPart() {
return method.getMethod();
}
});
}
};
}
}