Home | History | Annotate | Download | only in systemui 
      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 
     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 
     39     public <T> T injectMockDependency(Class<T> cls) {
     40         final T mock = mock(cls);
     41         injectTestDependency(cls, mock);
     42         return mock;
     43     }
     44 
     45     public <T> void injectTestDependency(DependencyKey<T> key, T obj) {
     46         mObjs.put(key, obj);
     47     }
     48 
     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
     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 
     63     public <T> boolean hasInstantiatedDependency(Class<T> key) {
     64         return mObjs.containsKey(key) || mInstantiatedObjects.contains(key);
     65     }
     66 }
     67