12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061 |
- // Copyright 2017 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/fake_content_suggestions_provider_observer.h"
- #include <utility>
- #include "testing/gmock/include/gmock/gmock.h"
- #include "testing/gtest/include/gtest/gtest.h"
- namespace ntp_snippets {
- using testing::Eq;
- using testing::Not;
- FakeContentSuggestionsProviderObserver::
- FakeContentSuggestionsProviderObserver() = default;
- FakeContentSuggestionsProviderObserver::
- ~FakeContentSuggestionsProviderObserver() = default;
- void FakeContentSuggestionsProviderObserver::OnNewSuggestions(
- ContentSuggestionsProvider* provider,
- Category category,
- std::vector<ContentSuggestion> suggestions) {
- suggestions_[category] = std::move(suggestions);
- }
- void FakeContentSuggestionsProviderObserver::OnCategoryStatusChanged(
- ContentSuggestionsProvider* provider,
- Category category,
- CategoryStatus new_status) {
- statuses_[category] = new_status;
- }
- void FakeContentSuggestionsProviderObserver::OnSuggestionInvalidated(
- ContentSuggestionsProvider* provider,
- const ContentSuggestion::ID& suggestion_id) {
- FAIL() << "not implemented.";
- }
- const std::map<Category, CategoryStatus, Category::CompareByID>&
- FakeContentSuggestionsProviderObserver::statuses() const {
- return statuses_;
- }
- CategoryStatus FakeContentSuggestionsProviderObserver::StatusForCategory(
- Category category) const {
- auto it = statuses_.find(category);
- EXPECT_THAT(it, Not(Eq(statuses_.end())));
- return it->second;
- }
- const std::vector<ContentSuggestion>&
- FakeContentSuggestionsProviderObserver::SuggestionsForCategory(
- Category category) {
- return suggestions_[category];
- }
- } // namespace ntp_snippets
|