diff options
| -rw-r--r-- | chrome/browser/safe_browsing/safe_browsing_service_browsertest.cc | 11 |
1 files changed, 8 insertions, 3 deletions
diff --git a/chrome/browser/safe_browsing/safe_browsing_service_browsertest.cc b/chrome/browser/safe_browsing/safe_browsing_service_browsertest.cc index b734071..6c49d54 100644 --- a/chrome/browser/safe_browsing/safe_browsing_service_browsertest.cc +++ b/chrome/browser/safe_browsing/safe_browsing_service_browsertest.cc @@ -11,6 +11,7 @@ #include "base/metrics/histogram.h" #include "crypto/sha2.h" #include "chrome/browser/browser_process.h" +#include "chrome/browser/prerender/prerender_manager.h" #include "chrome/browser/safe_browsing/safe_browsing_database.h" #include "chrome/browser/safe_browsing/safe_browsing_service.h" #include "chrome/browser/safe_browsing/safe_browsing_util.h" @@ -394,23 +395,27 @@ const char kPrefetchMalwarePage[] = "files/safe_browsing/prefetch_malware.html"; // This test confirms that prefetches don't themselves get the // interstitial treatment. -// FLAKY: http://crbug.com/80719 -IN_PROC_BROWSER_TEST_F(SafeBrowsingServiceTest, FLAKY_Prefetch) { +IN_PROC_BROWSER_TEST_F(SafeBrowsingServiceTest, Prefetch) { GURL url = test_server()->GetURL(kPrefetchMalwarePage); GURL malware_url = test_server()->GetURL(kMalwarePage); class SetPrefetchForTest { public: explicit SetPrefetchForTest(bool prefetch) - : old_prefetch_state_(ResourceDispatcherHost::is_prefetch_enabled()) { + : old_prefetch_state_(ResourceDispatcherHost::is_prefetch_enabled()), + old_prerender_mode_(prerender::PrerenderManager::GetMode()) { ResourceDispatcherHost::set_is_prefetch_enabled(prefetch); + prerender::PrerenderManager::SetMode( + prerender::PrerenderManager::PRERENDER_MODE_DISABLED); } ~SetPrefetchForTest() { ResourceDispatcherHost::set_is_prefetch_enabled(old_prefetch_state_); + prerender::PrerenderManager::SetMode(old_prerender_mode_); } private: bool old_prefetch_state_; + prerender::PrerenderManager::PrerenderManagerMode old_prerender_mode_; } set_prefetch_for_test(true); // Even though we have added this uri to the safebrowsing database and |