Add option to turn screen off for sensor tests

Change-Id: I9b3ae318a45eccc7beb0e4d29ba94a58fcada137

1 files changed, 6 insertions, 1 deletions

diff --git a/prod-tests/src/com/android/sensor/tests/SingleSensorTests.java b/prod-tests/src/com/android/sensor/tests/SingleSensorTests.java
index 88bbb0bfb..c6389dc7b 100644
--- a/prod-tests/src/com/android/sensor/tests/SingleSensorTests.java
+++ b/prod-tests/src/com/android/sensor/tests/SingleSensorTests.java
@@ -41,6 +41,9 @@ public class SingleSensorTests extends InstrumentationTest {
             + "files will be pulled, uploaded, and deleted from the device.")
     private String mOutputPrefix = "single_sensor_";
 
+    @Option(name = "screen-off", description = "Whether to run the tests with the screen off")
+    private boolean mScreenOff = false;
+
     /**
      * Run the instrumentation tests, pull the test generated files, and clean up.
      */
@@ -49,7 +52,9 @@ public class SingleSensorTests extends InstrumentationTest {
         Assert.assertNotNull(getDevice());
 
         try {
-            turnScreenOff();
+            if (mScreenOff) {
+                turnScreenOff();
+            }
             super.run(listener);
             pullFiles(listener);
         } finally {