Provide assumed favicon for intents if service provider page was never visited.

chrome/browser/history/thumbnail_database_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 <algorithm>
 #include <vector>
 
 #include "base/basictypes.h"
 #include "base/command_line.h"
 #include "base/file_path.h"
@@ skipping 415 matching lines @@
   EXPECT_TRUE(db.HasMappingFor(id2));
   EXPECT_FALSE(db.HasMappingFor(id3));
 
   // Remove all mappings
   db.DeleteIconMappings(page_url);
   EXPECT_FALSE(db.HasMappingFor(id1));
   EXPECT_FALSE(db.HasMappingFor(id2));
   EXPECT_FALSE(db.HasMappingFor(id3));
 }
 
+TEST_F(ThumbnailDatabaseTest, CloneIconMapping) {
+  ThumbnailDatabase db;
+  ASSERT_EQ(sql::INIT_OK, db.Init(file_name_, NULL, NULL));
+  db.BeginTransaction();
+
+  std::vector<unsigned char> data(blob1, blob1 + sizeof(blob1));
+  scoped_refptr<RefCountedBytes> favicon(new RefCountedBytes(data));
+
+  // Add a favicon which will have icon_mappings
+  FaviconID id1 = db.AddFavicon(GURL("http://google.com"), FAVICON);
+  EXPECT_NE(id1, 0);
+  base::Time time = base::Time::Now();
+  db.SetFavicon(id1, favicon, time);
+
+  // Add another type of favicon
+  FaviconID id2 = db.AddFavicon(GURL("http://www.google.com/icon"), TOUCH_ICON);
+  EXPECT_NE(id2, 0);
+  time = base::Time::Now();
+  db.SetFavicon(id2, favicon, time);
+
+  // Add 3rd favicon
+  FaviconID id3 = db.AddFavicon(GURL("http://www.google.com/icon"), TOUCH_ICON);
+  EXPECT_NE(id3, 0);
+  time = base::Time::Now();
+  db.SetFavicon(id3, favicon, time);
+
+  GURL page1_url("http://page1.com");
+  EXPECT_TRUE(db.AddIconMapping(page1_url, id1));
+  EXPECT_TRUE(db.AddIconMapping(page1_url, id2));
+
+  GURL page2_url("http://page2.com");
+  EXPECT_TRUE(db.AddIconMapping(page2_url, id3));
+
+  // Test we do nothing with existing mappings.
+  std::vector<IconMapping> icon_mapping;
+  EXPECT_TRUE(db.GetIconMappingsForPageURL(page2_url, &icon_mapping));
+  ASSERT_EQ(1U, icon_mapping.size());
+
+  EXPECT_TRUE(db.CloneIconMapping(page1_url, page2_url));
+
+  icon_mapping.clear();
+  EXPECT_TRUE(db.GetIconMappingsForPageURL(page2_url, &icon_mapping));
+  ASSERT_EQ(1U, icon_mapping.size());
+  EXPECT_EQ(page2_url, icon_mapping[0].page_url);
+  EXPECT_EQ(id3, icon_mapping[0].icon_id);
+
+  // Test we clone if the new page has no mappings.
+  GURL page3_url("http://page3.com");
+  EXPECT_TRUE(db.CloneIconMapping(page1_url, page3_url));
+
+  icon_mapping.clear();
+  EXPECT_TRUE(db.GetIconMappingsForPageURL(page3_url, &icon_mapping));
+
+  ASSERT_EQ(2U, icon_mapping.size());
+  if (icon_mapping[0].icon_id == id2)
+    std::swap(icon_mapping[0], icon_mapping[1]);
+  EXPECT_EQ(page3_url, icon_mapping[0].page_url);
+  EXPECT_EQ(id1, icon_mapping[0].icon_id);
+  EXPECT_EQ(page3_url, icon_mapping[1].page_url);
+  EXPECT_EQ(id2, icon_mapping[1].icon_id);
+}
+
 TEST_F(IconMappingMigrationTest, TestIconMappingMigration) {
   HistoryDatabase history_db;
   ASSERT_TRUE(history_db.db_.Open(history_db_name_));
   history_db.BeginTransaction();
 
   const GURL icon1 = GURL("http://www.google.com/favicon.ico");
   const GURL icon2 = GURL("http://www.yahoo.com/favicon.ico");
 
   ThumbnailDatabase db;
   ASSERT_EQ(sql::INIT_OK, db.Init(thumbnail_db_name_, NULL, &history_db));
@@ skipping 35 matching lines @@
   ASSERT_TRUE(db.GetFavicon(
       icon_mappings[0].icon_id, &time, &out_data, &out_icon_url));
   EXPECT_EQ(icon2, out_icon_url);
 
   // Test a page without icon
   GURL page_url4 = GURL("http://www.google.com/blank.html");
   EXPECT_FALSE(db.GetIconMappingsForPageURL(page_url4, NULL));
 }
 
 }  // namespace history