Refine startBatteryDetailPage method to support history data

Bug: 177406865
Test: make SettingsRoboTests
Test: make SettingsGoogleRoboTests
Change-Id: Iee3b6b353c4f92bdad37fac55adad58b4f205bf6

tests/robotests/src/com/android/settings/fuelgauge/BatteryChartPreferenceControllerTest.java

@@ -63,6 +63,7 @@ public final class BatteryChartPreferenceControllerTest {
     @Mock private BatteryHistEntry mBatteryHistEntry;
     @Mock private BatteryChartView mBatteryChartView;
     @Mock private PowerGaugePreference mPowerGaugePreference;
+    @Mock private BatteryUtils mBatteryUtils;
 
     private Context mContext;
     private BatteryDiffEntry mBatteryDiffEntry;
@@ -127,6 +128,12 @@ public final class BatteryChartPreferenceControllerTest {
         assertThat(mBatteryChartPreferenceController.mPreferenceCache).isEmpty();
     }
 
+    @Test
+    public void testOnDestroy_removeAllPreferenceFromPreferenceGroup() {
+        mBatteryChartPreferenceController.onDestroy();
+        verify(mAppListGroup).removeAll();
+    }
+
     @Test
     public void testSetBatteryHistoryMap_createExpectedKeysAndLevels() {
         mBatteryChartPreferenceController.setBatteryHistoryMap(
@@ -266,6 +273,34 @@ public final class BatteryChartPreferenceControllerTest {
         assertThat(pref.getTitle()).isEqualTo(appLabel);
         assertThat(pref.getIcon()).isEqualTo(mDrawable);
         assertThat(pref.getOrder()).isEqualTo(1);
+        assertThat(pref.getBatteryDiffEntry()).isSameInstanceAs(mBatteryDiffEntry);
+    }
+
+    @Test
+    public void testHandlePreferenceTreeClick_notPowerGaugePreference_returnFalse() {
+        assertThat(mBatteryChartPreferenceController.handlePreferenceTreeClick(mAppListGroup))
+            .isFalse();
+    }
+
+    @Test
+    public void testHandlePreferenceTreeClick_validPackageName_returnTrue() {
+        doReturn(false).when(mBatteryHistEntry).isAppEntry();
+        doReturn(mBatteryDiffEntry).when(mPowerGaugePreference).getBatteryDiffEntry();
+
+        assertThat(mBatteryChartPreferenceController.handlePreferenceTreeClick(
+            mPowerGaugePreference)).isTrue();
+    }
+
+    @Test
+    public void testHandlePreferenceTreeClick_appEntryWithInvalidPackage_returnFalse() {
+        mBatteryChartPreferenceController.mBatteryUtils = mBatteryUtils;
+        doReturn(true).when(mBatteryHistEntry).isAppEntry();
+        doReturn(BatteryUtils.UID_NULL).when(mBatteryUtils)
+            .getPackageUid(mBatteryHistEntry.mPackageName);
+        doReturn(mBatteryDiffEntry).when(mPowerGaugePreference).getBatteryDiffEntry();
+
+        assertThat(mBatteryChartPreferenceController.handlePreferenceTreeClick(
+            mPowerGaugePreference)).isFalse();
     }
 
     private Map<Long, List<BatteryHistEntry>> createBatteryHistoryMap(int size) {

tests/robotests/src/com/android/settings/fuelgauge/BatteryDiffEntryTest.java

@@ -30,6 +30,8 @@ import android.os.SystemBatteryConsumer;
 import android.os.UserManager;
 import android.os.UserHandle;
 
+import com.android.settings.R;
+
 import org.junit.Before;
 import org.junit.Test;
 import org.junit.runner.RunWith;
@@ -118,6 +120,7 @@ public final class BatteryDiffEntryTest {
         final BatteryDiffEntry entry = createBatteryDiffEntry(10, batteryHistEntry);
 
         assertThat(entry.getAppLabel()).isEqualTo("Ambient display");
+        assertThat(entry.getAppIconId()).isEqualTo(R.drawable.ic_settings_aod);
         assertThat(BatteryDiffEntry.sResourceCache).isEmpty();
     }
 
@@ -134,6 +137,7 @@ public final class BatteryDiffEntryTest {
 
         assertThat(entry.getAppLabel()).isEqualTo("Removed user");
         assertThat(entry.getAppIcon()).isNull();
+        assertThat(entry.getAppIconId()).isEqualTo(0);
         assertThat(BatteryDiffEntry.sResourceCache).isEmpty();
     }
 
@@ -154,6 +158,7 @@ public final class BatteryDiffEntryTest {
         final BatteryDiffEntry entry = createBatteryDiffEntry(10, batteryHistEntry);
 
         assertThat(entry.getAppLabel()).isEqualTo(expectedAppLabel);
+        assertThat(entry.getAppIconId()).isEqualTo(0);
         assertThat(BatteryDiffEntry.sResourceCache).hasSize(1);
         // Verifies the app label in the cache.
         final BatteryEntry.NameAndIcon nameAndIcon =

tests/robotests/src/com/android/settings/fuelgauge/BatteryHistEntryTest.java

@@ -167,6 +167,30 @@ public final class BatteryHistEntryTest {
         assertThat(batteryHistEntry.getKey()).isEqualTo("S|1");
     }
 
+    @Test
+    public void testIsAppEntry_returnExpectedResult() {
+        assertThat(createEntry(ConvertUtils.CONSUMER_TYPE_SYSTEM_BATTERY).isAppEntry())
+            .isFalse();
+        assertThat(createEntry(ConvertUtils.CONSUMER_TYPE_USER_BATTERY).isAppEntry())
+            .isFalse();
+        assertThat(createEntry(ConvertUtils.CONSUMER_TYPE_UID_BATTERY).isAppEntry())
+            .isTrue();
+    }
+
+    @Test
+    public void testIsUserEntry_returnExpectedResult() {
+        assertThat(createEntry(ConvertUtils.CONSUMER_TYPE_SYSTEM_BATTERY).isUserEntry())
+            .isFalse();
+        assertThat(createEntry(ConvertUtils.CONSUMER_TYPE_USER_BATTERY).isUserEntry())
+            .isTrue();
+        assertThat(createEntry(ConvertUtils.CONSUMER_TYPE_UID_BATTERY).isUserEntry())
+            .isFalse();
+    }
+
+    private static BatteryHistEntry createEntry(int consumerType) {
+        return new BatteryHistEntry(getContentValuesWithType(consumerType));
+    }
+
     private static ContentValues getContentValuesWithType(int consumerType) {
         final ContentValues values = new ContentValues();
         values.put("consumerType", Integer.valueOf(consumerType));