For prerendering, keep track of all the intermediate redirects, and hook into...

chrome/browser/prerender/prerender_resource_handler_unittest.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 "chrome/browser/prerender/prerender_resource_handler.h"
 #include "chrome/common/resource_response.h"
 #include "net/http/http_response_headers.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace {
@@ skipping 82 matching lines @@
                 mock_handler_,
                 NewCallback(
                     this,
                     &PrerenderResourceHandlerTest::SetLastHandledURL)))),
         ui_thread_(BrowserThread::UI, &loop_),
         default_url_("http://www.prerender.com") {
     pre_handler_->set_prerender_duration(prerender_duration_);
     pre_handler_->set_get_current_time_function(&FixedGetCurrentTime);
   }
 
-  void SetLastHandledURL(const GURL& url) {
+  void SetLastHandledURL(const GURL& url, const std::vector<GURL>& alias_urls) {
     last_handled_url_ = url;
+    alias_urls_ = alias_urls;
   }
 
   // Common logic shared by many of the tests
   void StartPrerendering(const std::string& mime_type,
                          const char* headers) {
     int request_id = 1;
     bool defer = false;
     EXPECT_TRUE(pre_handler_->OnWillStart(request_id, default_url_, &defer));
     EXPECT_FALSE(defer);
     scoped_refptr<ResourceResponse> response(new ResourceResponse);
     response->response_head.request_time = FixedGetCurrentTime();
     response->response_head.response_time = FixedGetCurrentTime();
     response->response_head.mime_type = mime_type;
     response->response_head.headers = CreateResponseHeaders(headers);
     EXPECT_TRUE(last_handled_url_.is_empty());
 
     // Start the response. If it is able to prerender, a task will
     // be posted to loop_ (masquerading as the UI thread), and
     // |SetLastHandledURL| will be called.
     EXPECT_TRUE(pre_handler_->OnResponseStarted(request_id, response));
     loop_.RunAllPending();
   }
 
+  // Test whether a given URL is part of alias_urls_.
+  bool ContainsAliasURL(const GURL& url) {
+    return std::find(alias_urls_.begin(), alias_urls_.end(), url)
+        != alias_urls_.end();
+  }
+
   base::TimeDelta prerender_duration_;
   scoped_refptr<MockResourceHandler> mock_handler_;
   scoped_refptr<PrerenderResourceHandler> pre_handler_;
   MessageLoop loop_;
   BrowserThread ui_thread_;
   GURL last_handled_url_;
   GURL default_url_;
+  std::vector<GURL> alias_urls_;
 };
 
 namespace {
 
 TEST_F(PrerenderResourceHandlerTest, NoOp) {
 }
 
 // Tests that a valid HTML resource will correctly get diverted
 // to the PrerenderManager.
 TEST_F(PrerenderResourceHandlerTest, Prerender) {
@@ skipping 56 matching lines @@
   response->response_head.mime_type = "text/html";
   response->response_head.request_time = FixedGetCurrentTime();
   response->response_head.response_time = FixedGetCurrentTime();
   response->response_head.headers = CreateResponseHeaders(
       "HTTP/1.1 200 OK\n"
       "cache-control: max-age=86400\n");
   EXPECT_TRUE(pre_handler_->OnResponseStarted(request_id, response));
   EXPECT_TRUE(last_handled_url_.is_empty());
   loop_.RunAllPending();
   EXPECT_EQ(url_redirect, last_handled_url_);
+  EXPECT_EQ(true, ContainsAliasURL(url_redirect));
+  EXPECT_EQ(true, ContainsAliasURL(default_url_));
+  EXPECT_EQ(2, static_cast<int>(alias_urls_.size()));
 }
 
 }
-