Add TopSites::GetTemporaryThumbnailScore().

chrome/browser/history/top_sites_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/file_util.h"
 #include "base/format_macros.h"
 #include "base/memory/scoped_temp_dir.h"
 #include "base/path_service.h"
 #include "base/string_util.h"
@@ skipping 1077 matching lines @@
   // Don't store thumbnails for Javascript URLs.
   EXPECT_FALSE(top_sites()->SetPageThumbnail(invalid_url,
                                              thumbnail,
                                              medium_score));
   // Store thumbnails for unknown (but valid) URLs temporarily - calls
   // AddTemporaryThumbnail.
   EXPECT_TRUE(top_sites()->SetPageThumbnail(unknown_url,
                                             thumbnail,
                                             medium_score));
 
-  // We shouldn't get the thumnail back though (the url isn't in to sites yet).
+  // The URL is unknown.
+  EXPECT_FALSE(top_sites()->IsKnownURL(unknown_url));
+  // We should be able to get the thumnail back even if the url isn't in
+  // to sites yet.
   scoped_refptr<RefCountedBytes> out;
-  EXPECT_FALSE(top_sites()->GetPageThumbnail(unknown_url, &out));
+  EXPECT_TRUE(top_sites()->GetPageThumbnail(unknown_url, &out));
+  EXPECT_TRUE(ThumbnailEqualsBytes(thumbnail, out.get()));
+  ThumbnailScore out_score;
+  EXPECT_TRUE(top_sites()->GetPageThumbnailScore(unknown_url, &out_score));
+  EXPECT_TRUE(medium_score.Equals(out_score));
 
   std::vector<MostVisitedURL> list;
 
   MostVisitedURL mv;
   mv.url = unknown_url;
   mv.redirects.push_back(mv.url);
   mv.redirects.push_back(url1a);
   mv.redirects.push_back(url1b);
   list.push_back(mv);
 
-  // Update URLs. This should result in using thumbnail.
+  // Update URLs. This should result in making the URL known.
   SetTopSites(list);
 
-  ASSERT_TRUE(top_sites()->GetPageThumbnail(unknown_url, &out));
-  EXPECT_TRUE(ThumbnailEqualsBytes(thumbnail, out.get()));
+  EXPECT_TRUE(top_sites()->IsKnownURL(unknown_url));
 }
 
 // Tests variations of blacklisting.
 TEST_F(TopSitesTest, Blacklisting) {
   MostVisitedURLList pages;
   MostVisitedURL url, url1;
   url.url = GURL("http://bbc.com/");
   url.redirects.push_back(url.url);
   pages.push_back(url);
   url1.url = GURL("http://google.com/");
@@ skipping 203 matching lines @@
   WaitForTopSites();
   EXPECT_FALSE(IsTopSitesLoaded());
 
   // Load history, which should make TopSites finish loading too.
   profile()->CreateHistoryService(false, false);
   profile()->BlockUntilTopSitesLoaded();
   EXPECT_TRUE(IsTopSitesLoaded());
 }
 
 }  // namespace history