Home | History | Annotate | Download | only in history 
      1 // Copyright (c) 2011 The Chromium Authors. All rights reserved.
      2 // Use of this source code is governed by a BSD-style license that can be
      3 // found in the LICENSE file.
      4 
      5 #include <algorithm>
      6 #include <vector>
      7 
      8 #include "base/basictypes.h"
      9 #include "base/command_line.h"
     10 #include "base/file_path.h"
     11 #include "base/file_util.h"
     12 #include "base/memory/ref_counted_memory.h"
     13 #include "base/memory/scoped_temp_dir.h"
     14 #include "base/path_service.h"
     15 #include "chrome/browser/history/history_database.h"
     16 #include "chrome/browser/history/history_unittest_base.h"
     17 #include "chrome/browser/history/thumbnail_database.h"
     18 #include "chrome/common/chrome_constants.h"
     19 #include "chrome/common/chrome_paths.h"
     20 #include "chrome/common/thumbnail_score.h"
     21 #include "chrome/test/testing_profile.h"
     22 #include "chrome/tools/profiles/thumbnail-inl.h"
     23 #include "googleurl/src/gurl.h"
     24 #include "testing/gtest/include/gtest/gtest.h"
     25 #include "third_party/skia/include/core/SkBitmap.h"
     26 #include "ui/gfx/codec/jpeg_codec.h"
     27 
     28 using base::Time;
     29 using base::TimeDelta;
     30 
     31 namespace history {
     32 
     33 namespace {
     34 
     35 // data we'll put into the thumbnail database
     36 static const unsigned char blob1[] =
     37     "12346102356120394751634516591348710478123649165419234519234512349134";
     38 static const unsigned char blob2[] =
     39     "goiwuegrqrcomizqyzkjalitbahxfjytrqvpqeroicxmnlkhlzunacxaneviawrtxcywhgef";
     40 static const unsigned char blob3[] =
     41     "3716871354098370776510470746794707624107647054607467847164027";
     42 const double kBoringness = 0.25;
     43 const double kWorseBoringness = 0.50;
     44 const double kBetterBoringness = 0.10;
     45 const double kTotallyBoring = 1.0;
     46 
     47 const int64 kPage1 = 1234;
     48 
     49 }  // namespace
     50 
     51 class ThumbnailDatabaseTest : public testing::Test {
     52  public:
     53   ThumbnailDatabaseTest() {
     54   }
     55   ~ThumbnailDatabaseTest() {
     56   }
     57 
     58  protected:
     59   virtual void SetUp() {
     60     // Get a temporary directory for the test DB files.
     61     ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
     62 
     63     file_name_ = temp_dir_.path().AppendASCII("TestThumbnails.db");
     64     new_file_name_ = temp_dir_.path().AppendASCII("TestFavicons.db");
     65     history_db_name_ = temp_dir_.path().AppendASCII("TestHistory.db");
     66     google_bitmap_.reset(
     67         gfx::JPEGCodec::Decode(kGoogleThumbnail, sizeof(kGoogleThumbnail)));
     68   }
     69 
     70   scoped_ptr<SkBitmap> google_bitmap_;
     71 
     72   ScopedTempDir temp_dir_;
     73   FilePath file_name_;
     74   FilePath new_file_name_;
     75   FilePath history_db_name_;
     76 };
     77 
     78 class IconMappingMigrationTest : public HistoryUnitTestBase {
     79  public:
     80   IconMappingMigrationTest() {
     81   }
     82   ~IconMappingMigrationTest() {
     83   }
     84 
     85  protected:
     86   virtual void SetUp() {
     87     profile_.reset(new TestingProfile);
     88 
     89     FilePath data_path;
     90     ASSERT_TRUE(PathService::Get(chrome::DIR_TEST_DATA, &data_path));
     91     data_path = data_path.AppendASCII("History");
     92 
     93     history_db_name_ = profile_->GetPath().Append(chrome::kHistoryFilename);
     94     // Set up history and thumbnails as they would be before migration.
     95     ASSERT_NO_FATAL_FAILURE(
     96         ExecuteSQLScript(data_path.AppendASCII("history.20.sql"),
     97                          history_db_name_));
     98     thumbnail_db_name_ =
     99         profile_->GetPath().Append(chrome::kThumbnailsFilename);
    100     ASSERT_NO_FATAL_FAILURE(
    101         ExecuteSQLScript(data_path.AppendASCII("thumbnails.3.sql"),
    102                          thumbnail_db_name_));
    103   }
    104 
    105  protected:
    106   FilePath history_db_name_;
    107   FilePath thumbnail_db_name_;
    108 
    109  private:
    110   scoped_ptr<TestingProfile> profile_;
    111 };
    112 
    113 TEST_F(ThumbnailDatabaseTest, GetFaviconAfterMigrationToTopSites) {
    114   ThumbnailDatabase db;
    115   ASSERT_EQ(sql::INIT_OK, db.Init(file_name_, NULL, NULL));
    116   db.BeginTransaction();
    117 
    118   std::vector<unsigned char> data(blob1, blob1 + sizeof(blob1));
    119   scoped_refptr<RefCountedBytes> favicon(new RefCountedBytes(data));
    120 
    121   GURL url("http://google.com");
    122   FaviconID id = db.AddFavicon(url, FAVICON);
    123   base::Time time = base::Time::Now();
    124   db.SetFavicon(id, favicon, time);
    125   EXPECT_TRUE(db.RenameAndDropThumbnails(file_name_, new_file_name_));
    126 
    127   base::Time time_out;
    128   std::vector<unsigned char> favicon_out;
    129   GURL url_out;
    130   EXPECT_TRUE(db.GetFavicon(id, &time_out, &favicon_out, &url_out));
    131   EXPECT_EQ(url, url_out);
    132   EXPECT_EQ(time.ToTimeT(), time_out.ToTimeT());
    133   ASSERT_EQ(data.size(), favicon_out.size());
    134   EXPECT_TRUE(std::equal(data.begin(),
    135                          data.end(),
    136                          favicon_out.begin()));
    137 }
    138 
    139 TEST_F(ThumbnailDatabaseTest, AddIconMapping) {
    140   ThumbnailDatabase db;
    141   ASSERT_EQ(sql::INIT_OK, db.Init(file_name_, NULL, NULL));
    142   db.BeginTransaction();
    143 
    144   std::vector<unsigned char> data(blob1, blob1 + sizeof(blob1));
    145   scoped_refptr<RefCountedBytes> favicon(new RefCountedBytes(data));
    146 
    147   GURL url("http://google.com");
    148   FaviconID id = db.AddFavicon(url, TOUCH_ICON);
    149   EXPECT_NE(0, id);
    150   base::Time time = base::Time::Now();
    151   db.SetFavicon(id, favicon, time);
    152 
    153   EXPECT_NE(0, db.AddIconMapping(url, id));
    154   std::vector<IconMapping> icon_mapping;
    155   EXPECT_TRUE(db.GetIconMappingsForPageURL(url, &icon_mapping));
    156   EXPECT_EQ(1u, icon_mapping.size());
    157   EXPECT_EQ(url, icon_mapping.front().page_url);
    158   EXPECT_EQ(id, icon_mapping.front().icon_id);
    159 }
    160 
    161 TEST_F(ThumbnailDatabaseTest, UpdateIconMapping) {
    162   ThumbnailDatabase db;
    163   ASSERT_EQ(sql::INIT_OK, db.Init(file_name_, NULL, NULL));
    164   db.BeginTransaction();
    165 
    166   std::vector<unsigned char> data(blob1, blob1 + sizeof(blob1));
    167   scoped_refptr<RefCountedBytes> favicon(new RefCountedBytes(data));
    168 
    169   GURL url("http://google.com");
    170   FaviconID id = db.AddFavicon(url, TOUCH_ICON);
    171   base::Time time = base::Time::Now();
    172   db.SetFavicon(id, favicon, time);
    173 
    174   EXPECT_TRUE(0 < db.AddIconMapping(url, id));
    175   std::vector<IconMapping> icon_mapping;
    176   EXPECT_TRUE(db.GetIconMappingsForPageURL(url, &icon_mapping));
    177   ASSERT_EQ(1u, icon_mapping.size());
    178   EXPECT_EQ(url, icon_mapping.front().page_url);
    179   EXPECT_EQ(id, icon_mapping.front().icon_id);
    180 
    181   GURL url1("http://www.google.com/");
    182   FaviconID new_id = db.AddFavicon(url1, TOUCH_ICON);
    183   EXPECT_TRUE(db.UpdateIconMapping(icon_mapping.front().mapping_id, new_id));
    184 
    185   icon_mapping.clear();
    186   EXPECT_TRUE(db.GetIconMappingsForPageURL(url, &icon_mapping));
    187   ASSERT_EQ(1u, icon_mapping.size());
    188   EXPECT_EQ(url, icon_mapping.front().page_url);
    189   EXPECT_EQ(new_id, icon_mapping.front().icon_id);
    190   EXPECT_NE(id, icon_mapping.front().icon_id);
    191 }
    192 
    193 TEST_F(ThumbnailDatabaseTest, DeleteIconMappings) {
    194   ThumbnailDatabase db;
    195   ASSERT_EQ(sql::INIT_OK, db.Init(file_name_, NULL, NULL));
    196   db.BeginTransaction();
    197 
    198   std::vector<unsigned char> data(blob1, blob1 + sizeof(blob1));
    199   scoped_refptr<RefCountedBytes> favicon(new RefCountedBytes(data));
    200 
    201   GURL url("http://google.com");
    202   FaviconID id = db.AddFavicon(url, TOUCH_ICON);
    203   base::Time time = base::Time::Now();
    204   db.SetFavicon(id, favicon, time);
    205   EXPECT_TRUE(0 < db.AddIconMapping(url, id));
    206 
    207   FaviconID id2 = db.AddFavicon(url, FAVICON);
    208   db.SetFavicon(id2, favicon, time);
    209   EXPECT_TRUE(0 < db.AddIconMapping(url, id2));
    210   ASSERT_NE(id, id2);
    211 
    212   std::vector<IconMapping> icon_mapping;
    213   EXPECT_TRUE(db.GetIconMappingsForPageURL(url, &icon_mapping));
    214   ASSERT_EQ(2u, icon_mapping.size());
    215   EXPECT_EQ(icon_mapping.front().icon_type, TOUCH_ICON);
    216   EXPECT_TRUE(db.GetIconMappingForPageURL(url, FAVICON, NULL));
    217 
    218   db.DeleteIconMappings(url);
    219 
    220   EXPECT_FALSE(db.GetIconMappingsForPageURL(url, NULL));
    221   EXPECT_FALSE(db.GetIconMappingForPageURL(url, FAVICON, NULL));
    222 }
    223 
    224 TEST_F(ThumbnailDatabaseTest, GetIconMappingsForPageURL) {
    225   ThumbnailDatabase db;
    226   ASSERT_EQ(sql::INIT_OK, db.Init(file_name_, NULL, NULL));
    227   db.BeginTransaction();
    228 
    229   std::vector<unsigned char> data(blob1, blob1 + sizeof(blob1));
    230   scoped_refptr<RefCountedBytes> favicon(new RefCountedBytes(data));
    231 
    232   GURL url("http://google.com");
    233 
    234   FaviconID id1 = db.AddFavicon(url, TOUCH_ICON);
    235   base::Time time = base::Time::Now();
    236   db.SetFavicon(id1, favicon, time);
    237   EXPECT_TRUE(0 < db.AddIconMapping(url, id1));
    238 
    239   FaviconID id2 = db.AddFavicon(url, FAVICON);
    240   EXPECT_NE(id1, id2);
    241   db.SetFavicon(id2, favicon, time);
    242   EXPECT_TRUE(0 < db.AddIconMapping(url, id2));
    243 
    244   std::vector<IconMapping> icon_mapping;
    245   EXPECT_TRUE(db.GetIconMappingsForPageURL(url, &icon_mapping));
    246   ASSERT_EQ(2u, icon_mapping.size());
    247   EXPECT_NE(icon_mapping[0].icon_id, icon_mapping[1].icon_id);
    248   EXPECT_TRUE(icon_mapping[0].icon_id == id1 && icon_mapping[1].icon_id == id2);
    249 }
    250 
    251 TEST_F(ThumbnailDatabaseTest, UpgradeToVersion4) {
    252   ThumbnailDatabase db;
    253   ASSERT_EQ(sql::INIT_OK, db.Init(file_name_, NULL, NULL));
    254   db.BeginTransaction();
    255 
    256   const char* name = "favicons";
    257   std::string sql;
    258   sql.append("DROP TABLE IF EXISTS ");
    259   sql.append(name);
    260   EXPECT_TRUE(db.db_.Execute(sql.c_str()));
    261 
    262   sql.resize(0);
    263   sql.append("CREATE TABLE ");
    264   sql.append(name);
    265   sql.append("("
    266              "id INTEGER PRIMARY KEY,"
    267              "url LONGVARCHAR NOT NULL,"
    268              "last_updated INTEGER DEFAULT 0,"
    269              "image_data BLOB)");
    270   EXPECT_TRUE(db.db_.Execute(sql.c_str()));
    271 
    272   EXPECT_TRUE(db.UpgradeToVersion4());
    273 
    274   std::vector<unsigned char> data(blob1, blob1 + sizeof(blob1));
    275   scoped_refptr<RefCountedBytes> favicon(new RefCountedBytes(data));
    276 
    277   GURL url("http://google.com");
    278   FaviconID id = db.AddFavicon(url, TOUCH_ICON);
    279   base::Time time = base::Time::Now();
    280   db.SetFavicon(id, favicon, time);
    281 
    282   EXPECT_TRUE(0 < db.AddIconMapping(url, id));
    283   IconMapping icon_mapping;
    284   EXPECT_TRUE(db.GetIconMappingForPageURL(url, TOUCH_ICON, &icon_mapping));
    285   EXPECT_EQ(url, icon_mapping.page_url);
    286   EXPECT_EQ(id, icon_mapping.icon_id);
    287 }
    288 
    289 TEST_F(ThumbnailDatabaseTest, TemporayIconMapping) {
    290   ThumbnailDatabase db;
    291 
    292   ASSERT_EQ(sql::INIT_OK, db.Init(file_name_, NULL, NULL));
    293 
    294   db.BeginTransaction();
    295 
    296   EXPECT_TRUE(db.InitTemporaryIconMappingTable());
    297 
    298   std::vector<unsigned char> data(blob1, blob1 + sizeof(blob1));
    299   scoped_refptr<RefCountedBytes> favicon(new RefCountedBytes(data));
    300 
    301   GURL url("http://google.com");
    302   FaviconID id = db.AddFavicon(url, FAVICON);
    303   base::Time time = base::Time::Now();
    304   db.SetFavicon(id, favicon, time);
    305 
    306   db.AddToTemporaryIconMappingTable(url, id);
    307   db.CommitTemporaryIconMappingTable();
    308   IconMapping icon_mapping;
    309   EXPECT_TRUE(db.GetIconMappingForPageURL(url, FAVICON, &icon_mapping));
    310   EXPECT_EQ(id, icon_mapping.icon_id);
    311   EXPECT_EQ(url, icon_mapping.page_url);
    312 }
    313 
    314 TEST_F(ThumbnailDatabaseTest, GetIconMappingsForPageURLForReturnOrder) {
    315   ThumbnailDatabase db;
    316   ASSERT_EQ(sql::INIT_OK, db.Init(file_name_, NULL, NULL));
    317   db.BeginTransaction();
    318 
    319   // Add a favicon
    320   std::vector<unsigned char> data(blob1, blob1 + sizeof(blob1));
    321   scoped_refptr<RefCountedBytes> favicon(new RefCountedBytes(data));
    322 
    323   GURL url("http://google.com");
    324   FaviconID id = db.AddFavicon(url, FAVICON);
    325   base::Time time = base::Time::Now();
    326   db.SetFavicon(id, favicon, time);
    327 
    328   EXPECT_NE(0, db.AddIconMapping(url, id));
    329   std::vector<IconMapping> icon_mapping;
    330   EXPECT_TRUE(db.GetIconMappingsForPageURL(url, &icon_mapping));
    331 
    332   EXPECT_EQ(url, icon_mapping.front().page_url);
    333   EXPECT_EQ(id, icon_mapping.front().icon_id);
    334   EXPECT_EQ(FAVICON, icon_mapping.front().icon_type);
    335 
    336   // Add a touch icon
    337   std::vector<unsigned char> data2(blob2, blob2 + sizeof(blob2));
    338   scoped_refptr<RefCountedBytes> favicon2(new RefCountedBytes(data));
    339 
    340   FaviconID id2 = db.AddFavicon(url, TOUCH_ICON);
    341   db.SetFavicon(id2, favicon2, time);
    342   EXPECT_NE(0, db.AddIconMapping(url, id2));
    343 
    344   icon_mapping.clear();
    345   EXPECT_TRUE(db.GetIconMappingsForPageURL(url, &icon_mapping));
    346 
    347   EXPECT_EQ(url, icon_mapping.front().page_url);
    348   EXPECT_EQ(id2, icon_mapping.front().icon_id);
    349   EXPECT_EQ(TOUCH_ICON, icon_mapping.front().icon_type);
    350 
    351   // Add a touch precomposed icon
    352   scoped_refptr<RefCountedBytes> favicon3(new RefCountedBytes(data2));
    353 
    354   FaviconID id3 = db.AddFavicon(url, TOUCH_PRECOMPOSED_ICON);
    355   db.SetFavicon(id3, favicon3, time);
    356   EXPECT_NE(0, db.AddIconMapping(url, id3));
    357 
    358   icon_mapping.clear();
    359   EXPECT_TRUE(db.GetIconMappingsForPageURL(url, &icon_mapping));
    360 
    361   EXPECT_EQ(url, icon_mapping.front().page_url);
    362   EXPECT_EQ(id3, icon_mapping.front().icon_id);
    363   EXPECT_EQ(TOUCH_PRECOMPOSED_ICON, icon_mapping.front().icon_type);
    364 }
    365 
    366 TEST_F(ThumbnailDatabaseTest, HasMappingFor) {
    367   ThumbnailDatabase db;
    368   ASSERT_EQ(sql::INIT_OK, db.Init(file_name_, NULL, NULL));
    369   db.BeginTransaction();
    370 
    371   std::vector<unsigned char> data(blob1, blob1 + sizeof(blob1));
    372   scoped_refptr<RefCountedBytes> favicon(new RefCountedBytes(data));
    373 
    374   // Add a favicon which will have icon_mappings
    375   FaviconID id1 = db.AddFavicon(GURL("http://google.com"), FAVICON);
    376   EXPECT_NE(id1, 0);
    377   base::Time time = base::Time::Now();
    378   db.SetFavicon(id1, favicon, time);
    379 
    380   // Add another type of favicon
    381   FaviconID id2 = db.AddFavicon(GURL("http://www.google.com/icon"), TOUCH_ICON);
    382   EXPECT_NE(id2, 0);
    383   time = base::Time::Now();
    384   db.SetFavicon(id2, favicon, time);
    385 
    386   // Add 3rd favicon
    387   FaviconID id3 = db.AddFavicon(GURL("http://www.google.com/icon"), TOUCH_ICON);
    388   EXPECT_NE(id3, 0);
    389   time = base::Time::Now();
    390   db.SetFavicon(id3, favicon, time);
    391 
    392   // Add 2 icon mapping
    393   GURL page_url("http://www.google.com");
    394   EXPECT_TRUE(db.AddIconMapping(page_url, id1));
    395   EXPECT_TRUE(db.AddIconMapping(page_url, id2));
    396 
    397   EXPECT_TRUE(db.HasMappingFor(id1));
    398   EXPECT_TRUE(db.HasMappingFor(id2));
    399   EXPECT_FALSE(db.HasMappingFor(id3));
    400 
    401   // Remove all mappings
    402   db.DeleteIconMappings(page_url);
    403   EXPECT_FALSE(db.HasMappingFor(id1));
    404   EXPECT_FALSE(db.HasMappingFor(id2));
    405   EXPECT_FALSE(db.HasMappingFor(id3));
    406 }
    407 
    408 TEST_F(IconMappingMigrationTest, TestIconMappingMigration) {
    409   HistoryDatabase history_db;
    410   ASSERT_TRUE(history_db.db_.Open(history_db_name_));
    411   history_db.BeginTransaction();
    412 
    413   const GURL icon1 = GURL("http://www.google.com/favicon.ico");
    414   const GURL icon2 = GURL("http://www.yahoo.com/favicon.ico");
    415 
    416   ThumbnailDatabase db;
    417   ASSERT_EQ(sql::INIT_OK, db.Init(thumbnail_db_name_, NULL, &history_db));
    418   db.BeginTransaction();
    419 
    420   // Migration should be done.
    421   // Test one icon_mapping.
    422   GURL page_url1 = GURL("http://google.com/");
    423   std::vector<IconMapping> icon_mappings;
    424   EXPECT_TRUE(db.GetIconMappingsForPageURL(page_url1, &icon_mappings));
    425   ASSERT_EQ(1u, icon_mappings.size());
    426   EXPECT_EQ(FAVICON, icon_mappings[0].icon_type);
    427   EXPECT_EQ(page_url1, icon_mappings[0].page_url);
    428   EXPECT_EQ(1, icon_mappings[0].icon_id);
    429   base::Time time;
    430   std::vector<unsigned char> out_data;
    431   GURL out_icon_url;
    432   ASSERT_TRUE(db.GetFavicon(
    433       icon_mappings[0].icon_id, &time, &out_data, &out_icon_url));
    434   EXPECT_EQ(icon1, out_icon_url);
    435 
    436   // Test a page which has the same icon.
    437   GURL page_url3 = GURL("http://www.google.com/");
    438   icon_mappings.clear();
    439   EXPECT_TRUE(db.GetIconMappingsForPageURL(page_url3, &icon_mappings));
    440   ASSERT_EQ(1u, icon_mappings.size());
    441   EXPECT_EQ(FAVICON, icon_mappings[0].icon_type);
    442   EXPECT_EQ(page_url3, icon_mappings[0].page_url);
    443   EXPECT_EQ(1, icon_mappings[0].icon_id);
    444 
    445   // Test a icon_mapping with different IconID.
    446   GURL page_url2 = GURL("http://yahoo.com/");
    447   icon_mappings.clear();
    448   EXPECT_TRUE(db.GetIconMappingsForPageURL(page_url2, &icon_mappings));
    449   ASSERT_EQ(1u, icon_mappings.size());
    450   EXPECT_EQ(FAVICON, icon_mappings[0].icon_type);
    451   EXPECT_EQ(page_url2, icon_mappings[0].page_url);
    452   EXPECT_EQ(2, icon_mappings[0].icon_id);
    453   ASSERT_TRUE(db.GetFavicon(
    454       icon_mappings[0].icon_id, &time, &out_data, &out_icon_url));
    455   EXPECT_EQ(icon2, out_icon_url);
    456 
    457   // Test a page without icon
    458   GURL page_url4 = GURL("http://www.google.com/blank.html");
    459   EXPECT_FALSE(db.GetIconMappingsForPageURL(page_url4, NULL));
    460 }
    461 
    462 }  // namespace history
    463