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         mContext = context;
29         if (SystemUIFactory.getInstance() == null) {
30             SystemUIFactory.createFromConfig(context);
31         }
32         start();
33     }
34 
injectMockDependency(Class<T> cls)35     public <T> T injectMockDependency(Class<T> cls) {
36         final T mock = mock(cls);
37         injectTestDependency(cls, mock);
38         return mock;
39     }
40 
injectTestDependency(DependencyKey<T> key, T obj)41     public <T> void injectTestDependency(DependencyKey<T> key, T obj) {
42         mObjs.put(key, obj);
43     }
44 
injectTestDependency(Class<T> key, T obj)45     public <T> void injectTestDependency(Class<T> key, T obj) {
46         mObjs.put(key, obj);
47     }
48 
49     @Override
createDependency(Object key)50     protected <T> T createDependency(Object key) {
51         if (mObjs.containsKey(key)) return (T) mObjs.get(key);
52         mInstantiatedObjects.add(key);
53         return super.createDependency(key);
54     }
55 
hasInstantiatedDependency(Class<T> key)56     public <T> boolean hasInstantiatedDependency(Class<T> key) {
57         return mObjs.containsKey(key) || mInstantiatedObjects.contains(key);
58     }
59 }
60