Merge "Encapsulate BatterySipper and BatteryConsumer in BatteryEntry" into sc-dev
This commit is contained in:
committed by
Android (Google) Code Review
commit
60cc9e507b
@@ -47,7 +47,6 @@ import androidx.loader.app.LoaderManager;
|
||||
import androidx.preference.Preference;
|
||||
import androidx.recyclerview.widget.RecyclerView;
|
||||
|
||||
import com.android.internal.os.BatterySipper;
|
||||
import com.android.internal.os.BatteryStatsHelper;
|
||||
import com.android.settings.R;
|
||||
import com.android.settings.SettingsActivity;
|
||||
@@ -114,12 +113,8 @@ public class AdvancedPowerUsageDetailTest {
|
||||
@Mock
|
||||
private BatteryEntry mBatteryEntry;
|
||||
@Mock
|
||||
private BatterySipper mBatterySipper;
|
||||
@Mock
|
||||
private BatteryStatsHelper mBatteryStatsHelper;
|
||||
@Mock
|
||||
private BatteryStats.Uid mUid;
|
||||
@Mock
|
||||
private PackageManager mPackageManager;
|
||||
@Mock
|
||||
private AppOpsManager mAppOpsManager;
|
||||
@@ -140,6 +135,7 @@ public class AdvancedPowerUsageDetailTest {
|
||||
MockitoAnnotations.initMocks(this);
|
||||
|
||||
mContext = spy(RuntimeEnvironment.application);
|
||||
when(mContext.getPackageName()).thenReturn("foo");
|
||||
FakeFeatureFactory.setupForTest();
|
||||
|
||||
mFragment = spy(new AdvancedPowerUsageDetail());
|
||||
@@ -168,19 +164,13 @@ public class AdvancedPowerUsageDetailTest {
|
||||
doReturn(mEntityHeaderController).when(mEntityHeaderController)
|
||||
.setSummary(nullable(String.class));
|
||||
|
||||
doReturn(UID).when(mBatterySipper).getUid();
|
||||
when(mBatteryEntry.getUid()).thenReturn(UID);
|
||||
when(mBatteryEntry.getLabel()).thenReturn(APP_LABEL);
|
||||
doReturn(BACKGROUND_TIME_US).when(mUid).getProcessStateTime(
|
||||
eq(BatteryStats.Uid.PROCESS_STATE_BACKGROUND), anyLong(), anyInt());
|
||||
doReturn(PROCSTATE_TOP_TIME_US).when(mUid).getProcessStateTime(
|
||||
eq(BatteryStats.Uid.PROCESS_STATE_TOP), anyLong(), anyInt());
|
||||
doReturn(mForegroundActivityTimer).when(mUid).getForegroundActivityTimer();
|
||||
doReturn(FOREGROUND_ACTIVITY_TIME_US).when(mForegroundActivityTimer)
|
||||
.getTotalTimeLocked(anyLong(), anyInt());
|
||||
ReflectionHelpers.setField(mBatteryEntry, "sipper", mBatterySipper);
|
||||
when(mBatteryEntry.getTimeInBackgroundMs(any(BatteryUtils.class)))
|
||||
.thenReturn(BACKGROUND_TIME_MS);
|
||||
when(mBatteryEntry.getTimeInForegroundMs(any(BatteryUtils.class)))
|
||||
.thenReturn(FOREGROUND_TIME_MS);
|
||||
mBatteryEntry.iconId = ICON_ID;
|
||||
mBatterySipper.uidObj = mUid;
|
||||
mBatterySipper.drainType = BatterySipper.DrainType.APP;
|
||||
|
||||
mFragment.mHeaderPreference = mHeaderPreference;
|
||||
mFragment.mState = mState;
|
||||
@@ -200,6 +190,7 @@ public class AdvancedPowerUsageDetailTest {
|
||||
|
||||
Answer<Void> callable = invocation -> {
|
||||
mBundle = captor.getValue().getBundleExtra(EXTRA_SHOW_FRAGMENT_ARGUMENTS);
|
||||
System.out.println("mBundle = " + mBundle);
|
||||
return null;
|
||||
};
|
||||
doAnswer(callable).when(mActivity).startActivityAsUser(captor.capture(),
|
||||
@@ -274,27 +265,10 @@ public class AdvancedPowerUsageDetailTest {
|
||||
.isEqualTo(USAGE_PERCENT);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testStartBatteryDetailPage_typeNotApp_hasBasicData() {
|
||||
mBatterySipper.drainType = BatterySipper.DrainType.PHONE;
|
||||
mBatterySipper.usageTimeMs = PHONE_FOREGROUND_TIME_MS;
|
||||
|
||||
AdvancedPowerUsageDetail.startBatteryDetailPage(mActivity, mBatteryUtils, mFragment,
|
||||
mBatteryStatsHelper, 0, mBatteryEntry, USAGE_PERCENT);
|
||||
|
||||
assertThat(mBundle.getInt(AdvancedPowerUsageDetail.EXTRA_UID)).isEqualTo(UID);
|
||||
assertThat(mBundle.getLong(AdvancedPowerUsageDetail.EXTRA_FOREGROUND_TIME))
|
||||
.isEqualTo(PHONE_FOREGROUND_TIME_MS);
|
||||
assertThat(mBundle.getLong(AdvancedPowerUsageDetail.EXTRA_BACKGROUND_TIME))
|
||||
.isEqualTo(PHONE_BACKGROUND_TIME_MS);
|
||||
assertThat(mBundle.getString(AdvancedPowerUsageDetail.EXTRA_POWER_USAGE_PERCENT))
|
||||
.isEqualTo(USAGE_PERCENT);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testStartBatteryDetailPage_NormalApp() {
|
||||
mBatterySipper.mPackages = PACKAGE_NAME;
|
||||
mBatteryEntry.defaultPackageName = PACKAGE_NAME[0];
|
||||
when(mBatteryEntry.getDefaultPackageName()).thenReturn(PACKAGE_NAME[0]);
|
||||
|
||||
AdvancedPowerUsageDetail.startBatteryDetailPage(mActivity, mBatteryUtils, mFragment,
|
||||
mBatteryStatsHelper, 0, mBatteryEntry, USAGE_PERCENT);
|
||||
|
||||
@@ -304,7 +278,8 @@ public class AdvancedPowerUsageDetailTest {
|
||||
|
||||
@Test
|
||||
public void testStartBatteryDetailPage_SystemApp() {
|
||||
mBatterySipper.mPackages = null;
|
||||
when(mBatteryEntry.getDefaultPackageName()).thenReturn(null);
|
||||
|
||||
AdvancedPowerUsageDetail.startBatteryDetailPage(mActivity, mBatteryUtils, mFragment,
|
||||
mBatteryStatsHelper, 0, mBatteryEntry, USAGE_PERCENT);
|
||||
|
||||
@@ -316,8 +291,8 @@ public class AdvancedPowerUsageDetailTest {
|
||||
@Test
|
||||
public void testStartBatteryDetailPage_WorkApp() {
|
||||
final int appUid = 1010019;
|
||||
mBatterySipper.mPackages = PACKAGE_NAME;
|
||||
doReturn(appUid).when(mBatterySipper).getUid();
|
||||
doReturn(appUid).when(mBatteryEntry).getUid();
|
||||
|
||||
AdvancedPowerUsageDetail.startBatteryDetailPage(mActivity, mBatteryUtils, mFragment,
|
||||
mBatteryStatsHelper, 0, mBatteryEntry, USAGE_PERCENT);
|
||||
|
||||
@@ -326,8 +301,7 @@ public class AdvancedPowerUsageDetailTest {
|
||||
|
||||
@Test
|
||||
public void testStartBatteryDetailPage_typeUser_startByCurrentUser() {
|
||||
mBatterySipper.drainType = BatterySipper.DrainType.USER;
|
||||
mBatterySipper.userId = 10;
|
||||
when(mBatteryEntry.isUserEntry()).thenReturn(true);
|
||||
|
||||
final int currentUser = 20;
|
||||
ShadowActivityManager.setCurrentUser(currentUser);
|
||||
@@ -364,18 +338,6 @@ public class AdvancedPowerUsageDetailTest {
|
||||
assertThat(mBundle.getInt(AdvancedPowerUsageDetail.EXTRA_UID)).isEqualTo(UID);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testStartBatteryDetailPage_defaultPackageNull_chooseFromBatterySipper() {
|
||||
mBatteryEntry.defaultPackageName = null;
|
||||
mBatteryEntry.sipper.mPackages = PACKAGE_NAME;
|
||||
|
||||
AdvancedPowerUsageDetail.startBatteryDetailPage(mActivity, mBatteryUtils, mFragment,
|
||||
mBatteryStatsHelper, 0, mBatteryEntry, USAGE_PERCENT);
|
||||
|
||||
assertThat(mBundle.getString(AdvancedPowerUsageDetail.EXTRA_PACKAGE_NAME))
|
||||
.isEqualTo(PACKAGE_NAME[0]);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testInitPreference_hasCorrectSummary() {
|
||||
Bundle bundle = new Bundle(4);
|
||||
|
@@ -17,6 +17,8 @@ package com.android.settings.fuelgauge;
|
||||
|
||||
import static com.google.common.truth.Truth.assertThat;
|
||||
|
||||
import static org.mockito.ArgumentMatchers.any;
|
||||
import static org.mockito.ArgumentMatchers.eq;
|
||||
import static org.mockito.Mockito.mock;
|
||||
import static org.mockito.Mockito.when;
|
||||
|
||||
@@ -24,8 +26,12 @@ import android.content.Context;
|
||||
import android.content.pm.ApplicationInfo;
|
||||
import android.content.pm.PackageManager;
|
||||
import android.content.pm.PackageManager.NameNotFoundException;
|
||||
import android.os.BatteryConsumer;
|
||||
import android.os.BatteryStats;
|
||||
import android.os.Handler;
|
||||
import android.os.Process;
|
||||
import android.os.SystemBatteryConsumer;
|
||||
import android.os.UidBatteryConsumer;
|
||||
import android.os.UserManager;
|
||||
|
||||
import com.android.internal.os.BatterySipper;
|
||||
@@ -61,6 +67,9 @@ public class BatteryEntryTest {
|
||||
@Mock private Handler mockHandler;
|
||||
@Mock private PackageManager mockPackageManager;
|
||||
@Mock private UserManager mockUserManager;
|
||||
@Mock private UidBatteryConsumer mUidBatteryConsumer;
|
||||
@Mock private SystemBatteryConsumer mSystemBatteryConsumer;
|
||||
@Mock BatteryUtils mBatteryUtils;
|
||||
|
||||
@Before
|
||||
public void stubContextToReturnMockPackageManager() {
|
||||
@@ -80,7 +89,7 @@ public class BatteryEntryTest {
|
||||
|
||||
private BatteryEntry createBatteryEntryForApp() {
|
||||
return new BatteryEntry(mockContext, mockHandler, mockUserManager, createSipperForApp(),
|
||||
null);
|
||||
null, null);
|
||||
}
|
||||
|
||||
private BatterySipper createSipperForApp() {
|
||||
@@ -90,11 +99,6 @@ public class BatteryEntryTest {
|
||||
return sipper;
|
||||
}
|
||||
|
||||
private BatteryEntry createBatteryEntryForSystem() {
|
||||
return new BatteryEntry(mockContext, mockHandler, mockUserManager, createSipperForSystem(),
|
||||
null);
|
||||
}
|
||||
|
||||
private BatterySipper createSipperForSystem() {
|
||||
BatterySipper sipper =
|
||||
new BatterySipper(DrainType.APP, new FakeUid(SYSTEM_UID), 0 /* power use */);
|
||||
@@ -103,11 +107,15 @@ public class BatteryEntryTest {
|
||||
return sipper;
|
||||
}
|
||||
|
||||
private BatterySipper createNonAppSipper() {
|
||||
return new BatterySipper(DrainType.IDLE, null, 0 /* power use */);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void batteryEntryForApp_shouldSetDefaultPackageNameAndLabel() throws Exception {
|
||||
BatteryEntry entry = createBatteryEntryForApp();
|
||||
|
||||
assertThat(entry.defaultPackageName).isEqualTo(APP_DEFAULT_PACKAGE_NAME);
|
||||
assertThat(entry.getDefaultPackageName()).isEqualTo(APP_DEFAULT_PACKAGE_NAME);
|
||||
assertThat(entry.getLabel()).isEqualTo(APP_LABEL);
|
||||
}
|
||||
|
||||
@@ -146,7 +154,7 @@ public class BatteryEntryTest {
|
||||
final BatterySipper batterySipper = mock(BatterySipper.class);
|
||||
batterySipper.drainType = DrainType.AMBIENT_DISPLAY;
|
||||
final BatteryEntry entry = new BatteryEntry(RuntimeEnvironment.application, mockHandler,
|
||||
mockUserManager, batterySipper, null);
|
||||
mockUserManager, batterySipper, null, null);
|
||||
|
||||
assertThat(entry.iconId).isEqualTo(R.drawable.ic_settings_aod);
|
||||
assertThat(entry.name).isEqualTo("Ambient display");
|
||||
@@ -154,17 +162,82 @@ public class BatteryEntryTest {
|
||||
|
||||
@Test
|
||||
public void extractPackageFromSipper_systemSipper_returnSystemPackage() {
|
||||
BatteryEntry entry = createBatteryEntryForSystem();
|
||||
|
||||
assertThat(entry.extractPackagesFromSipper(entry.sipper))
|
||||
assertThat(BatteryEntry.extractPackagesFromSipper(createSipperForSystem()))
|
||||
.isEqualTo(new String[] {ANDROID_PACKAGE});
|
||||
}
|
||||
|
||||
@Test
|
||||
public void extractPackageFromSipper_normalSipper_returnDefaultPackage() {
|
||||
BatteryEntry entry = createBatteryEntryForApp();
|
||||
BatterySipper sipper = createSipperForApp();
|
||||
assertThat(BatteryEntry.extractPackagesFromSipper(sipper)).isEqualTo(sipper.mPackages);
|
||||
}
|
||||
|
||||
assertThat(entry.extractPackagesFromSipper(entry.sipper)).isEqualTo(entry.sipper.mPackages);
|
||||
@Test
|
||||
public void getTimeInForegroundMs_app() {
|
||||
final BatteryEntry entry = new BatteryEntry(RuntimeEnvironment.application, mockHandler,
|
||||
mockUserManager, null, mUidBatteryConsumer, null);
|
||||
|
||||
when(mUidBatteryConsumer.getTimeInStateMs(UidBatteryConsumer.STATE_FOREGROUND))
|
||||
.thenReturn(100L);
|
||||
|
||||
assertThat(entry.getTimeInForegroundMs(mBatteryUtils)).isEqualTo(100L);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void getTimeInForegroundMs_systemConsumer() {
|
||||
final BatteryEntry entry = new BatteryEntry(RuntimeEnvironment.application, mockHandler,
|
||||
mockUserManager, createNonAppSipper(), mSystemBatteryConsumer, null);
|
||||
|
||||
when(mSystemBatteryConsumer.getUsageDurationMillis(BatteryConsumer.TIME_COMPONENT_USAGE))
|
||||
.thenReturn(100L);
|
||||
|
||||
assertThat(entry.getTimeInForegroundMs(mBatteryUtils)).isEqualTo(100L);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void getTimeInForegroundMs_useSipper() {
|
||||
final BatterySipper batterySipper = createSipperForApp();
|
||||
final BatteryEntry entry = new BatteryEntry(RuntimeEnvironment.application, mockHandler,
|
||||
mockUserManager, batterySipper, null, null);
|
||||
|
||||
when(mBatteryUtils.getProcessTimeMs(eq(BatteryUtils.StatusType.FOREGROUND),
|
||||
any(BatteryStats.Uid.class), eq(BatteryStats.STATS_SINCE_CHARGED)))
|
||||
.thenReturn(100L);
|
||||
assertThat(entry.getTimeInForegroundMs(mBatteryUtils)).isEqualTo(100L);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void getTimeInBackgroundMs_app() {
|
||||
final BatteryEntry entry = new BatteryEntry(RuntimeEnvironment.application, mockHandler,
|
||||
mockUserManager, null, mUidBatteryConsumer, null);
|
||||
|
||||
when(mUidBatteryConsumer.getTimeInStateMs(UidBatteryConsumer.STATE_BACKGROUND))
|
||||
.thenReturn(100L);
|
||||
|
||||
assertThat(entry.getTimeInBackgroundMs(mBatteryUtils)).isEqualTo(100L);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void getTimeInBackgroundMs_systemConsumer() {
|
||||
final BatteryEntry entry = new BatteryEntry(RuntimeEnvironment.application, mockHandler,
|
||||
mockUserManager, createNonAppSipper(), mSystemBatteryConsumer, null);
|
||||
|
||||
when(mSystemBatteryConsumer.getUsageDurationMillis(BatteryConsumer.TIME_COMPONENT_USAGE))
|
||||
.thenReturn(100L);
|
||||
|
||||
assertThat(entry.getTimeInBackgroundMs(mBatteryUtils)).isEqualTo(0);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void getTimeInBackgroundMs_useSipper() {
|
||||
final BatterySipper batterySipper = createSipperForApp();
|
||||
final BatteryEntry entry = new BatteryEntry(RuntimeEnvironment.application, mockHandler,
|
||||
mockUserManager, batterySipper, null, null);
|
||||
|
||||
when(mBatteryUtils.getProcessTimeMs(eq(BatteryUtils.StatusType.BACKGROUND),
|
||||
any(BatteryStats.Uid.class), eq(BatteryStats.STATS_SINCE_CHARGED)))
|
||||
.thenReturn(100L);
|
||||
assertThat(entry.getTimeInBackgroundMs(mBatteryUtils)).isEqualTo(100L);
|
||||
}
|
||||
|
||||
@Test
|
||||
|
Reference in New Issue
Block a user