1. Create and use the TopSites database file....

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 "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/history/top_sites_database.h"
+#include "chrome/browser/history/history_notifications.h"
 #include "chrome/test/testing_profile.h"
 #include "chrome/tools/profiles/thumbnail-inl.h"
 #include "gfx/codec/jpeg_codec.h"
 #include "googleurl/src/gurl.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "third_party/skia/include/core/SkBitmap.h"
 
 
 namespace history {
 
 static const unsigned char kBlob[] =
     "12346102356120394751634516591348710478123649165419234519234512349134";
 
 class TopSitesTest : public testing::Test {
  public:
   TopSitesTest() {
   }
   ~TopSitesTest() {
   }
 
   TopSites& top_sites() { return *top_sites_; }
+  Profile& profile() {return *profile_;}
   FilePath& file_name() { return file_name_; }
   RefCountedBytes* google_thumbnail() { return google_thumbnail_; }
   RefCountedBytes* random_thumbnail() { return random_thumbnail_; }
   RefCountedBytes* weewar_thumbnail() { return weewar_thumbnail_; }
 
   virtual void SetUp() {
     profile_.reset(new TestingProfile);
     top_sites_ = new TopSites(profile_.get());
 
     ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
@@ skipping 40 matching lines @@
   }
 
  private:
   scoped_refptr<TopSites> top_sites_;
   scoped_ptr<TestingProfile> profile_;
   ScopedTempDir temp_dir_;
   FilePath file_name_;  // Database filename.
   scoped_refptr<RefCountedBytes> google_thumbnail_;
   scoped_refptr<RefCountedBytes> random_thumbnail_;
   scoped_refptr<RefCountedBytes> weewar_thumbnail_;
+  MessageLoop message_loop_;
 
   DISALLOW_COPY_AND_ASSIGN(TopSitesTest);
 };
 
 // A mockup of a HistoryService used for testing TopSites.
 class MockHistoryServiceImpl : public TopSites::MockHistoryService {
  public:
   // Calls the callback directly with the results.
   HistoryService::Handle QueryMostVisitedURLs(
       int result_count, int days_back,
       CancelableRequestConsumerBase* consumer,
       HistoryService::QueryMostVisitedURLsCallback* callback) {
     callback->Run(CancelableRequestProvider::Handle(0),  // Handle is unused.
                   most_visited_urls_);
     delete callback;
     return 0;
   }
 
   // Add a page to the end of the pages list.
   void AppendMockPage(const GURL& url,
                       const string16& title) {
     MostVisitedURL page;
     page.url = url;
     page.title = title;
     page.redirects = RedirectList();
     page.redirects.push_back(url);
     most_visited_urls_.push_back(page);
   }
 
+  // Removes the last URL in the list.
+  void RemoveMostVisitedURL() {
+    most_visited_urls_.pop_back();
+  }
+
  private:
   MostVisitedURLList most_visited_urls_;
 };
 
 
 // A mockup of a TopSitesDatabase used for testing TopSites.
 class MockTopSitesDatabaseImpl : public TopSitesDatabase {
  public:
   virtual MostVisitedURLList GetTopURLs() {
     // Return a copy of the vector.
@@ skipping 470 matching lines @@
                                            high_score));
   // Check that the thumbnail was updated.
   top_sites().GetPageThumbnail(google1_url, &out_1);
   EXPECT_FALSE(ThumbnailsAreEqual(out_1, out_2.thumbnail));
   // Read the new thumbnail from the DB - should match what's in TopSites.
   db->GetPageThumbnail(url2, &out_2);
   EXPECT_TRUE(ThumbnailsAreEqual(out_1, out_2.thumbnail));
   EXPECT_TRUE(high_score.Equals(out_2.thumbnail_score));
 }
 
+TEST_F(TopSitesTest, DeleteNotifications) {
+  GURL google1_url("http://google.com");
+  GURL google2_url("http://google.com/redirect");
+  GURL google3_url("http://www.google.com");
+  string16 google_title(ASCIIToUTF16("Google"));
+  GURL news_url("http://news.google.com");
+  string16 news_title(ASCIIToUTF16("Google News"));
+
+  MockHistoryServiceImpl hs;
+
+  top_sites().Init(file_name());
+
+  hs.AppendMockPage(google1_url, google_title);
+  hs.AppendMockPage(news_url, news_title);
+  top_sites().SetMockHistoryService(&hs);
+
+  top_sites().StartQueryForMostVisited();
+
+  MostVisitedURLList result = top_sites().GetMostVisitedURLs();
+  ASSERT_EQ(2u, result.size());
+
+  hs.RemoveMostVisitedURL();
+
+  history::URLsDeletedDetails details;
+  details.all_history = false;
+  top_sites().Observe(NotificationType::HISTORY_URLS_DELETED,
+                      Source<Profile> (&profile()),
+                      (const NotificationDetails&)details);
+
+  result = top_sites().GetMostVisitedURLs();
+  ASSERT_EQ(1u, result.size());
+  EXPECT_EQ(google_title, result[0].title);
+
+  hs.RemoveMostVisitedURL();
+  details.all_history = true;
+  top_sites().Observe(NotificationType::HISTORY_URLS_DELETED,
+                      Source<Profile> (&profile()),
+                      (const NotificationDetails&)details);
+  result = top_sites().GetMostVisitedURLs();
+  ASSERT_EQ(0u, result.size());
+}
+
+TEST_F(TopSitesTest, GetUpdateDelay) {
+  top_sites().last_num_urls_changed_ = 0;
+  EXPECT_EQ(60, top_sites().GetUpdateDelay().InMinutes());
+
+  top_sites().last_num_urls_changed_ = 3;
+  EXPECT_EQ(52, top_sites().GetUpdateDelay().InMinutes());
+
+  top_sites().last_num_urls_changed_ = 20;
+  EXPECT_EQ(1, top_sites().GetUpdateDelay().InMinutes());
+}
+
 }  // namespace history