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 6943a36f9490cad1de67489b7f06b5e84196c942..206b016e3ec78a7b3ce62b41c62f7539c4860696 100644 |
--- a/components/ntp_snippets/content_suggestions_service_unittest.cc |
+++ b/components/ntp_snippets/content_suggestions_service_unittest.cc |
@@ -13,6 +13,8 @@ |
#include "base/message_loop/message_loop.h" |
#include "base/run_loop.h" |
#include "base/strings/string_number_conversions.h" |
+#include "base/strings/utf_string_conversions.h" |
+#include "components/ntp_snippets/category_info.h" |
#include "components/ntp_snippets/category_status.h" |
#include "components/ntp_snippets/content_suggestion.h" |
#include "components/ntp_snippets/content_suggestions_provider.h" |
@@ -71,6 +73,10 @@ class MockProvider : public ContentSuggestionsProvider { |
return statuses_[category.id()]; |
} |
+ CategoryInfo GetCategoryInfo(Category category) { |
+ return CategoryInfo(base::ASCIIToUTF16("Section title")); |
+ } |
+ |
void FireSuggestionsChanged(Category category, std::vector<int> numbers) { |
observer()->OnNewSuggestions(this, category, CreateSuggestions(numbers)); |
} |