// Copyright (c) 2011 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 #include #include #include "testing/gtest/include/gtest/gtest.h" #include "base/callback.h" #include "base/memory/ref_counted.h" #include "base/memory/scoped_ptr.h" #include "base/message_loop.h" #include "base/string16.h" #include "base/synchronization/waitable_event.h" #include "base/task.h" #include "base/time.h" #include "base/utf_string_conversions.h" #include "chrome/browser/autofill/autofill_common_test.h" #include "chrome/browser/sync/abstract_profile_sync_service_test.h" #include "chrome/browser/sync/engine/model_changing_syncer_command.h" #include "chrome/browser/sync/engine/syncapi.h" #include "chrome/browser/sync/glue/autofill_change_processor.h" #include "chrome/browser/sync/glue/autofill_data_type_controller.h" #include "chrome/browser/sync/glue/autofill_model_associator.h" #include "chrome/browser/sync/glue/autofill_profile_change_processor.h" #include "chrome/browser/sync/glue/autofill_profile_data_type_controller.h" #include "chrome/browser/sync/glue/autofill_profile_model_associator.h" #include "chrome/browser/sync/glue/data_type_controller.h" #include "chrome/browser/sync/profile_sync_factory.h" #include "chrome/browser/sync/profile_sync_service.h" #include "chrome/browser/sync/profile_sync_test_util.h" #include "chrome/browser/sync/protocol/autofill_specifics.pb.h" #include "chrome/browser/sync/syncable/autofill_migration.h" #include "chrome/browser/sync/syncable/directory_manager.h" #include "chrome/browser/sync/syncable/model_type.h" #include "chrome/browser/sync/syncable/syncable.h" #include "chrome/browser/sync/test_profile_sync_service.h" #include "chrome/browser/webdata/autofill_change.h" #include "chrome/browser/webdata/autofill_entry.h" #include "chrome/browser/webdata/autofill_table.h" #include "chrome/browser/webdata/web_database.h" #include "chrome/common/net/gaia/gaia_constants.h" #include "chrome/test/sync/engine/test_id_factory.h" #include "content/browser/browser_thread.h" #include "content/common/notification_source.h" #include "content/common/notification_type.h" #include "testing/gmock/include/gmock/gmock.h" using base::Time; using base::WaitableEvent; using browser_sync::AutofillChangeProcessor; using browser_sync::AutofillDataTypeController; using browser_sync::AutofillModelAssociator; using browser_sync::AutofillProfileChangeProcessor; using browser_sync::AutofillProfileDataTypeController; using browser_sync::AutofillProfileModelAssociator; using browser_sync::DataTypeController; using browser_sync::GROUP_DB; using browser_sync::kAutofillTag; using browser_sync::SyncBackendHostForProfileSyncTest; using browser_sync::UnrecoverableErrorHandler; using syncable::CREATE_NEW_UPDATE_ITEM; using syncable::AUTOFILL; using syncable::BASE_VERSION; using syncable::CREATE; using syncable::GET_BY_SERVER_TAG; using syncable::INVALID; using syncable::MutableEntry; using syncable::OriginalEntries; using syncable::SERVER_PARENT_ID; using syncable::SERVER_SPECIFICS; using syncable::SPECIFICS; using syncable::UNITTEST; using syncable::WriterTag; using syncable::WriteTransaction; using testing::_; using testing::DoAll; using testing::DoDefault; using testing::ElementsAre; using testing::Eq; using testing::Invoke; using testing::Mock; using testing::Return; using testing::SaveArg; using testing::SetArgumentPointee; namespace syncable { class Id; } class AutofillTableMock : public AutofillTable { public: AutofillTableMock() : AutofillTable(NULL, NULL) {} MOCK_METHOD2(RemoveFormElement, bool(const string16& name, const string16& value)); // NOLINT MOCK_METHOD1(GetAllAutofillEntries, bool(std::vector* entries)); // NOLINT MOCK_METHOD3(GetAutofillTimestamps, bool(const string16& name, // NOLINT const string16& value, std::vector* timestamps)); MOCK_METHOD1(UpdateAutofillEntries, bool(const std::vector&)); // NOLINT MOCK_METHOD1(GetAutofillProfiles, bool(std::vector*)); // NOLINT MOCK_METHOD1(UpdateAutofillProfile, bool(const AutofillProfile&)); // NOLINT MOCK_METHOD1(AddAutofillProfile, bool(const AutofillProfile&)); // NOLINT MOCK_METHOD1(RemoveAutofillProfile, bool(const std::string&)); // NOLINT }; class WebDatabaseFake : public WebDatabase { public: explicit WebDatabaseFake(AutofillTable* autofill_table) : autofill_table_(autofill_table) {} virtual AutofillTable* GetAutofillTable() { return autofill_table_; } private: AutofillTable* autofill_table_; }; class ProfileSyncServiceAutofillTest; template syncable::ModelType GetModelType() { return syncable::UNSPECIFIED; } template<> syncable::ModelType GetModelType() { return syncable::AUTOFILL; } template<> syncable::ModelType GetModelType() { return syncable::AUTOFILL_PROFILE; } class WebDataServiceFake : public WebDataService { public: explicit WebDataServiceFake(WebDatabase* web_database) : web_database_(web_database) {} virtual bool IsDatabaseLoaded() { return true; } virtual WebDatabase* GetDatabase() { return web_database_; } private: WebDatabase* web_database_; }; ACTION_P4(MakeAutofillSyncComponents, service, wd, pdm, dtc) { EXPECT_TRUE(BrowserThread::CurrentlyOn(BrowserThread::DB)); if (!BrowserThread::CurrentlyOn(BrowserThread::DB)) return ProfileSyncFactory::SyncComponents(NULL, NULL); AutofillModelAssociator* model_associator = new AutofillModelAssociator(service, wd, pdm); AutofillChangeProcessor* change_processor = new AutofillChangeProcessor(model_associator, wd, pdm, dtc); return ProfileSyncFactory::SyncComponents(model_associator, change_processor); } ACTION_P4(MakeAutofillProfileSyncComponents, service, wd, pdm, dtc) { EXPECT_TRUE(BrowserThread::CurrentlyOn(BrowserThread::DB)); if (!BrowserThread::CurrentlyOn(BrowserThread::DB)) return ProfileSyncFactory::SyncComponents(NULL, NULL); AutofillProfileModelAssociator* model_associator = new AutofillProfileModelAssociator(service, wd, pdm); AutofillProfileChangeProcessor* change_processor = new AutofillProfileChangeProcessor(model_associator, wd, pdm, dtc); return ProfileSyncFactory::SyncComponents(model_associator, change_processor); } class AbstractAutofillFactory { public: virtual AutofillDataTypeController* CreateDataTypeController( ProfileSyncFactory *factory, ProfileMock* profile, ProfileSyncService* service) = 0; virtual void SetExpectation(ProfileSyncFactoryMock* factory, ProfileSyncService* service, WebDatabase* wd, PersonalDataManager* pdm, DataTypeController* dtc) = 0; virtual ~AbstractAutofillFactory() {} }; class AutofillEntryFactory : public AbstractAutofillFactory { public: browser_sync::AutofillDataTypeController* CreateDataTypeController( ProfileSyncFactory *factory, ProfileMock* profile, ProfileSyncService* service) { return new AutofillDataTypeController(factory, profile); } void SetExpectation(ProfileSyncFactoryMock* factory, ProfileSyncService* service, WebDatabase* wd, PersonalDataManager* pdm, DataTypeController* dtc) { EXPECT_CALL(*factory, CreateAutofillSyncComponents(_,_,_,_)). WillOnce(MakeAutofillSyncComponents(service, wd, pdm, dtc)); } }; class AutofillProfileFactory : public AbstractAutofillFactory { public: browser_sync::AutofillDataTypeController* CreateDataTypeController( ProfileSyncFactory *factory, ProfileMock* profile, ProfileSyncService* service) { return new AutofillProfileDataTypeController(factory, profile); } void SetExpectation(ProfileSyncFactoryMock* factory, ProfileSyncService* service, WebDatabase* wd, PersonalDataManager* pdm, DataTypeController* dtc) { EXPECT_CALL(*factory, CreateAutofillProfileSyncComponents(_,_,_,_)). WillOnce(MakeAutofillProfileSyncComponents(service, wd, pdm, dtc)); } }; class PersonalDataManagerMock: public PersonalDataManager { public: MOCK_CONST_METHOD0(IsDataLoaded, bool()); MOCK_METHOD0(LoadProfiles, void()); MOCK_METHOD0(LoadCreditCards, void()); MOCK_METHOD0(Refresh, void()); }; template class AddAutofillTask; class ProfileSyncServiceAutofillTest : public AbstractProfileSyncServiceTest { protected: ProfileSyncServiceAutofillTest() {} AutofillProfileFactory profile_factory_; AutofillEntryFactory entry_factory_; AbstractAutofillFactory* GetFactory(syncable::ModelType type) { if (type == syncable::AUTOFILL) { return &entry_factory_; } else if (type == syncable::AUTOFILL_PROFILE) { return &profile_factory_; } else { NOTREACHED(); return NULL; } } virtual void SetUp() { AbstractProfileSyncServiceTest::SetUp(); profile_.CreateRequestContext(); web_database_.reset(new WebDatabaseFake(&autofill_table_)); web_data_service_ = new WebDataServiceFake(web_database_.get()); personal_data_manager_ = new PersonalDataManagerMock(); EXPECT_CALL(*personal_data_manager_, LoadProfiles()).Times(1); EXPECT_CALL(*personal_data_manager_, LoadCreditCards()).Times(1); personal_data_manager_->Init(&profile_); notification_service_ = new ThreadNotificationService(&db_thread_); notification_service_->Init(); } virtual void TearDown() { service_.reset(); notification_service_->TearDown(); profile_.ResetRequestContext(); AbstractProfileSyncServiceTest::TearDown(); } void StartSyncService(Task* task, bool will_fail_association, syncable::ModelType type) { AbstractAutofillFactory* factory = GetFactory(type); service_.reset( new TestProfileSyncService(&factory_, &profile_, "test_user", false, task)); EXPECT_CALL(profile_, GetProfileSyncService()).WillRepeatedly( Return(service_.get())); AutofillDataTypeController* data_type_controller = factory->CreateDataTypeController(&factory_, &profile_, service_.get()); SyncBackendHostForProfileSyncTest:: SetDefaultExpectationsForWorkerCreation(&profile_); factory->SetExpectation(&factory_, service_.get(), web_database_.get(), personal_data_manager_.get(), data_type_controller); EXPECT_CALL(factory_, CreateDataTypeManager(_, _)). WillOnce(ReturnNewDataTypeManager()); EXPECT_CALL(profile_, GetWebDataService(_)). WillOnce(Return(web_data_service_.get())); EXPECT_CALL(profile_, GetPersonalDataManager()). WillRepeatedly(Return(personal_data_manager_.get())); EXPECT_CALL(*personal_data_manager_, IsDataLoaded()). WillRepeatedly(Return(true)); // We need tokens to get the tests going token_service_.IssueAuthTokenForTest( GaiaConstants::kSyncService, "token"); EXPECT_CALL(profile_, GetTokenService()). WillRepeatedly(Return(&token_service_)); service_->RegisterDataTypeController(data_type_controller); service_->Initialize(); MessageLoop::current()->Run(); } bool AddAutofillSyncNode(const AutofillEntry& entry) { sync_api::WriteTransaction trans(service_->GetUserShare()); sync_api::ReadNode autofill_root(&trans); if (!autofill_root.InitByTagLookup(browser_sync::kAutofillTag)) return false; sync_api::WriteNode node(&trans); std::string tag = AutofillModelAssociator::KeyToTag(entry.key().name(), entry.key().value()); if (!node.InitUniqueByCreation(syncable::AUTOFILL, autofill_root, tag)) return false; AutofillChangeProcessor::WriteAutofillEntry(entry, &node); return true; } bool AddAutofillSyncNode(const AutofillProfile& profile) { sync_api::WriteTransaction trans(service_->GetUserShare()); sync_api::ReadNode autofill_root(&trans); if (!autofill_root.InitByTagLookup(browser_sync::kAutofillProfileTag)) return false; sync_api::WriteNode node(&trans); std::string tag = profile.guid(); if (!node.InitUniqueByCreation(syncable::AUTOFILL_PROFILE, autofill_root, tag)) return false; AutofillProfileChangeProcessor::WriteAutofillProfile(profile, &node); return true; } bool GetAutofillEntriesFromSyncDB(std::vector* entries, std::vector* profiles) { sync_api::ReadTransaction trans(service_->GetUserShare()); sync_api::ReadNode autofill_root(&trans); if (!autofill_root.InitByTagLookup(browser_sync::kAutofillTag)) return false; int64 child_id = autofill_root.GetFirstChildId(); while (child_id != sync_api::kInvalidId) { sync_api::ReadNode child_node(&trans); if (!child_node.InitByIdLookup(child_id)) return false; const sync_pb::AutofillSpecifics& autofill( child_node.GetAutofillSpecifics()); if (autofill.has_value()) { AutofillKey key(UTF8ToUTF16(autofill.name()), UTF8ToUTF16(autofill.value())); std::vector timestamps; int timestamps_count = autofill.usage_timestamp_size(); for (int i = 0; i < timestamps_count; ++i) { timestamps.push_back(Time::FromInternalValue( autofill.usage_timestamp(i))); } entries->push_back(AutofillEntry(key, timestamps)); } else if (autofill.has_profile()) { AutofillProfile p; p.set_guid(autofill.profile().guid()); AutofillProfileModelAssociator::OverwriteProfileWithServerData(&p, autofill.profile()); profiles->push_back(p); } child_id = child_node.GetSuccessorId(); } return true; } bool GetAutofillProfilesFromSyncDBUnderProfileNode( std::vector* profiles) { sync_api::ReadTransaction trans(service_->GetUserShare()); sync_api::ReadNode autofill_root(&trans); if (!autofill_root.InitByTagLookup(browser_sync::kAutofillProfileTag)) return false; int64 child_id = autofill_root.GetFirstChildId(); while (child_id != sync_api::kInvalidId) { sync_api::ReadNode child_node(&trans); if (!child_node.InitByIdLookup(child_id)) return false; const sync_pb::AutofillProfileSpecifics& autofill( child_node.GetAutofillProfileSpecifics()); AutofillProfile p; p.set_guid(autofill.guid()); AutofillProfileModelAssociator::OverwriteProfileWithServerData(&p, autofill); profiles->push_back(p); child_id = child_node.GetSuccessorId(); } return true; } void SetIdleChangeProcessorExpectations() { EXPECT_CALL(autofill_table_, RemoveFormElement(_, _)).Times(0); EXPECT_CALL(autofill_table_, GetAutofillTimestamps(_, _, _)).Times(0); EXPECT_CALL(autofill_table_, UpdateAutofillEntries(_)).Times(0); } static AutofillEntry MakeAutofillEntry(const char* name, const char* value, time_t timestamp0, time_t timestamp1) { std::vector