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); |