1/* 2 * Copyright (C) 2012 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 */ 16package com.android.test.runner; 17 18import android.app.Instrumentation; 19 20import com.android.test.runner.junit3.AndroidJUnit3Builder; 21import com.android.test.runner.junit4.AndroidJUnit4Builder; 22 23import org.junit.internal.builders.AllDefaultPossibilitiesBuilder; 24import org.junit.runner.Runner; 25import org.junit.runners.model.RunnerBuilder; 26 27/** 28 * A {@link RunnerBuilder} that can handle all types of tests. 29 */ 30class AndroidRunnerBuilder extends AllDefaultPossibilitiesBuilder { 31 32 private final Instrumentation mInstr; 33 private final AndroidJUnit3Builder mAndroidJUnit3Builder; 34 private final AndroidJUnit4Builder mAndroidJUnit4Builder; 35 36 public AndroidRunnerBuilder(boolean canUseSuiteMethod, Instrumentation instr, 37 boolean skipExecution) { 38 super(canUseSuiteMethod); 39 mInstr = instr; 40 mAndroidJUnit3Builder = new AndroidJUnit3Builder(mInstr, skipExecution); 41 mAndroidJUnit4Builder = new AndroidJUnit4Builder(mInstr, skipExecution); 42 } 43 44 @Override 45 public Runner runnerForClass(Class<?> testClass) throws Throwable { 46 Runner runner = mAndroidJUnit3Builder.safeRunnerForClass(testClass); 47 if (runner != null) { 48 return runner; 49 } 50 runner = mAndroidJUnit4Builder.safeRunnerForClass(testClass); 51 if (runner != null) { 52 return runner; 53 } 54 return super.runnerForClass(testClass); 55 } 56 57} 58