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 com.android.phone; 18 19 import android.content.Context; 20 21 import com.android.services.telephony.Log; 22 23 /** 24 * Helper for Mockito-based test cases. 25 */ 26 public final class MockitoHelper { 27 28 private static final String TAG = "MockitoHelper"; 29 private static final String DEXCACHE = "dexmaker.dexcache"; 30 31 private ClassLoader mOriginalClassLoader; 32 private Thread mContextThread; 33 34 /** 35 * Creates a new helper, which in turn will set the context classloader so it can load Mockito 36 * resources. 37 * 38 * @param packageClass test case class 39 */ setUp(Context context, Class<?> packageClass)40 public void setUp(Context context, Class<?> packageClass) throws Exception { 41 // makes a copy of the context classloader 42 mContextThread = Thread.currentThread(); 43 mOriginalClassLoader = mContextThread.getContextClassLoader(); 44 ClassLoader newClassLoader = packageClass.getClassLoader(); 45 Log.v(TAG, "Changing context classloader from " + mOriginalClassLoader 46 + " to " + newClassLoader); 47 mContextThread.setContextClassLoader(newClassLoader); 48 String dexCache = context.getCacheDir().toString(); 49 Log.v(this, "Setting property %s to %s", DEXCACHE, dexCache); 50 System.setProperty(DEXCACHE, dexCache); 51 } 52 53 /** 54 * Restores the context classloader to the previous value. 55 */ tearDown()56 public void tearDown() throws Exception { 57 Log.v(TAG, "Restoring context classloader to " + mOriginalClassLoader); 58 mContextThread.setContextClassLoader(mOriginalClassLoader); 59 System.clearProperty(DEXCACHE); 60 } 61 }