1 /* 2 * Copyright (C) 2016 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.settings.security.trustagent; 18 19 import static com.google.common.truth.Truth.assertThat; 20 import static org.mockito.Mockito.when; 21 22 import android.content.pm.PackageManager; 23 import android.content.pm.ResolveInfo; 24 import android.content.pm.ServiceInfo; 25 26 import com.android.settings.testutils.SettingsRobolectricTestRunner; 27 28 import org.junit.Before; 29 import org.junit.Test; 30 import org.junit.runner.RunWith; 31 import org.mockito.Mock; 32 import org.mockito.MockitoAnnotations; 33 34 @RunWith(SettingsRobolectricTestRunner.class) 35 public class TrustAgentManagerTest { 36 37 private static final String CANNED_PACKAGE_NAME = "com.test.package"; 38 39 @Mock 40 private PackageManager mPackageManager; 41 42 private TrustAgentManager mTrustAgentManager; 43 44 @Before 45 public void setUp() { 46 MockitoAnnotations.initMocks(this); 47 mTrustAgentManager = new TrustAgentManager(); 48 } 49 50 @Test 51 public void shouldProvideTrust_doesProvideTrustWithPermission() { 52 when(mPackageManager.checkPermission(TrustAgentManager.PERMISSION_PROVIDE_AGENT, 53 CANNED_PACKAGE_NAME)).thenReturn(PackageManager.PERMISSION_GRANTED); 54 55 ServiceInfo serviceInfo = new ServiceInfo(); 56 serviceInfo.packageName = CANNED_PACKAGE_NAME; 57 ResolveInfo resolveInfo = new ResolveInfo(); 58 resolveInfo.serviceInfo = serviceInfo; 59 60 assertThat(mTrustAgentManager.shouldProvideTrust(resolveInfo, mPackageManager)).isTrue(); 61 } 62 63 @Test 64 public void shouldProvideTrust_doesNotProvideTrustWithoutPermission() { 65 when(mPackageManager.checkPermission(TrustAgentManager.PERMISSION_PROVIDE_AGENT, 66 CANNED_PACKAGE_NAME)).thenReturn(PackageManager.PERMISSION_DENIED); 67 68 ServiceInfo serviceInfo = new ServiceInfo(); 69 serviceInfo.packageName = CANNED_PACKAGE_NAME; 70 ResolveInfo resolveInfo = new ResolveInfo(); 71 resolveInfo.serviceInfo = serviceInfo; 72 73 assertThat(mTrustAgentManager.shouldProvideTrust(resolveInfo, mPackageManager)).isFalse(); 74 } 75 } 76