1 /* 2 * Copyright (C) 2017 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; 18 19 import static org.mockito.Matchers.any; 20 import static org.mockito.Matchers.eq; 21 import static org.mockito.Mockito.doReturn; 22 import static org.mockito.Mockito.never; 23 import static org.mockito.Mockito.verify; 24 import static org.mockito.Mockito.when; 25 26 import android.app.AppOpsManager; 27 import android.app.admin.DeviceAdminInfo; 28 import android.content.Context; 29 30 import com.android.internal.logging.nano.MetricsProto; 31 import com.android.settings.fuelgauge.BatteryUtils; 32 import com.android.settings.testutils.FakeFeatureFactory; 33 import com.android.settings.testutils.SettingsRobolectricTestRunner; 34 35 import org.junit.Before; 36 import org.junit.Test; 37 import org.junit.runner.RunWith; 38 import org.mockito.Answers; 39 import org.mockito.Mock; 40 import org.mockito.MockitoAnnotations; 41 import org.robolectric.Robolectric; 42 43 @RunWith(SettingsRobolectricTestRunner.class) 44 public class DeviceAdminAddTest { 45 private static final int UID = 12345; 46 private static final String PACKAGE_NAME = "com.android.test.device.admin"; 47 48 @Mock(answer = Answers.RETURNS_DEEP_STUBS) 49 private DeviceAdminInfo mDeviceAdmin; 50 @Mock 51 private BatteryUtils mBatteryUtils; 52 private FakeFeatureFactory mFeatureFactory; 53 private DeviceAdminAdd mDeviceAdminAdd; 54 55 @Before setUp()56 public void setUp() { 57 MockitoAnnotations.initMocks(this); 58 59 mFeatureFactory = FakeFeatureFactory.setupForTest(); 60 mDeviceAdminAdd = Robolectric.buildActivity(DeviceAdminAdd.class).get(); 61 62 doReturn(UID).when(mBatteryUtils).getPackageUid(PACKAGE_NAME); 63 when(mDeviceAdmin.getComponent().getPackageName()).thenReturn(PACKAGE_NAME); 64 mDeviceAdminAdd.mDeviceAdmin = mDeviceAdmin; 65 } 66 67 @Test logSpecialPermissionChange()68 public void logSpecialPermissionChange() { 69 mDeviceAdminAdd.logSpecialPermissionChange(true, "app"); 70 verify(mFeatureFactory.metricsFeatureProvider).action(any(Context.class), 71 eq(MetricsProto.MetricsEvent.APP_SPECIAL_PERMISSION_ADMIN_ALLOW), eq("app")); 72 73 mDeviceAdminAdd.logSpecialPermissionChange(false, "app"); 74 verify(mFeatureFactory.metricsFeatureProvider).action(any(Context.class), 75 eq(MetricsProto.MetricsEvent.APP_SPECIAL_PERMISSION_ADMIN_DENY), eq("app")); 76 } 77 78 @Test unrestrictAppIfPossible_appRestricted_unrestrictApp()79 public void unrestrictAppIfPossible_appRestricted_unrestrictApp() { 80 doReturn(true).when(mBatteryUtils).isForceAppStandbyEnabled(UID, PACKAGE_NAME); 81 82 mDeviceAdminAdd.unrestrictAppIfPossible(mBatteryUtils); 83 84 verify(mBatteryUtils).setForceAppStandby(UID, PACKAGE_NAME, AppOpsManager.MODE_ALLOWED); 85 } 86 87 @Test unrestrictAppIfPossible_appUnrestricted_doNothing()88 public void unrestrictAppIfPossible_appUnrestricted_doNothing() { 89 doReturn(false).when(mBatteryUtils).isForceAppStandbyEnabled(UID, PACKAGE_NAME); 90 91 mDeviceAdminAdd.unrestrictAppIfPossible(mBatteryUtils); 92 93 verify(mBatteryUtils, never()).setForceAppStandby(UID, PACKAGE_NAME, 94 AppOpsManager.MODE_ALLOWED); 95 } 96 } 97