From f55e87c36bab24f8619b42f3425b3236a85c50b9 Mon Sep 17 00:00:00 2001
From: vadimt <vadimt@google.com>
Date: Mon, 6 May 2019 16:13:48 -0700
Subject: [PATCH] Fixing AppPredictionsUITests

Bug: 131772711
Bug: 131188880
Change-Id: Ic6556b26464766294ced7f85af716b0a4eb51ca1
---
 .../src/com/android/quickstep/AppPredictionsUITests.java     | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/quickstep/tests/src/com/android/quickstep/AppPredictionsUITests.java b/quickstep/tests/src/com/android/quickstep/AppPredictionsUITests.java
index 1817135504..ffe3633353 100644
--- a/quickstep/tests/src/com/android/quickstep/AppPredictionsUITests.java
+++ b/quickstep/tests/src/com/android/quickstep/AppPredictionsUITests.java
@@ -40,7 +40,6 @@ import com.android.launcher3.model.AppLaunchTracker;
 
 import org.junit.After;
 import org.junit.Before;
-import org.junit.Ignore;
 import org.junit.Test;
 import org.junit.runner.RunWith;
 
@@ -70,6 +69,8 @@ public class AppPredictionsUITests extends AbstractQuickStepTest {
         // Disable app tracker
         AppLaunchTracker.INSTANCE.initializeForTesting(new AppLaunchTracker());
 
+        PredictionUiStateManager.INSTANCE.initializeForTesting(null);
+
         mCallback = PredictionUiStateManager.INSTANCE.get(mTargetContext).appPredictorCallback(
                 Client.HOME);
 
@@ -85,7 +86,6 @@ public class AppPredictionsUITests extends AbstractQuickStepTest {
      * Test that prediction UI is updated as soon as we get predictions from the system
      */
     @Test
-    @Ignore // b/131772711: this really fails (when being run as a part of the whole test suite)!
     public void testPredictionExistsInAllApps() {
         mActivityMonitor.startLauncher();
         mLauncher.pressHome().switchToAllApps();
@@ -118,7 +118,6 @@ public class AppPredictionsUITests extends AbstractQuickStepTest {
     }
 
     @Test
-    @Ignore // b/131772711 - this was failing in the lab
     public void testPredictionsDisabled() {
         mActivityMonitor.startLauncher();
         sendPredictionUpdate();