Popular sites on the NTP: add thumbnails!

components/suggestions/image_manager_unittest.cc

 // Copyright 2014 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 <string>
 
 #include "base/files/file_path.h"
 #include "base/run_loop.h"
 #include "base/thread_task_runner_handle.h"
 #include "components/leveldb_proto/proto_database.h"
@@ skipping 131 matching lines @@
 
   image_manager_->Initialize(suggestions_profile);
 
   GURL output;
   EXPECT_TRUE(image_manager_->GetImageURL(GURL(kTestUrl1), &output));
   EXPECT_EQ(GURL(kTestImagePath), output);
 
   EXPECT_FALSE(image_manager_->GetImageURL(GURL("http://b.com"), &output));
 }
 
+TEST_F(ImageManagerTest, AddImageURL) {
+  InitializeDefaultImageMapAndDatabase(image_manager_.get(), fake_db_);
+
+  GURL output;
+  // Try a URL the ImageManager doesn't know about.
+  ASSERT_FALSE(image_manager_->GetImageURL(GURL(kTestUrl2), &output));
+
+  // Explicitly add the URL and try again.
+  image_manager_->AddImageURL(GURL(kTestUrl2), GURL(kTestImagePath));
+  EXPECT_TRUE(image_manager_->GetImageURL(GURL(kTestUrl2), &output));
+  EXPECT_EQ(GURL(kTestImagePath), output);
+}
+
 TEST_F(ImageManagerTest, GetImageForURLNetwork) {
   InitializeDefaultImageMapAndDatabase(image_manager_.get(), fake_db_);
 
   // We expect the fetcher to go to network and call the callback.
   EXPECT_CALL(*mock_image_fetcher_, StartOrQueueNetworkRequest(_, _, _));
 
   // Fetch existing URL.
   base::RunLoop run_loop;
   image_manager_->GetImageForURL(GURL(kTestUrl1),
                                  base::Bind(&ImageManagerTest::OnImageAvailable,
@@ skipping 32 matching lines @@
   image_manager_->GetImageForURL(GURL(kTestUrl1),
                                  base::Bind(&ImageManagerTest::OnImageAvailable,
                                             base::Unretained(this), &run_loop));
   run_loop.Run();
 
   EXPECT_EQ(0, num_callback_null_called_);
   EXPECT_EQ(1, num_callback_valid_called_);
 }
 
 }  // namespace suggestions