Addition of new batterystats options

Adding clear battery stats and enabling full battery stats options
to device setup

Change-Id: Ib65abfd49c98dd90a29f6ab8d941c608e3c0e52a

1 files changed, 17 insertions, 0 deletions

diff --git a/src/com/android/tradefed/targetprep/DeviceSetup.java b/src/com/android/tradefed/targetprep/DeviceSetup.java
index ead692bda..2475846f4 100644
--- a/src/com/android/tradefed/targetprep/DeviceSetup.java
+++ b/src/com/android/tradefed/targetprep/DeviceSetup.java
@@ -190,6 +190,12 @@ public class DeviceSetup implements ITargetPreparer, ITargetCleaner {
     protected Integer mBatterySaverTrigger = null;
     // settings put global low_power_trigger_level $N
 
+    @Option(name = "enable-full-battery-stats-history",
+            description = "Enable full history for batterystats. This option is only " +
+            "applicable for L+")
+    protected boolean mEnableFullBatteryStatsHistory = false;
+    // dumpsys batterystats --enable full-history
+
     @Option(name = "disable-doze",
             description = "Disable device from going into doze mode. This option is only " +
             "applicable for M+")
@@ -539,6 +545,10 @@ public class DeviceSetup implements ITargetPreparer, ITargetCleaner {
             mGlobalSettings.put("low_power_trigger_level", Integer.toString(mBatterySaverTrigger));
         }
 
+        if (mEnableFullBatteryStatsHistory) {
+            mRunCommandAfterSettings.add("dumpsys batterystats --enable full-history");
+        }
+
         if (mDisableDoze) {
             mRunCommandAfterSettings.add("dumpsys deviceidle disable");
         }
@@ -1029,6 +1039,13 @@ public class DeviceSetup implements ITargetPreparer, ITargetCleaner {
     /**
      * Exposed for unit testing
      */
+    protected void setEnableFullBatteryStatsHistory(boolean enableFullBatteryStatsHistory) {
+        mEnableFullBatteryStatsHistory = enableFullBatteryStatsHistory;
+    }
+
+    /**
+     * Exposed for unit testing
+     */
     protected void setDisableDoze(boolean disableDoze) {
         mDisableDoze = disableDoze;
     }