Index: components/ntp_tiles/icon_cacher_impl_unittest.cc |
diff --git a/components/ntp_tiles/icon_cacher_impl_unittest.cc b/components/ntp_tiles/icon_cacher_impl_unittest.cc |
index 4a2ddc2513d15138b61251ba2eb978d72ae774fe..60e0561822b5134c48ad49d96c3e6568d1ed800e 100644 |
--- a/components/ntp_tiles/icon_cacher_impl_unittest.cc |
+++ b/components/ntp_tiles/icon_cacher_impl_unittest.cc |
@@ -18,6 +18,7 @@ |
#include "components/favicon/core/favicon_client.h" |
#include "components/favicon/core/favicon_service_impl.h" |
#include "components/favicon/core/favicon_util.h" |
+#include "components/favicon/core/large_icon_service.h" |
#include "components/history/core/browser/history_database_params.h" |
#include "components/history/core/browser/history_service.h" |
#include "components/image_fetcher/core/image_decoder.h" |
@@ -104,8 +105,12 @@ class IconCacherTest : public ::testing::Test { |
GURL()), // thumbnail, unused |
image_fetcher_(new ::testing::StrictMock<MockImageFetcher>), |
image_decoder_(new ::testing::StrictMock<MockImageDecoder>), |
+ task_runner_(new base::TestSimpleTaskRunner()), |
favicon_service_(/*favicon_client=*/nullptr, &history_service_), |
- task_runner_(new base::TestSimpleTaskRunner()) { |
+ large_icon_service_( |
+ &favicon_service_, |
+ task_runner_, |
+ base::MakeUnique<::testing::StrictMock<MockImageFetcher>>()) { |
CHECK(history_dir_.CreateUniqueTempDir()); |
CHECK(history_service_.Init( |
history::HistoryDatabaseParams(history_dir_.GetPath(), 0, 0))); |
@@ -184,8 +189,9 @@ class IconCacherTest : public ::testing::Test { |
std::unique_ptr<MockImageDecoder> image_decoder_; |
base::ScopedTempDir history_dir_; |
history::HistoryService history_service_; |
- favicon::FaviconServiceImpl favicon_service_; |
scoped_refptr<base::TestSimpleTaskRunner> task_runner_; |
+ favicon::FaviconServiceImpl favicon_service_; |
+ favicon::LargeIconService large_icon_service_; |
NiceMock<MockResourceDelegate> mock_resource_delegate_; |
}; |
@@ -223,8 +229,9 @@ TEST_F(IconCacherTest, LargeCached) { |
} |
PreloadIcon(site_.url, site_.large_icon_url, favicon_base::TOUCH_ICON, 128, |
128); |
- IconCacherImpl cacher(&favicon_service_, std::move(image_fetcher_)); |
- cacher.StartFetch(site_, done.Get(), done.Get()); |
+ IconCacherImpl cacher(&favicon_service_, &large_icon_service_, |
+ std::move(image_fetcher_)); |
+ cacher.StartFetchPopularSites(site_, done.Get(), done.Get()); |
WaitForTasksToFinish(); |
EXPECT_FALSE(IconIsCachedFor(site_.url, favicon_base::FAVICON)); |
EXPECT_TRUE(IconIsCachedFor(site_.url, favicon_base::TOUCH_ICON)); |
@@ -245,8 +252,9 @@ TEST_F(IconCacherTest, LargeNotCachedAndFetchSucceeded) { |
EXPECT_CALL(done, Run()).WillOnce(Quit(&loop)); |
} |
- IconCacherImpl cacher(&favicon_service_, std::move(image_fetcher_)); |
- cacher.StartFetch(site_, done.Get(), done.Get()); |
+ IconCacherImpl cacher(&favicon_service_, &large_icon_service_, |
+ std::move(image_fetcher_)); |
+ cacher.StartFetchPopularSites(site_, done.Get(), done.Get()); |
loop.Run(); |
EXPECT_FALSE(IconIsCachedFor(site_.url, favicon_base::FAVICON)); |
EXPECT_TRUE(IconIsCachedFor(site_.url, favicon_base::TOUCH_ICON)); |
@@ -269,8 +277,9 @@ TEST_F(IconCacherTest, SmallNotCachedAndFetchSucceeded) { |
EXPECT_CALL(done, Run()).WillOnce(Quit(&loop)); |
} |
- IconCacherImpl cacher(&favicon_service_, std::move(image_fetcher_)); |
- cacher.StartFetch(site_, done.Get(), done.Get()); |
+ IconCacherImpl cacher(&favicon_service_, &large_icon_service_, |
+ std::move(image_fetcher_)); |
+ cacher.StartFetchPopularSites(site_, done.Get(), done.Get()); |
loop.Run(); |
EXPECT_TRUE(IconIsCachedFor(site_.url, favicon_base::FAVICON)); |
EXPECT_FALSE(IconIsCachedFor(site_.url, favicon_base::TOUCH_ICON)); |
@@ -290,8 +299,9 @@ TEST_F(IconCacherTest, LargeNotCachedAndFetchFailed) { |
.WillOnce(FailFetch()); |
} |
- IconCacherImpl cacher(&favicon_service_, std::move(image_fetcher_)); |
- cacher.StartFetch(site_, done.Get(), done.Get()); |
+ IconCacherImpl cacher(&favicon_service_, &large_icon_service_, |
+ std::move(image_fetcher_)); |
+ cacher.StartFetchPopularSites(site_, done.Get(), done.Get()); |
WaitForTasksToFinish(); |
EXPECT_FALSE(IconIsCachedFor(site_.url, favicon_base::FAVICON)); |
EXPECT_FALSE(IconIsCachedFor(site_.url, favicon_base::TOUCH_ICON)); |
@@ -302,8 +312,9 @@ TEST_F(IconCacherTest, HandlesEmptyCallbacksNicely) { |
EXPECT_CALL(*image_fetcher_, SetDesiredImageFrameSize(_)); |
ON_CALL(*image_fetcher_, StartOrQueueNetworkRequest(_, _, _)) |
.WillByDefault(PassFetch(128, 128)); |
- IconCacherImpl cacher(&favicon_service_, std::move(image_fetcher_)); |
- cacher.StartFetch(site_, base::Closure(), base::Closure()); |
+ IconCacherImpl cacher(&favicon_service_, &large_icon_service_, |
+ std::move(image_fetcher_)); |
+ cacher.StartFetchPopularSites(site_, base::Closure(), base::Closure()); |
WaitForTasksToFinish(); |
} |
@@ -338,10 +349,11 @@ TEST_F(IconCacherTest, ProvidesDefaultIconAndSucceedsWithFetching) { |
EXPECT_CALL(icon_available, Run()).WillOnce(Quit(&fetch_loop)); |
} |
- IconCacherImpl cacher(&favicon_service_, std::move(image_fetcher_)); |
+ IconCacherImpl cacher(&favicon_service_, &large_icon_service_, |
+ std::move(image_fetcher_)); |
site_.default_icon_resource = 12345; |
- cacher.StartFetch(site_, icon_available.Get(), |
- preliminary_icon_available.Get()); |
+ cacher.StartFetchPopularSites(site_, icon_available.Get(), |
+ preliminary_icon_available.Get()); |
default_loop.Run(); // Wait for the default image. |
EXPECT_THAT(GetCachedIconFor(site_.url, favicon_base::TOUCH_ICON).Size(), |