Merge "Unrestrict app if it been set as device admin app"

This commit is contained in:
TreeHugger Robot
2018-07-03 02:26:16 +00:00
committed by Android (Google) Code Review
2 changed files with 51 additions and 0 deletions

View File

@@ -18,23 +18,37 @@ package com.android.settings.applications.specialaccess.deviceadmin;
import static org.mockito.Matchers.any;
import static org.mockito.Matchers.eq;
import static org.mockito.Mockito.doReturn;
import static org.mockito.Mockito.never;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;
import android.app.AppOpsManager;
import android.app.admin.DeviceAdminInfo;
import android.content.Context;
import com.android.internal.logging.nano.MetricsProto;
import com.android.settings.fuelgauge.BatteryUtils;
import com.android.settings.testutils.FakeFeatureFactory;
import com.android.settings.testutils.SettingsRobolectricTestRunner;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.Answers;
import org.mockito.Mock;
import org.mockito.MockitoAnnotations;
import org.robolectric.Robolectric;
@RunWith(SettingsRobolectricTestRunner.class)
public class DeviceAdminAddTest {
private static final int UID = 12345;
private static final String PACKAGE_NAME = "com.android.test.device.admin";
@Mock(answer = Answers.RETURNS_DEEP_STUBS)
private DeviceAdminInfo mDeviceAdmin;
@Mock
private BatteryUtils mBatteryUtils;
private FakeFeatureFactory mFeatureFactory;
private DeviceAdminAdd mDeviceAdminAdd;
@@ -44,6 +58,10 @@ public class DeviceAdminAddTest {
mFeatureFactory = FakeFeatureFactory.setupForTest();
mDeviceAdminAdd = Robolectric.buildActivity(DeviceAdminAdd.class).get();
doReturn(UID).when(mBatteryUtils).getPackageUid(PACKAGE_NAME);
when(mDeviceAdmin.getComponent().getPackageName()).thenReturn(PACKAGE_NAME);
mDeviceAdminAdd.mDeviceAdmin = mDeviceAdmin;
}
@Test
@@ -56,4 +74,23 @@ public class DeviceAdminAddTest {
verify(mFeatureFactory.metricsFeatureProvider).action(any(Context.class),
eq(MetricsProto.MetricsEvent.APP_SPECIAL_PERMISSION_ADMIN_DENY), eq("app"));
}
@Test
public void unrestrictAppIfPossible_appRestricted_unrestrictApp() {
doReturn(true).when(mBatteryUtils).isForceAppStandbyEnabled(UID, PACKAGE_NAME);
mDeviceAdminAdd.unrestrictAppIfPossible(mBatteryUtils);
verify(mBatteryUtils).setForceAppStandby(UID, PACKAGE_NAME, AppOpsManager.MODE_ALLOWED);
}
@Test
public void unrestrictAppIfPossible_appUnrestricted_doNothing() {
doReturn(false).when(mBatteryUtils).isForceAppStandbyEnabled(UID, PACKAGE_NAME);
mDeviceAdminAdd.unrestrictAppIfPossible(mBatteryUtils);
verify(mBatteryUtils, never()).setForceAppStandby(UID, PACKAGE_NAME,
AppOpsManager.MODE_ALLOWED);
}
}