From ef7f9d2458d547df4b47808f19e549eee9af76ce Mon Sep 17 00:00:00 2001
From: Maksymilian Osowski <maxosowski@google.com>
Date: Tue, 3 Aug 2010 10:27:19 +0100
Subject: [PATCH] Added wakelock to keep the screen from sleeping when
 executing tests.

Change-Id: Ib6a0ff2a7c906f500868269d6a617553733d33b1
---
 tests/DumpRenderTree2/AndroidManifest.xml             |  1 +
 .../android/dumprendertree2/LayoutTestsExecutor.java  | 11 +++++++++++
 2 files changed, 12 insertions(+)

diff --git a/tests/DumpRenderTree2/AndroidManifest.xml b/tests/DumpRenderTree2/AndroidManifest.xml
index 1213674b6eb92..9f6097a960ac5 100644
--- a/tests/DumpRenderTree2/AndroidManifest.xml
+++ b/tests/DumpRenderTree2/AndroidManifest.xml
@@ -41,4 +41,5 @@ limitations under the License.
     <uses-permission android:name="android.permission.INTERNET" />
     <uses-permission android:name="android.permission.WRITE_SDCARD" />
     <uses-permission android:name="android.permission.WRITE_EXTERNAL_STORAGE" />
+    <uses-permission android:name="android.permission.WAKE_LOCK" />
 </manifest>
\ No newline at end of file
diff --git a/tests/DumpRenderTree2/src/com/android/dumprendertree2/LayoutTestsExecutor.java b/tests/DumpRenderTree2/src/com/android/dumprendertree2/LayoutTestsExecutor.java
index 73026bc95a8d7..fe00a926009dc 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);