1 /* 2 * Copyright (C) 2017 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file 5 * except in compliance with the License. You may obtain a copy of the License at 6 * 7 * http://www.apache.org/licenses/LICENSE-2.0 8 * 9 * Unless required by applicable law or agreed to in writing, software distributed under the 10 * License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY 11 * KIND, either express or implied. See the License for the specific language governing 12 * permissions and limitations under the License. 13 */ 14 15 package com.android.systemui; 16 17 import static org.mockito.Mockito.mock; 18 19 import android.util.ArrayMap; 20 import android.util.ArraySet; 21 import android.util.Log; 22 23 public class TestableDependency extends Dependency { 24 private static final String TAG = "TestableDependency"; 25 26 private final ArrayMap<Object, Object> mObjs = new ArrayMap<>(); 27 private final ArraySet<Object> mInstantiatedObjects = new ArraySet<>(); 28 private final Dependency mParent; 29 TestableDependency(Dependency parent)30 public TestableDependency(Dependency parent) { 31 mParent = parent; 32 } 33 injectMockDependency(Class<T> cls)34 public <T> T injectMockDependency(Class<T> cls) { 35 final T mock = mock(cls); 36 injectTestDependency(cls, mock); 37 return mock; 38 } 39 injectTestDependency(DependencyKey<T> key, T obj)40 public <T> void injectTestDependency(DependencyKey<T> key, T obj) { 41 mObjs.put(key, obj); 42 } 43 injectTestDependency(Class<T> key, T obj)44 public <T> void injectTestDependency(Class<T> key, T obj) { 45 if (mInstantiatedObjects.contains(key)) { 46 Log.d(TAG, key + " was already initialized but overriding with testDependency."); 47 } 48 mObjs.put(key, obj); 49 } 50 51 @Override createDependency(Object key)52 public <T> T createDependency(Object key) { 53 if (mObjs.containsKey(key)) return (T) mObjs.get(key); 54 55 mInstantiatedObjects.add(key); 56 return mParent.createDependency(key); 57 } 58 59 @Override autoRegisterModulesForDump()60 protected boolean autoRegisterModulesForDump() { 61 return false; 62 } 63 hasInstantiatedDependency(Class<T> key)64 public <T> boolean hasInstantiatedDependency(Class<T> key) { 65 return mObjs.containsKey(key) || mInstantiatedObjects.contains(key); 66 } 67 } 68