Index: chrome/android/javatests/src/org/chromium/chrome/browser/prerender/PrerenderServiceTest.java |
diff --git a/chrome/android/javatests/src/org/chromium/chrome/browser/prerender/PrerenderServiceTest.java b/chrome/android/javatests/src/org/chromium/chrome/browser/prerender/PrerenderServiceTest.java |
index d19f4d02e674d413115bffec4ee9f08d6233fe90..3bc16c2d23feb5f3e37cb6a88df14fe1c011e460 100644 |
--- a/chrome/android/javatests/src/org/chromium/chrome/browser/prerender/PrerenderServiceTest.java |
+++ b/chrome/android/javatests/src/org/chromium/chrome/browser/prerender/PrerenderServiceTest.java |
@@ -160,7 +160,7 @@ public class PrerenderServiceTest extends |
} |
}); |
// TODO(yusufo): We should be using the NotificationCenter for checking the page loading. |
- CriteriaHelper.pollForCriteria(new Criteria() { |
+ CriteriaHelper.pollInstrumentationThread(new Criteria() { |
@Override |
public boolean isSatisfied() { |
return chromeActivity.getActivityTab() != null |
@@ -177,7 +177,7 @@ public class PrerenderServiceTest extends |
} |
private void assertServiceHasPrerenderedUrl(final String url) throws InterruptedException { |
- CriteriaHelper.pollForUIThreadCriteria(new Criteria() { |
+ CriteriaHelper.pollUiThread(new Criteria() { |
@Override |
public boolean isSatisfied() { |
return WarmupManager.getInstance().hasPrerenderedUrl(url); |