1/* 2 * Copyright (C) 2015 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 android.support.v7.util; 18 19import org.junit.Assert; 20import org.junit.Before; 21import org.junit.Rule; 22 23import android.app.Instrumentation; 24import android.support.annotation.UiThread; 25import android.support.test.InstrumentationRegistry; 26import android.support.test.rule.ActivityTestRule; 27import android.support.v7.widget.TestActivity; 28import android.test.ActivityInstrumentationTestCase2; 29 30abstract public class BaseThreadedTest { 31 @Rule 32 public ActivityTestRule<TestActivity> mActivityRule = new ActivityTestRule<>( 33 TestActivity.class); 34 35 public final void setUp() throws Exception{ 36 try { 37 getInstrumentation().runOnMainSync(new Runnable() { 38 @Override 39 public void run() { 40 setUpUi(); 41 } 42 }); 43 } catch (Throwable throwable) { 44 Assert.fail(throwable.getMessage()); 45 } 46 } 47 48 public Instrumentation getInstrumentation() { 49 return InstrumentationRegistry.getInstrumentation(); 50 } 51 52 public void runTestOnUiThread(final Runnable test) { 53 final Throwable[] error = new Throwable[1]; 54 getInstrumentation().runOnMainSync(new Runnable() { 55 @Override 56 public void run() { 57 try { 58 test.run(); 59 } catch (Throwable t) { 60 error[0] = t; 61 } 62 } 63 }); 64 Assert.assertNull(error[0]); 65 } 66 67 @UiThread 68 protected abstract void setUpUi(); 69} 70