Inject CanAddURLToHistory into TopSitesImpl

chrome/browser/history/thumbnail_database_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 <algorithm>
 #include <vector>
 
 #include "base/basictypes.h"
 #include "base/files/file_enumerator.h"
 #include "base/files/file_path.h"
@@ skipping 369 matching lines @@
                                kBlob2));
 
   // The ones not retained should be missing.
   EXPECT_FALSE(db.GetFaviconIDForFaviconURL(kPageUrl2, false, NULL));
   EXPECT_FALSE(db.GetFaviconIDForFaviconURL(kPageUrl4, false, NULL));
 
   // Schema should be the same.
   EXPECT_EQ(original_schema, db.db_.GetSchema());
 }
 
+// Test that RetainDataForPageUrls() expires retained favicons.
+TEST_F(ThumbnailDatabaseTest, RetainDataForPageUrlsExpiresRetainedFavicons) {
+  ThumbnailDatabase db(NULL);
+  ASSERT_EQ(sql::INIT_OK, db.Init(file_name_));
+  db.BeginTransaction();
+
+  scoped_refptr<base::RefCountedStaticMemory> favicon1(
+      new base::RefCountedStaticMemory(kBlob1, sizeof(kBlob1)));
+  favicon_base::FaviconID kept_id = db.AddFavicon(
+      kIconUrl1, favicon_base::FAVICON, favicon1, base::Time::Now(),
+      gfx::Size());
+  db.AddIconMapping(kPageUrl1, kept_id);
+
+  EXPECT_TRUE(db.RetainDataForPageUrls(std::vector<GURL>(1u, kPageUrl1)));
+
+  favicon_base::FaviconID new_favicon_id = db.GetFaviconIDForFaviconURL(
+      kIconUrl1, favicon_base::FAVICON, nullptr);
+  ASSERT_NE(0, new_favicon_id);
+  std::vector<FaviconBitmap> new_favicon_bitmaps;
+  db.GetFaviconBitmaps(new_favicon_id, &new_favicon_bitmaps);
+
+  ASSERT_EQ(1u, new_favicon_bitmaps.size());
+  EXPECT_EQ(0, new_favicon_bitmaps[0].last_updated.ToInternalValue());
+}
+
 // Tests that deleting a favicon deletes the favicon row and favicon bitmap
 // rows from the database.
 TEST_F(ThumbnailDatabaseTest, DeleteFavicon) {
   ThumbnailDatabase db(NULL);
   ASSERT_EQ(sql::INIT_OK, db.Init(file_name_));
   db.BeginTransaction();
 
   std::vector<unsigned char> data1(kBlob1, kBlob1 + sizeof(kBlob1));
   scoped_refptr<base::RefCountedBytes> favicon1(
       new base::RefCountedBytes(data1));
@@ skipping 692 matching lines @@
     ThumbnailDatabase db(NULL);
     ASSERT_EQ(sql::INIT_OK, db.Init(db_path));
 
     // Verify that the resulting schema is correct, whether it
     // involved razing the file or fixing things in place.
     VerifyTablesAndColumns(&db.db_);
   }
 }
 
 }  // namespace history