Added wakelock to keep the screen from sleeping when executing tests.

Change-Id: Ib6a0ff2a7c906f500868269d6a617553733d33b1

1 files changed, 11 insertions, 0 deletions

diff --git a/tests/DumpRenderTree2/src/com/android/dumprendertree2/LayoutTestsExecutor.java b/tests/DumpRenderTree2/src/com/android/dumprendertree2/LayoutTestsExecutor.java
index 73026bc..fe00a92 100644
--- a/tests/DumpRenderTree2/src/com/android/dumprendertree2/LayoutTestsExecutor.java
+++ b/tests/DumpRenderTree2/src/com/android/dumprendertree2/LayoutTestsExecutor.java
@@ -28,7 +28,9 @@ import android.os.Handler;
 import android.os.IBinder;
 import android.os.Message;
 import android.os.Messenger;
+import android.os.PowerManager;
 import android.os.RemoteException;
+import android.os.PowerManager.WakeLock;
 import android.util.Log;
 import android.view.Window;
 import android.webkit.ConsoleMessage;
@@ -103,6 +105,8 @@ public class LayoutTestsExecutor extends Activity {
     private boolean mCanOpenWindows;
     private boolean mDumpDatabaseCallbacks;
 
+    private WakeLock mScreenDimLock;
+
     /** COMMUNICATION WITH ManagerService */
 
     private Messenger mManagerServiceMessenger;
@@ -239,6 +243,11 @@ public class LayoutTestsExecutor extends Activity {
         mCurrentTestIndex = intent.getIntExtra(EXTRA_TEST_INDEX, -1);
         mTotalTestCount = mCurrentTestIndex + mTestsList.size();
 
+        PowerManager pm = (PowerManager) getSystemService(Context.POWER_SERVICE);
+        mScreenDimLock = pm.newWakeLock(PowerManager.SCREEN_DIM_WAKE_LOCK
+                | PowerManager.ON_AFTER_RELEASE, "WakeLock in LayoutTester");
+        mScreenDimLock.acquire();
+
         bindService(new Intent(this, ManagerService.class), mServiceConnection,
                 Context.BIND_AUTO_CREATE);
     }
@@ -390,6 +399,8 @@ public class LayoutTestsExecutor extends Activity {
     }
 
     private void onAllTestsFinished() {
+        mScreenDimLock.release();
+
         try {
             Message serviceMsg =
                     Message.obtain(null, ManagerService.MSG_ALL_TESTS_FINISHED);