Another attempt to fix espresso on Travis

Author: Josh Deffibaugh

build.gradle

@@ -83,5 +83,4 @@ task testAllModules << {
 
 testAllModules.dependsOn(':android-sdk:connectedAndroidTest', ':android-sdk:test',
         ':event-handler:connectedAndroidTest', ':event-handler:test',
-        ':user-experiment-record:connectedAndroidTest', ':shared:connectedAndroidTest',
-        ':test-app:connectedAndroidTest')
+        ':user-experiment-record:connectedAndroidTest', ':shared:connectedAndroidTest')

test-app/build.gradle

@@ -9,7 +9,7 @@ android {
         targetSdkVersion target_sdk_version
         versionCode 1
         versionName "1.0"
-        testInstrumentationRunner "com.optimizely.ab.android.test_app.TestRunner"
+        testInstrumentationRunner "android.support.test.runner.AndroidJUnitRunner"
     }
     testOptions {
         unitTests.returnDefaultValues = true

test-app/src/androidTest/java/com/optimizely/ab/android/test_app/MainActivityEspressoTest.java

@@ -25,6 +25,7 @@
 import android.support.test.rule.ActivityTestRule;
 import android.support.test.runner.AndroidJUnit4;
 import android.util.Pair;
+import android.view.WindowManager;
 
 import com.optimizely.ab.android.event_handler.EventIntentService;
 import com.optimizely.ab.android.sdk.DataFileService;
@@ -122,7 +123,26 @@ protected void after() {
                     CountingIdlingResourceManager.clearEvents();
                 }
             })
-            .around(activityTestRule);
+            .around(activityTestRule)
+            .around(new ExternalResource() {
+                @Override
+                protected void before() throws Throwable {
+                    super.before();
+                    Runnable wakeUpDevice = new Runnable() {
+                        public void run() {
+                            activityTestRule.getActivity().getWindow().addFlags(WindowManager.LayoutParams.FLAG_TURN_SCREEN_ON |
+                                    WindowManager.LayoutParams.FLAG_SHOW_WHEN_LOCKED |
+                                    WindowManager.LayoutParams.FLAG_KEEP_SCREEN_ON);
+                        }
+                    };
+                    activityTestRule.getActivity().runOnUiThread(wakeUpDevice);
+                }
+
+                @Override
+                protected void after() {
+                    super.after();
+                }
+            });
 
     @Test
     public void experimentActivationForWhitelistUser() throws InterruptedException {