Index: chrome/android/javatests/src/org/chromium/chrome/browser/customtabs/CustomTabsConnectionTest.java |
diff --git a/chrome/android/javatests/src/org/chromium/chrome/browser/customtabs/CustomTabsConnectionTest.java b/chrome/android/javatests/src/org/chromium/chrome/browser/customtabs/CustomTabsConnectionTest.java |
index 503975dd7ccf817c29833d722ea1828f4d04231a..c14c9a96c685e2f7eaa44133ab357332420fdf9e 100644 |
--- a/chrome/android/javatests/src/org/chromium/chrome/browser/customtabs/CustomTabsConnectionTest.java |
+++ b/chrome/android/javatests/src/org/chromium/chrome/browser/customtabs/CustomTabsConnectionTest.java |
@@ -23,6 +23,7 @@ import org.chromium.base.ThreadUtils; |
import org.chromium.base.library_loader.LibraryLoader; |
import org.chromium.base.library_loader.LibraryProcessType; |
import org.chromium.base.test.util.Restriction; |
+import org.chromium.chrome.browser.WarmupManager; |
import org.chromium.chrome.browser.preferences.PrefServiceBridge; |
import org.chromium.content_public.browser.WebContents; |
@@ -57,6 +58,12 @@ public class CustomTabsConnectionTest extends InstrumentationTestCase { |
protected void tearDown() throws Exception { |
super.tearDown(); |
CustomTabsTestUtils.cleanupSessions(mCustomTabsConnection); |
+ ThreadUtils.runOnUiThreadBlocking(new Runnable() { |
+ @Override |
+ public void run() { |
+ WarmupManager.getInstance().destroySpareWebContents(); |
+ } |
+ }); |
} |
/** |
@@ -102,8 +109,12 @@ public class CustomTabsConnectionTest extends InstrumentationTestCase { |
ThreadUtils.runOnUiThreadBlocking(new Runnable() { |
@Override |
public void run() { |
- assertNotNull(mCustomTabsConnection.takeSpareWebContents()); |
- assertNull(mCustomTabsConnection.takeSpareWebContents()); |
+ WarmupManager warmupManager = WarmupManager.getInstance(); |
+ assertTrue(warmupManager.hasSpareWebContents()); |
+ WebContents webContents = warmupManager.takeSpareWebContents(false, false); |
+ assertNotNull(webContents); |
+ assertFalse(warmupManager.hasSpareWebContents()); |
+ webContents.destroy(); |
} |
}); |
} |
@@ -116,7 +127,7 @@ public class CustomTabsConnectionTest extends InstrumentationTestCase { |
@Override |
public void run() { |
assertSpareWebContentsNotNullAndDestroy(); |
- assertNull(mCustomTabsConnection.takeSpareWebContents()); |
+ assertFalse(WarmupManager.getInstance().hasSpareWebContents()); |
} |
}); |
assertTrue(mCustomTabsConnection.warmup(0)); |
@@ -135,7 +146,7 @@ public class CustomTabsConnectionTest extends InstrumentationTestCase { |
ThreadUtils.runOnUiThreadBlocking(new Runnable() { |
@Override |
public void run() { |
- assertNull(mCustomTabsConnection.takeSpareWebContents()); |
+ assertFalse(WarmupManager.getInstance().hasSpareWebContents()); |
String referrer = |
mCustomTabsConnection.getReferrerForSession(token).getUrl(); |
WebContents webContents = |
@@ -284,7 +295,7 @@ public class CustomTabsConnectionTest extends InstrumentationTestCase { |
ThreadUtils.runOnUiThreadBlocking(new Runnable() { |
@Override |
public void run() { |
- assertNull(mCustomTabsConnection.takeSpareWebContents()); |
+ assertNull(WarmupManager.getInstance().takeSpareWebContents(false, false)); |
String referrer = mCustomTabsConnection.getReferrerForSession(token).getUrl(); |
assertNotNull(mCustomTabsConnection.takePrerenderedUrl(token, URL, referrer)); |
} |
@@ -298,7 +309,7 @@ public class CustomTabsConnectionTest extends InstrumentationTestCase { |
ThreadUtils.runOnUiThreadBlocking(new Runnable() { |
@Override |
public void run() { |
- assertNull(mCustomTabsConnection.takeSpareWebContents()); |
+ assertNull(WarmupManager.getInstance().takeSpareWebContents(false, false)); |
} |
}); |
assertTrue(mCustomTabsConnection.mayLaunchUrl(token, null, null, null)); |
@@ -313,7 +324,7 @@ public class CustomTabsConnectionTest extends InstrumentationTestCase { |
} |
private void assertSpareWebContentsNotNullAndDestroy() { |
- WebContents webContents = mCustomTabsConnection.takeSpareWebContents(); |
+ WebContents webContents = WarmupManager.getInstance().takeSpareWebContents(false, false); |
assertNotNull(webContents); |
webContents.destroy(); |
} |
@@ -517,7 +528,7 @@ public class CustomTabsConnectionTest extends InstrumentationTestCase { |
ThreadUtils.runOnUiThreadBlocking(new Runnable() { |
@Override |
public void run() { |
- assertNull(mCustomTabsConnection.takeSpareWebContents()); |
+ assertNull(WarmupManager.getInstance().takeSpareWebContents(false, false)); |
String referrer = mCustomTabsConnection.getReferrerForSession(token).getUrl(); |
WebContents prerender = mCustomTabsConnection.takePrerenderedUrl( |
token, URL, referrer); |