2 * Copyright (C) 2017 The Android Open Source Project
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
8 * http://www.apache.org/licenses/LICENSE-2.0
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
17 package com.android.settings;
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;
26 import android.app.AppOpsManager;
27 import android.app.admin.DeviceAdminInfo;
28 import android.content.Context;
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;
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;
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";
48 @Mock(answer = Answers.RETURNS_DEEP_STUBS)
49 private DeviceAdminInfo mDeviceAdmin;
51 private BatteryUtils mBatteryUtils;
52 private FakeFeatureFactory mFeatureFactory;
53 private DeviceAdminAdd mDeviceAdminAdd;
57 MockitoAnnotations.initMocks(this);
59 mFeatureFactory = FakeFeatureFactory.setupForTest();
60 mDeviceAdminAdd = Robolectric.buildActivity(DeviceAdminAdd.class).get();
62 doReturn(UID).when(mBatteryUtils).getPackageUid(PACKAGE_NAME);
63 when(mDeviceAdmin.getComponent().getPackageName()).thenReturn(PACKAGE_NAME);
64 mDeviceAdminAdd.mDeviceAdmin = mDeviceAdmin;
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"));
73 mDeviceAdminAdd.logSpecialPermissionChange(false, "app");
74 verify(mFeatureFactory.metricsFeatureProvider).action(any(Context.class),
75 eq(MetricsProto.MetricsEvent.APP_SPECIAL_PERMISSION_ADMIN_DENY), eq("app"));
79 public void unrestrictAppIfPossible_appRestricted_unrestrictApp() {
80 doReturn(true).when(mBatteryUtils).isForceAppStandbyEnabled(UID, PACKAGE_NAME);
82 mDeviceAdminAdd.unrestrictAppIfPossible(mBatteryUtils);
84 verify(mBatteryUtils).setForceAppStandby(UID, PACKAGE_NAME, AppOpsManager.MODE_ALLOWED);
88 public void unrestrictAppIfPossible_appUnrestricted_doNothing() {
89 doReturn(false).when(mBatteryUtils).isForceAppStandbyEnabled(UID, PACKAGE_NAME);
91 mDeviceAdminAdd.unrestrictAppIfPossible(mBatteryUtils);
93 verify(mBatteryUtils, never()).setForceAppStandby(UID, PACKAGE_NAME,
94 AppOpsManager.MODE_ALLOWED);