// 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 "base/basictypes.h" #include "base/file_util.h" #include "base/memory/ref_counted.h" #include "base/memory/scoped_ptr.h" #include "base/memory/scoped_vector.h" #include "base/message_loop.h" #include "base/path_service.h" #include "base/stl_util-inl.h" #include "base/string16.h" #include "base/string_util.h" #include "base/synchronization/waitable_event.h" #include "base/time.h" #include "base/utf_string_conversions.h" #include "chrome/browser/autofill/autofill_profile.h" #include "chrome/browser/autofill/credit_card.h" #include "chrome/browser/webdata/autofill_change.h" #include "chrome/browser/webdata/autofill_entry.h" #include "chrome/browser/webdata/web_data_service.h" #include "chrome/browser/webdata/web_data_service_test_util.h" #include "chrome/common/chrome_paths.h" #include "chrome/common/guid.h" #include "chrome/test/thread_observer_helper.h" #include "content/browser/browser_thread.h" #include "content/common/notification_details.h" #include "content/common/notification_service.h" #include "content/common/notification_type.h" #include "testing/gmock/include/gmock/gmock.h" #include "testing/gtest/include/gtest/gtest.h" #include "webkit/glue/form_field.h" using base::Time; using base::TimeDelta; using base::WaitableEvent; using testing::_; using testing::DoDefault; using testing::ElementsAreArray; using testing::Pointee; using testing::Property; typedef std::vector AutofillChangeList; static const int kWebDataServiceTimeoutSeconds = 8; ACTION_P(SignalEvent, event) { event->Signal(); } class AutofillDBThreadObserverHelper : public DBThreadObserverHelper { protected: virtual void RegisterObservers() { registrar_.Add(&observer_, NotificationType::AUTOFILL_ENTRIES_CHANGED, NotificationService::AllSources()); registrar_.Add(&observer_, NotificationType::AUTOFILL_PROFILE_CHANGED, NotificationService::AllSources()); registrar_.Add(&observer_, NotificationType::AUTOFILL_CREDIT_CARD_CHANGED, NotificationService::AllSources()); } }; class WebDataServiceTest : public testing::Test { public: WebDataServiceTest() : ui_thread_(BrowserThread::UI, &message_loop_), db_thread_(BrowserThread::DB) {} protected: virtual void SetUp() { db_thread_.Start(); PathService::Get(chrome::DIR_TEST_DATA, &profile_dir_); const std::string test_profile = "WebDataServiceTest"; profile_dir_ = profile_dir_.AppendASCII(test_profile); file_util::Delete(profile_dir_, true); file_util::CreateDirectory(profile_dir_); wds_ = new WebDataService(); wds_->Init(profile_dir_); } virtual void TearDown() { if (wds_.get()) wds_->Shutdown(); file_util::Delete(profile_dir_, true); db_thread_.Stop(); MessageLoop::current()->PostTask(FROM_HERE, new MessageLoop::QuitTask); MessageLoop::current()->Run(); } MessageLoopForUI message_loop_; BrowserThread ui_thread_; BrowserThread db_thread_; FilePath profile_dir_; scoped_refptr wds_; }; class WebDataServiceAutofillTest : public WebDataServiceTest { public: WebDataServiceAutofillTest() : WebDataServiceTest(), unique_id1_(1), unique_id2_(2), test_timeout_(TimeDelta::FromSeconds(kWebDataServiceTimeoutSeconds)), done_event_(false, false) {} protected: virtual void SetUp() { WebDataServiceTest::SetUp(); name1_ = ASCIIToUTF16("name1"); name2_ = ASCIIToUTF16("name2"); value1_ = ASCIIToUTF16("value1"); value2_ = ASCIIToUTF16("value2"); observer_helper_ = new AutofillDBThreadObserverHelper(); observer_helper_->Init(); } virtual void TearDown() { // Release this first so it can get destructed on the db thread. observer_helper_ = NULL; WebDataServiceTest::TearDown(); } void AppendFormField(const string16& name, const string16& value, std::vector* form_fields) { form_fields->push_back( webkit_glue::FormField(string16(), name, value, string16(), 0, false)); } string16 name1_; string16 name2_; string16 value1_; string16 value2_; int unique_id1_, unique_id2_; const TimeDelta test_timeout_; scoped_refptr observer_helper_; WaitableEvent done_event_; }; TEST_F(WebDataServiceAutofillTest, FormFillAdd) { const AutofillChange expected_changes[] = { AutofillChange(AutofillChange::ADD, AutofillKey(name1_, value1_)), AutofillChange(AutofillChange::ADD, AutofillKey(name2_, value2_)) }; // This will verify that the correct notification is triggered, // passing the correct list of autofill keys in the details. EXPECT_CALL( *observer_helper_->observer(), Observe(NotificationType(NotificationType::AUTOFILL_ENTRIES_CHANGED), Source(wds_.get()), Property(&Details::ptr, Pointee(ElementsAreArray(expected_changes))))). WillOnce(SignalEvent(&done_event_)); std::vector form_fields; AppendFormField(name1_, value1_, &form_fields); AppendFormField(name2_, value2_, &form_fields); wds_->AddFormFields(form_fields); // The event will be signaled when the mock observer is notified. done_event_.TimedWait(test_timeout_); AutofillWebDataServiceConsumer > consumer; WebDataService::Handle handle; static const int limit = 10; handle = wds_->GetFormValuesForElementName( name1_, string16(), limit, &consumer); // The message loop will exit when the consumer is called. MessageLoop::current()->Run(); EXPECT_EQ(handle, consumer.handle()); ASSERT_EQ(1U, consumer.result().size()); EXPECT_EQ(value1_, consumer.result()[0]); } TEST_F(WebDataServiceAutofillTest, FormFillRemoveOne) { // First add some values to autofill. EXPECT_CALL(*observer_helper_->observer(), Observe(_, _, _)). WillOnce(SignalEvent(&done_event_)); std::vector form_fields; AppendFormField(name1_, value1_, &form_fields); wds_->AddFormFields(form_fields); // The event will be signaled when the mock observer is notified. done_event_.TimedWait(test_timeout_); // This will verify that the correct notification is triggered, // passing the correct list of autofill keys in the details. const AutofillChange expected_changes[] = { AutofillChange(AutofillChange::REMOVE, AutofillKey(name1_, value1_)) }; EXPECT_CALL( *observer_helper_->observer(), Observe(NotificationType(NotificationType::AUTOFILL_ENTRIES_CHANGED), Source(wds_.get()), Property(&Details::ptr, Pointee(ElementsAreArray(expected_changes))))). WillOnce(SignalEvent(&done_event_)); wds_->RemoveFormValueForElementName(name1_, value1_); // The event will be signaled when the mock observer is notified. done_event_.TimedWait(test_timeout_); } TEST_F(WebDataServiceAutofillTest, FormFillRemoveMany) { TimeDelta one_day(TimeDelta::FromDays(1)); Time t = Time::Now(); EXPECT_CALL(*observer_helper_->observer(), Observe(_, _, _)). WillOnce(SignalEvent(&done_event_)); std::vector form_fields; AppendFormField(name1_, value1_, &form_fields); AppendFormField(name2_, value2_, &form_fields); wds_->AddFormFields(form_fields); // The event will be signaled when the mock observer is notified. done_event_.TimedWait(test_timeout_); // This will verify that the correct notification is triggered, // passing the correct list of autofill keys in the details. const AutofillChange expected_changes[] = { AutofillChange(AutofillChange::REMOVE, AutofillKey(name1_, value1_)), AutofillChange(AutofillChange::REMOVE, AutofillKey(name2_, value2_)) }; EXPECT_CALL( *observer_helper_->observer(), Observe(NotificationType(NotificationType::AUTOFILL_ENTRIES_CHANGED), Source(wds_.get()), Property(&Details::ptr, Pointee(ElementsAreArray(expected_changes))))). WillOnce(SignalEvent(&done_event_)); wds_->RemoveFormElementsAddedBetween(t, t + one_day); // The event will be signaled when the mock observer is notified. done_event_.TimedWait(test_timeout_); } TEST_F(WebDataServiceAutofillTest, ProfileAdd) { AutofillProfile profile; // Check that GUID-based notification was sent. const AutofillProfileChange expected_change( AutofillProfileChange::ADD, profile.guid(), &profile); EXPECT_CALL( *observer_helper_->observer(), Observe(NotificationType(NotificationType::AUTOFILL_PROFILE_CHANGED), Source(wds_.get()), Property(&Details::ptr, Pointee(expected_change)))). WillOnce(SignalEvent(&done_event_)); wds_->AddAutofillProfile(profile); done_event_.TimedWait(test_timeout_); // Check that it was added. AutofillWebDataServiceConsumer > consumer; WebDataService::Handle handle = wds_->GetAutofillProfiles(&consumer); MessageLoop::current()->Run(); EXPECT_EQ(handle, consumer.handle()); ASSERT_EQ(1U, consumer.result().size()); EXPECT_EQ(profile, *consumer.result()[0]); STLDeleteElements(&consumer.result()); } TEST_F(WebDataServiceAutofillTest, ProfileRemove) { AutofillProfile profile; // Add a profile. EXPECT_CALL(*observer_helper_->observer(), Observe(_, _, _)). Times(1). WillOnce(SignalEvent(&done_event_)); wds_->AddAutofillProfile(profile); done_event_.TimedWait(test_timeout_); // Check that it was added. AutofillWebDataServiceConsumer > consumer; WebDataService::Handle handle = wds_->GetAutofillProfiles(&consumer); MessageLoop::current()->Run(); EXPECT_EQ(handle, consumer.handle()); ASSERT_EQ(1U, consumer.result().size()); EXPECT_EQ(profile, *consumer.result()[0]); STLDeleteElements(&consumer.result()); // Check that GUID-based notification was sent. const AutofillProfileChange expected_change( AutofillProfileChange::REMOVE, profile.guid(), NULL); EXPECT_CALL( *observer_helper_->observer(), Observe(NotificationType(NotificationType::AUTOFILL_PROFILE_CHANGED), Source(wds_.get()), Property(&Details::ptr, Pointee(expected_change)))). WillOnce(SignalEvent(&done_event_)); // Remove the profile. wds_->RemoveAutofillProfile(profile.guid()); done_event_.TimedWait(test_timeout_); // Check that it was removed. AutofillWebDataServiceConsumer > consumer2; WebDataService::Handle handle2 = wds_->GetAutofillProfiles(&consumer2); MessageLoop::current()->Run(); EXPECT_EQ(handle2, consumer2.handle()); ASSERT_EQ(0U, consumer2.result().size()); } TEST_F(WebDataServiceAutofillTest, ProfileUpdate) { AutofillProfile profile1; profile1.SetInfo(NAME_FIRST, ASCIIToUTF16("Abe")); AutofillProfile profile2; profile2.SetInfo(NAME_FIRST, ASCIIToUTF16("Alice")); EXPECT_CALL(*observer_helper_->observer(), Observe(_, _, _)). WillOnce(DoDefault()). WillOnce(SignalEvent(&done_event_)); wds_->AddAutofillProfile(profile1); wds_->AddAutofillProfile(profile2); done_event_.TimedWait(test_timeout_); // Check that they were added. AutofillWebDataServiceConsumer > consumer; WebDataService::Handle handle = wds_->GetAutofillProfiles(&consumer); MessageLoop::current()->Run(); EXPECT_EQ(handle, consumer.handle()); ASSERT_EQ(2U, consumer.result().size()); EXPECT_EQ(profile1, *consumer.result()[0]); EXPECT_EQ(profile2, *consumer.result()[1]); STLDeleteElements(&consumer.result()); AutofillProfile profile1_changed(profile1); profile1_changed.SetInfo(NAME_FIRST, ASCIIToUTF16("Bill")); const AutofillProfileChange expected_change( AutofillProfileChange::UPDATE, profile1.guid(), &profile1_changed); EXPECT_CALL( *observer_helper_->observer(), Observe(NotificationType(NotificationType::AUTOFILL_PROFILE_CHANGED), Source(wds_.get()), Property(&Details::ptr, Pointee(expected_change)))). WillOnce(SignalEvent(&done_event_)); // Update the profile. wds_->UpdateAutofillProfile(profile1_changed); done_event_.TimedWait(test_timeout_); // Check that the updates were made. AutofillWebDataServiceConsumer > consumer2; WebDataService::Handle handle2 = wds_->GetAutofillProfiles(&consumer2); MessageLoop::current()->Run(); EXPECT_EQ(handle2, consumer2.handle()); ASSERT_EQ(2U, consumer2.result().size()); EXPECT_NE(profile1, *consumer2.result()[0]); EXPECT_EQ(profile1_changed, *consumer2.result()[0]); EXPECT_EQ(profile2, *consumer2.result()[1]); STLDeleteElements(&consumer2.result()); } TEST_F(WebDataServiceAutofillTest, CreditAdd) { CreditCard card; const AutofillCreditCardChange expected_change( AutofillCreditCardChange::ADD, card.guid(), &card); EXPECT_CALL( *observer_helper_->observer(), Observe( NotificationType(NotificationType::AUTOFILL_CREDIT_CARD_CHANGED), Source(wds_.get()), Property(&Details::ptr, Pointee(expected_change)))). WillOnce(SignalEvent(&done_event_)); wds_->AddCreditCard(card); done_event_.TimedWait(test_timeout_); // Check that it was added. AutofillWebDataServiceConsumer > consumer; WebDataService::Handle handle = wds_->GetCreditCards(&consumer); MessageLoop::current()->Run(); EXPECT_EQ(handle, consumer.handle()); ASSERT_EQ(1U, consumer.result().size()); EXPECT_EQ(card, *consumer.result()[0]); STLDeleteElements(&consumer.result()); } TEST_F(WebDataServiceAutofillTest, CreditCardRemove) { CreditCard credit_card; // Add a credit card. EXPECT_CALL(*observer_helper_->observer(), Observe(_, _, _)). WillOnce(SignalEvent(&done_event_)); wds_->AddCreditCard(credit_card); done_event_.TimedWait(test_timeout_); // Check that it was added. AutofillWebDataServiceConsumer > consumer; WebDataService::Handle handle = wds_->GetCreditCards(&consumer); MessageLoop::current()->Run(); EXPECT_EQ(handle, consumer.handle()); ASSERT_EQ(1U, consumer.result().size()); EXPECT_EQ(credit_card, *consumer.result()[0]); STLDeleteElements(&consumer.result()); // Remove the credit card. const AutofillCreditCardChange expected_change( AutofillCreditCardChange::REMOVE, credit_card.guid(), NULL); EXPECT_CALL( *observer_helper_->observer(), Observe( NotificationType(NotificationType::AUTOFILL_CREDIT_CARD_CHANGED), Source(wds_.get()), Property(&Details::ptr, Pointee(expected_change)))). WillOnce(SignalEvent(&done_event_)); wds_->RemoveCreditCard(credit_card.guid()); done_event_.TimedWait(test_timeout_); // Check that it was removed. AutofillWebDataServiceConsumer > consumer2; WebDataService::Handle handle2 = wds_->GetCreditCards(&consumer2); MessageLoop::current()->Run(); EXPECT_EQ(handle2, consumer2.handle()); ASSERT_EQ(0U, consumer2.result().size()); } TEST_F(WebDataServiceAutofillTest, CreditUpdate) { CreditCard card1; card1.SetInfo(CREDIT_CARD_NAME, ASCIIToUTF16("Abe")); CreditCard card2; card2.SetInfo(CREDIT_CARD_NAME, ASCIIToUTF16("Alice")); EXPECT_CALL(*observer_helper_->observer(), Observe(_, _, _)). Times(2). WillOnce(DoDefault()). WillOnce(SignalEvent(&done_event_)); wds_->AddCreditCard(card1); wds_->AddCreditCard(card2); done_event_.TimedWait(test_timeout_); // Check that they got added. AutofillWebDataServiceConsumer > consumer; WebDataService::Handle handle = wds_->GetCreditCards(&consumer); MessageLoop::current()->Run(); EXPECT_EQ(handle, consumer.handle()); ASSERT_EQ(2U, consumer.result().size()); EXPECT_EQ(card1, *consumer.result()[0]); EXPECT_EQ(card2, *consumer.result()[1]); STLDeleteElements(&consumer.result()); CreditCard card1_changed(card1); card1_changed.SetInfo(CREDIT_CARD_NAME, ASCIIToUTF16("Bill")); const AutofillCreditCardChange expected_change( AutofillCreditCardChange::UPDATE, card1.guid(), &card1_changed); EXPECT_CALL( *observer_helper_->observer(), Observe( NotificationType(NotificationType::AUTOFILL_CREDIT_CARD_CHANGED), Source(wds_.get()), Property(&Details::ptr, Pointee(expected_change)))). WillOnce(SignalEvent(&done_event_)); wds_->UpdateCreditCard(card1_changed); done_event_.TimedWait(test_timeout_); // Check that the updates were made. AutofillWebDataServiceConsumer > consumer2; WebDataService::Handle handle2 = wds_->GetCreditCards(&consumer2); MessageLoop::current()->Run(); EXPECT_EQ(handle2, consumer2.handle()); ASSERT_EQ(2U, consumer2.result().size()); EXPECT_NE(card1, *consumer2.result()[0]); EXPECT_EQ(card1_changed, *consumer2.result()[0]); EXPECT_EQ(card2, *consumer2.result()[1]); STLDeleteElements(&consumer2.result()); } TEST_F(WebDataServiceAutofillTest, AutofillRemoveModifiedBetween) { // Add a profile. EXPECT_CALL(*observer_helper_->observer(), Observe(_, _, _)). Times(1). WillOnce(SignalEvent(&done_event_)); AutofillProfile profile; wds_->AddAutofillProfile(profile); done_event_.TimedWait(test_timeout_); // Check that it was added. AutofillWebDataServiceConsumer > profile_consumer; WebDataService::Handle handle = wds_->GetAutofillProfiles(&profile_consumer); MessageLoop::current()->Run(); EXPECT_EQ(handle, profile_consumer.handle()); ASSERT_EQ(1U, profile_consumer.result().size()); EXPECT_EQ(profile, *profile_consumer.result()[0]); STLDeleteElements(&profile_consumer.result()); // Add a credit card. EXPECT_CALL(*observer_helper_->observer(), Observe(_, _, _)). WillOnce(SignalEvent(&done_event_)); CreditCard credit_card; wds_->AddCreditCard(credit_card); done_event_.TimedWait(test_timeout_); // Check that it was added. AutofillWebDataServiceConsumer > card_consumer; handle = wds_->GetCreditCards(&card_consumer); MessageLoop::current()->Run(); EXPECT_EQ(handle, card_consumer.handle()); ASSERT_EQ(1U, card_consumer.result().size()); EXPECT_EQ(credit_card, *card_consumer.result()[0]); STLDeleteElements(&card_consumer.result()); // Check that GUID-based notification was sent for the profile. const AutofillProfileChange expected_profile_change( AutofillProfileChange::REMOVE, profile.guid(), NULL); EXPECT_CALL( *observer_helper_->observer(), Observe(NotificationType(NotificationType::AUTOFILL_PROFILE_CHANGED), Source(wds_.get()), Property(&Details::ptr, Pointee(expected_profile_change)))). WillOnce(SignalEvent(&done_event_)); // Check that GUID-based notification was sent for the credit card. const AutofillCreditCardChange expected_card_change( AutofillCreditCardChange::REMOVE, credit_card.guid(), NULL); EXPECT_CALL( *observer_helper_->observer(), Observe( NotificationType(NotificationType::AUTOFILL_CREDIT_CARD_CHANGED), Source(wds_.get()), Property(&Details::ptr, Pointee(expected_card_change)))). WillOnce(SignalEvent(&done_event_)); // Remove the profile using time range of "all time". wds_->RemoveAutofillProfilesAndCreditCardsModifiedBetween(Time(), Time()); done_event_.TimedWait(test_timeout_); done_event_.TimedWait(test_timeout_); // Check that the profile was removed. AutofillWebDataServiceConsumer > profile_consumer2; WebDataService::Handle handle2 = wds_->GetAutofillProfiles(&profile_consumer2); MessageLoop::current()->Run(); EXPECT_EQ(handle2, profile_consumer2.handle()); ASSERT_EQ(0U, profile_consumer2.result().size()); // Check that the credit card was removed. AutofillWebDataServiceConsumer > card_consumer2; handle2 = wds_->GetCreditCards(&card_consumer2); MessageLoop::current()->Run(); EXPECT_EQ(handle2, card_consumer2.handle()); ASSERT_EQ(0U, card_consumer2.result().size()); }