| Index: chrome/browser/search/suggestions/suggestions_service_unittest.cc
|
| diff --git a/chrome/browser/search/suggestions/suggestions_service_unittest.cc b/chrome/browser/search/suggestions/suggestions_service_unittest.cc
|
| index 889ad2501850754ffd4be24fbc84d830adcb9fa7..395eb4d50b3455341963039a09718e541155f23f 100644
|
| --- a/chrome/browser/search/suggestions/suggestions_service_unittest.cc
|
| +++ b/chrome/browser/search/suggestions/suggestions_service_unittest.cc
|
| @@ -14,6 +14,7 @@
|
| #include "base/prefs/pref_service.h"
|
| #include "base/strings/utf_string_conversions.h"
|
| #include "chrome/browser/search/suggestions/blacklist_store.h"
|
| +#include "chrome/browser/search/suggestions/image_manager.h"
|
| #include "chrome/browser/search/suggestions/proto/suggestions.pb.h"
|
| #include "chrome/browser/search/suggestions/suggestions_service_factory.h"
|
| #include "chrome/browser/search/suggestions/suggestions_store.h"
|
| @@ -32,6 +33,7 @@ using testing::DoAll;
|
| using ::testing::Eq;
|
| using ::testing::Return;
|
| using testing::SetArgPointee;
|
| +using ::testing::NiceMock;
|
| using ::testing::StrictMock;
|
| using ::testing::_;
|
|
|
| @@ -100,12 +102,12 @@ class MockSuggestionsStore : public suggestions::SuggestionsStore {
|
| MOCK_METHOD0(ClearSuggestions, void());
|
| };
|
|
|
| -class MockThumbnailManager : public suggestions::ThumbnailManager {
|
| +class MockImageManager : public suggestions::ImageManager {
|
| public:
|
| - MockThumbnailManager() {}
|
| - virtual ~MockThumbnailManager() {}
|
| - MOCK_METHOD1(InitializeThumbnailMap, void(const SuggestionsProfile&));
|
| - MOCK_METHOD2(GetPageThumbnail,
|
| + MockImageManager() {}
|
| + virtual ~MockImageManager() {}
|
| + MOCK_METHOD1(Initialize, void(const SuggestionsProfile&));
|
| + MOCK_METHOD2(GetImageForURL,
|
| void(const GURL&,
|
| base::Callback<void(const GURL&, const SkBitmap*)>));
|
| };
|
| @@ -184,11 +186,11 @@ class SuggestionsServiceTest : public testing::Test {
|
| // SuggestionsStore in |mock_suggestions_store_|.
|
| SuggestionsService* CreateSuggestionsServiceWithMocks() {
|
| mock_suggestions_store_ = new StrictMock<MockSuggestionsStore>();
|
| - mock_thumbnail_manager_ = new StrictMock<MockThumbnailManager>();
|
| + mock_thumbnail_manager_ = new NiceMock<MockImageManager>();
|
| mock_blacklist_store_ = new MockBlacklistStore();
|
| return new SuggestionsService(
|
| request_context_, scoped_ptr<SuggestionsStore>(mock_suggestions_store_),
|
| - scoped_ptr<ThumbnailManager>(mock_thumbnail_manager_),
|
| + scoped_ptr<ImageManager>(mock_thumbnail_manager_),
|
| scoped_ptr<BlacklistStore>(mock_blacklist_store_));
|
| }
|
|
|
| @@ -248,7 +250,7 @@ class SuggestionsServiceTest : public testing::Test {
|
| net::FakeURLFetcherFactory factory_;
|
| // Only used if the SuggestionsService is built with mocks. Not owned.
|
| MockSuggestionsStore* mock_suggestions_store_;
|
| - MockThumbnailManager* mock_thumbnail_manager_;
|
| + MockImageManager* mock_thumbnail_manager_;
|
| MockBlacklistStore* mock_blacklist_store_;
|
| scoped_refptr<net::TestURLRequestContextGetter> request_context_;
|
|
|
|
|