Add a unit test for the BookmarkSuggestionsProvider.

components/ntp_snippets/remote/remote_suggestions_provider_impl_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/remote_suggestions_provider_impl.h"
 
 #include <memory>
 #include <utility>
 #include <vector>
 
@@ skipping 13 matching lines @@
 #include "base/threading/thread_task_runner_handle.h"
 #include "base/time/time.h"
 #include "components/image_fetcher/image_decoder.h"
 #include "components/image_fetcher/image_fetcher.h"
 #include "components/image_fetcher/image_fetcher_delegate.h"
 #include "components/ntp_snippets/category.h"
 #include "components/ntp_snippets/category_info.h"
 #include "components/ntp_snippets/category_rankers/category_ranker.h"
 #include "components/ntp_snippets/category_rankers/constant_category_ranker.h"
 #include "components/ntp_snippets/category_rankers/mock_category_ranker.h"
+#include "components/ntp_snippets/fake_content_suggestions_provider_observer.h"
 #include "components/ntp_snippets/ntp_snippets_constants.h"
 #include "components/ntp_snippets/pref_names.h"
 #include "components/ntp_snippets/remote/ntp_snippet.h"
 #include "components/ntp_snippets/remote/ntp_snippets_fetcher.h"
 #include "components/ntp_snippets/remote/persistent_scheduler.h"
 #include "components/ntp_snippets/remote/remote_suggestions_database.h"
 #include "components/ntp_snippets/remote/test_utils.h"
 #include "components/ntp_snippets/user_classifier.h"
 #include "components/prefs/testing_pref_service.h"
 #include "components/signin/core/browser/fake_profile_oauth2_token_service.h"
@@ skipping 292 matching lines @@
  public:
   MOCK_METHOD1(SetImageFetcherDelegate, void(ImageFetcherDelegate*));
   MOCK_METHOD1(SetDataUseServiceName, void(DataUseServiceName));
   MOCK_METHOD3(
       StartOrQueueNetworkRequest,
       void(const std::string&,
            const GURL&,
            base::Callback<void(const std::string&, const gfx::Image&)>));
 };
 
-class FakeContentSuggestionsProviderObserver
-    : public ContentSuggestionsProvider::Observer {
- public:
-  FakeContentSuggestionsProviderObserver() = default;
-
-  void OnNewSuggestions(ContentSuggestionsProvider* provider,
-                        Category category,
-                        std::vector<ContentSuggestion> suggestions) override {
-    suggestions_[category] = std::move(suggestions);
-  }
-
-  void OnCategoryStatusChanged(ContentSuggestionsProvider* provider,
-                               Category category,
-                               CategoryStatus new_status) override {
-    statuses_[category] = new_status;
-  }
-
-  void OnSuggestionInvalidated(
-      ContentSuggestionsProvider* provider,
-      const ContentSuggestion::ID& suggestion_id) override {}
-
-  const std::map<Category, CategoryStatus, Category::CompareByID>& statuses()
-      const {
-    return statuses_;
-  }
-
-  CategoryStatus StatusForCategory(Category category) const {
-    auto it = statuses_.find(category);
-    EXPECT_THAT(it, Not(Eq(statuses_.end())));
-    return it->second;
-  }
-
-  const std::vector<ContentSuggestion>& SuggestionsForCategory(
-      Category category) {
-    return suggestions_[category];
-  }
-
- private:
-  std::map<Category, CategoryStatus, Category::CompareByID> statuses_;
-  std::map<Category, std::vector<ContentSuggestion>, Category::CompareByID>
-      suggestions_;
-
-  DISALLOW_COPY_AND_ASSIGN(FakeContentSuggestionsProviderObserver);
-};
-
 class FakeImageDecoder : public image_fetcher::ImageDecoder {
  public:
   FakeImageDecoder() {}
   ~FakeImageDecoder() override = default;
   void DecodeImage(
       const std::string& image_data,
       const image_fetcher::ImageDecodedCallback& callback) override {
     callback.Run(decoded_image_);
   }
 
@@ skipping 1454 matching lines @@
                          RemoteSuggestionsProvider::ProviderStatus)>::Call,
                      base::Unretained(&status_callback))));
 
   // Should be called when becoming disabled.
   EXPECT_CALL(status_callback,
               Call(RemoteSuggestionsProvider::ProviderStatus::INACTIVE));
   service->EnterState(RemoteSuggestionsProviderImpl::State::DISABLED);
 }
 
 }  // namespace ntp_snippets