1 /*
2  * Copyright (C) 2014 The Android Open Source Project
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *      http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 
17 package com.android.providers.tv;
18 
19 import android.content.ComponentName;
20 import android.content.ContentResolver;
21 import android.content.Context;
22 import android.content.pm.ApplicationInfo;
23 import android.content.pm.PackageManager;
24 import android.content.pm.ServiceInfo;
25 import android.content.res.Resources;
26 import android.content.SharedPreferences;
27 import android.test.IsolatedContext;
28 import android.test.RenamingDelegatingContext;
29 import android.test.mock.MockContext;
30 import android.test.mock.MockPackageManager;
31 
32 import java.util.Arrays;
33 import java.util.Set;
34 import java.util.HashSet;
35 
36 class MockTvProviderContext extends IsolatedContext {
37     private final Context mBase;
38     private final MockPackageManagerForTesting mMockPackageManager =
39             new MockPackageManagerForTesting();
40 
41     private final Set<String> rejectedPermissions = new HashSet<>();
42 
MockTvProviderContext(ContentResolver resolver, Context base)43     MockTvProviderContext(ContentResolver resolver, Context base) {
44         super(resolver, new RenamingDelegatingContext(new MockContext(), base, "test."));
45         mBase = base;
46     }
47 
48     @Override
getResources()49     public Resources getResources() {
50         return mBase.getResources();
51     }
52 
53     @Override
getPackageManager()54     public PackageManager getPackageManager() {
55         return mMockPackageManager;
56     }
57 
58     @Override
getPackageName()59     public String getPackageName() {
60         return mBase.getPackageName();
61     }
62 
63     @Override
getApplicationInfo()64     public ApplicationInfo getApplicationInfo() {
65         return mBase.getApplicationInfo();
66     }
67 
68     @Override
checkCallingOrSelfPermission(String permission)69     public int checkCallingOrSelfPermission(String permission) {
70         return rejectedPermissions.contains(permission) ? PackageManager.PERMISSION_DENIED
71                 : PackageManager.PERMISSION_GRANTED;
72     }
73 
74     @Override
getSharedPreferences(String name, int mode)75     public SharedPreferences getSharedPreferences(String name, int mode) {
76         return mBase.getSharedPreferences(name, mode);
77     }
78 
grantOrRejectPermission(String permission, boolean granted)79     public void grantOrRejectPermission(String permission, boolean granted) {
80         if (granted) {
81             rejectedPermissions.remove(permission);
82         } else {
83             rejectedPermissions.add(permission);
84         }
85     }
86 
setExistingPackages(String... packages)87     public void setExistingPackages(String... packages) {
88         mMockPackageManager.setExistingPackages(new HashSet<>(Arrays.asList(packages)));
89     }
90 
91     private static class MockPackageManagerForTesting extends MockPackageManager {
92         private Set<String> mExistingPackages;
93         @Override
getServiceInfo(ComponentName className, int flags)94         public ServiceInfo getServiceInfo(ComponentName className, int flags) {
95             return null;
96         }
97 
98         @Override
getApplicationInfo(String packageName, int flags)99         public ApplicationInfo getApplicationInfo(String packageName, int flags)
100                 throws NameNotFoundException {
101             if (mExistingPackages != null && mExistingPackages.contains(packageName)) {
102                 return new ApplicationInfo();
103             }
104             throw new NameNotFoundException();
105         }
106 
setExistingPackages(Set<String> existingPackages)107         public void setExistingPackages(Set<String> existingPackages) {
108             mExistingPackages = existingPackages;
109         }
110     }
111 }
112