| Index: chrome/android/javatests/src/org/chromium/chrome/browser/partnercustomizations/PartnerDisableIncognitoModeIntegrationTest.java
|
| diff --git a/chrome/android/javatests/src/org/chromium/chrome/browser/partnercustomizations/PartnerDisableIncognitoModeIntegrationTest.java b/chrome/android/javatests/src/org/chromium/chrome/browser/partnercustomizations/PartnerDisableIncognitoModeIntegrationTest.java
|
| index 26d6ff2fcf2ab2031cfae007977dcd6816a27d0b..cebba5892946d093aad447f6d67df930d02cebc4 100644
|
| --- a/chrome/android/javatests/src/org/chromium/chrome/browser/partnercustomizations/PartnerDisableIncognitoModeIntegrationTest.java
|
| +++ b/chrome/android/javatests/src/org/chromium/chrome/browser/partnercustomizations/PartnerDisableIncognitoModeIntegrationTest.java
|
| @@ -70,7 +70,7 @@ public class PartnerDisableIncognitoModeIntegrationTest extends
|
|
|
| private void waitForParentalControlsEnabledState(final boolean parentalControlsEnabled)
|
| throws InterruptedException {
|
| - CriteriaHelper.pollForUIThreadCriteria(new Criteria() {
|
| + CriteriaHelper.pollUiThread(new Criteria() {
|
| @Override
|
| public boolean isSatisfied() {
|
| // areParentalControlsEnabled is updated on a background thread, so we
|
| @@ -162,7 +162,7 @@ public class PartnerDisableIncognitoModeIntegrationTest extends
|
| toggleActivityForegroundState();
|
| waitForParentalControlsEnabledState(true);
|
|
|
| - CriteriaHelper.pollForCriteria(Criteria.equals(0, new Callable<Integer>() {
|
| + CriteriaHelper.pollInstrumentationThread(Criteria.equals(0, new Callable<Integer>() {
|
| @Override
|
| public Integer call() {
|
| return incognitoTabsCount();
|
|
|