New pinning algorithm.

chrome/browser/history/top_sites_unittest.cc

 // Copyright (c) 2009 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 "app/l10n_util.h"
 #include "base/scoped_temp_dir.h"
 #include "base/string_util.h"
 #include "chrome/browser/history/top_sites.h"
 #include "chrome/common/chrome_paths.h"
 #include "chrome/browser/dom_ui/most_visited_handler.h"
@@ skipping 1097 matching lines @@
   top_sites().GetMostVisitedURLs(
       &c,
       NewCallback(static_cast<TopSitesTest*>(this),
                   &TopSitesTest::OnTopSitesAvailable));
 
   ASSERT_EQ(4u, urls().size());
   EXPECT_EQ("http://bbc.com/", urls()[0].url.spec());
   EXPECT_EQ("http://google.com/", urls()[1].url.spec());
   EXPECT_EQ(welcome_url(), urls()[2].url);
   EXPECT_EQ(themes_url(), urls()[3].url);
+
+  top_sites().AddPinnedURL(GURL("http://bbc.com"), 1);
+  top_sites().AddPinnedURL(themes_url(), 0);
+  top_sites().GetMostVisitedURLs(
+      &c,
+      NewCallback(static_cast<TopSitesTest*>(this),
+                  &TopSitesTest::OnTopSitesAvailable));
+
+  ASSERT_EQ(4u, urls().size());
+  EXPECT_EQ(themes_url(), urls()[0].url);
+  EXPECT_EQ("http://bbc.com/", urls()[1].url.spec());
+  EXPECT_EQ("http://google.com/", urls()[2].url.spec());
+  EXPECT_EQ(welcome_url(), urls()[3].url);
+
+  top_sites().RemovePinnedURL(GURL("http://bbc.com"));
+  top_sites().RemovePinnedURL(themes_url());
+
+  top_sites().AddPinnedURL(welcome_url(), 1);
+  top_sites().AddPinnedURL(GURL("http://bbc.com"), 3);
+
+  top_sites().GetMostVisitedURLs(
+      &c,
+      NewCallback(static_cast<TopSitesTest*>(this),
+                  &TopSitesTest::OnTopSitesAvailable));
+
+  ASSERT_EQ(4u, urls().size());
+  EXPECT_EQ("http://google.com/", urls()[0].url.spec());
+  EXPECT_EQ(welcome_url(), urls()[1].url);
+  EXPECT_EQ(themes_url(), urls()[2].url);
+  EXPECT_EQ("http://bbc.com/", urls()[3].url.spec());
 }
 
 TEST_F(TopSitesTest, BlacklistingAndPinnedURLs) {
   ChromeThread db_loop(ChromeThread::DB, MessageLoop::current());
   MostVisitedURLList pages;
   CancelableRequestConsumer c;
   top_sites().GetMostVisitedURLs(
       &c,
       NewCallback(static_cast<TopSitesTest*>(this),
                   &TopSitesTest::OnTopSitesAvailable));
@@ skipping 80 matching lines @@
 
   top_sites().RemoveBlacklistedURL(store_url);
   EXPECT_EQ(6, top_sites().GetIndexForChromeStore(pages));
 
   top_sites().AddPinnedURL(GURL("http://bbc.com"), 2);
   // All pinned - can't add store.
   EXPECT_EQ(-1, top_sites().GetIndexForChromeStore(pages));
 }
 
 }  // namespace history