OLD | NEW |
(Empty) | |
| 1 // Copyright 2017 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. |
| 4 |
| 5 #include "chrome/browser/search/one_google_bar/one_google_bar_service.h" |
| 6 |
| 7 #include <memory> |
| 8 #include <utility> |
| 9 #include <vector> |
| 10 |
| 11 #include "base/macros.h" |
| 12 #include "base/memory/ptr_util.h" |
| 13 #include "base/optional.h" |
| 14 #include "chrome/browser/search/one_google_bar/one_google_bar_data.h" |
| 15 #include "chrome/browser/search/one_google_bar/one_google_bar_fetcher.h" |
| 16 #include "components/signin/core/browser/account_tracker_service.h" |
| 17 #include "components/signin/core/browser/fake_profile_oauth2_token_service.h" |
| 18 #include "components/signin/core/browser/fake_signin_manager.h" |
| 19 #include "components/signin/core/browser/test_signin_client.h" |
| 20 #include "components/sync_preferences/testing_pref_service_syncable.h" |
| 21 #include "testing/gmock/include/gmock/gmock.h" |
| 22 #include "testing/gtest/include/gtest/gtest.h" |
| 23 |
| 24 using testing::Eq; |
| 25 using testing::StrictMock; |
| 26 |
| 27 class FakeOneGoogleBarFetcher : public OneGoogleBarFetcher { |
| 28 public: |
| 29 void Fetch(OneGoogleCallback callback) override { |
| 30 callbacks_.push_back(std::move(callback)); |
| 31 } |
| 32 |
| 33 size_t GetCallbackCount() const { return callbacks_.size(); } |
| 34 |
| 35 void RespondToAllCallbacks(const base::Optional<OneGoogleBarData>& data) { |
| 36 for (OneGoogleCallback& callback : callbacks_) { |
| 37 std::move(callback).Run(data); |
| 38 } |
| 39 callbacks_.clear(); |
| 40 } |
| 41 |
| 42 private: |
| 43 std::vector<OneGoogleCallback> callbacks_; |
| 44 }; |
| 45 |
| 46 class MockOneGoogleBarServiceObserver : public OneGoogleBarServiceObserver { |
| 47 public: |
| 48 MOCK_METHOD0(OnOneGoogleBarDataChanged, void()); |
| 49 }; |
| 50 |
| 51 class OneGoogleBarServiceTest : public testing::Test { |
| 52 public: |
| 53 OneGoogleBarServiceTest() |
| 54 : signin_client_(&pref_service_), |
| 55 signin_manager_(&signin_client_, &account_tracker_) { |
| 56 SigninManagerBase::RegisterProfilePrefs(pref_service_.registry()); |
| 57 SigninManagerBase::RegisterPrefs(pref_service_.registry()); |
| 58 |
| 59 auto fetcher = base::MakeUnique<FakeOneGoogleBarFetcher>(); |
| 60 fetcher_ = fetcher.get(); |
| 61 service_ = base::MakeUnique<OneGoogleBarService>(&signin_manager_, |
| 62 std::move(fetcher)); |
| 63 } |
| 64 |
| 65 FakeOneGoogleBarFetcher* fetcher() { return fetcher_; } |
| 66 OneGoogleBarService* service() { return service_.get(); } |
| 67 |
| 68 private: |
| 69 sync_preferences::TestingPrefServiceSyncable pref_service_; |
| 70 TestSigninClient signin_client_; |
| 71 AccountTrackerService account_tracker_; |
| 72 FakeSigninManagerBase signin_manager_; |
| 73 |
| 74 // Owned by the service. |
| 75 FakeOneGoogleBarFetcher* fetcher_; |
| 76 |
| 77 std::unique_ptr<OneGoogleBarService> service_; |
| 78 }; |
| 79 |
| 80 TEST_F(OneGoogleBarServiceTest, RefreshesOnRequest) { |
| 81 ASSERT_THAT(service()->one_google_bar_data(), Eq(base::nullopt)); |
| 82 |
| 83 // Request a refresh. That should arrive at the fetcher. |
| 84 service()->Refresh(); |
| 85 EXPECT_THAT(fetcher()->GetCallbackCount(), Eq(1u)); |
| 86 |
| 87 // Fulfill it. |
| 88 OneGoogleBarData data; |
| 89 data.bar_html = "<div></div>"; |
| 90 fetcher()->RespondToAllCallbacks(data); |
| 91 EXPECT_THAT(service()->one_google_bar_data(), Eq(data)); |
| 92 |
| 93 // Request another refresh. |
| 94 service()->Refresh(); |
| 95 EXPECT_THAT(fetcher()->GetCallbackCount(), Eq(1u)); |
| 96 |
| 97 // For now, the old data should still be there. |
| 98 EXPECT_THAT(service()->one_google_bar_data(), Eq(data)); |
| 99 |
| 100 // Fulfill the second request. |
| 101 OneGoogleBarData other_data; |
| 102 other_data.bar_html = "<div>Different!</div>"; |
| 103 fetcher()->RespondToAllCallbacks(other_data); |
| 104 EXPECT_THAT(service()->one_google_bar_data(), Eq(other_data)); |
| 105 } |
| 106 |
| 107 TEST_F(OneGoogleBarServiceTest, NotifiesObserverOnChanges) { |
| 108 ASSERT_THAT(service()->one_google_bar_data(), Eq(base::nullopt)); |
| 109 |
| 110 StrictMock<MockOneGoogleBarServiceObserver> observer; |
| 111 service()->AddObserver(&observer); |
| 112 |
| 113 // Empty result from a fetch doesn't change anything (it's already empty), so |
| 114 // should not result in a notification. |
| 115 service()->Refresh(); |
| 116 fetcher()->RespondToAllCallbacks(base::nullopt); |
| 117 |
| 118 // Non-empty response should result in a notification. |
| 119 service()->Refresh(); |
| 120 OneGoogleBarData data; |
| 121 data.bar_html = "<div></div>"; |
| 122 EXPECT_CALL(observer, OnOneGoogleBarDataChanged()); |
| 123 fetcher()->RespondToAllCallbacks(data); |
| 124 EXPECT_THAT(service()->one_google_bar_data(), Eq(data)); |
| 125 |
| 126 // Non-empty but identical response should not result in another notification. |
| 127 service()->Refresh(); |
| 128 OneGoogleBarData identical_data = data; |
| 129 fetcher()->RespondToAllCallbacks(identical_data); |
| 130 |
| 131 // Different response should result in a notification. |
| 132 service()->Refresh(); |
| 133 OneGoogleBarData other_data; |
| 134 data.bar_html = "<div>Different</div>"; |
| 135 EXPECT_CALL(observer, OnOneGoogleBarDataChanged()); |
| 136 fetcher()->RespondToAllCallbacks(other_data); |
| 137 EXPECT_THAT(service()->one_google_bar_data(), Eq(other_data)); |
| 138 |
| 139 // Finally, an empty response should result in a notification now. |
| 140 service()->Refresh(); |
| 141 EXPECT_CALL(observer, OnOneGoogleBarDataChanged()); |
| 142 fetcher()->RespondToAllCallbacks(base::nullopt); |
| 143 EXPECT_THAT(service()->one_google_bar_data(), Eq(base::nullopt)); |
| 144 |
| 145 service()->RemoveObserver(&observer); |
| 146 } |
| 147 |
| 148 #if !defined(OS_CHROMEOS) |
| 149 |
| 150 // Like OneGoogleBarServiceTest, but it has a FakeSigninManager (rather than |
| 151 // FakeSigninManagerBase), so it can simulate sign-in and sign-out. |
| 152 class OneGoogleBarServiceSignInTest : public testing::Test { |
| 153 public: |
| 154 OneGoogleBarServiceSignInTest() |
| 155 : signin_client_(&pref_service_), |
| 156 signin_manager_(&signin_client_, |
| 157 &token_service_, |
| 158 &account_tracker_, |
| 159 nullptr) { |
| 160 SigninManagerBase::RegisterProfilePrefs(pref_service_.registry()); |
| 161 SigninManagerBase::RegisterPrefs(pref_service_.registry()); |
| 162 AccountTrackerService::RegisterPrefs(pref_service_.registry()); |
| 163 |
| 164 account_tracker_.Initialize(&signin_client_); |
| 165 |
| 166 auto fetcher = base::MakeUnique<FakeOneGoogleBarFetcher>(); |
| 167 fetcher_ = fetcher.get(); |
| 168 service_ = base::MakeUnique<OneGoogleBarService>(&signin_manager_, |
| 169 std::move(fetcher)); |
| 170 } |
| 171 |
| 172 void SignIn() { signin_manager_.SignIn("account", "username", "pass"); } |
| 173 void SignOut() { signin_manager_.ForceSignOut(); } |
| 174 |
| 175 FakeOneGoogleBarFetcher* fetcher() { return fetcher_; } |
| 176 OneGoogleBarService* service() { return service_.get(); } |
| 177 |
| 178 private: |
| 179 sync_preferences::TestingPrefServiceSyncable pref_service_; |
| 180 TestSigninClient signin_client_; |
| 181 FakeProfileOAuth2TokenService token_service_; |
| 182 AccountTrackerService account_tracker_; |
| 183 FakeSigninManager signin_manager_; |
| 184 |
| 185 // Owned by the service. |
| 186 FakeOneGoogleBarFetcher* fetcher_; |
| 187 |
| 188 std::unique_ptr<OneGoogleBarService> service_; |
| 189 }; |
| 190 |
| 191 TEST_F(OneGoogleBarServiceSignInTest, ResetsOnSignIn) { |
| 192 // Load some data. |
| 193 service()->Refresh(); |
| 194 OneGoogleBarData data; |
| 195 data.bar_html = "<div></div>"; |
| 196 fetcher()->RespondToAllCallbacks(data); |
| 197 ASSERT_THAT(service()->one_google_bar_data(), Eq(data)); |
| 198 |
| 199 // Sign in. This should clear the cached data. |
| 200 SignIn(); |
| 201 EXPECT_THAT(service()->one_google_bar_data(), Eq(base::nullopt)); |
| 202 } |
| 203 |
| 204 TEST_F(OneGoogleBarServiceSignInTest, ResetsOnSignOut) { |
| 205 SignIn(); |
| 206 |
| 207 // Load some data. |
| 208 service()->Refresh(); |
| 209 OneGoogleBarData data; |
| 210 data.bar_html = "<div></div>"; |
| 211 fetcher()->RespondToAllCallbacks(data); |
| 212 ASSERT_THAT(service()->one_google_bar_data(), Eq(data)); |
| 213 |
| 214 // Sign out. This should clear the cached data. |
| 215 SignOut(); |
| 216 EXPECT_THAT(service()->one_google_bar_data(), Eq(base::nullopt)); |
| 217 } |
| 218 |
| 219 #endif // OS_CHROMEOS |
OLD | NEW |