Index: android_webview/javatests/src/org/chromium/android_webview/test/SafeBrowsingTest.java |
diff --git a/android_webview/javatests/src/org/chromium/android_webview/test/SafeBrowsingTest.java b/android_webview/javatests/src/org/chromium/android_webview/test/SafeBrowsingTest.java |
index 4672eaa302b1ddd750368b2375c225bdb5e2cad8..d2fdde79c20f9bf6272c7825b8fd57a04ceb8fe6 100644 |
--- a/android_webview/javatests/src/org/chromium/android_webview/test/SafeBrowsingTest.java |
+++ b/android_webview/javatests/src/org/chromium/android_webview/test/SafeBrowsingTest.java |
@@ -426,13 +426,18 @@ public class SafeBrowsingTest extends AwTestBase { |
@CommandLineFlags.Add(AwSwitches.WEBVIEW_ENABLE_SAFEBROWSING_SUPPORT) |
public void testSafeBrowsingShowsNetworkErrorForOddSizedViews() throws Throwable { |
mAwContents.setCanShowBigInterstitial(false); |
+ loadGreenPage(); |
+ int count = mWebContentsObserver.getAttachedInterstitialPageHelper().getCallCount(); |
final String responseUrl = mTestServer.getURL(MALWARE_HTML_PATH); |
- OnReceivedError2Helper errorHelper = mContentsClient.getOnReceivedError2Helper(); |
- int errorCount = errorHelper.getCallCount(); |
loadUrlAsync(mAwContents, responseUrl); |
- errorHelper.waitForCallback(errorCount); |
- assertEquals(ErrorCodeConversionHelper.ERROR_UNKNOWN, errorHelper.getError().errorCode); |
- assertEquals("Network error is for the malicious page", responseUrl, |
- errorHelper.getRequest().url); |
+ mWebContentsObserver.getAttachedInterstitialPageHelper().waitForCallback(count); |
+ assertTrue("Original page should not be showing", |
+ GREEN_PAGE_BACKGROUND_COLOR |
+ != GraphicsTestUtils.getPixelColorAtCenterOfView( |
+ mAwContents, mContainerView)); |
+ assertTrue("Target page should not be visible", |
+ MALWARE_PAGE_BACKGROUND_COLOR |
+ != GraphicsTestUtils.getPixelColorAtCenterOfView( |
+ mAwContents, mContainerView)); |
} |
} |