Add per-section clearing and dismissed suggestions to snippets-internals

components/ntp_snippets/ntp_snippets_service.h

 // 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.
 
 #ifndef COMPONENTS_NTP_SNIPPETS_NTP_SNIPPETS_SERVICE_H_
 #define COMPONENTS_NTP_SNIPPETS_NTP_SNIPPETS_SERVICE_H_
 
 #include <stddef.h>
 
 #include <memory>
@@ skipping 91 matching lines @@
   // for requests triggered by the user.
   void FetchSnippets(bool force_request);
 
   // Fetches snippets from the server for specified hosts (overriding
   // suggestions from the suggestion service) and adds them to the current ones.
   // Only called from chrome://snippets-internals, DO NOT USE otherwise!
   // Ignored while |loaded()| is false.
   void FetchSnippetsFromHosts(const std::set<std::string>& hosts,
                               bool force_request);
 
-  // Available snippets.
-  const NTPSnippet::PtrVector& snippets() const { return snippets_; }
-
-  // Returns the list of snippets previously dismissed by the user (that are
-  // not expired yet).
-  const NTPSnippet::PtrVector& dismissed_snippets() const {
-    return dismissed_snippets_;
-  }
-
   const NTPSnippetsFetcher* snippets_fetcher() const {
     return snippets_fetcher_.get();
   }
 
   // Returns a reason why the service is disabled, or DisabledReason::NONE
   // if it's not.
   DisabledReason disabled_reason() const {
     return snippets_status_service_->disabled_reason();
   }
 
   // (Re)schedules the periodic fetching of snippets. This is necessary because
   // the schedule depends on the time of day.
   void RescheduleFetching();
 
   // ContentSuggestionsProvider implementation
-  // TODO(pke): At some point reorder the implementations in the .cc file
-  // accordingly.
   std::vector<Category> GetProvidedCategories() override;
   CategoryStatus GetCategoryStatus(Category category) override;
   void DismissSuggestion(const std::string& suggestion_id) override;
   void FetchSuggestionImage(const std::string& suggestion_id,
                             const ImageFetchedCallback& callback) override;
-  void ClearCachedSuggestionsForDebugging() override;
-  void ClearDismissedSuggestionsForDebugging() override;
+  void ClearCachedSuggestionsForDebugging(Category category) override;
+  std::vector<ContentSuggestion> GetDismissedSuggestionsForDebugging(
+      Category category) override;
+  void ClearDismissedSuggestionsForDebugging(Category category) override;
 
   // Returns the lists of suggestion hosts the snippets are restricted to.
   std::set<std::string> GetSuggestionsHosts() const;
 
   // Returns the maximum number of snippets that will be shown at once.
   static int GetMaxSnippetCountForTesting();
 
+  // Available snippets, only for unit tests.
+  const NTPSnippet::PtrVector& GetSnippetsForTesting() const {
+    return snippets_;
+  }
+
  private:
   friend class NTPSnippetsServiceTest;
   FRIEND_TEST_ALL_PREFIXES(NTPSnippetsServiceTest, StatusChanges);
 
   // Possible state transitions:
   //       NOT_INITED --------+
   //       /       \          |
   //      v         v         |
   //   READY <--> DISABLED    |
   //       \       /          |
@@ skipping 144 matching lines @@
   bool fetch_after_load_;
 
   const Category provided_category_;
 
   DISALLOW_COPY_AND_ASSIGN(NTPSnippetsService);
 };
 
 }  // namespace ntp_snippets
 
 #endif  // COMPONENTS_NTP_SNIPPETS_NTP_SNIPPETS_SERVICE_H_