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 android.content.Context;
18 import android.testing.LeakCheck;
19 import android.testing.TestableContext;
20 import android.util.ArrayMap;
21 
22 public class SysuiTestableContext extends TestableContext implements SysUiServiceProvider {
23 
24     private ArrayMap<Class<?>, Object> mComponents;
25 
SysuiTestableContext(Context base)26     public SysuiTestableContext(Context base) {
27         super(base);
28     }
29 
SysuiTestableContext(Context base, LeakCheck check)30     public SysuiTestableContext(Context base, LeakCheck check) {
31         super(base, check);
32     }
33 
34     @SuppressWarnings("unchecked")
getComponent(Class<T> interfaceType)35     public <T> T getComponent(Class<T> interfaceType) {
36         return (T) (mComponents != null ? mComponents.get(interfaceType) : null);
37     }
38 
putComponent(Class<T> interfaceType, C component)39     public <T, C extends T> void putComponent(Class<T> interfaceType, C component) {
40         if (mComponents == null) mComponents = new ArrayMap<>();
41         mComponents.put(interfaceType, component);
42     }
43 }
44