Index: components/autofill/browser/webdata/web_data_service_unittest.cc |
diff --git a/components/autofill/browser/webdata/web_data_service_unittest.cc b/components/autofill/browser/webdata/web_data_service_unittest.cc |
index 362f828c660dba1ab15059b2623c440ac3b2e014..a6d0cae701537f792bca7f0cc86b146433e92c87 100644 |
--- a/components/autofill/browser/webdata/web_data_service_unittest.cc |
+++ b/components/autofill/browser/webdata/web_data_service_unittest.cc |
@@ -20,6 +20,7 @@ |
#include "components/autofill/browser/autofill_country.h" |
#include "components/autofill/browser/autofill_profile.h" |
#include "components/autofill/browser/credit_card.h" |
+#include "components/autofill/browser/webdata/autofill_backend_delegate.h" |
#include "components/autofill/browser/webdata/autofill_change.h" |
#include "components/autofill/browser/webdata/autofill_entry.h" |
#include "components/autofill/browser/webdata/autofill_table.h" |
@@ -128,24 +129,18 @@ class WebDataServiceAutofillTest : public WebDataServiceTest { |
value1_ = ASCIIToUTF16("value1"); |
value2_ = ASCIIToUTF16("value2"); |
- void(AutofillWebDataService::*add_observer_func)( |
- AutofillWebDataServiceObserverOnDBThread*) = |
- &AutofillWebDataService::AddObserver; |
- BrowserThread::PostTask( |
- BrowserThread::DB, |
- FROM_HERE, |
- base::Bind(add_observer_func, wds_, &observer_)); |
+ wds_->GetDelegateOnDB( |
+ base::Bind(&WebDataServiceAutofillTest::OnGetDelegate, |
+ base::Unretained(this))); |
WaitForDatabaseThread(); |
} |
virtual void TearDown() { |
- void(AutofillWebDataService::*remove_observer_func)( |
- AutofillWebDataServiceObserverOnDBThread*) = |
- &AutofillWebDataService::RemoveObserver; |
BrowserThread::PostTask( |
BrowserThread::DB, |
FROM_HERE, |
- base::Bind(remove_observer_func, wds_, &observer_)); |
+ base::Bind(&WebDataServiceAutofillTest::ClearDelegate, |
+ base::Unretained(this))); |
WaitForDatabaseThread(); |
WebDataServiceTest::TearDown(); |
@@ -160,6 +155,16 @@ class WebDataServiceAutofillTest : public WebDataServiceTest { |
form_fields->push_back(field); |
} |
+ void OnGetDelegate(AutofillBackendDelegate* delegate) { |
+ delegate_ = delegate; |
+ delegate_->AddObserver(&observer_); |
+ } |
+ |
+ void ClearDelegate() { |
+ delegate_->RemoveObserver(&observer_); |
+ delegate_ = NULL; |
+ } |
+ |
base::string16 name1_; |
base::string16 name2_; |
base::string16 value1_; |
@@ -167,6 +172,7 @@ class WebDataServiceAutofillTest : public WebDataServiceTest { |
int unique_id1_, unique_id2_; |
const TimeDelta test_timeout_; |
testing::NiceMock<MockAutofillWebDataServiceObserver> observer_; |
+ AutofillBackendDelegate* delegate_; |
WaitableEvent done_event_; |
}; |