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 6f4780e83bf92611b9797d21cabd25a9341fda98..81a7918254108472dd7ac98a15af17c4add75244 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,11 @@ class MockProvider : public ContentSuggestionsProvider { |
return statuses_[category.id()]; |
} |
+ CategoryInfo GetCategoryInfo(Category category) { |
+ return CategoryInfo(base::ASCIIToUTF16("Section title"), |
+ ContentSuggestionsCardLayout::FULL_CARD); |
+ } |
+ |
void FireSuggestionsChanged(Category category, std::vector<int> numbers) { |
observer()->OnNewSuggestions(this, category, CreateSuggestions(numbers)); |
} |