Torne (Richard Coles) | a36e592 | 2013-08-05 13:57:33 +0100 | [diff] [blame] | 1 | // Copyright 2013 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 "base/bind.h" |
| 6 | #include "base/message_loop/message_loop.h" |
| 7 | #include "base/run_loop.h" |
| 8 | #include "base/strings/utf_string_conversions.h" |
| 9 | #include "base/threading/sequenced_worker_pool.h" |
| 10 | #include "chrome/browser/managed_mode/managed_user_refresh_token_fetcher.h" |
| 11 | #include "chrome/browser/managed_mode/managed_user_registration_utility.h" |
| 12 | #include "chrome/browser/managed_mode/managed_user_sync_service.h" |
| 13 | #include "chrome/browser/managed_mode/managed_user_sync_service_factory.h" |
| 14 | #include "chrome/browser/prefs/scoped_user_pref_update.h" |
| 15 | #include "chrome/common/pref_names.h" |
| 16 | #include "chrome/test/base/testing_pref_service_syncable.h" |
| 17 | #include "chrome/test/base/testing_profile.h" |
| 18 | #include "content/public/browser/browser_thread.h" |
| 19 | #include "google_apis/gaia/google_service_auth_error.h" |
| 20 | #include "sync/api/sync_change.h" |
| 21 | #include "sync/api/sync_error_factory_mock.h" |
| 22 | #include "sync/protocol/sync.pb.h" |
| 23 | #include "testing/gtest/include/gtest/gtest.h" |
| 24 | |
| 25 | using sync_pb::ManagedUserSpecifics; |
| 26 | using syncer::MANAGED_USERS; |
| 27 | using syncer::SyncChange; |
| 28 | using syncer::SyncChangeList; |
| 29 | using syncer::SyncChangeProcessor; |
| 30 | using syncer::SyncData; |
| 31 | using syncer::SyncDataList; |
| 32 | using syncer::SyncError; |
| 33 | using syncer::SyncErrorFactory; |
| 34 | using syncer::SyncMergeResult; |
| 35 | |
| 36 | namespace { |
| 37 | |
| 38 | const char kManagedUserToken[] = "managedusertoken"; |
| 39 | |
| 40 | class MockChangeProcessor : public SyncChangeProcessor { |
| 41 | public: |
| 42 | MockChangeProcessor() {} |
| 43 | virtual ~MockChangeProcessor() {} |
| 44 | |
| 45 | // SyncChangeProcessor implementation: |
| 46 | virtual SyncError ProcessSyncChanges( |
| 47 | const tracked_objects::Location& from_here, |
| 48 | const SyncChangeList& change_list) OVERRIDE; |
| 49 | |
| 50 | const SyncChangeList& changes() const { return change_list_; } |
| 51 | |
| 52 | private: |
| 53 | SyncChangeList change_list_; |
| 54 | }; |
| 55 | |
| 56 | SyncError MockChangeProcessor::ProcessSyncChanges( |
| 57 | const tracked_objects::Location& from_here, |
| 58 | const SyncChangeList& change_list) { |
| 59 | change_list_ = change_list; |
| 60 | return SyncError(); |
| 61 | } |
| 62 | |
| 63 | class MockManagedUserRefreshTokenFetcher |
| 64 | : public ManagedUserRefreshTokenFetcher { |
| 65 | public: |
| 66 | MockManagedUserRefreshTokenFetcher() {} |
| 67 | virtual ~MockManagedUserRefreshTokenFetcher() {} |
| 68 | |
| 69 | // ManagedUserRefreshTokenFetcher implementation: |
| 70 | virtual void Start(const std::string& managed_user_id, |
Torne (Richard Coles) | a36e592 | 2013-08-05 13:57:33 +0100 | [diff] [blame] | 71 | const std::string& device_name, |
| 72 | const TokenCallback& callback) OVERRIDE { |
| 73 | GoogleServiceAuthError error(GoogleServiceAuthError::NONE); |
| 74 | callback.Run(error, kManagedUserToken); |
| 75 | } |
| 76 | }; |
| 77 | |
| 78 | } // namespace |
| 79 | |
| 80 | class ManagedUserRegistrationUtilityTest : public ::testing::Test { |
| 81 | public: |
| 82 | ManagedUserRegistrationUtilityTest(); |
| 83 | virtual ~ManagedUserRegistrationUtilityTest(); |
| 84 | |
| 85 | virtual void TearDown() OVERRIDE; |
| 86 | |
| 87 | protected: |
| 88 | scoped_ptr<SyncChangeProcessor> CreateChangeProcessor(); |
| 89 | scoped_ptr<SyncErrorFactory> CreateErrorFactory(); |
| 90 | SyncData CreateRemoteData(const std::string& id, const std::string& name); |
| 91 | |
| 92 | SyncMergeResult StartInitialSync(); |
| 93 | |
| 94 | ManagedUserRegistrationUtility::RegistrationCallback |
| 95 | GetRegistrationCallback(); |
| 96 | |
| 97 | void Acknowledge(); |
| 98 | |
| 99 | PrefService* prefs() { return profile_.GetTestingPrefService(); } |
| 100 | ManagedUserSyncService* service() { return service_; } |
| 101 | MockChangeProcessor* change_processor() { return change_processor_; } |
| 102 | |
| 103 | bool received_callback() const { return received_callback_; } |
| 104 | const GoogleServiceAuthError& error() const { return error_; } |
| 105 | const std::string& token() const { return token_; } |
| 106 | |
| 107 | private: |
| 108 | void OnManagedUserRegistered(const GoogleServiceAuthError& error, |
| 109 | const std::string& token); |
| 110 | |
| 111 | base::MessageLoop message_loop_; |
| 112 | base::RunLoop run_loop_; |
| 113 | base::WeakPtrFactory<ManagedUserRegistrationUtilityTest> weak_ptr_factory_; |
| 114 | TestingProfile profile_; |
| 115 | ManagedUserSyncService* service_; |
| 116 | |
| 117 | // Owned by the ManagedUserSyncService. |
| 118 | MockChangeProcessor* change_processor_; |
| 119 | |
| 120 | // A unique ID for creating "remote" Sync data. |
| 121 | int64 sync_data_id_; |
| 122 | |
| 123 | // Whether OnManagedUserRegistered has been called. |
| 124 | bool received_callback_; |
| 125 | |
| 126 | // Hold the registration result (either an error, or a token). |
| 127 | GoogleServiceAuthError error_; |
| 128 | std::string token_; |
| 129 | }; |
| 130 | |
| 131 | ManagedUserRegistrationUtilityTest::ManagedUserRegistrationUtilityTest() |
| 132 | : weak_ptr_factory_(this), |
| 133 | change_processor_(NULL), |
| 134 | sync_data_id_(0), |
| 135 | received_callback_(false), |
| 136 | error_(GoogleServiceAuthError::NUM_STATES) { |
| 137 | service_ = ManagedUserSyncServiceFactory::GetForProfile(&profile_); |
| 138 | } |
| 139 | |
| 140 | ManagedUserRegistrationUtilityTest::~ManagedUserRegistrationUtilityTest() { |
| 141 | EXPECT_FALSE(weak_ptr_factory_.HasWeakPtrs()); |
| 142 | } |
| 143 | |
| 144 | void ManagedUserRegistrationUtilityTest::TearDown() { |
| 145 | content::BrowserThread::GetBlockingPool()->FlushForTesting(); |
| 146 | base::RunLoop().RunUntilIdle(); |
| 147 | } |
| 148 | |
| 149 | scoped_ptr<SyncChangeProcessor> |
| 150 | ManagedUserRegistrationUtilityTest::CreateChangeProcessor() { |
| 151 | EXPECT_FALSE(change_processor_); |
| 152 | change_processor_ = new MockChangeProcessor(); |
| 153 | return scoped_ptr<SyncChangeProcessor>(change_processor_); |
| 154 | } |
| 155 | |
| 156 | scoped_ptr<SyncErrorFactory> |
| 157 | ManagedUserRegistrationUtilityTest::CreateErrorFactory() { |
| 158 | return scoped_ptr<SyncErrorFactory>(new syncer::SyncErrorFactoryMock()); |
| 159 | } |
| 160 | |
| 161 | SyncMergeResult ManagedUserRegistrationUtilityTest::StartInitialSync() { |
| 162 | SyncDataList initial_sync_data; |
| 163 | SyncMergeResult result = |
| 164 | service()->MergeDataAndStartSyncing(MANAGED_USERS, |
| 165 | initial_sync_data, |
| 166 | CreateChangeProcessor(), |
| 167 | CreateErrorFactory()); |
| 168 | EXPECT_FALSE(result.error().IsSet()); |
| 169 | return result; |
| 170 | } |
| 171 | |
| 172 | ManagedUserRegistrationUtility::RegistrationCallback |
| 173 | ManagedUserRegistrationUtilityTest::GetRegistrationCallback() { |
| 174 | return base::Bind( |
| 175 | &ManagedUserRegistrationUtilityTest::OnManagedUserRegistered, |
| 176 | weak_ptr_factory_.GetWeakPtr()); |
| 177 | } |
| 178 | |
| 179 | void ManagedUserRegistrationUtilityTest::Acknowledge() { |
| 180 | SyncChangeList new_changes; |
| 181 | const SyncChangeList& changes = change_processor()->changes(); |
| 182 | for (SyncChangeList::const_iterator it = changes.begin(); it != changes.end(); |
| 183 | ++it) { |
| 184 | EXPECT_EQ(SyncChange::ACTION_ADD, it->change_type()); |
| 185 | ::sync_pb::EntitySpecifics specifics = it->sync_data().GetSpecifics(); |
| 186 | EXPECT_FALSE(specifics.managed_user().acknowledged()); |
| 187 | specifics.mutable_managed_user()->set_acknowledged(true); |
| 188 | new_changes.push_back( |
| 189 | SyncChange(FROM_HERE, SyncChange::ACTION_UPDATE, |
| 190 | SyncData::CreateRemoteData(++sync_data_id_, |
| 191 | specifics, |
| 192 | base::Time()))); |
| 193 | } |
| 194 | service()->ProcessSyncChanges(FROM_HERE, new_changes); |
| 195 | |
| 196 | run_loop_.Run(); |
| 197 | } |
| 198 | |
| 199 | void ManagedUserRegistrationUtilityTest::OnManagedUserRegistered( |
| 200 | const GoogleServiceAuthError& error, |
| 201 | const std::string& token) { |
| 202 | received_callback_ = true; |
| 203 | error_ = error; |
| 204 | token_ = token; |
| 205 | run_loop_.Quit(); |
| 206 | } |
| 207 | |
| 208 | TEST_F(ManagedUserRegistrationUtilityTest, Register) { |
| 209 | StartInitialSync(); |
| 210 | scoped_ptr<ManagedUserRefreshTokenFetcher> token_fetcher( |
| 211 | new MockManagedUserRefreshTokenFetcher); |
| 212 | ManagedUserRegistrationUtility registration_utility(prefs(), |
| 213 | token_fetcher.Pass(), |
| 214 | service()); |
| 215 | registration_utility.Register( |
Ben Murdoch | 2385ea3 | 2013-08-06 11:01:04 +0100 | [diff] [blame] | 216 | ManagedUserRegistrationUtility::GenerateNewManagedUserId(), |
Torne (Richard Coles) | a36e592 | 2013-08-05 13:57:33 +0100 | [diff] [blame] | 217 | ManagedUserRegistrationInfo(ASCIIToUTF16("Dug")), |
| 218 | GetRegistrationCallback()); |
| 219 | EXPECT_EQ(1u, prefs()->GetDictionary(prefs::kManagedUsers)->size()); |
| 220 | Acknowledge(); |
| 221 | |
| 222 | EXPECT_TRUE(received_callback()); |
| 223 | EXPECT_EQ(GoogleServiceAuthError::NONE, error().state()); |
| 224 | EXPECT_FALSE(token().empty()); |
| 225 | } |
| 226 | |
| 227 | TEST_F(ManagedUserRegistrationUtilityTest, RegisterBeforeInitialSync) { |
| 228 | scoped_ptr<ManagedUserRefreshTokenFetcher> token_fetcher( |
| 229 | new MockManagedUserRefreshTokenFetcher); |
| 230 | ManagedUserRegistrationUtility registration_utility(prefs(), |
| 231 | token_fetcher.Pass(), |
| 232 | service()); |
| 233 | registration_utility.Register( |
Ben Murdoch | 2385ea3 | 2013-08-06 11:01:04 +0100 | [diff] [blame] | 234 | ManagedUserRegistrationUtility::GenerateNewManagedUserId(), |
Torne (Richard Coles) | a36e592 | 2013-08-05 13:57:33 +0100 | [diff] [blame] | 235 | ManagedUserRegistrationInfo(ASCIIToUTF16("Nemo")), |
| 236 | GetRegistrationCallback()); |
| 237 | EXPECT_EQ(1u, prefs()->GetDictionary(prefs::kManagedUsers)->size()); |
| 238 | StartInitialSync(); |
| 239 | Acknowledge(); |
| 240 | |
| 241 | EXPECT_TRUE(received_callback()); |
| 242 | EXPECT_EQ(GoogleServiceAuthError::NONE, error().state()); |
| 243 | EXPECT_FALSE(token().empty()); |
| 244 | } |
| 245 | |
| 246 | TEST_F(ManagedUserRegistrationUtilityTest, SyncServiceShutdownBeforeRegFinish) { |
| 247 | StartInitialSync(); |
| 248 | scoped_ptr<ManagedUserRefreshTokenFetcher> token_fetcher( |
| 249 | new MockManagedUserRefreshTokenFetcher); |
| 250 | ManagedUserRegistrationUtility registration_utility(prefs(), |
| 251 | token_fetcher.Pass(), |
| 252 | service()); |
| 253 | registration_utility.Register( |
Ben Murdoch | 2385ea3 | 2013-08-06 11:01:04 +0100 | [diff] [blame] | 254 | ManagedUserRegistrationUtility::GenerateNewManagedUserId(), |
Torne (Richard Coles) | a36e592 | 2013-08-05 13:57:33 +0100 | [diff] [blame] | 255 | ManagedUserRegistrationInfo(ASCIIToUTF16("Remy")), |
| 256 | GetRegistrationCallback()); |
| 257 | EXPECT_EQ(1u, prefs()->GetDictionary(prefs::kManagedUsers)->size()); |
| 258 | service()->Shutdown(); |
| 259 | EXPECT_EQ(0u, prefs()->GetDictionary(prefs::kManagedUsers)->size()); |
| 260 | EXPECT_TRUE(received_callback()); |
| 261 | EXPECT_EQ(GoogleServiceAuthError::REQUEST_CANCELED, error().state()); |
| 262 | EXPECT_EQ(std::string(), token()); |
| 263 | } |
| 264 | |
| 265 | TEST_F(ManagedUserRegistrationUtilityTest, StopSyncingBeforeRegFinish) { |
| 266 | StartInitialSync(); |
| 267 | scoped_ptr<ManagedUserRefreshTokenFetcher> token_fetcher( |
| 268 | new MockManagedUserRefreshTokenFetcher); |
| 269 | ManagedUserRegistrationUtility registration_utility(prefs(), |
| 270 | token_fetcher.Pass(), |
| 271 | service()); |
| 272 | registration_utility.Register( |
Ben Murdoch | 2385ea3 | 2013-08-06 11:01:04 +0100 | [diff] [blame] | 273 | ManagedUserRegistrationUtility::GenerateNewManagedUserId(), |
Torne (Richard Coles) | a36e592 | 2013-08-05 13:57:33 +0100 | [diff] [blame] | 274 | ManagedUserRegistrationInfo(ASCIIToUTF16("Mike")), |
| 275 | GetRegistrationCallback()); |
| 276 | EXPECT_EQ(1u, prefs()->GetDictionary(prefs::kManagedUsers)->size()); |
| 277 | service()->StopSyncing(MANAGED_USERS); |
| 278 | EXPECT_EQ(0u, prefs()->GetDictionary(prefs::kManagedUsers)->size()); |
| 279 | EXPECT_TRUE(received_callback()); |
| 280 | EXPECT_EQ(GoogleServiceAuthError::REQUEST_CANCELED, error().state()); |
| 281 | EXPECT_EQ(std::string(), token()); |
| 282 | } |