Cancel prerenders from Omnibox if we navigate to a different URL than predicted

chrome/browser/prerender/prerender_manager_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 "base/command_line.h"
 #include "base/memory/scoped_vector.h"
 #include "base/message_loop.h"
 #include "base/time.h"
 #include "chrome/browser/prerender/prerender_contents.h"
 #include "chrome/browser/prerender/prerender_manager.h"
 #include "chrome/browser/prerender/prerender_origin.h"
+#include "chrome/common/chrome_switches.h"
 #include "chrome/test/base/testing_browser_process.h"
+#include "chrome/test/base/testing_profile.h"
 #include "content/browser/renderer_host/render_view_host.h"
 #include "content/test/test_browser_thread.h"
 #include "googleurl/src/gurl.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 using content::BrowserThread;
 
 namespace prerender {
 
 namespace {
 
 class DummyPrerenderContents : public PrerenderContents {
  public:
   DummyPrerenderContents(PrerenderManager* prerender_manager,
                          PrerenderTracker* prerender_tracker,
                          const GURL& url,
+                         Origin origin,
                          FinalStatus expected_final_status)
       : PrerenderContents(prerender_manager, prerender_tracker,
                           NULL, url, content::Referrer(),
-                          ORIGIN_LINK_REL_PRERENDER,
-                          PrerenderManager::kNoExperiment),
+                          origin, PrerenderManager::kNoExperiment),
         has_started_(false),
         expected_final_status_(expected_final_status) {
   }
 
   virtual ~DummyPrerenderContents() {
     EXPECT_EQ(expected_final_status_, final_status());
   }
 
   virtual void StartPrerendering(
       const RenderViewHost* source_render_view_host,
@@ skipping 16 matching lines @@
   FinalStatus expected_final_status() const { return expected_final_status_; }
 
  private:
   bool has_started_;
   FinalStatus expected_final_status_;
 };
 
 class TestPrerenderManager : public PrerenderManager {
  public:
   explicit TestPrerenderManager(PrerenderTracker* prerender_tracker)
-      : PrerenderManager(NULL, prerender_tracker),
+      : PrerenderManager(&profile_, prerender_tracker),
         time_(base::Time::Now()),
         time_ticks_(base::TimeTicks::Now()),
         next_prerender_contents_(NULL),
         prerender_tracker_(prerender_tracker) {
     set_rate_limit_enabled(false);
   }
 
   virtual ~TestPrerenderManager() {
     if (next_prerender_contents()) {
       next_prerender_contents_.release()->Destroy(
@@ skipping 17 matching lines @@
 
   void AdvanceTimeTicks(base::TimeDelta delta) {
     time_ticks_ += delta;
   }
 
   DummyPrerenderContents* CreateNextPrerenderContents(
       const GURL& url,
       FinalStatus expected_final_status) {
     DummyPrerenderContents* prerender_contents =
         new DummyPrerenderContents(this, prerender_tracker_, url,
+                                   ORIGIN_LINK_REL_PRERENDER,
                                    expected_final_status);
     SetNextPrerenderContents(prerender_contents);
     return prerender_contents;
   }
 
   DummyPrerenderContents* CreateNextPrerenderContents(
       const GURL& url,
+      Origin origin,
+      FinalStatus expected_final_status) {
+    DummyPrerenderContents* prerender_contents =
+        new DummyPrerenderContents(this, prerender_tracker_, url,
+                                   origin, expected_final_status);
+    SetNextPrerenderContents(prerender_contents);
+    return prerender_contents;
+  }
+
+  DummyPrerenderContents* CreateNextPrerenderContents(
+      const GURL& url,
       const std::vector<GURL>& alias_urls,
       FinalStatus expected_final_status) {
     DummyPrerenderContents* prerender_contents =
         new DummyPrerenderContents(this, prerender_tracker_, url,
+                                   ORIGIN_LINK_REL_PRERENDER,
                                    expected_final_status);
     for (std::vector<GURL>::const_iterator it = alias_urls.begin();
          it != alias_urls.end();
          ++it) {
       EXPECT_TRUE(prerender_contents->AddAliasURL(*it));
     }
     SetNextPrerenderContents(prerender_contents);
     return prerender_contents;
   }
 
@@ skipping 27 matching lines @@
   virtual base::TimeTicks GetCurrentTimeTicks() const OVERRIDE {
     return time_ticks_;
   }
 
   virtual PrerenderContents* CreatePrerenderContents(
       const GURL& url,
       const content::Referrer& referrer,
       Origin origin,
       uint8 experiment_id) OVERRIDE {
     DCHECK(next_prerender_contents_.get());
+    DCHECK_EQ(next_prerender_contents_->prerender_url(), url);
+    DCHECK_EQ(next_prerender_contents_->origin(), origin);
     return next_prerender_contents_.release();
   }
 
   base::Time time_;
   base::TimeTicks time_ticks_;
   scoped_ptr<PrerenderContents> next_prerender_contents_;
   // PrerenderContents with an |expected_final_status| of FINAL_STATUS_USED,
   // tracked so they will be automatically deleted.
   ScopedVector<PrerenderContents> used_prerender_contents_;
 
   PrerenderTracker* prerender_tracker_;
+
+  TestingProfile profile_;
 };
 
 class RestorePrerenderMode {
  public:
   RestorePrerenderMode() : prev_mode_(PrerenderManager::GetMode()) {
   }
 
   ~RestorePrerenderMode() { PrerenderManager::SetMode(prev_mode_); }
  private:
   PrerenderManager::PrerenderManagerMode prev_mode_;
 };
 
 }  // namespace
 
 class PrerenderManagerTest : public testing::Test {
  public:
   PrerenderManagerTest() : ui_thread_(BrowserThread::UI, &message_loop_),
                            prerender_manager_(
                                new TestPrerenderManager(prerender_tracker())) {
+    // Enable omnibox prerendering.
+    CommandLine::ForCurrentProcess()->AppendSwitchASCII(
+        switches::kPrerenderFromOmnibox,
+        switches::kPrerenderFromOmniboxSwitchValueEnabled);
   }
 
   TestPrerenderManager* prerender_manager() {
     return prerender_manager_.get();
   }
 
  private:
   PrerenderTracker* prerender_tracker() {
     return g_browser_process->prerender_tracker();
   }
@@ skipping 317 matching lines @@
       prerender_manager()->CreateNextPrerenderContents(
           url,
           FINAL_STATUS_CACHE_OR_HISTORY_CLEARED);
   EXPECT_TRUE(prerender_manager()->AddSimplePrerender(url));
   EXPECT_TRUE(prerender_contents->has_started());
   prerender_manager()->ClearData(PrerenderManager::CLEAR_PRERENDER_CONTENTS);
   DummyPrerenderContents* null = NULL;
   EXPECT_EQ(null, prerender_manager()->GetEntry(url));
 }
 
+// Make sure canceling works as expected.
+TEST_F(PrerenderManagerTest, CancelAllTest) {
+  const DummyPrerenderContents* null = NULL;
+  GURL url("http://www.google.com/");
+  DummyPrerenderContents* prerender_contents =
+      prerender_manager()->CreateNextPrerenderContents(
+          url, FINAL_STATUS_CANCELLED);
+  EXPECT_TRUE(prerender_manager()->AddSimplePrerender(url));
+  EXPECT_TRUE(prerender_contents->has_started());
+  prerender_manager()->CancelAllPrerenders();
+  EXPECT_EQ(null, prerender_manager()->GetEntry(url));
+}
+
+// Make sure canceling for omnibox works as expected.
+TEST_F(PrerenderManagerTest, CancelOmniboxTest) {
+  const DummyPrerenderContents* null = NULL;
+  // Check canceling removes the Omnibox url.
+  {
+    GURL url("http://www.google.com/");
+    DummyPrerenderContents* prerender_contents =
+        prerender_manager()->CreateNextPrerenderContents(
+            url, ORIGIN_OMNIBOX, FINAL_STATUS_CANCELLED);
+    EXPECT_TRUE(prerender_manager()->AddPrerenderFromOmnibox(url, NULL));
+    EXPECT_TRUE(prerender_contents->has_started());
+    prerender_manager()->CancelOmniboxPrerenders();
+    EXPECT_EQ(null, prerender_manager()->GetEntry(url));
+  }
+
+  // Check canceling does not remove a Link url.
+  {
+    GURL url("http://www.google.com/");
+    DummyPrerenderContents* prerender_contents =
+        prerender_manager()->CreateNextPrerenderContents(
+            url, ORIGIN_LINK_REL_PRERENDER, FINAL_STATUS_CANCELLED);
+    EXPECT_TRUE(prerender_manager()->AddSimplePrerender(url));
+    EXPECT_TRUE(prerender_contents->has_started());
+    prerender_manager()->CancelOmniboxPrerenders();
+    EXPECT_NE(null, prerender_manager()->GetEntry(url));
+  }
+
+}
+
 }  // namespace prerender