Broadcast prerender URLs to all render processes.

chrome/renderer/prerender/prerender_dispatcher_unittest.cc

 // Copyright (c) 2012 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/renderer/prerender/prerender_dispatcher.h"
 
 #include <map>
 #include <utility>
 
 #include "base/compiler_specific.h"
@@ skipping 27 matching lines @@
 
   const std::map<int, WebPrerender>& prerenders() const {
     return prerender_dispatcher_.prerenders_;
   }
 
   int StartPrerender(const GURL& url) {
     DCHECK_EQ(0u, prerender_dispatcher_.prerenders_.count(g_next_prerender_id));
     prerender_dispatcher_.prerenders_[g_next_prerender_id] = WebPrerender();
 
     prerender_dispatcher_.OnPrerenderStart(g_next_prerender_id);
-    prerender_dispatcher_.OnPrerenderAddAlias(g_next_prerender_id, url);
+    prerender_dispatcher_.OnPrerenderAddAlias(url);
     return g_next_prerender_id++;
   }
 
-  void AddAliasToPrerender(int prerender_id, const GURL& url) {
-    prerender_dispatcher_.OnPrerenderAddAlias(prerender_id, url);
+  void AddAliasToPrerender(const GURL& url) {
+    prerender_dispatcher_.OnPrerenderAddAlias(url);
+  }
+
+  void RemoveAliasFromPrerender(const GURL& url) {
+    std::vector<GURL> urls;
+    urls.push_back(url);
+    prerender_dispatcher_.OnPrerenderRemoveAliases(urls);
   }
 
   void StopPrerender(int prerender_id) {
     prerender_dispatcher_.OnPrerenderStop(prerender_id);
   }
 
   int GetCountForURL(const GURL& url) const {
     return prerender_dispatcher_.running_prerender_urls_.count(url);
   }
 
@@ skipping 13 matching lines @@
   EXPECT_TRUE(IsPrerenderURL(foo_url));
   EXPECT_EQ(1, GetCountForURL(foo_url));
 }
 
 TEST_F(PrerenderDispatcherTest, PrerenderDispatcherMultipleAdd) {
   GURL foo_url = GURL("http://foo.com");
   GURL bar_url = GURL("http://bar.com");
 
   EXPECT_FALSE(IsPrerenderURL(foo_url));
   EXPECT_FALSE(IsPrerenderURL(bar_url));
-  int foo_id = StartPrerender(foo_url);
+  StartPrerender(foo_url);
   EXPECT_TRUE(IsPrerenderURL(foo_url));
   EXPECT_FALSE(IsPrerenderURL(bar_url));
 
-  AddAliasToPrerender(foo_id, foo_url);
+  AddAliasToPrerender(foo_url);
   EXPECT_TRUE(IsPrerenderURL(foo_url));
   EXPECT_FALSE(IsPrerenderURL(bar_url));
   EXPECT_EQ(2, GetCountForURL(foo_url));
 
   StartPrerender(bar_url);
   EXPECT_TRUE(IsPrerenderURL(foo_url));
   EXPECT_TRUE(IsPrerenderURL(bar_url));
   EXPECT_EQ(2, GetCountForURL(foo_url));
   EXPECT_EQ(1, GetCountForURL(bar_url));
 }
 
 TEST_F(PrerenderDispatcherTest, PrerenderDispatcherSingleRemove) {
   GURL foo_url = GURL("http://foo.com");
   EXPECT_FALSE(IsPrerenderURL(foo_url));
   int foo_id = StartPrerender(foo_url);
   EXPECT_TRUE(IsPrerenderURL(foo_url));
   StopPrerender(foo_id);
+  EXPECT_TRUE(IsPrerenderURL(foo_url));
+  EXPECT_EQ(1, GetCountForURL(foo_url));
+  RemoveAliasFromPrerender(foo_url);
   EXPECT_FALSE(IsPrerenderURL(foo_url));
   EXPECT_EQ(0, GetCountForURL(foo_url));
 }
+
+TEST_F(PrerenderDispatcherTest, PrerenderDispatcherTooManyRemoves) {
+  GURL foo_url = GURL("http://foo.com");
+  EXPECT_FALSE(IsPrerenderURL(foo_url));
+  int foo_id = StartPrerender(foo_url);
+  EXPECT_TRUE(IsPrerenderURL(foo_url));
+  StopPrerender(foo_id);
+  EXPECT_TRUE(IsPrerenderURL(foo_url));
+  EXPECT_EQ(1, GetCountForURL(foo_url));
+  RemoveAliasFromPrerender(foo_url);
+  EXPECT_FALSE(IsPrerenderURL(foo_url));
+  EXPECT_EQ(0, GetCountForURL(foo_url));
+  RemoveAliasFromPrerender(foo_url);
+  EXPECT_FALSE(IsPrerenderURL(foo_url));
+  EXPECT_EQ(0, GetCountForURL(foo_url));
+}
+
+TEST_F(PrerenderDispatcherTest, PrerenderDispatcherMultipleRemoves) {
+  GURL foo_url = GURL("http://foo.com");
+  EXPECT_FALSE(IsPrerenderURL(foo_url));
+  int foo_id = StartPrerender(foo_url);
+  EXPECT_TRUE(IsPrerenderURL(foo_url));
+  AddAliasToPrerender(foo_url);
+  StopPrerender(foo_id);
+  EXPECT_TRUE(IsPrerenderURL(foo_url));
+  EXPECT_EQ(2, GetCountForURL(foo_url));
+  RemoveAliasFromPrerender(foo_url);
+  EXPECT_TRUE(IsPrerenderURL(foo_url));
+  EXPECT_EQ(1, GetCountForURL(foo_url));
+  RemoveAliasFromPrerender(foo_url);
+  EXPECT_FALSE(IsPrerenderURL(foo_url));
+  EXPECT_EQ(0, GetCountForURL(foo_url));
+}
 
 }  // end namespace prerender