| 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..dd14b8bd6b7d398b06ace43278868f1e9ac7a4ba 100644
|
| --- a/components/ntp_snippets/content_suggestions_service_unittest.cc
|
| +++ b/components/ntp_snippets/content_suggestions_service_unittest.cc
|
| @@ -88,6 +88,7 @@ class MockProvider : public ContentSuggestionsProvider {
|
| void(base::Time begin,
|
| base::Time end,
|
| const base::Callback<bool(const GURL& url)>& filter));
|
| + MOCK_METHOD2(FetchMore, void(const Category&, FetchedMoreCallback));
|
| MOCK_METHOD1(ClearCachedSuggestions, void(Category category));
|
| MOCK_METHOD2(GetDismissedSuggestionsForDebugging,
|
| void(Category category,
|
| @@ -575,6 +576,15 @@ TEST_F(ContentSuggestionsServiceTest, ShouldForwardClearHistory) {
|
| service()->ClearHistory(begin, end, filter);
|
| }
|
|
|
| +TEST_F(ContentSuggestionsServiceTest, ShouldForwardFetchMore) {
|
| + Category category = FromKnownCategory(KnownCategories::ARTICLES);
|
| + MockProvider* provider = RegisterProvider(category);
|
| + provider->FireCategoryStatusChangedWithCurrentStatus(category);
|
| + EXPECT_CALL(*provider, FetchMore(category, _));
|
| + service()->FetchMore(category,
|
| + ContentSuggestionsService::FetchedMoreCallback());
|
| +}
|
| +
|
| TEST_F(ContentSuggestionsServiceTest, DismissAndRestoreCategory) {
|
| // Register a category with one suggestion.
|
| Category category = FromKnownCategory(KnownCategories::ARTICLES);
|
|
|