1/* 2 * Copyright (C) 2013 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 17package com.android.uiautomator.testrunner; 18 19import android.test.AndroidTestRunner; 20import android.test.InstrumentationTestRunner; 21 22import com.android.uiautomator.core.Tracer; 23 24import junit.framework.AssertionFailedError; 25import junit.framework.Test; 26import junit.framework.TestListener; 27 28/** 29 * Test runner for {@link UiAutomatorTestCase}s. Such tests are executed 30 * on the device and have access to an applications context. 31 */ 32public class UiAutomatorInstrumentationTestRunner extends InstrumentationTestRunner { 33 34 @Override 35 public void onStart() { 36 // process runner arguments before test starts 37 String traceType = getArguments().getString("traceOutputMode"); 38 if(traceType != null) { 39 Tracer.Mode mode = Tracer.Mode.valueOf(Tracer.Mode.class, traceType); 40 if (mode == Tracer.Mode.FILE || mode == Tracer.Mode.ALL) { 41 String filename = getArguments().getString("traceLogFilename"); 42 if (filename == null) { 43 throw new RuntimeException("Name of log file not specified. " + 44 "Please specify it using traceLogFilename parameter"); 45 } 46 Tracer.getInstance().setOutputFilename(filename); 47 } 48 Tracer.getInstance().setOutputMode(mode); 49 } 50 super.onStart(); 51 } 52 53 @Override 54 protected AndroidTestRunner getAndroidTestRunner() { 55 AndroidTestRunner testRunner = super.getAndroidTestRunner(); 56 testRunner.addTestListener(new TestListener() { 57 @Override 58 public void startTest(Test test) { 59 if (test instanceof UiAutomatorTestCase) { 60 ((UiAutomatorTestCase)test).initialize(getArguments()); 61 } 62 } 63 64 @Override 65 public void endTest(Test test) { 66 } 67 68 @Override 69 public void addFailure(Test test, AssertionFailedError e) { 70 } 71 72 @Override 73 public void addError(Test test, Throwable t) { 74 } 75 }); 76 return testRunner; 77 } 78} 79