Popular sites on the NTP: add thumbnails!

components/suggestions/suggestions_service_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 "components/suggestions/suggestions_service.h"
 
 #include <map>
 #include <sstream>
 
 #include "base/bind.h"
@@ skipping 117 matching lines @@
 };
 
 class MockImageManager : public suggestions::ImageManager {
  public:
   MockImageManager() {}
   virtual ~MockImageManager() {}
   MOCK_METHOD1(Initialize, void(const SuggestionsProfile&));
   MOCK_METHOD2(GetImageForURL,
                void(const GURL&,
                     base::Callback<void(const GURL&, const SkBitmap*)>));
+  MOCK_METHOD2(AddImageURL, void(const GURL&, const GURL&));
 };
 
 class MockBlacklistStore : public suggestions::BlacklistStore {
  public:
   MOCK_METHOD1(BlacklistUrl, bool(const GURL&));
   MOCK_METHOD0(IsEmpty, bool());
   MOCK_METHOD1(GetTimeUntilReadyForUpload, bool(base::TimeDelta*));
   MOCK_METHOD2(GetTimeUntilURLReadyForUpload,
                bool(const GURL&, base::TimeDelta*));
   MOCK_METHOD1(GetCandidateForUpload, bool(GURL*));
@@ skipping 477 matching lines @@
 TEST_F(SuggestionsServiceTest, CheckDefaultTimeStamps) {
   scoped_ptr<SuggestionsService> suggestions_service(
       CreateSuggestionsServiceWithMocks());
   SuggestionsProfile suggestions =
       CreateSuggestionsProfileWithExpiryTimestamps();
   suggestions_service->SetDefaultExpiryTimestamp(&suggestions,
                                                  kTestDefaultExpiry);
   EXPECT_EQ(kTestSetExpiry, suggestions.suggestions(0).expiry_ts());
   EXPECT_EQ(kTestDefaultExpiry, suggestions.suggestions(1).expiry_ts());
 }
+
+TEST_F(SuggestionsServiceTest, GetPageThumbnail) {
+  scoped_ptr<SuggestionsService> suggestions_service(
+      CreateSuggestionsServiceWithMocks());
+  ASSERT_TRUE(suggestions_service != NULL);
+
+  GURL test_url(kTestUrl);
+  GURL thumbnail_url("https://www.thumbnails.com/thumb.jpg");
+  base::Callback<void(const GURL&, const SkBitmap*)> dummy_callback;
+
+  EXPECT_CALL(*mock_thumbnail_manager_, GetImageForURL(test_url, _));
+  suggestions_service->GetPageThumbnail(test_url, dummy_callback);
+
+  EXPECT_CALL(*mock_thumbnail_manager_, AddImageURL(test_url, thumbnail_url));
+  EXPECT_CALL(*mock_thumbnail_manager_, GetImageForURL(test_url, _));
+  suggestions_service->GetPageThumbnailWithURL(test_url, thumbnail_url,
+                                               dummy_callback);
+
+}
+
 }  // namespace suggestions