Ntp: show AllDismissedItem when all sections have been dismissed.

components/ntp_snippets/content_suggestions_service_unittest.cc

 // Copyright 2016 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/content_suggestions_service.h"
 
 #include <memory>
 #include <utility>
 #include <vector>
 
@@ skipping 146 matching lines @@
       ADD_FAILURE() << "Suggestion number " << number
                     << " not present, though expected";
     }
   }
 
   const std::map<Category, ContentSuggestionsProvider*, Category::CompareByID>&
   providers() {
     return service()->providers_by_category_;
   }
 
+  const std::map<Category, ContentSuggestionsProvider*, Category::CompareByID>&
+  dismissed_providers() {
+    return service()->dismissed_providers_by_category_;
+  }
+
   CategoryFactory* category_factory() { return service()->category_factory(); }
 
   Category FromKnownCategory(KnownCategories known_category) {
     return service()->category_factory()->FromKnownCategory(known_category);
   }
 
   Category FromRemoteCategory(int remote_category) {
     return service()->category_factory()->FromRemoteCategory(remote_category);
   }
 
@@ skipping 417 matching lines @@
 TEST_F(ContentSuggestionsServiceTest, ShouldForwardClearHistory) {
   Category category = FromKnownCategory(KnownCategories::DOWNLOADS);
   MockProvider* provider = RegisterProvider(category);
   base::Time begin = base::Time::FromTimeT(123),
              end = base::Time::FromTimeT(456);
   EXPECT_CALL(*provider, ClearHistory(begin, end, _));
   base::Callback<bool(const GURL& url)> filter;
   service()->ClearHistory(begin, end, filter);
 }
 
+TEST_F(ContentSuggestionsServiceTest, DismissAndRestoreCategory) {
+  // Register a category with one suggestion.
+  Category category = FromKnownCategory(KnownCategories::ARTICLES);
+  MockProvider* provider = RegisterProvider(category);
+  provider->FireCategoryStatusChangedWithCurrentStatus(category);
+  provider->FireSuggestionsChanged(category, CreateSuggestions(category, {42}));
+
+  EXPECT_THAT(service()->GetCategories(), ElementsAre(category));
+  EXPECT_THAT(service()->GetCategoryStatus(category),
+              Eq(CategoryStatus::AVAILABLE));
+  ExpectThatSuggestionsAre(category, {42});
+  EXPECT_THAT(providers().count(category), Eq(1ul));
+  EXPECT_THAT(dismissed_providers(), IsEmpty());
+
+  // Dismissing the category clears the suggestions for it.
+  service()->DismissCategory(category);
+
+  EXPECT_THAT(service()->GetCategories(), IsEmpty());
+  EXPECT_THAT(service()->GetCategoryStatus(category),
+              Eq(CategoryStatus::NOT_PROVIDED));
+  EXPECT_THAT(service()->GetSuggestionsForCategory(category), IsEmpty());
+  EXPECT_THAT(providers(), IsEmpty());
+  EXPECT_THAT(dismissed_providers().count(category), Eq(1ul));
+
+  // Restoring the dismissed category makes it available again but it is still
+  // empty.
+  service()->RestoreDismissedCategories();
+
+  EXPECT_THAT(service()->GetCategories(), ElementsAre(category));
+  EXPECT_THAT(service()->GetCategoryStatus(category),
+              Eq(CategoryStatus::AVAILABLE));
+  EXPECT_THAT(service()->GetSuggestionsForCategory(category), IsEmpty());
+  EXPECT_THAT(providers().count(category), Eq(1ul));
+  EXPECT_THAT(dismissed_providers(), IsEmpty());
+}
+
 }  // namespace ntp_snippets