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 ba95a177d622614b12d79040babe2bb2b2d21a41..378604f640941c7dfae106c74a3bf0615436323c 100644 |
--- a/components/ntp_snippets/content_suggestions_service_unittest.cc |
+++ b/components/ntp_snippets/content_suggestions_service_unittest.cc |
@@ -11,6 +11,8 @@ |
#include "base/macros.h" |
#include "base/memory/ptr_util.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" |
@@ -68,6 +70,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)); |
} |