OLD | NEW |
1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "components/ntp_snippets/content_suggestions_service.h" | 5 #include "components/ntp_snippets/content_suggestions_service.h" |
6 | 6 |
7 #include <memory> | 7 #include <memory> |
8 #include <utility> | 8 #include <utility> |
9 #include <vector> | 9 #include <vector> |
10 | 10 |
(...skipping 70 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
81 } | 81 } |
82 | 82 |
83 void FireSuggestionInvalidated(const ContentSuggestion::ID& suggestion_id) { | 83 void FireSuggestionInvalidated(const ContentSuggestion::ID& suggestion_id) { |
84 observer()->OnSuggestionInvalidated(this, suggestion_id); | 84 observer()->OnSuggestionInvalidated(this, suggestion_id); |
85 } | 85 } |
86 | 86 |
87 MOCK_METHOD3(ClearHistory, | 87 MOCK_METHOD3(ClearHistory, |
88 void(base::Time begin, | 88 void(base::Time begin, |
89 base::Time end, | 89 base::Time end, |
90 const base::Callback<bool(const GURL& url)>& filter)); | 90 const base::Callback<bool(const GURL& url)>& filter)); |
| 91 MOCK_METHOD2(FetchMore, void(const Category&, const FetchedMoreCallback&)); |
91 MOCK_METHOD1(ClearCachedSuggestions, void(Category category)); | 92 MOCK_METHOD1(ClearCachedSuggestions, void(Category category)); |
92 MOCK_METHOD2(GetDismissedSuggestionsForDebugging, | 93 MOCK_METHOD2(GetDismissedSuggestionsForDebugging, |
93 void(Category category, | 94 void(Category category, |
94 const DismissedSuggestionsCallback& callback)); | 95 const DismissedSuggestionsCallback& callback)); |
95 MOCK_METHOD1(ClearDismissedSuggestionsForDebugging, void(Category category)); | 96 MOCK_METHOD1(ClearDismissedSuggestionsForDebugging, void(Category category)); |
96 MOCK_METHOD1(DismissSuggestion, | 97 MOCK_METHOD1(DismissSuggestion, |
97 void(const ContentSuggestion::ID& suggestion_id)); | 98 void(const ContentSuggestion::ID& suggestion_id)); |
98 MOCK_METHOD2(FetchSuggestionImage, | 99 MOCK_METHOD2(FetchSuggestionImage, |
99 void(const ContentSuggestion::ID& suggestion_id, | 100 void(const ContentSuggestion::ID& suggestion_id, |
100 const ImageFetchedCallback& callback)); | 101 const ImageFetchedCallback& callback)); |
(...skipping 467 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
568 TEST_F(ContentSuggestionsServiceTest, ShouldForwardClearHistory) { | 569 TEST_F(ContentSuggestionsServiceTest, ShouldForwardClearHistory) { |
569 Category category = FromKnownCategory(KnownCategories::DOWNLOADS); | 570 Category category = FromKnownCategory(KnownCategories::DOWNLOADS); |
570 MockProvider* provider = RegisterProvider(category); | 571 MockProvider* provider = RegisterProvider(category); |
571 base::Time begin = base::Time::FromTimeT(123), | 572 base::Time begin = base::Time::FromTimeT(123), |
572 end = base::Time::FromTimeT(456); | 573 end = base::Time::FromTimeT(456); |
573 EXPECT_CALL(*provider, ClearHistory(begin, end, _)); | 574 EXPECT_CALL(*provider, ClearHistory(begin, end, _)); |
574 base::Callback<bool(const GURL& url)> filter; | 575 base::Callback<bool(const GURL& url)> filter; |
575 service()->ClearHistory(begin, end, filter); | 576 service()->ClearHistory(begin, end, filter); |
576 } | 577 } |
577 | 578 |
| 579 TEST_F(ContentSuggestionsServiceTest, ShouldForwardFetchMore) { |
| 580 Category category = FromKnownCategory(KnownCategories::ARTICLES); |
| 581 MockProvider* provider = RegisterProvider(category); |
| 582 provider->FireCategoryStatusChangedWithCurrentStatus(category); |
| 583 EXPECT_CALL(*provider, FetchMore(category, _)); |
| 584 service()->FetchMore(category, |
| 585 ContentSuggestionsService::FetchedMoreCallback()); |
| 586 } |
| 587 |
578 TEST_F(ContentSuggestionsServiceTest, DismissAndRestoreCategory) { | 588 TEST_F(ContentSuggestionsServiceTest, DismissAndRestoreCategory) { |
579 // Register a category with one suggestion. | 589 // Register a category with one suggestion. |
580 Category category = FromKnownCategory(KnownCategories::ARTICLES); | 590 Category category = FromKnownCategory(KnownCategories::ARTICLES); |
581 MockProvider* provider = RegisterProvider(category); | 591 MockProvider* provider = RegisterProvider(category); |
582 provider->FireCategoryStatusChangedWithCurrentStatus(category); | 592 provider->FireCategoryStatusChangedWithCurrentStatus(category); |
583 provider->FireSuggestionsChanged(category, CreateSuggestions(category, {42})); | 593 provider->FireSuggestionsChanged(category, CreateSuggestions(category, {42})); |
584 | 594 |
585 EXPECT_THAT(service()->GetCategories(), ElementsAre(category)); | 595 EXPECT_THAT(service()->GetCategories(), ElementsAre(category)); |
586 EXPECT_THAT(service()->GetCategoryStatus(category), | 596 EXPECT_THAT(service()->GetCategoryStatus(category), |
587 Eq(CategoryStatus::AVAILABLE)); | 597 Eq(CategoryStatus::AVAILABLE)); |
(...skipping 95 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
683 provider = RegisterProvider(category); | 693 provider = RegisterProvider(category); |
684 provider->FireCategoryStatusChangedWithCurrentStatus(category); | 694 provider->FireCategoryStatusChangedWithCurrentStatus(category); |
685 EXPECT_TRUE(service()->IsCategoryDismissed(category)); | 695 EXPECT_TRUE(service()->IsCategoryDismissed(category)); |
686 | 696 |
687 service()->RestoreDismissedCategories(); | 697 service()->RestoreDismissedCategories(); |
688 EXPECT_FALSE(service()->IsCategoryDismissed(category)); | 698 EXPECT_FALSE(service()->IsCategoryDismissed(category)); |
689 EXPECT_THAT(providers().find(category)->second, Eq(provider)); | 699 EXPECT_THAT(providers().find(category)->second, Eq(provider)); |
690 } | 700 } |
691 | 701 |
692 } // namespace ntp_snippets | 702 } // namespace ntp_snippets |
OLD | NEW |