Merge "Hide uninstalled app in restrict app dialog" into pi-dev
This commit is contained in:
@@ -17,19 +17,25 @@
|
||||
package com.android.settings.fuelgauge.batterytip.detectors;
|
||||
|
||||
import static com.google.common.truth.Truth.assertThat;
|
||||
|
||||
import static org.mockito.Matchers.any;
|
||||
import static org.mockito.Matchers.anyInt;
|
||||
import static org.mockito.Matchers.anyLong;
|
||||
import static org.mockito.Matchers.eq;
|
||||
import static org.mockito.Mockito.doReturn;
|
||||
import static org.mockito.Mockito.doThrow;
|
||||
import static org.mockito.Mockito.spy;
|
||||
|
||||
import android.content.Context;
|
||||
import android.content.pm.ApplicationInfo;
|
||||
import android.content.pm.PackageManager;
|
||||
|
||||
import com.android.settings.fuelgauge.batterytip.AnomalyDatabaseHelper;
|
||||
import com.android.settings.fuelgauge.batterytip.AppInfo;
|
||||
import com.android.settings.fuelgauge.batterytip.BatteryDatabaseManager;
|
||||
import com.android.settings.fuelgauge.batterytip.BatteryTipPolicy;
|
||||
import com.android.settings.fuelgauge.batterytip.tips.BatteryTip;
|
||||
import com.android.settings.fuelgauge.batterytip.tips.RestrictAppTip;
|
||||
import com.android.settings.testutils.DatabaseTestUtils;
|
||||
import com.android.settings.testutils.SettingsRobolectricTestRunner;
|
||||
|
||||
@@ -48,24 +54,40 @@ import java.util.List;
|
||||
public class RestrictAppDetectorTest {
|
||||
|
||||
private static final String PACKAGE_NAME = "com.android.app";
|
||||
private static final String UNINSTALLED_PACKAGE_NAME = "com.android.uninstalled";
|
||||
private Context mContext;
|
||||
private BatteryTipPolicy mPolicy;
|
||||
private RestrictAppDetector mRestrictAppDetector;
|
||||
private List<AppInfo> mAppInfoList;
|
||||
private AppInfo mAppInfo;
|
||||
@Mock
|
||||
private BatteryDatabaseManager mBatteryDatabaseManager;
|
||||
@Mock
|
||||
private PackageManager mPackageManager;
|
||||
@Mock
|
||||
private ApplicationInfo mApplicationInfo;
|
||||
|
||||
@Before
|
||||
public void setUp() {
|
||||
public void setUp() throws PackageManager.NameNotFoundException {
|
||||
MockitoAnnotations.initMocks(this);
|
||||
|
||||
mAppInfoList = new ArrayList<>();
|
||||
mAppInfoList.add(new AppInfo.Builder().setPackageName(PACKAGE_NAME).build());
|
||||
mAppInfo = new AppInfo.Builder()
|
||||
.setPackageName(PACKAGE_NAME)
|
||||
.build();
|
||||
mAppInfoList.add(mAppInfo);
|
||||
|
||||
mContext = RuntimeEnvironment.application;
|
||||
mContext = spy(RuntimeEnvironment.application);
|
||||
mPolicy = spy(new BatteryTipPolicy(mContext));
|
||||
mRestrictAppDetector = new RestrictAppDetector(mContext, mPolicy);
|
||||
mRestrictAppDetector.mBatteryDatabaseManager = mBatteryDatabaseManager;
|
||||
|
||||
doReturn(mPackageManager).when(mContext).getPackageManager();
|
||||
doReturn(mApplicationInfo).when(mPackageManager).getApplicationInfo(eq(PACKAGE_NAME),
|
||||
anyInt());
|
||||
doReturn(PACKAGE_NAME).when(mApplicationInfo).loadLabel(any());
|
||||
doThrow(new PackageManager.NameNotFoundException()).when(
|
||||
mPackageManager).getApplicationInfo(eq(UNINSTALLED_PACKAGE_NAME), anyInt());
|
||||
}
|
||||
|
||||
@After
|
||||
@@ -76,7 +98,7 @@ public class RestrictAppDetectorTest {
|
||||
@Test
|
||||
public void testDetect_hasAnomaly_tipNew() {
|
||||
doReturn(mAppInfoList).when(mBatteryDatabaseManager)
|
||||
.queryAllAnomalies(anyLong(), eq(AnomalyDatabaseHelper.State.NEW));
|
||||
.queryAllAnomalies(anyLong(), eq(AnomalyDatabaseHelper.State.NEW));
|
||||
|
||||
assertThat(mRestrictAppDetector.detect().getState()).isEqualTo(BatteryTip.StateType.NEW);
|
||||
}
|
||||
@@ -84,21 +106,34 @@ public class RestrictAppDetectorTest {
|
||||
@Test
|
||||
public void testDetect_hasAutoHandledAnomaly_tipHandled() {
|
||||
doReturn(new ArrayList<AppInfo>()).when(mBatteryDatabaseManager)
|
||||
.queryAllAnomalies(anyLong(), eq(AnomalyDatabaseHelper.State.NEW));
|
||||
.queryAllAnomalies(anyLong(), eq(AnomalyDatabaseHelper.State.NEW));
|
||||
doReturn(mAppInfoList).when(mBatteryDatabaseManager)
|
||||
.queryAllAnomalies(anyLong(), eq(AnomalyDatabaseHelper.State.AUTO_HANDLED));
|
||||
.queryAllAnomalies(anyLong(), eq(AnomalyDatabaseHelper.State.AUTO_HANDLED));
|
||||
|
||||
assertThat(mRestrictAppDetector.detect().getState())
|
||||
.isEqualTo(BatteryTip.StateType.HANDLED);
|
||||
.isEqualTo(BatteryTip.StateType.HANDLED);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testDetect_hasUninstalledAnomaly_removeIt() {
|
||||
mAppInfoList.add(new AppInfo.Builder()
|
||||
.setPackageName(UNINSTALLED_PACKAGE_NAME)
|
||||
.build());
|
||||
doReturn(mAppInfoList).when(mBatteryDatabaseManager)
|
||||
.queryAllAnomalies(anyLong(), eq(AnomalyDatabaseHelper.State.NEW));
|
||||
|
||||
final RestrictAppTip restrictAppTip = (RestrictAppTip) mRestrictAppDetector.detect();
|
||||
assertThat(restrictAppTip.getState()).isEqualTo(BatteryTip.StateType.NEW);
|
||||
assertThat(restrictAppTip.getRestrictAppList()).containsExactly(mAppInfo);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testDetect_noAnomaly_tipInvisible() {
|
||||
doReturn(new ArrayList<AppInfo>()).when(mBatteryDatabaseManager)
|
||||
.queryAllAnomalies(anyLong(), anyInt());
|
||||
.queryAllAnomalies(anyLong(), anyInt());
|
||||
|
||||
assertThat(mRestrictAppDetector.detect().getState())
|
||||
.isEqualTo(BatteryTip.StateType.INVISIBLE);
|
||||
.isEqualTo(BatteryTip.StateType.INVISIBLE);
|
||||
}
|
||||
|
||||
@Test
|
||||
|
Reference in New Issue
Block a user