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.content.Context; 20 import android.util.ArrayMap; 21 import android.util.ArraySet; 22 23 public class TestableDependency extends Dependency { 24 private final ArrayMap<Object, Object> mObjs = new ArrayMap<>(); 25 private final ArraySet<Object> mInstantiatedObjects = new ArraySet<>(); 26 TestableDependency(Context context)27 public TestableDependency(Context context) { 28 if (context instanceof SysuiTestableContext) { 29 mComponents = ((SysuiTestableContext) context).getComponents(); 30 } 31 mContext = context; 32 SystemUIFactory.createFromConfig(context); 33 SystemUIFactory.getInstance().getRootComponent() 34 .createDependency() 35 .createSystemUI(this); 36 start(); 37 } 38 injectMockDependency(Class<T> cls)39 public <T> T injectMockDependency(Class<T> cls) { 40 final T mock = mock(cls); 41 injectTestDependency(cls, mock); 42 return mock; 43 } 44 injectTestDependency(DependencyKey<T> key, T obj)45 public <T> void injectTestDependency(DependencyKey<T> key, T obj) { 46 mObjs.put(key, obj); 47 } 48 injectTestDependency(Class<T> key, T obj)49 public <T> void injectTestDependency(Class<T> key, T obj) { 50 if (mInstantiatedObjects.contains(key)) { 51 throw new IllegalStateException(key + " was already initialized"); 52 } 53 mObjs.put(key, obj); 54 } 55 56 @Override createDependency(Object key)57 protected <T> T createDependency(Object key) { 58 if (mObjs.containsKey(key)) return (T) mObjs.get(key); 59 mInstantiatedObjects.add(key); 60 return super.createDependency(key); 61 } 62 hasInstantiatedDependency(Class<T> key)63 public <T> boolean hasInstantiatedDependency(Class<T> key) { 64 return mObjs.containsKey(key) || mInstantiatedObjects.contains(key); 65 } 66 } 67