📰 Persist category dismissals

components/ntp_snippets/remote/ntp_snippets_service_unittest.cc

 // Copyright 2015 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "components/ntp_snippets/remote/ntp_snippets_service.h"
 
 #include <memory>
 #include <utility>
 #include <vector>
 
@@ skipping 292 matching lines @@
   FakeContentSuggestionsProviderObserver()
       : loaded_(base::WaitableEvent::ResetPolicy::MANUAL,
                 base::WaitableEvent::InitialState::NOT_SIGNALED) {}
 
   void OnNewSuggestions(ContentSuggestionsProvider* provider,
                         Category category,
                         std::vector<ContentSuggestion> suggestions) override {
     suggestions_[category] = std::move(suggestions);
   }
 
+  void OnNewSuggestionBatch(ContentSuggestionsProvider* provider,
+                            SuggestionBatch suggestions) override {
+    for (auto it = suggestions.begin(); it != suggestions.end(); ++it) {
+      suggestions_[it->first] = std::move(it->second);
+    }
+  }
+
   void OnCategoryStatusChanged(ContentSuggestionsProvider* provider,
                                Category category,
                                CategoryStatus new_status) override {
     if (category.IsKnownCategory(KnownCategories::ARTICLES) &&
         IsCategoryStatusAvailable(new_status)) {
       loaded_.Signal();
     }
     statuses_[category] = new_status;
   }
 
@@ skipping 949 matching lines @@
         base::StringPrintf("http://localhost/snippet-id-%d", i)));
   }
   LoadFromJSONString(service.get(), GetTestJson(suggestions));
   // TODO(tschumann): We should probably trim out any additional results and
   // only serve the MaxSnippetCount items.
   EXPECT_THAT(service->GetSnippetsForTesting(articles_category()),
               SizeIs(service->GetMaxSnippetCountForTesting() + 1));
 }
 
 }  // namespace ntp_snippets