| Index: chrome/browser/sync/glue/autofill_data_type_controller_unittest.cc
|
| diff --git a/chrome/browser/sync/glue/autofill_data_type_controller_unittest.cc b/chrome/browser/sync/glue/autofill_data_type_controller_unittest.cc
|
| index 1609a9f4d188b891c3538a5463957dcd9a983d97..6383c31001a363a082a1560e5b3b8d05a8d3fe83 100644
|
| --- a/chrome/browser/sync/glue/autofill_data_type_controller_unittest.cc
|
| +++ b/chrome/browser/sync/glue/autofill_data_type_controller_unittest.cc
|
| @@ -129,6 +129,33 @@ class FakeWebDataService : public WebDataService {
|
| DISALLOW_COPY_AND_ASSIGN(FakeWebDataService);
|
| };
|
|
|
| +class MockWebDataServiceWrapper : public WebDataServiceWrapper {
|
| + public:
|
| + static ProfileKeyedService* Build(Profile* profile) {
|
| + return new MockWebDataServiceWrapper();
|
| + }
|
| +
|
| + MockWebDataServiceWrapper() {
|
| + fake_web_data_service_ = new FakeWebDataService();
|
| + }
|
| +
|
| + void Shutdown() OVERRIDE {
|
| + fake_web_data_service_->ShutdownOnUIThread();
|
| + }
|
| +
|
| + scoped_refptr<WebDataService> GetWebData() OVERRIDE {
|
| + return fake_web_data_service_;
|
| + }
|
| +
|
| + ~MockWebDataServiceWrapper() {
|
| + fake_web_data_service_ = NULL;
|
| + }
|
| +
|
| + private:
|
| + scoped_refptr<FakeWebDataService> fake_web_data_service_;
|
| +
|
| +};
|
| +
|
| class SyncAutofillDataTypeControllerTest : public testing::Test {
|
| public:
|
| SyncAutofillDataTypeControllerTest()
|
| @@ -147,7 +174,7 @@ class SyncAutofillDataTypeControllerTest : public testing::Test {
|
| WillRepeatedly(Return(change_processor_.get()));
|
|
|
| WebDataServiceFactory::GetInstance()->SetTestingFactory(
|
| - &profile_, BuildWebDataService);
|
| + &profile_, MockWebDataServiceWrapper::Build);
|
|
|
| autofill_dtc_ =
|
| new AutofillDataTypeController(&profile_sync_factory_,
|
| @@ -175,11 +202,6 @@ class SyncAutofillDataTypeControllerTest : public testing::Test {
|
| change_processor_ = NULL;
|
| }
|
|
|
| - static scoped_refptr<RefcountedProfileKeyedService>
|
| - BuildWebDataService(Profile* profile) {
|
| - return new FakeWebDataService();
|
| - }
|
| -
|
| void BlockForDBThread() {
|
| base::RunLoop run_loop;
|
| ASSERT_TRUE(BrowserThread::PostTaskAndReply(BrowserThread::DB, FROM_HERE,
|
|
|