Add an option to disable Doze mode

Change-Id: I75f0f774153c5b9f8b5a03db93e869e1014c8344

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 ae27b0e8b..424e64c55 100644
--- a/src/com/android/tradefed/targetprep/DeviceSetup.java
+++ b/src/com/android/tradefed/targetprep/DeviceSetup.java
@@ -192,6 +192,12 @@ public class DeviceSetup implements ITargetPreparer, ITargetCleaner {
     protected Integer mBatterySaverTrigger = null;
     // settings put global low_power_trigger_level $N
 
+    @Option(name = "disable-doze",
+            description = "Disable device from going into doze mode. This option is only " +
+            "applicable for M+")
+    protected boolean mDisableDoze = false;
+    // dumpsys deviceidle disable
+
     // Time
     @Option(name = "auto-update-time",
             description = "Turn auto update time on or off")
@@ -523,6 +529,10 @@ public class DeviceSetup implements ITargetPreparer, ITargetCleaner {
             mGlobalSettings.put("low_power_trigger_level", Integer.toString(mBatterySaverTrigger));
         }
 
+        if (mDisableDoze) {
+            mRunCommandAfterSettings.add("dumpsys deviceidle disable");
+        }
+
         setSettingForBinaryState(mAutoUpdateTime, mSystemSettings, "auto_time", "1", "0");
 
         setSettingForBinaryState(mAutoUpdateTimezone, mSystemSettings, "auto_timezone", "1", "0");
@@ -969,6 +979,13 @@ public class DeviceSetup implements ITargetPreparer, ITargetCleaner {
     /**
      * Exposed for unit testing
      */
+    protected void setDisableDoze(boolean disableDoze) {
+        mDisableDoze = disableDoze;
+    }
+
+    /**
+     * Exposed for unit testing
+     */
     protected void setAutoUpdateTime(BinaryState autoUpdateTime) {
         mAutoUpdateTime = autoUpdateTime;
     }