PrerenderContents uses RESOURCE_RECEIVED_REDIRECT notification.

chrome/browser/prerender/prerender_browsertest.cc

 // Copyright (c) 2011 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "base/command_line.h"
 #include "base/path_service.h"
 #include "base/string_util.h"
 #include "chrome/browser/prerender/prerender_contents.h"
 #include "chrome/browser/prerender/prerender_manager.h"
 #include "chrome/browser/profiles/profile.h"
@@ skipping 19 matching lines @@
 // the page behaves as expected when prerendered.
 //
 // The prerendered page is then displayed on a tab.  The Javascript function
 // DidDisplayPass() is called, and returns true if the page behaved as it
 // should while being displayed.
 
 namespace prerender {
 
 namespace {
 
-bool CreateRedirect(const std::string& dest_url, std::string* redirect_path) {
-  std::vector<net::TestServer::StringPair> replacement_text;
-  replacement_text.push_back(make_pair("REPLACE_WITH_URL", dest_url));
-  return net::TestServer::GetFilePathWithReplacements(
-      "prerender_redirect.html",
-      replacement_text,
-      redirect_path);
+std::string CreateClientRedirect(const std::string& dest_url) {
+  const char* const kClientRedirectBase = "client-redirect?";
+  return kClientRedirectBase + dest_url;
+}
+
+std::string CreateServerRedirect(const std::string& dest_url) {
+  const char* const kServerRedirectBase = "server-redirect?";
+  return kServerRedirectBase + dest_url;
 }
 
 // PrerenderContents that stops the UI message loop on DidStopLoading().
 class TestPrerenderContents : public PrerenderContents {
  public:
   TestPrerenderContents(
       PrerenderManager* prerender_manager, Profile* profile, const GURL& url,
       const std::vector<GURL>& alias_urls,
       const GURL& referrer,
       int number_of_loads,
@@ skipping 116 matching lines @@
     // Check if page behaved as expected when actually displayed.
     bool display_test_result = false;
     ASSERT_TRUE(ui_test_utils::ExecuteJavaScriptAndExtractBool(
         browser()->GetSelectedTabContents()->render_view_host(), L"",
         L"window.domAutomationController.send(DidDisplayPass())",
         &display_test_result));
     EXPECT_TRUE(display_test_result);
   }
 
   bool UrlIsInPrerenderManager(const std::string& html_file) {
-    GURL dest_url = UrlForHtmlFile(html_file);
-
+    GURL dest_url = test_server()->GetURL(html_file);
     return (prerender_manager()->FindEntry(dest_url) != NULL);
   }
 
   bool UrlIsPendingInPrerenderManager(const std::string& html_file) {
-    GURL dest_url = UrlForHtmlFile(html_file);
-
+    GURL dest_url = test_server()->GetURL(html_file);
     return (prerender_manager()->FindPendingEntry(dest_url) != NULL);
   }
 
   void set_use_https_src(bool use_https_src_server) {
     use_https_src_server_ = use_https_src_server;
   }
 
   TaskManagerModel* model() const {
     return TaskManager::GetInstance()->model();
   }
 
+  void set_dest_url(const GURL& dest_url) { dest_url_ = dest_url; }
+
  private:
   void PrerenderTestURLImpl(
       const std::string& html_file,
       const std::deque<FinalStatus>& expected_final_status_queue,
       int total_navigations) {
     ASSERT_TRUE(test_server()->Start());
-    dest_url_ = UrlForHtmlFile(html_file);
+    dest_url_ = test_server()->GetURL(html_file);
 
     std::vector<net::TestServer::StringPair> replacement_text;
     replacement_text.push_back(
         make_pair("REPLACE_WITH_PREFETCH_URL", dest_url_.spec()));
     std::string replacement_path;
     ASSERT_TRUE(net::TestServer::GetFilePathWithReplacements(
         "files/prerender/prerender_loader.html",
         replacement_text,
         &replacement_path));
 
@@ skipping 52 matching lines @@
         break;
     }
   }
 
   PrerenderManager* prerender_manager() const {
     Profile* profile = browser()->GetSelectedTabContents()->profile();
     PrerenderManager* prerender_manager = profile->GetPrerenderManager();
     return prerender_manager;
   }
 
-  // Non-const as test_server()->GetURL() is not const
-  GURL UrlForHtmlFile(const std::string& html_file) {
-    std::string dest_path = "files/prerender/";
-    dest_path.append(html_file);
-    return test_server()->GetURL(dest_path);
-  }
-
   WaitForLoadPrerenderContentsFactory* prc_factory_;
   GURL dest_url_;
   bool use_https_src_server_;
 };
 
 // Checks that a page is correctly prerendered in the case of a
 // <link rel=prefetch> tag and then loaded into a tab in response to a
 // navigation.
 IN_PROC_BROWSER_TEST_F(PrerenderBrowserTest, PrerenderPage) {
-  PrerenderTestURL("prerender_page.html", FINAL_STATUS_USED, 1);
+  PrerenderTestURL("files/prerender/prerender_page.html", FINAL_STATUS_USED, 1);
   NavigateToDestURL();
 }
 
 // Checks that the prerendering of a page is canceled correctly when a
 // Javascript alert is called.
 IN_PROC_BROWSER_TEST_F(PrerenderBrowserTest, PrerenderAlertBeforeOnload) {
-  PrerenderTestURL("prerender_alert_before_onload.html",
-                   FINAL_STATUS_JAVASCRIPT_ALERT, 1);
+  PrerenderTestURL("files/prerender/prerender_alert_before_onload.html",
+                   FINAL_STATUS_JAVASCRIPT_ALERT,
+                   1);
 }
 
 // Checks that the prerendering of a page is canceled correctly when a
 // Javascript alert is called.
 IN_PROC_BROWSER_TEST_F(PrerenderBrowserTest, PrerenderAlertAfterOnload) {
-  PrerenderTestURL("prerender_alert_after_onload.html",
-                   FINAL_STATUS_JAVASCRIPT_ALERT, 1);
+  PrerenderTestURL("files/prerender/prerender_alert_after_onload.html",
+                   FINAL_STATUS_JAVASCRIPT_ALERT,
+                   1);
 }
 
 // Checks that plugins are not loaded while a page is being preloaded, but
 // are loaded when the page is displayed.
 IN_PROC_BROWSER_TEST_F(PrerenderBrowserTest, PrerenderDelayLoadPlugin) {
-  PrerenderTestURL("plugin_delay_load.html", FINAL_STATUS_USED, 1);
+  PrerenderTestURL("files/prerender/plugin_delay_load.html",
+                   FINAL_STATUS_USED,
+                   1);
   NavigateToDestURL();
 }
 
 // Checks that plugins in an iframe are not loaded while a page is
 // being preloaded, but are loaded when the page is displayed.
 IN_PROC_BROWSER_TEST_F(PrerenderBrowserTest, PrerenderIframeDelayLoadPlugin) {
-  PrerenderTestURL("prerender_iframe_plugin_delay_load.html",
-                   FINAL_STATUS_USED, 1);
+  PrerenderTestURL("files/prerender/prerender_iframe_plugin_delay_load.html",
+                   FINAL_STATUS_USED,
+                   1);
   NavigateToDestURL();
 }
 
 // Renders a page that contains a prerender link to a page that contains an
 // iframe with a source that requires http authentication. This should not
 // prerender successfully.
 IN_PROC_BROWSER_TEST_F(PrerenderBrowserTest, PrerenderHttpAuthentication) {
-  PrerenderTestURL("prerender_http_auth_container.html",
-                   FINAL_STATUS_AUTH_NEEDED, 1);
+  PrerenderTestURL("files/prerender/prerender_http_auth_container.html",
+                   FINAL_STATUS_AUTH_NEEDED,
+                   1);
 }
 
-// Checks that HTML redirects work with prerendering - specifically, checks the
-// page is used and plugins aren't loaded.
-IN_PROC_BROWSER_TEST_F(PrerenderBrowserTest, PrerenderRedirect) {
+// Checks that client-issued redirects work with prerendering - specifically,
+// checks the page is used and plugins aren't loaded. This version navigates to
+// the first page rather than the second page.
+IN_PROC_BROWSER_TEST_F(PrerenderBrowserTest,
+                       PrerenderClientRedirectNavigateToFirst) {
+  PrerenderTestURL(
+      CreateClientRedirect("files/prerender/prerender_page.html"),
+      FINAL_STATUS_USED,
+      2);
+  NavigateToDestURL();
+}
+
+// Checks that client-issued redirects work with prerendering - specifically,
+// checks the page is used and plugins aren't loaded. This version navigates to
+// the second page rather than the first page.
+IN_PROC_BROWSER_TEST_F(PrerenderBrowserTest,
+                       PrerenderClientRedirectNavigateToSecond) {
   std::string redirect_path;
-  ASSERT_TRUE(CreateRedirect("prerender_page.html", &redirect_path));
-  PrerenderTestURL(redirect_path, FINAL_STATUS_USED, 2);
+  PrerenderTestURL(
+      CreateClientRedirect("files/prerender/prerender_page.html"),
+      FINAL_STATUS_USED,
+      2);
+  set_dest_url(test_server()->GetURL("files/prerender/prerender_page.html"));
+  NavigateToDestURL();
+}
+// Checks that client-issued redirects work with prerendering - specifically,
+// checks the page is used and plugins aren't loaded. This version navigates to
+// the first page rather than the second page.
+IN_PROC_BROWSER_TEST_F(PrerenderBrowserTest,
+                       PrerenderServerRedirectNavigateToFirst) {
+  PrerenderTestURL(
+      CreateServerRedirect("files/prerender/prerender_page.html"),
+      FINAL_STATUS_USED,
+      1);
+  NavigateToDestURL();
+}
+
+// Checks that client-issued redirects work with prerendering - specifically,
+// checks the page is used and plugins aren't loaded. This version navigates to
+// the second page rather than the first page.
+IN_PROC_BROWSER_TEST_F(PrerenderBrowserTest,
+                       PrerenderServerRedirectNavigateToSecond) {
+  std::string redirect_path;
+  PrerenderTestURL(
+      CreateServerRedirect("files/prerender/prerender_page.html"),
+      FINAL_STATUS_USED,
+      1);
+  set_dest_url(test_server()->GetURL("files/prerender/prerender_page.html"));
   NavigateToDestURL();
 }
 
 // Prerenders a page that contains an automatic download triggered through an
 // iframe. This should not prerender successfully.
 IN_PROC_BROWSER_TEST_F(PrerenderBrowserTest, PrerenderDownloadIFrame) {
-  PrerenderTestURL("prerender_download_iframe.html", FINAL_STATUS_DOWNLOAD, 1);
+  PrerenderTestURL("files/prerender/prerender_download_iframe.html",
+                   FINAL_STATUS_DOWNLOAD,
+                   1);
 }
 
 // Prerenders a page that contains an automatic download triggered through
 // Javascript changing the window.location. This should not prerender
 // successfully.
 IN_PROC_BROWSER_TEST_F(PrerenderBrowserTest, PrerenderDownloadLocation) {
-  std::string redirect_path;
-  ASSERT_TRUE(CreateRedirect("../download-test1.lib", &redirect_path));
-  PrerenderTestURL(redirect_path, FINAL_STATUS_DOWNLOAD, 1);
+  PrerenderTestURL(CreateClientRedirect("files/download-test1.lib"),
+                   FINAL_STATUS_DOWNLOAD,
+                   1);
 }
 
 // Prerenders a page that contains an automatic download triggered through a
-// <meta http-equiv="refresh"> tag. This should not prerender successfully.
-IN_PROC_BROWSER_TEST_F(PrerenderBrowserTest, PrerenderDownloadRefresh) {
-  PrerenderTestURL("prerender_download_refresh.html", FINAL_STATUS_DOWNLOAD, 1);
+// client-issued redirect. This should not prerender successfully.
+IN_PROC_BROWSER_TEST_F(PrerenderBrowserTest, PrerenderDownloadClientRedirect) {
+  PrerenderTestURL("files/prerender/prerender_download_refresh.html",
+                   FINAL_STATUS_DOWNLOAD,
+                   1);
 }
 
 // Checks that the referrer is set when prerendering.
 IN_PROC_BROWSER_TEST_F(PrerenderBrowserTest, PrerenderReferrer) {
-  PrerenderTestURL("prerender_referrer.html", FINAL_STATUS_USED, 1);
+  PrerenderTestURL("files/prerender/prerender_referrer.html",
+                   FINAL_STATUS_USED,
+                   1);
   NavigateToDestURL();
 }
 
 // Checks that the referrer is not set when prerendering and the source page is
 // HTTPS.
 IN_PROC_BROWSER_TEST_F(PrerenderBrowserTest, PrerenderNoSSLReferrer) {
   set_use_https_src(true);
-  PrerenderTestURL("prerender_no_referrer.html", FINAL_STATUS_USED, 1);
+  PrerenderTestURL("files/prerender/prerender_no_referrer.html",
+                   FINAL_STATUS_USED,
+                   1);
   NavigateToDestURL();
 }
 
 // Checks that popups on a prerendered page cause cancellation.
 IN_PROC_BROWSER_TEST_F(PrerenderBrowserTest, PrerenderPopup) {
-  PrerenderTestURL("prerender_popup.html", FINAL_STATUS_CREATE_NEW_WINDOW, 1);
+  PrerenderTestURL("files/prerender/prerender_popup.html",
+                   FINAL_STATUS_CREATE_NEW_WINDOW,
+                   1);
 }
 
-// Test that page-based redirects to https will cancel prerenders.
-// Disabled, http://crbug.com/73580
-IN_PROC_BROWSER_TEST_F(PrerenderBrowserTest, PrerenderRedirectToHttps) {
+// Test that client redirects to https will cancel prerenders.
+IN_PROC_BROWSER_TEST_F(PrerenderBrowserTest, PrerenderClientRedirectToHttps) {
   net::TestServer https_server(net::TestServer::TYPE_HTTPS,
                                FilePath(FILE_PATH_LITERAL("chrome/test/data")));
   ASSERT_TRUE(https_server.Start());
   GURL https_url = https_server.GetURL("files/prerender/prerender_page.html");
-  std::string redirect_path;
-  ASSERT_TRUE(CreateRedirect(https_url.spec(), &redirect_path));
-  PrerenderTestURL(redirect_path, FINAL_STATUS_HTTPS, 1);
+  PrerenderTestURL(CreateClientRedirect(https_url.spec()),
+                   FINAL_STATUS_HTTPS,
+                   1);
 }
 
 // Checks that renderers using excessive memory will be terminated.
 // Disabled, http://crbug.com/77870.
 IN_PROC_BROWSER_TEST_F(PrerenderBrowserTest,
                        DISABLED_PrerenderExcessiveMemory) {
-  PrerenderTestURL("prerender_excessive_memory.html",
-                   FINAL_STATUS_MEMORY_LIMIT_EXCEEDED, 1);
+  PrerenderTestURL("files/prerender/prerender_excessive_memory.html",
+                   FINAL_STATUS_MEMORY_LIMIT_EXCEEDED,
+                   1);
 }
 
 // Checks that we don't prerender in an infinite loop.
 // Disabled, http://crbug.com/77870.
 IN_PROC_BROWSER_TEST_F(PrerenderBrowserTest, DISABLED_PrerenderInfiniteLoop) {
   const char* const kHtmlFileA = "prerender_infinite_a.html";
   const char* const kHtmlFileB = "prerender_infinite_b.html";
 
   std::deque<FinalStatus> expected_final_status_queue;
   expected_final_status_queue.push_back(FINAL_STATUS_USED);
   expected_final_status_queue.push_back(FINAL_STATUS_APP_TERMINATING);
 
   PrerenderTestURL(kHtmlFileA, expected_final_status_queue, 1);
 
   // Next url should be in pending list but not an active entry.
   EXPECT_FALSE(UrlIsInPrerenderManager(kHtmlFileB));
   EXPECT_TRUE(UrlIsPendingInPrerenderManager(kHtmlFileB));
 
   NavigateToDestURL();
 
   // Make sure the PrerenderContents for the next url is now in the manager
   // and not pending.
   EXPECT_TRUE(UrlIsInPrerenderManager(kHtmlFileB));
   EXPECT_FALSE(UrlIsPendingInPrerenderManager(kHtmlFileB));
 }
 
 // Checks that we don't prerender in an infinite loop and multiple links are
 // handled correctly.
-IN_PROC_BROWSER_TEST_F(PrerenderBrowserTest, FLAKY_PrerenderInfiniteLoopMultiple) {
-  const char* const kHtmlFileA = "prerender_infinite_a_multiple.html";
-  const char* const kHtmlFileB = "prerender_infinite_b_multiple.html";
-  const char* const kHtmlFileC = "prerender_infinite_c_multiple.html";
+IN_PROC_BROWSER_TEST_F(PrerenderBrowserTest,
+                       FLAKY_PrerenderInfiniteLoopMultiple) {
+  const char* const kHtmlFileA =
+      "files/prerender/prerender_infinite_a_multiple.html";
+  const char* const kHtmlFileB =
+      "files/prerender/prerender_infinite_b_multiple.html";
+  const char* const kHtmlFileC =
+      "files/prerender/prerender_infinite_c_multiple.html";
 
   // We need to set the final status to expect here before starting any
   // prerenders. We set them on a queue so whichever we see first is expected to
   // be evicted, and the second should stick around until we exit.
   std::deque<FinalStatus> expected_final_status_queue;
   expected_final_status_queue.push_back(FINAL_STATUS_USED);
   expected_final_status_queue.push_back(FINAL_STATUS_EVICTED);
   expected_final_status_queue.push_back(FINAL_STATUS_APP_TERMINATING);
 
   PrerenderTestURL(kHtmlFileA, expected_final_status_queue, 1);
@@ skipping 16 matching lines @@
   EXPECT_FALSE(UrlIsPendingInPrerenderManager(kHtmlFileB));
   EXPECT_FALSE(UrlIsPendingInPrerenderManager(kHtmlFileC));
 }
 
 IN_PROC_BROWSER_TEST_F(PrerenderBrowserTest, TaskManager) {
   // Show the task manager. This populates the model.
   browser()->window()->ShowTaskManager();
 
   // Start with two resources.
   EXPECT_EQ(2, model()->ResourceCount());
-  PrerenderTestURL("prerender_page.html", FINAL_STATUS_USED, 1);
+  PrerenderTestURL("files/prerender/prerender_page.html", FINAL_STATUS_USED, 1);
 
   // The prerender makes three.
   EXPECT_EQ(3, model()->ResourceCount());
 
   // It shouldn't have a TabContents associated with it.
   ASSERT_TRUE(model()->GetResourceTabContents(1) == NULL);
 
   // The prefix should be "Prerender:"
   string16 prefix =
       l10n_util::GetStringFUTF16(IDS_TASK_MANAGER_PRERENDER_PREFIX,
                                  string16());
   ASSERT_TRUE(StartsWith(model()->GetResourceTitle(1), prefix, true));
 
   NavigateToDestURL();
 
   // Prerender task should be killed and removed from the Task Manager.
   EXPECT_EQ(2, model()->ResourceCount());
 }
 
 // Checks that prerenderers will terminate when an audio tag is encountered.
 IN_PROC_BROWSER_TEST_F(PrerenderBrowserTest, PrerenderHTML5Audio) {
-  PrerenderTestURL("prerender_html5_audio.html", FINAL_STATUS_HTML5_MEDIA, 1);
+  PrerenderTestURL("files/prerender/prerender_html5_audio.html",
+                   FINAL_STATUS_HTML5_MEDIA,
+                   1);
 }
 
 // Checks that prerenderers will terminate when a video tag is encountered.
 IN_PROC_BROWSER_TEST_F(PrerenderBrowserTest, PrerenderHTML5Video) {
-  PrerenderTestURL("prerender_html5_video.html", FINAL_STATUS_HTML5_MEDIA, 1);
+  PrerenderTestURL("files/prerender/prerender_html5_video.html",
+                   FINAL_STATUS_HTML5_MEDIA,
+                   1);
 }
 
 // Checks that prerenderers will terminate when a video tag is inserted via
 // javascript.
 IN_PROC_BROWSER_TEST_F(PrerenderBrowserTest, PrerenderHTML5VideoJs) {
-  PrerenderTestURL("prerender_html5_video_script.html",
-                   FINAL_STATUS_HTML5_MEDIA, 1);
+  PrerenderTestURL("files/prerender/prerender_html5_video_script.html",
+                   FINAL_STATUS_HTML5_MEDIA,
+                   1);
 }
 
 }  // namespace prerender