Merge "Clean up BatteryAppListPreferenceController"
This commit is contained in:
committed by
Android (Google) Code Review
commit
be679b68a6
@@ -216,41 +216,6 @@ public class BatteryUtilsTest {
|
||||
assertThat(time).isEqualTo(0);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testShouldHideSystemConsumer_TypeIdle_ReturnTrue() {
|
||||
assertThat(mBatteryUtils.shouldHideDevicePowerComponent(mAggregateBatteryConsumer,
|
||||
BatteryConsumer.POWER_COMPONENT_IDLE)).isTrue();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testShouldHideSystemConsumer_TypeMobileRadio_ReturnTrue() {
|
||||
assertThat(mBatteryUtils.shouldHideDevicePowerComponent(mAggregateBatteryConsumer,
|
||||
BatteryConsumer.POWER_COMPONENT_MOBILE_RADIO)).isTrue();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testShouldHideSystemConsumer_TypeScreen_ReturnTrue() {
|
||||
assertThat(mBatteryUtils.shouldHideDevicePowerComponent(mAggregateBatteryConsumer,
|
||||
BatteryConsumer.POWER_COMPONENT_SCREEN)).isTrue();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testShouldHideSystemConsumer_TypeBluetooth_ReturnTrue() {
|
||||
assertThat(mBatteryUtils.shouldHideDevicePowerComponent(mAggregateBatteryConsumer,
|
||||
BatteryConsumer.POWER_COMPONENT_BLUETOOTH)).isTrue(); }
|
||||
|
||||
@Test
|
||||
public void testShouldHideSystemConsumer_TypeWifi_ReturnTrue() {
|
||||
assertThat(mBatteryUtils.shouldHideDevicePowerComponent(mAggregateBatteryConsumer,
|
||||
BatteryConsumer.POWER_COMPONENT_WIFI)).isTrue();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testShouldHideSystemConsumer_OtherType_ReturnFalse() {
|
||||
assertThat(mBatteryUtils.shouldHideDevicePowerComponent(mAggregateBatteryConsumer,
|
||||
BatteryConsumer.POWER_COMPONENT_FLASHLIGHT)).isFalse();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testCalculateBatteryPercent() {
|
||||
assertThat(mBatteryUtils.calculateBatteryPercent(BATTERY_SYSTEM_USAGE, TOTAL_BATTERY_USAGE,
|
||||
|
@@ -1,148 +0,0 @@
|
||||
/*
|
||||
* Copyright (C) 2022 The Android Open Source Project
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package com.android.settings.fuelgauge.batteryusage;
|
||||
|
||||
import static com.google.common.truth.Truth.assertThat;
|
||||
|
||||
import static org.mockito.Mockito.anyInt;
|
||||
import static org.mockito.Mockito.doReturn;
|
||||
import static org.mockito.Mockito.spy;
|
||||
import static org.mockito.Mockito.when;
|
||||
|
||||
import android.content.Context;
|
||||
import android.content.pm.PackageManager;
|
||||
import android.content.res.Resources;
|
||||
import android.os.Process;
|
||||
import android.os.UserManager;
|
||||
import android.text.format.DateUtils;
|
||||
|
||||
import androidx.preference.PreferenceGroup;
|
||||
|
||||
import com.android.settings.R;
|
||||
import com.android.settings.SettingsActivity;
|
||||
import com.android.settings.core.InstrumentedPreferenceFragment;
|
||||
import com.android.settings.fuelgauge.BatteryUtils;
|
||||
import com.android.settings.testutils.FakeFeatureFactory;
|
||||
|
||||
import org.junit.Before;
|
||||
import org.junit.Test;
|
||||
import org.junit.runner.RunWith;
|
||||
import org.mockito.Mock;
|
||||
import org.mockito.MockitoAnnotations;
|
||||
import org.robolectric.RobolectricTestRunner;
|
||||
import org.robolectric.RuntimeEnvironment;
|
||||
|
||||
@RunWith(RobolectricTestRunner.class)
|
||||
public class BatteryAppListPreferenceControllerTest {
|
||||
|
||||
private static final String KEY_APP_LIST = "app_list";
|
||||
|
||||
@Mock
|
||||
private SettingsActivity mSettingsActivity;
|
||||
@Mock
|
||||
private PreferenceGroup mAppListGroup;
|
||||
@Mock
|
||||
private InstrumentedPreferenceFragment mFragment;
|
||||
@Mock
|
||||
private BatteryUtils mBatteryUtils;
|
||||
@Mock
|
||||
private PackageManager mPackageManager;
|
||||
@Mock
|
||||
private UserManager mUserManager;
|
||||
@Mock
|
||||
private BatteryEntry mBatteryEntry;
|
||||
|
||||
private Context mContext;
|
||||
private PowerGaugePreference mPreference;
|
||||
private BatteryAppListPreferenceController mPreferenceController;
|
||||
private FakeFeatureFactory mFeatureFactory;
|
||||
|
||||
@Before
|
||||
public void setUp() {
|
||||
MockitoAnnotations.initMocks(this);
|
||||
|
||||
mFeatureFactory = FakeFeatureFactory.setupForTest();
|
||||
mContext = spy(RuntimeEnvironment.application);
|
||||
final Resources resources = spy(mContext.getResources());
|
||||
when(mContext.getResources()).thenReturn(resources);
|
||||
when(mContext.getPackageManager()).thenReturn(mPackageManager);
|
||||
when(mContext.getApplicationContext()).thenReturn(mContext);
|
||||
when(mContext.getSystemService(UserManager.class)).thenReturn(mUserManager);
|
||||
when(mUserManager.getProfileIdsWithDisabled(anyInt())).thenReturn(new int[]{});
|
||||
|
||||
mPreference = new PowerGaugePreference(mContext);
|
||||
|
||||
mPreferenceController = new BatteryAppListPreferenceController(mContext, KEY_APP_LIST, null,
|
||||
mSettingsActivity, mFragment);
|
||||
mPreferenceController.mBatteryUtils = mBatteryUtils;
|
||||
mPreferenceController.mAppListGroup = mAppListGroup;
|
||||
|
||||
BatteryAppListPreferenceController.sConfig =
|
||||
new BatteryAppListPreferenceController.Config() {
|
||||
@Override
|
||||
public boolean shouldShowBatteryAttributionList(Context context) {
|
||||
return true;
|
||||
}
|
||||
};
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testSetUsageSummary_timeLessThanOneMinute_doNotSetSummary() {
|
||||
when(mBatteryEntry.getTimeInForegroundMs()).thenReturn(59 * DateUtils.SECOND_IN_MILLIS);
|
||||
|
||||
mPreferenceController.setUsageSummary(mPreference, mBatteryEntry);
|
||||
assertThat(mPreference.getSummary()).isNull();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testSetUsageSummary_systemProcessUid_doNotSetSummary() {
|
||||
when(mBatteryEntry.getTimeInForegroundMs()).thenReturn(DateUtils.MINUTE_IN_MILLIS);
|
||||
when(mBatteryEntry.getUid()).thenReturn(Process.SYSTEM_UID);
|
||||
|
||||
mPreferenceController.setUsageSummary(mPreference, mBatteryEntry);
|
||||
assertThat(mPreference.getSummary()).isNull();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testSetUsageSummary_timeMoreThanOneMinute_normalApp_setScreenSummary() {
|
||||
when(mBatteryEntry.getTimeInForegroundMs()).thenReturn(2 * DateUtils.MINUTE_IN_MILLIS);
|
||||
doReturn(mContext.getText(R.string.battery_used_for)).when(mFragment).getText(
|
||||
R.string.battery_used_for);
|
||||
doReturn(mContext).when(mFragment).getContext();
|
||||
|
||||
mPreferenceController.setUsageSummary(mPreference, mBatteryEntry);
|
||||
|
||||
assertThat(mPreference.getSummary().toString()).isEqualTo("Used for 2 min");
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testSetUsageSummary_timeMoreThanOneMinute_hiddenApp_setUsedSummary() {
|
||||
when(mBatteryEntry.getTimeInForegroundMs()).thenReturn(2 * DateUtils.MINUTE_IN_MILLIS);
|
||||
when(mBatteryEntry.isHidden()).thenReturn(true);
|
||||
|
||||
doReturn(mContext).when(mFragment).getContext();
|
||||
|
||||
mPreferenceController.setUsageSummary(mPreference, mBatteryEntry);
|
||||
|
||||
assertThat(mPreference.getSummary().toString()).isEqualTo("2 min");
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testNeverUseFakeData() {
|
||||
assertThat(BatteryAppListPreferenceController.USE_FAKE_DATA).isFalse();
|
||||
}
|
||||
}
|
@@ -30,7 +30,6 @@ import android.content.pm.ApplicationInfo;
|
||||
import android.content.pm.PackageManager;
|
||||
import android.content.pm.PackageManager.NameNotFoundException;
|
||||
import android.os.BatteryConsumer;
|
||||
import android.os.Handler;
|
||||
import android.os.Process;
|
||||
import android.os.UidBatteryConsumer;
|
||||
import android.os.UserBatteryConsumer;
|
||||
@@ -70,8 +69,6 @@ public class BatteryEntryTest {
|
||||
private Context mMockContext;
|
||||
private Context mContext;
|
||||
@Mock
|
||||
private Handler mMockHandler;
|
||||
@Mock
|
||||
private PackageManager mMockPackageManager;
|
||||
@Mock
|
||||
private UserManager mMockUserManager;
|
||||
@@ -102,7 +99,7 @@ public class BatteryEntryTest {
|
||||
UidBatteryConsumer consumer = mock(UidBatteryConsumer.class);
|
||||
when(consumer.getUid()).thenReturn(APP_UID);
|
||||
when(consumer.getPackageWithHighestDrain()).thenReturn(highDrainPackage);
|
||||
return new BatteryEntry(mMockContext, mMockHandler, mMockUserManager,
|
||||
return new BatteryEntry(mMockContext, mMockUserManager,
|
||||
consumer, false, APP_UID, packages, packageName);
|
||||
}
|
||||
|
||||
@@ -117,7 +114,7 @@ public class BatteryEntryTest {
|
||||
private BatteryEntry createUserBatteryConsumer(int userId) {
|
||||
UserBatteryConsumer consumer = mock(UserBatteryConsumer.class);
|
||||
when(consumer.getUserId()).thenReturn(userId);
|
||||
return new BatteryEntry(mMockContext, mMockHandler, mMockUserManager,
|
||||
return new BatteryEntry(mMockContext, mMockUserManager,
|
||||
consumer, false, 0, null, null);
|
||||
}
|
||||
|
||||
@@ -182,7 +179,7 @@ public class BatteryEntryTest {
|
||||
when(mUidBatteryConsumer.getTimeInStateMs(UidBatteryConsumer.STATE_FOREGROUND))
|
||||
.thenReturn(100L);
|
||||
|
||||
final BatteryEntry entry = new BatteryEntry(RuntimeEnvironment.application, mMockHandler,
|
||||
final BatteryEntry entry = new BatteryEntry(RuntimeEnvironment.application,
|
||||
mMockUserManager, mUidBatteryConsumer, false, 0, null, null);
|
||||
|
||||
assertThat(entry.getTimeInForegroundMs()).isEqualTo(100L);
|
||||
@@ -201,7 +198,7 @@ public class BatteryEntryTest {
|
||||
when(mUidBatteryConsumer.getTimeInStateMs(UidBatteryConsumer.STATE_BACKGROUND))
|
||||
.thenReturn(100L);
|
||||
|
||||
final BatteryEntry entry = new BatteryEntry(RuntimeEnvironment.application, mMockHandler,
|
||||
final BatteryEntry entry = new BatteryEntry(RuntimeEnvironment.application,
|
||||
mMockUserManager, mUidBatteryConsumer, false, 0, null, null);
|
||||
|
||||
assertThat(entry.getTimeInBackgroundMs()).isEqualTo(100L);
|
||||
@@ -217,8 +214,6 @@ public class BatteryEntryTest {
|
||||
|
||||
@Test
|
||||
public void testUidCache_switchLocale_shouldCleanCache() {
|
||||
BatteryEntry.stopRequestQueue();
|
||||
|
||||
Locale.setDefault(new Locale("en_US"));
|
||||
BatteryEntry.sUidCache.put(Integer.toString(APP_UID), null);
|
||||
assertThat(BatteryEntry.sUidCache).isNotEmpty();
|
||||
|
@@ -20,7 +20,6 @@ import static com.google.common.truth.Truth.assertThat;
|
||||
|
||||
import static org.mockito.ArgumentMatchers.any;
|
||||
import static org.mockito.Mockito.doReturn;
|
||||
import static org.mockito.Mockito.never;
|
||||
import static org.mockito.Mockito.spy;
|
||||
import static org.mockito.Mockito.verify;
|
||||
import static org.mockito.Mockito.when;
|
||||
@@ -59,8 +58,6 @@ public final class BatteryUsageDataLoaderTest {
|
||||
@Mock
|
||||
private BatteryUsageStats mBatteryUsageStats;
|
||||
@Mock
|
||||
private BatteryAppListPreferenceController mMockBatteryAppListController;
|
||||
@Mock
|
||||
private BatteryEntry mMockBatteryEntry;
|
||||
@Captor
|
||||
private ArgumentCaptor<BatteryUsageStatsQuery> mStatsQueryCaptor;
|
||||
@@ -69,7 +66,6 @@ public final class BatteryUsageDataLoaderTest {
|
||||
public void setUp() {
|
||||
MockitoAnnotations.initMocks(this);
|
||||
mContext = spy(RuntimeEnvironment.application);
|
||||
BatteryUsageDataLoader.sController = mMockBatteryAppListController;
|
||||
doReturn(mContext).when(mContext).getApplicationContext();
|
||||
doReturn(mBatteryStatsManager).when(mContext).getSystemService(
|
||||
Context.BATTERY_STATS_SERVICE);
|
||||
@@ -84,8 +80,7 @@ public final class BatteryUsageDataLoaderTest {
|
||||
batteryEntryList.add(mMockBatteryEntry);
|
||||
when(mBatteryStatsManager.getBatteryUsageStats(mStatsQueryCaptor.capture()))
|
||||
.thenReturn(mBatteryUsageStats);
|
||||
when(mMockBatteryAppListController.getBatteryEntryList(mBatteryUsageStats, true))
|
||||
.thenReturn(batteryEntryList);
|
||||
BatteryUsageDataLoader.sFakeBatteryEntryListSupplier = () -> batteryEntryList;
|
||||
|
||||
BatteryUsageDataLoader.loadUsageData(mContext);
|
||||
|
||||
@@ -93,24 +88,6 @@ public final class BatteryUsageDataLoaderTest {
|
||||
assertThat(queryFlags
|
||||
& BatteryUsageStatsQuery.FLAG_BATTERY_USAGE_STATS_INCLUDE_HISTORY)
|
||||
.isNotEqualTo(0);
|
||||
verify(mMockBatteryAppListController)
|
||||
.getBatteryEntryList(mBatteryUsageStats, /*showAllApps=*/ true);
|
||||
verify(mMockContentResolver).insert(any(), any());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void loadUsageData_nullBatteryUsageStats_notLoadBatteryEntryData() {
|
||||
when(mBatteryStatsManager.getBatteryUsageStats(mStatsQueryCaptor.capture()))
|
||||
.thenReturn(null);
|
||||
|
||||
BatteryUsageDataLoader.loadUsageData(mContext);
|
||||
|
||||
final int queryFlags = mStatsQueryCaptor.getValue().getFlags();
|
||||
assertThat(queryFlags
|
||||
& BatteryUsageStatsQuery.FLAG_BATTERY_USAGE_STATS_INCLUDE_HISTORY)
|
||||
.isNotEqualTo(0);
|
||||
verify(mMockBatteryAppListController, never())
|
||||
.getBatteryEntryList(mBatteryUsageStats, /*showAllApps=*/ true);
|
||||
verify(mMockContentResolver).insert(any(), any());
|
||||
}
|
||||
|
||||
@@ -118,8 +95,7 @@ public final class BatteryUsageDataLoaderTest {
|
||||
public void loadUsageData_nullBatteryEntryList_insertFakeDataIntoProvider() {
|
||||
when(mBatteryStatsManager.getBatteryUsageStats(mStatsQueryCaptor.capture()))
|
||||
.thenReturn(mBatteryUsageStats);
|
||||
when(mMockBatteryAppListController.getBatteryEntryList(mBatteryUsageStats, true))
|
||||
.thenReturn(null);
|
||||
BatteryUsageDataLoader.sFakeBatteryEntryListSupplier = () -> null;
|
||||
|
||||
BatteryUsageDataLoader.loadUsageData(mContext);
|
||||
|
||||
@@ -130,8 +106,7 @@ public final class BatteryUsageDataLoaderTest {
|
||||
public void loadUsageData_emptyBatteryEntryList_insertFakeDataIntoProvider() {
|
||||
when(mBatteryStatsManager.getBatteryUsageStats(mStatsQueryCaptor.capture()))
|
||||
.thenReturn(mBatteryUsageStats);
|
||||
when(mMockBatteryAppListController.getBatteryEntryList(mBatteryUsageStats, true))
|
||||
.thenReturn(new ArrayList<BatteryEntry>());
|
||||
BatteryUsageDataLoader.sFakeBatteryEntryListSupplier = () -> new ArrayList<BatteryEntry>();
|
||||
|
||||
BatteryUsageDataLoader.loadUsageData(mContext);
|
||||
|
||||
|
Reference in New Issue
Block a user