com.intropro.prairie.junit.PrairieRunner Maven / Gradle / Ivy
/**
* 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 com.intropro.prairie.junit;
import com.intropro.prairie.unit.common.DependencyResolver;
import com.intropro.prairie.unit.common.exception.DestroyUnitException;
import com.intropro.prairie.unit.common.exception.InitUnitException;
import org.junit.runners.BlockJUnit4ClassRunner;
import org.junit.runners.model.FrameworkMethod;
import org.junit.runners.model.InitializationError;
import org.junit.runners.model.Statement;
/**
* Created by presidentio on 04.09.15.
*/
public class PrairieRunner extends BlockJUnit4ClassRunner {
private Class clazz;
private DependencyResolver dependencyResolver;
public PrairieRunner(Class klass) throws InitializationError {
super(klass);
this.clazz = klass;
dependencyResolver = new DependencyResolver();
}
@Override
protected Statement withBefores(FrameworkMethod method, Object target, Statement statement) {
return new BeforeStatement(super.withBefores(method, target, statement), target);
}
@Override
protected Statement withAfters(FrameworkMethod method, Object target, Statement statement) {
return new AfterStatement(super.withAfters(method, target, statement), target);
}
@Override
protected Statement withBeforeClasses(Statement statement) {
return new BeforeClassStatement(super.withBeforeClasses(statement), clazz);
}
@Override
protected Statement withAfterClasses(Statement statement) {
return new AfterClassStatement(super.withAfterClasses(statement), clazz);
}
private class BeforeStatement extends Statement {
private Statement statement;
private Object target;
public BeforeStatement(Statement statement, Object target) {
this.statement = statement;
this.target = target;
}
@Override
public void evaluate() throws Throwable {
try {
dependencyResolver.resolve(target);
} catch (Throwable e) {
throw new InitUnitException(e);
}
statement.evaluate();
}
}
private class AfterStatement extends Statement {
private Statement statement;
private Object target;
public AfterStatement(Statement statement, Object target) {
this.statement = statement;
this.target = target;
}
@Override
public void evaluate() throws Throwable {
Throwable error = null;
try {
statement.evaluate();
} catch (InitUnitException e) {
throw e;
} catch (Throwable e) {
error = e;
}
try {
dependencyResolver.destroy(target);
} catch (DestroyUnitException e) {
if (error == null) {
throw e;
} else {
e.printStackTrace();
}
}
if (error != null) {
throw error;
}
}
}
private class BeforeClassStatement extends Statement {
private Statement statement;
private Class clazz;
public BeforeClassStatement(Statement statement, Class clazz) {
this.statement = statement;
this.clazz = clazz;
}
@Override
public void evaluate() throws Throwable {
try {
dependencyResolver.resolveStatic(clazz);
} catch (Throwable e) {
throw new InitUnitException(e);
}
statement.evaluate();
}
}
private class AfterClassStatement extends Statement {
private Statement statement;
private Class clazz;
public AfterClassStatement(Statement statement, Class clazz) {
this.statement = statement;
this.clazz = clazz;
}
@Override
public void evaluate() throws Throwable {
Throwable error = null;
try {
statement.evaluate();
} catch (InitUnitException e) {
throw e;
} catch (Throwable e) {
error = e;
}
try {
dependencyResolver.destroyStatic(clazz);
} catch (DestroyUnitException e) {
e.printStackTrace();
}
if (error != null) {
throw error;
}
}
}
}