| Index: components/ntp_snippets/content_suggestions_service_unittest.cc
|
| diff --git a/components/ntp_snippets/content_suggestions_service_unittest.cc b/components/ntp_snippets/content_suggestions_service_unittest.cc
|
| index 99d7a3fd0259b478e397167124398b0371f48247..978d01b3ae409035dc4197a69a192d01126a7188 100644
|
| --- a/components/ntp_snippets/content_suggestions_service_unittest.cc
|
| +++ b/components/ntp_snippets/content_suggestions_service_unittest.cc
|
| @@ -88,6 +88,10 @@ class MockProvider : public ContentSuggestionsProvider {
|
| void(base::Time begin,
|
| base::Time end,
|
| const base::Callback<bool(const GURL& url)>& filter));
|
| + MOCK_METHOD3(Fetch,
|
| + void(const Category&,
|
| + const std::set<std::string>&,
|
| + FetchingCallback));
|
| MOCK_METHOD1(ClearCachedSuggestions, void(Category category));
|
| MOCK_METHOD2(GetDismissedSuggestionsForDebugging,
|
| void(Category category,
|
| @@ -575,6 +579,16 @@ TEST_F(ContentSuggestionsServiceTest, ShouldForwardClearHistory) {
|
| service()->ClearHistory(begin, end, filter);
|
| }
|
|
|
| +TEST_F(ContentSuggestionsServiceTest, ShouldForwardFetch) {
|
| + Category category = FromKnownCategory(KnownCategories::ARTICLES);
|
| + std::set<std::string> known_suggestions;
|
| + MockProvider* provider = RegisterProvider(category);
|
| + provider->FireCategoryStatusChangedWithCurrentStatus(category);
|
| + EXPECT_CALL(*provider, Fetch(category, known_suggestions, _));
|
| + service()->Fetch(category, known_suggestions,
|
| + ContentSuggestionsService::FetchingCallback());
|
| +}
|
| +
|
| TEST_F(ContentSuggestionsServiceTest, DismissAndRestoreCategory) {
|
| // Register a category with one suggestion.
|
| Category category = FromKnownCategory(KnownCategories::ARTICLES);
|
|
|