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 17package com.android.providers.tv; 18 19import android.content.ComponentName; 20import android.content.ContentResolver; 21import android.content.Context; 22import android.content.pm.ApplicationInfo; 23import android.content.pm.PackageManager; 24import android.content.pm.ServiceInfo; 25import android.content.res.Resources; 26import android.content.SharedPreferences; 27import android.test.IsolatedContext; 28import android.test.RenamingDelegatingContext; 29import android.test.mock.MockContext; 30import android.test.mock.MockPackageManager; 31 32import java.util.Set; 33import java.util.HashSet; 34 35class MockTvProviderContext extends IsolatedContext { 36 private final Context mBase; 37 private final MockPackageManager mMockPackageManager = new MockPackageManager() { 38 @Override 39 public ServiceInfo getServiceInfo(ComponentName className, int flags) { 40 return null; 41 } 42 }; 43 44 private final Set<String> rejectedPermissions = new HashSet<>(); 45 46 MockTvProviderContext(ContentResolver resolver, Context base) { 47 super(resolver, new RenamingDelegatingContext(new MockContext(), base, "test.")); 48 mBase = base; 49 } 50 51 @Override 52 public Resources getResources() { 53 return mBase.getResources(); 54 } 55 56 @Override 57 public PackageManager getPackageManager() { 58 return mMockPackageManager; 59 } 60 61 @Override 62 public String getPackageName() { 63 return mBase.getPackageName(); 64 } 65 66 @Override 67 public ApplicationInfo getApplicationInfo() { 68 return mBase.getApplicationInfo(); 69 } 70 71 @Override 72 public int checkCallingOrSelfPermission(String permission) { 73 return rejectedPermissions.contains(permission) ? PackageManager.PERMISSION_DENIED 74 : PackageManager.PERMISSION_GRANTED; 75 } 76 77 @Override 78 public SharedPreferences getSharedPreferences(String name, int mode) { 79 return mBase.getSharedPreferences(name, mode); 80 } 81 82 public void grantOrRejectPermission(String permission, boolean granted) { 83 if (granted) { 84 rejectedPermissions.remove(permission); 85 } else { 86 rejectedPermissions.add(permission); 87 } 88 } 89} 90