1 /* 2 * Copyright (C) 2016 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17 package vogar.target.junit; 18 19 import com.google.common.base.Function; 20 import org.junit.internal.builders.AnnotatedBuilder; 21 import org.junit.runner.RunWith; 22 import org.junit.runner.Runner; 23 import org.junit.runners.model.RunnerBuilder; 24 25 /** 26 * An {@link AnnotatedBuilder} that can map the {@link Runner} class specified in the 27 * {@link RunWith} annotation to a replacement one. 28 */ 29 public class MappingAnnotatedBuilder extends AnnotatedBuilder { 30 31 private final Function<Class<? extends Runner>, Class<? extends Runner>> mappingFunction; 32 MappingAnnotatedBuilder( RunnerBuilder suiteBuilder, Function<Class<? extends Runner>, Class<? extends Runner>> mappingFunction)33 public MappingAnnotatedBuilder( 34 RunnerBuilder suiteBuilder, 35 Function<Class<? extends Runner>, Class<? extends Runner>> mappingFunction) { 36 super(suiteBuilder); 37 this.mappingFunction = mappingFunction; 38 } 39 40 @Override runnerForClass(Class<?> testClass)41 public Runner runnerForClass(Class<?> testClass) throws Exception { 42 RunWith runWith = testClass.getAnnotation(RunWith.class); 43 if (runWith != null) { 44 Class<? extends Runner> runnerClass = runWith.value(); 45 46 runnerClass = mappingFunction.apply(runnerClass); 47 if (runnerClass != null) { 48 return buildRunner(runnerClass, testClass); 49 } 50 } 51 52 return null; 53 } 54 } 55