Merge "Add unit test for BatteryUsageStats"

2 files changed, 160 insertions, 0 deletions

diff --git a/core/tests/coretests/src/com/android/internal/os/BatteryStatsTests.java b/core/tests/coretests/src/com/android/internal/os/BatteryStatsTests.java
index 67f4c8a16da6..bd4154210cbf 100644
--- a/core/tests/coretests/src/com/android/internal/os/BatteryStatsTests.java
+++ b/core/tests/coretests/src/com/android/internal/os/BatteryStatsTests.java
@@ -38,6 +38,7 @@ import org.junit.runners.Suite;
         BatteryStatsTimeBaseTest.class,
         BatteryStatsTimerTest.class,
         BatteryStatsUidTest.class,
+        BatteryUsageStatsTest.class,
         BatteryStatsUserLifecycleTests.class,
         BluetoothPowerCalculatorTest.class,
         BstatsCpuTimesValidationTest.class,
diff --git a/core/tests/coretests/src/com/android/internal/os/BatteryUsageStatsTest.java b/core/tests/coretests/src/com/android/internal/os/BatteryUsageStatsTest.java
new file mode 100644
index 000000000000..96e9c4af21c2
--- /dev/null
+++ b/core/tests/coretests/src/com/android/internal/os/BatteryUsageStatsTest.java
@@ -0,0 +1,159 @@
+/*
+ * Copyright (C) 2020 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.internal.os;
+
+import static com.google.common.truth.Truth.assertThat;
+
+import static org.junit.Assert.fail;
+
+import android.os.BatteryConsumer;
+import android.os.BatteryUsageStats;
+import android.os.Parcel;
+import android.os.SystemBatteryConsumer;
+import android.os.UidBatteryConsumer;
+
+import androidx.test.filters.SmallTest;
+import androidx.test.runner.AndroidJUnit4;
+
+import org.junit.Test;
+import org.junit.runner.RunWith;
+
+import java.util.List;
+
+@SmallTest
+@RunWith(AndroidJUnit4.class)
+public class BatteryUsageStatsTest {
+
+    @Test
+    public void testBuilder() {
+        BatteryUsageStats batteryUsageStats = buildBatteryUsageStats();
+        validateBatteryUsageStats(batteryUsageStats);
+    }
+
+    @Test
+    public void testParcelability() {
+        final BatteryUsageStats outBatteryUsageStats = buildBatteryUsageStats();
+        final Parcel outParcel = Parcel.obtain();
+        outParcel.writeParcelable(outBatteryUsageStats, 0);
+        final byte[] bytes = outParcel.marshall();
+        outParcel.recycle();
+
+        final Parcel inParcel = Parcel.obtain();
+        inParcel.unmarshall(bytes, 0, bytes.length);
+        inParcel.setDataPosition(0);
+        final BatteryUsageStats inBatteryUsageStats =
+                inParcel.readParcelable(getClass().getClassLoader());
+        assertThat(inBatteryUsageStats).isNotNull();
+        validateBatteryUsageStats(inBatteryUsageStats);
+    }
+
+    private BatteryUsageStats buildBatteryUsageStats() {
+        final MockClocks clocks = new MockClocks();
+        final MockBatteryStatsImpl batteryStats = new MockBatteryStatsImpl(clocks);
+        final BatteryStatsImpl.Uid batteryStatsUid = batteryStats.getUidStatsLocked(2000);
+
+        final BatteryUsageStats.Builder builder = new BatteryUsageStats.Builder(1, 1, true);
+        builder.setConsumedPower(100);
+        builder.setDischargePercentage(20);
+
+        final UidBatteryConsumer.Builder uidBatteryConsumerBuilder =
+                builder.getOrCreateUidBatteryConsumerBuilder(batteryStatsUid);
+        uidBatteryConsumerBuilder.setPackageWithHighestDrain("foo");
+        uidBatteryConsumerBuilder.setConsumedPower(200);
+        uidBatteryConsumerBuilder.setConsumedPower(BatteryConsumer.POWER_COMPONENT_USAGE, 300);
+        uidBatteryConsumerBuilder.setConsumedPower(BatteryConsumer.POWER_COMPONENT_CPU, 400);
+        uidBatteryConsumerBuilder.setConsumedPowerForCustomComponent(
+                BatteryConsumer.FIRST_CUSTOM_POWER_COMPONENT_ID, 500);
+        uidBatteryConsumerBuilder.setConsumedPowerForCustomComponent(
+                BatteryConsumer.FIRST_MODELED_POWER_COMPONENT_ID
+                        + BatteryConsumer.POWER_COMPONENT_CPU, 510);
+        uidBatteryConsumerBuilder.setUsageDurationMillis(BatteryConsumer.TIME_COMPONENT_CPU, 600);
+        uidBatteryConsumerBuilder.setUsageDurationMillis(
+                BatteryConsumer.TIME_COMPONENT_CPU_FOREGROUND, 700);
+        uidBatteryConsumerBuilder.setUsageDurationForCustomComponentMillis(
+                BatteryConsumer.FIRST_CUSTOM_TIME_COMPONENT_ID, 800);
+
+        final SystemBatteryConsumer.Builder systemBatteryConsumerBuilder =
+                builder.getOrCreateSystemBatteryConsumerBuilder(
+                        SystemBatteryConsumer.DRAIN_TYPE_CAMERA);
+        systemBatteryConsumerBuilder.setConsumedPower(10000);
+        systemBatteryConsumerBuilder.setConsumedPower(BatteryConsumer.POWER_COMPONENT_CPU, 10100);
+        systemBatteryConsumerBuilder.setConsumedPowerForCustomComponent(
+                BatteryConsumer.FIRST_CUSTOM_POWER_COMPONENT_ID, 10200);
+        systemBatteryConsumerBuilder.setConsumedPowerForCustomComponent(
+                BatteryConsumer.FIRST_MODELED_POWER_COMPONENT_ID
+                        + BatteryConsumer.POWER_COMPONENT_CPU, 10210);
+        systemBatteryConsumerBuilder.setUsageDurationMillis(
+                BatteryConsumer.TIME_COMPONENT_CPU, 10300);
+        systemBatteryConsumerBuilder.setUsageDurationForCustomComponentMillis(
+                BatteryConsumer.FIRST_CUSTOM_TIME_COMPONENT_ID, 10400);
+
+        return builder.build();
+    }
+
+    public void validateBatteryUsageStats(BatteryUsageStats batteryUsageStats) {
+        assertThat(batteryUsageStats.getConsumedPower()).isEqualTo(100);
+        assertThat(batteryUsageStats.getDischargePercentage()).isEqualTo(20);
+
+        final List<UidBatteryConsumer> uidBatteryConsumers =
+                batteryUsageStats.getUidBatteryConsumers();
+        for (UidBatteryConsumer uidBatteryConsumer : uidBatteryConsumers) {
+            if (uidBatteryConsumer.getUid() == 2000) {
+                assertThat(uidBatteryConsumer.getPackageWithHighestDrain()).isEqualTo("foo");
+                assertThat(uidBatteryConsumer.getConsumedPower()).isEqualTo(200);
+                assertThat(uidBatteryConsumer.getConsumedPower(
+                        BatteryConsumer.POWER_COMPONENT_USAGE)).isEqualTo(300);
+                assertThat(uidBatteryConsumer.getConsumedPower(
+                        BatteryConsumer.POWER_COMPONENT_CPU)).isEqualTo(400);
+                assertThat(uidBatteryConsumer.getConsumedPowerForCustomComponent(
+                        BatteryConsumer.FIRST_CUSTOM_POWER_COMPONENT_ID)).isEqualTo(500);
+                assertThat(uidBatteryConsumer.getConsumedPowerForCustomComponent(
+                        BatteryConsumer.FIRST_MODELED_POWER_COMPONENT_ID
+                                + BatteryConsumer.POWER_COMPONENT_CPU)).isEqualTo(510);
+                assertThat(uidBatteryConsumer.getUsageDurationMillis(
+                        BatteryConsumer.TIME_COMPONENT_CPU)).isEqualTo(600);
+                assertThat(uidBatteryConsumer.getUsageDurationMillis(
+                        BatteryConsumer.TIME_COMPONENT_CPU_FOREGROUND)).isEqualTo(700);
+                assertThat(uidBatteryConsumer.getUsageDurationForCustomComponentMillis(
+                        BatteryConsumer.FIRST_CUSTOM_TIME_COMPONENT_ID)).isEqualTo(800);
+            } else {
+                fail("Unexpected UID " + uidBatteryConsumer.getUid());
+            }
+        }
+
+        final List<SystemBatteryConsumer> systemBatteryConsumers =
+                batteryUsageStats.getSystemBatteryConsumers();
+        for (SystemBatteryConsumer systemBatteryConsumer : systemBatteryConsumers) {
+            if (systemBatteryConsumer.getDrainType() == SystemBatteryConsumer.DRAIN_TYPE_CAMERA) {
+                assertThat(systemBatteryConsumer.getConsumedPower()).isEqualTo(10000);
+                assertThat(systemBatteryConsumer.getConsumedPower(
+                        BatteryConsumer.POWER_COMPONENT_CPU)).isEqualTo(10100);
+                assertThat(systemBatteryConsumer.getConsumedPowerForCustomComponent(
+                        BatteryConsumer.FIRST_CUSTOM_POWER_COMPONENT_ID)).isEqualTo(10200);
+                assertThat(systemBatteryConsumer.getConsumedPowerForCustomComponent(
+                        BatteryConsumer.FIRST_MODELED_POWER_COMPONENT_ID
+                                + BatteryConsumer.POWER_COMPONENT_CPU)).isEqualTo(10210);
+                assertThat(systemBatteryConsumer.getUsageDurationMillis(
+                        BatteryConsumer.TIME_COMPONENT_CPU)).isEqualTo(10300);
+                assertThat(systemBatteryConsumer.getUsageDurationForCustomComponentMillis(
+                        BatteryConsumer.FIRST_CUSTOM_TIME_COMPONENT_ID)).isEqualTo(10400);
+            } else {
+                fail("Unexpected drain type " + systemBatteryConsumer.getDrainType());
+            }
+        }
+    }
+}