OLD | NEW |
(Empty) | |
| 1 // Copyright (c) 2011 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/sync/glue/new_non_frontend_data_type_controller.h" |
| 6 |
| 7 #include "base/callback.h" |
| 8 #include "base/memory/scoped_ptr.h" |
| 9 #include "base/message_loop.h" |
| 10 #include "base/task.h" |
| 11 #include "base/test/test_timeouts.h" |
| 12 #include "base/tracked_objects.h" |
| 13 #include "base/synchronization/waitable_event.h" |
| 14 #include "chrome/browser/sync/api/syncable_service_mock.h" |
| 15 #include "chrome/browser/sync/engine/model_safe_worker.h" |
| 16 #include "chrome/browser/sync/glue/data_type_controller_mock.h" |
| 17 #include "chrome/browser/sync/glue/new_non_frontend_data_type_controller_mock.h" |
| 18 #include "chrome/browser/sync/glue/shared_change_processor_mock.h" |
| 19 #include "chrome/browser/sync/profile_sync_factory_mock.h" |
| 20 #include "chrome/browser/sync/profile_sync_service_mock.h" |
| 21 #include "chrome/test/base/profile_mock.h" |
| 22 #include "content/browser/browser_thread.h" |
| 23 #include "testing/gtest/include/gtest/gtest.h" |
| 24 |
| 25 using base::WaitableEvent; |
| 26 using browser_sync::GenericChangeProcessor; |
| 27 using browser_sync::SharedChangeProcessorMock; |
| 28 using browser_sync::DataTypeController; |
| 29 using browser_sync::GROUP_DB; |
| 30 using browser_sync::NewNonFrontendDataTypeController; |
| 31 using browser_sync::NewNonFrontendDataTypeControllerMock; |
| 32 using browser_sync::StartCallback; |
| 33 using syncable::AUTOFILL_PROFILE; |
| 34 using testing::_; |
| 35 using testing::DoAll; |
| 36 using testing::InvokeWithoutArgs; |
| 37 using testing::Return; |
| 38 using testing::SetArgumentPointee; |
| 39 using testing::StrictMock; |
| 40 |
| 41 namespace browser_sync { |
| 42 |
| 43 namespace { |
| 44 |
| 45 ACTION_P(WaitOnEvent, event) { |
| 46 event->Wait(); |
| 47 } |
| 48 |
| 49 ACTION_P(SignalEvent, event) { |
| 50 event->Signal(); |
| 51 } |
| 52 |
| 53 ACTION_P(SaveChangeProcessor, scoped_change_processor) { |
| 54 scoped_change_processor->reset(arg2); |
| 55 } |
| 56 |
| 57 ACTION_P(GetWeakPtrToSyncableService, syncable_service) { |
| 58 // Have to do this within an Action to ensure it's not evaluated on the wrong |
| 59 // thread. |
| 60 return syncable_service->AsWeakPtr(); |
| 61 } |
| 62 |
| 63 class NewNonFrontendDataTypeControllerFake |
| 64 : public NewNonFrontendDataTypeController { |
| 65 public: |
| 66 NewNonFrontendDataTypeControllerFake( |
| 67 ProfileSyncFactory* profile_sync_factory, |
| 68 Profile* profile, |
| 69 NewNonFrontendDataTypeControllerMock* mock) |
| 70 : NewNonFrontendDataTypeController(profile_sync_factory, |
| 71 profile), |
| 72 mock_(mock) {} |
| 73 |
| 74 virtual syncable::ModelType type() const OVERRIDE { |
| 75 return AUTOFILL_PROFILE; |
| 76 } |
| 77 virtual ModelSafeGroup model_safe_group() const OVERRIDE { |
| 78 return GROUP_DB; |
| 79 } |
| 80 |
| 81 protected: |
| 82 virtual base::WeakPtr<SyncableService> |
| 83 GetWeakPtrToSyncableService() const OVERRIDE { |
| 84 return profile_sync_factory()->GetAutofillProfileSyncableService(NULL); |
| 85 } |
| 86 virtual bool StartAssociationAsync() OVERRIDE { |
| 87 mock_->StartAssociationAsync(); |
| 88 return BrowserThread::PostTask(BrowserThread::DB, FROM_HERE, |
| 89 base::Bind(&NewNonFrontendDataTypeControllerFake::StartAssociation, |
| 90 this)); |
| 91 } |
| 92 |
| 93 // We mock the following methods because their default implementations do |
| 94 // nothing, but we still want to make sure they're called appropriately. |
| 95 virtual bool StartModels() OVERRIDE { |
| 96 return mock_->StartModels(); |
| 97 } |
| 98 virtual void StopModels() OVERRIDE { |
| 99 mock_->StopModels(); |
| 100 } |
| 101 virtual void StopLocalServiceAsync() OVERRIDE { |
| 102 mock_->StopLocalServiceAsync(); |
| 103 BrowserThread::PostTask(BrowserThread::DB, FROM_HERE, |
| 104 base::Bind(&NewNonFrontendDataTypeControllerFake::StopLocalService, |
| 105 this)); |
| 106 } |
| 107 virtual void RecordUnrecoverableError( |
| 108 const tracked_objects::Location& from_here, |
| 109 const std::string& message) OVERRIDE { |
| 110 mock_->RecordUnrecoverableError(from_here, message); |
| 111 } |
| 112 virtual void RecordAssociationTime(base::TimeDelta time) OVERRIDE { |
| 113 mock_->RecordAssociationTime(time); |
| 114 } |
| 115 virtual void RecordStartFailure(DataTypeController::StartResult result) |
| 116 OVERRIDE { |
| 117 mock_->RecordStartFailure(result); |
| 118 } |
| 119 private: |
| 120 NewNonFrontendDataTypeControllerMock* mock_; |
| 121 }; |
| 122 |
| 123 class NewNonFrontendDataTypeControllerTest : public testing::Test { |
| 124 public: |
| 125 NewNonFrontendDataTypeControllerTest() |
| 126 : ui_thread_(BrowserThread::UI, &message_loop_), |
| 127 db_thread_(BrowserThread::DB) {} |
| 128 |
| 129 virtual void SetUp() OVERRIDE { |
| 130 EXPECT_CALL(profile_, GetProfileSyncService()).WillRepeatedly( |
| 131 Return(&service_)); |
| 132 EXPECT_CALL(service_, GetUserShare()).WillRepeatedly( |
| 133 Return((sync_api::UserShare*)NULL)); |
| 134 db_thread_.Start(); |
| 135 profile_sync_factory_.reset(new ProfileSyncFactoryMock()); |
| 136 change_processor_ = new SharedChangeProcessorMock(); |
| 137 |
| 138 // Both of these are refcounted, so don't need to be released. |
| 139 dtc_mock_ = new StrictMock<NewNonFrontendDataTypeControllerMock>(); |
| 140 new_non_frontend_dtc_ = |
| 141 new NewNonFrontendDataTypeControllerFake(profile_sync_factory_.get(), |
| 142 &profile_, |
| 143 dtc_mock_.get()); |
| 144 } |
| 145 |
| 146 virtual void TearDown() OVERRIDE { |
| 147 db_thread_.Stop(); |
| 148 } |
| 149 |
| 150 void WaitForDTC() { |
| 151 WaitableEvent done(true, false); |
| 152 BrowserThread::PostTask(BrowserThread::DB, FROM_HERE, |
| 153 base::Bind(&NewNonFrontendDataTypeControllerTest::SignalDone, |
| 154 &done)); |
| 155 done.TimedWait(base::TimeDelta::FromMilliseconds( |
| 156 TestTimeouts::action_timeout_ms())); |
| 157 if (!done.IsSignaled()) { |
| 158 ADD_FAILURE() << "Timed out waiting for DB thread to finish."; |
| 159 } |
| 160 MessageLoop::current()->RunAllPending(); |
| 161 } |
| 162 |
| 163 protected: |
| 164 void SetStartExpectations() { |
| 165 EXPECT_CALL(*dtc_mock_, StartModels()).WillOnce(Return(true)); |
| 166 EXPECT_CALL(*profile_sync_factory_, |
| 167 GetAutofillProfileSyncableService(_)). |
| 168 WillOnce(GetWeakPtrToSyncableService(&syncable_service_)); |
| 169 EXPECT_CALL(*profile_sync_factory_, |
| 170 CreateSharedChangeProcessor()). |
| 171 WillOnce(Return(change_processor_.get())); |
| 172 } |
| 173 |
| 174 void SetAssociateExpectations() { |
| 175 EXPECT_CALL(*dtc_mock_, StartAssociationAsync()); |
| 176 EXPECT_CALL(*change_processor_, Connect(_,_,_,_)).WillOnce(Return(true)); |
| 177 EXPECT_CALL(*change_processor_, CryptoReadyIfNecessary(_)). |
| 178 WillOnce(Return(true)); |
| 179 EXPECT_CALL(*change_processor_, SyncModelHasUserCreatedNodes(_,_)). |
| 180 WillOnce(DoAll(SetArgumentPointee<1>(true), Return(true))); |
| 181 EXPECT_CALL(*change_processor_, GetSyncDataForType(_,_)). |
| 182 WillOnce(Return(SyncError())); |
| 183 EXPECT_CALL(syncable_service_, MergeDataAndStartSyncing(_,_,_)). |
| 184 WillOnce(DoAll(SaveChangeProcessor(&saved_change_processor_), |
| 185 Return(SyncError()))); |
| 186 EXPECT_CALL(*dtc_mock_, RecordAssociationTime(_)); |
| 187 } |
| 188 |
| 189 void SetActivateExpectations(DataTypeController::StartResult result) { |
| 190 EXPECT_CALL(service_, ActivateDataType(_, _, _)); |
| 191 EXPECT_CALL(start_callback_, Run(result,_)); |
| 192 } |
| 193 |
| 194 void SetStopExpectations() { |
| 195 EXPECT_CALL(*dtc_mock_, StopModels()); |
| 196 EXPECT_CALL(*change_processor_, Disconnect()).WillOnce(Return(true)); |
| 197 EXPECT_CALL(service_, DeactivateDataType(_)); |
| 198 EXPECT_CALL(*dtc_mock_, StopLocalServiceAsync()); |
| 199 EXPECT_CALL(syncable_service_, StopSyncing(_)); |
| 200 } |
| 201 |
| 202 void SetStartFailExpectations(DataTypeController::StartResult result) { |
| 203 EXPECT_CALL(*dtc_mock_, StopModels()); |
| 204 EXPECT_CALL(*dtc_mock_, RecordStartFailure(result)); |
| 205 EXPECT_CALL(start_callback_, Run(result,_)); |
| 206 } |
| 207 |
| 208 static void SignalDone(WaitableEvent* done) { |
| 209 done->Signal(); |
| 210 } |
| 211 |
| 212 MessageLoopForUI message_loop_; |
| 213 BrowserThread ui_thread_; |
| 214 BrowserThread db_thread_; |
| 215 ProfileMock profile_; |
| 216 scoped_ptr<ProfileSyncFactoryMock> profile_sync_factory_; |
| 217 ProfileSyncServiceMock service_; |
| 218 StartCallback start_callback_; |
| 219 // Must be destroyed after new_non_frontend_dtc_. |
| 220 SyncableServiceMock syncable_service_; |
| 221 scoped_refptr<NewNonFrontendDataTypeControllerFake> new_non_frontend_dtc_; |
| 222 scoped_refptr<NewNonFrontendDataTypeControllerMock> dtc_mock_; |
| 223 scoped_refptr<SharedChangeProcessorMock> change_processor_; |
| 224 scoped_ptr<SyncChangeProcessor> saved_change_processor_; |
| 225 }; |
| 226 |
| 227 TEST_F(NewNonFrontendDataTypeControllerTest, StartOk) { |
| 228 SetStartExpectations(); |
| 229 SetAssociateExpectations(); |
| 230 SetActivateExpectations(DataTypeController::OK); |
| 231 EXPECT_EQ(DataTypeController::NOT_RUNNING, new_non_frontend_dtc_->state()); |
| 232 new_non_frontend_dtc_->Start( |
| 233 NewCallback(&start_callback_, &StartCallback::Run)); |
| 234 WaitForDTC(); |
| 235 EXPECT_EQ(DataTypeController::RUNNING, new_non_frontend_dtc_->state()); |
| 236 } |
| 237 |
| 238 TEST_F(NewNonFrontendDataTypeControllerTest, StartFirstRun) { |
| 239 SetStartExpectations(); |
| 240 EXPECT_CALL(*dtc_mock_, StartAssociationAsync()); |
| 241 EXPECT_CALL(*change_processor_, Connect(_,_,_,_)).WillOnce(Return(true)); |
| 242 EXPECT_CALL(*change_processor_, CryptoReadyIfNecessary(_)). |
| 243 WillOnce(Return(true)); |
| 244 EXPECT_CALL(*change_processor_, SyncModelHasUserCreatedNodes(_,_)). |
| 245 WillOnce(DoAll(SetArgumentPointee<1>(false), Return(true))); |
| 246 EXPECT_CALL(*change_processor_, GetSyncDataForType(_,_)). |
| 247 WillOnce(Return(SyncError())); |
| 248 EXPECT_CALL(syncable_service_, MergeDataAndStartSyncing(_,_,_)). |
| 249 WillOnce(DoAll(SaveChangeProcessor(&saved_change_processor_), |
| 250 Return(SyncError()))); |
| 251 EXPECT_CALL(*dtc_mock_, RecordAssociationTime(_)); |
| 252 SetActivateExpectations(DataTypeController::OK_FIRST_RUN); |
| 253 EXPECT_EQ(DataTypeController::NOT_RUNNING, new_non_frontend_dtc_->state()); |
| 254 new_non_frontend_dtc_->Start( |
| 255 NewCallback(&start_callback_, &StartCallback::Run)); |
| 256 WaitForDTC(); |
| 257 EXPECT_EQ(DataTypeController::RUNNING, new_non_frontend_dtc_->state()); |
| 258 } |
| 259 |
| 260 TEST_F(NewNonFrontendDataTypeControllerTest, AbortDuringStartModels) { |
| 261 EXPECT_CALL(*dtc_mock_, StartModels()).WillOnce(Return(false)); |
| 262 SetStartFailExpectations(DataTypeController::ABORTED); |
| 263 EXPECT_EQ(DataTypeController::NOT_RUNNING, new_non_frontend_dtc_->state()); |
| 264 new_non_frontend_dtc_->Start( |
| 265 NewCallback(&start_callback_, &StartCallback::Run)); |
| 266 WaitForDTC(); |
| 267 EXPECT_EQ(DataTypeController::MODEL_STARTING, new_non_frontend_dtc_->state()); |
| 268 new_non_frontend_dtc_->Stop(); |
| 269 EXPECT_EQ(DataTypeController::NOT_RUNNING, new_non_frontend_dtc_->state()); |
| 270 } |
| 271 |
| 272 TEST_F(NewNonFrontendDataTypeControllerTest, StartAssociationFailed) { |
| 273 SetStartExpectations(); |
| 274 EXPECT_CALL(*dtc_mock_, StartAssociationAsync()); |
| 275 EXPECT_CALL(*change_processor_, Connect(_,_,_,_)).WillOnce(Return(true)); |
| 276 EXPECT_CALL(*change_processor_, CryptoReadyIfNecessary(_)). |
| 277 WillOnce(Return(true)); |
| 278 EXPECT_CALL(*change_processor_, SyncModelHasUserCreatedNodes(_,_)). |
| 279 WillOnce(DoAll(SetArgumentPointee<1>(true), Return(true))); |
| 280 EXPECT_CALL(*change_processor_, GetSyncDataForType(_,_)). |
| 281 WillOnce(Return(SyncError())); |
| 282 EXPECT_CALL(syncable_service_, MergeDataAndStartSyncing(_,_,_)). |
| 283 WillOnce(DoAll(SaveChangeProcessor(&saved_change_processor_), |
| 284 Return(SyncError(FROM_HERE, "failed", AUTOFILL_PROFILE)))); |
| 285 EXPECT_CALL(*dtc_mock_, RecordAssociationTime(_)); |
| 286 SetStartFailExpectations(DataTypeController::ASSOCIATION_FAILED); |
| 287 EXPECT_CALL(syncable_service_, StopSyncing(_)); |
| 288 // Set up association to fail with an association failed error. |
| 289 EXPECT_EQ(DataTypeController::NOT_RUNNING, new_non_frontend_dtc_->state()); |
| 290 new_non_frontend_dtc_->Start( |
| 291 NewCallback(&start_callback_, &StartCallback::Run)); |
| 292 WaitForDTC(); |
| 293 EXPECT_EQ(DataTypeController::DISABLED, new_non_frontend_dtc_->state()); |
| 294 } |
| 295 |
| 296 TEST_F(NewNonFrontendDataTypeControllerTest, |
| 297 StartAssociationTriggersUnrecoverableError) { |
| 298 SetStartExpectations(); |
| 299 SetStartFailExpectations(DataTypeController::UNRECOVERABLE_ERROR); |
| 300 // Set up association to fail with an unrecoverable error. |
| 301 EXPECT_CALL(*dtc_mock_, StartAssociationAsync()); |
| 302 EXPECT_CALL(*change_processor_, Connect(_,_,_,_)).WillOnce(Return(true)); |
| 303 EXPECT_CALL(*change_processor_, CryptoReadyIfNecessary(_)). |
| 304 WillRepeatedly(Return(true)); |
| 305 EXPECT_CALL(*change_processor_, SyncModelHasUserCreatedNodes(_,_)). |
| 306 WillRepeatedly(DoAll(SetArgumentPointee<1>(false), Return(false))); |
| 307 EXPECT_EQ(DataTypeController::NOT_RUNNING, new_non_frontend_dtc_->state()); |
| 308 new_non_frontend_dtc_->Start( |
| 309 NewCallback(&start_callback_, &StartCallback::Run)); |
| 310 WaitForDTC(); |
| 311 EXPECT_EQ(DataTypeController::NOT_RUNNING, new_non_frontend_dtc_->state()); |
| 312 } |
| 313 |
| 314 TEST_F(NewNonFrontendDataTypeControllerTest, StartAssociationCryptoNotReady) { |
| 315 SetStartExpectations(); |
| 316 SetStartFailExpectations(DataTypeController::NEEDS_CRYPTO); |
| 317 // Set up association to fail with a NEEDS_CRYPTO error. |
| 318 EXPECT_CALL(*dtc_mock_, StartAssociationAsync()); |
| 319 EXPECT_CALL(*change_processor_, Connect(_,_,_,_)).WillOnce(Return(true)); |
| 320 EXPECT_CALL(*change_processor_, CryptoReadyIfNecessary(_)). |
| 321 WillRepeatedly(Return(false)); |
| 322 EXPECT_EQ(DataTypeController::NOT_RUNNING, new_non_frontend_dtc_->state()); |
| 323 new_non_frontend_dtc_->Start( |
| 324 NewCallback(&start_callback_, &StartCallback::Run)); |
| 325 WaitForDTC(); |
| 326 EXPECT_EQ(DataTypeController::NOT_RUNNING, new_non_frontend_dtc_->state()); |
| 327 } |
| 328 |
| 329 // Trigger a Stop() call when we check if the model associator has user created |
| 330 // nodes. |
| 331 TEST_F(NewNonFrontendDataTypeControllerTest, AbortDuringAssociation) { |
| 332 WaitableEvent wait_for_db_thread_pause(false, false); |
| 333 WaitableEvent pause_db_thread(false, false); |
| 334 |
| 335 SetStartExpectations(); |
| 336 SetStartFailExpectations(DataTypeController::ABORTED); |
| 337 EXPECT_CALL(*dtc_mock_, StartAssociationAsync()); |
| 338 EXPECT_CALL(*change_processor_, Connect(_,_,_,_)).WillOnce(Return(true)); |
| 339 EXPECT_CALL(*change_processor_, CryptoReadyIfNecessary(_)). |
| 340 WillOnce(Return(true)); |
| 341 EXPECT_CALL(*change_processor_, SyncModelHasUserCreatedNodes(_,_)). |
| 342 WillOnce(DoAll( |
| 343 SignalEvent(&wait_for_db_thread_pause), |
| 344 WaitOnEvent(&pause_db_thread), |
| 345 SetArgumentPointee<1>(true), |
| 346 Return(true))); |
| 347 EXPECT_CALL(*change_processor_, GetSyncDataForType(_,_)). |
| 348 WillOnce(Return(SyncError(FROM_HERE, "Disconnected.", AUTOFILL_PROFILE))); |
| 349 EXPECT_CALL(*change_processor_, Disconnect()). |
| 350 WillOnce(DoAll(SignalEvent(&pause_db_thread), Return(true))); |
| 351 EXPECT_CALL(service_, DeactivateDataType(_)); |
| 352 EXPECT_CALL(*dtc_mock_, StopLocalServiceAsync()); |
| 353 EXPECT_CALL(syncable_service_, StopSyncing(_)); |
| 354 EXPECT_EQ(DataTypeController::NOT_RUNNING, new_non_frontend_dtc_->state()); |
| 355 new_non_frontend_dtc_->Start( |
| 356 NewCallback(&start_callback_, &StartCallback::Run)); |
| 357 wait_for_db_thread_pause.Wait(); |
| 358 new_non_frontend_dtc_->Stop(); |
| 359 WaitForDTC(); |
| 360 EXPECT_EQ(DataTypeController::NOT_RUNNING, new_non_frontend_dtc_->state()); |
| 361 } |
| 362 |
| 363 TEST_F(NewNonFrontendDataTypeControllerTest, Stop) { |
| 364 SetStartExpectations(); |
| 365 SetAssociateExpectations(); |
| 366 SetActivateExpectations(DataTypeController::OK); |
| 367 SetStopExpectations(); |
| 368 EXPECT_EQ(DataTypeController::NOT_RUNNING, new_non_frontend_dtc_->state()); |
| 369 new_non_frontend_dtc_->Start( |
| 370 NewCallback(&start_callback_, &StartCallback::Run)); |
| 371 WaitForDTC(); |
| 372 EXPECT_EQ(DataTypeController::RUNNING, new_non_frontend_dtc_->state()); |
| 373 new_non_frontend_dtc_->Stop(); |
| 374 EXPECT_EQ(DataTypeController::NOT_RUNNING, new_non_frontend_dtc_->state()); |
| 375 } |
| 376 |
| 377 TEST_F(NewNonFrontendDataTypeControllerTest, OnUnrecoverableError) { |
| 378 SetStartExpectations(); |
| 379 SetAssociateExpectations(); |
| 380 SetActivateExpectations(DataTypeController::OK); |
| 381 EXPECT_CALL(*dtc_mock_, RecordUnrecoverableError(_, "Test")); |
| 382 EXPECT_CALL(service_, OnUnrecoverableError(_,_)).WillOnce( |
| 383 InvokeWithoutArgs(new_non_frontend_dtc_.get(), |
| 384 &NewNonFrontendDataTypeController::Stop)); |
| 385 SetStopExpectations(); |
| 386 EXPECT_EQ(DataTypeController::NOT_RUNNING, new_non_frontend_dtc_->state()); |
| 387 new_non_frontend_dtc_->Start( |
| 388 NewCallback(&start_callback_, &StartCallback::Run)); |
| 389 WaitForDTC(); |
| 390 EXPECT_EQ(DataTypeController::RUNNING, new_non_frontend_dtc_->state()); |
| 391 // This should cause new_non_frontend_dtc_->Stop() to be called. |
| 392 BrowserThread::PostTask(BrowserThread::DB, FROM_HERE, |
| 393 base::Bind( |
| 394 &NewNonFrontendDataTypeControllerFake::OnUnrecoverableError, |
| 395 new_non_frontend_dtc_.get(), |
| 396 FROM_HERE, |
| 397 std::string("Test"))); |
| 398 WaitForDTC(); |
| 399 EXPECT_EQ(DataTypeController::NOT_RUNNING, new_non_frontend_dtc_->state()); |
| 400 } |
| 401 |
| 402 } // namespace |
| 403 |
| 404 } // namespace browser_sync |
OLD | NEW |