Chromium Code Reviews| Index: components/sync_driver/non_ui_model_type_controller_unittest.cc |
| diff --git a/components/sync_driver/non_ui_model_type_controller_unittest.cc b/components/sync_driver/non_ui_model_type_controller_unittest.cc |
| index a309c36a04f34b4ad6f19ab6b1e1b3928be3c6d8..629b719debcfad665518d0076cba3ea3b0c1c318 100644 |
| --- a/components/sync_driver/non_ui_model_type_controller_unittest.cc |
| +++ b/components/sync_driver/non_ui_model_type_controller_unittest.cc |
| @@ -54,8 +54,6 @@ class TestNonUIModelTypeController : public NonUIModelTypeController { |
| return model_task_runner_->PostTask(from_here, task); |
| } |
| - void InitializeProcessorInTest() { InitializeProcessor(); } |
| - |
| private: |
| ~TestNonUIModelTypeController() override {} |
| @@ -147,6 +145,30 @@ class MockBackendDataTypeConfigurer |
| } // namespace |
| +class FakeNonUIModelTypeService : public syncer_v2::FakeModelTypeService { |
|
skym
2016/03/24 16:08:52
This is a lot of boilerplate copy/pasted from ui_m
Gang Wu
2016/03/24 22:18:11
move duplicate code into FakeModelTypeService
|
| + public: |
| + FakeNonUIModelTypeService() : test_processor_(nullptr){}; |
| + ~FakeNonUIModelTypeService() override {} |
| + |
| + base::WeakPtr<syncer_v2::SharedModelTypeProcessor> SetUpProcessor( |
| + syncer_v2::ModelTypeChangeProcessor* processor) { |
| + test_processor_ = processor; |
| + return static_cast<syncer_v2::SharedModelTypeProcessor*>( |
| + GetOrCreateChangeProcessor()) |
| + ->AsWeakPtrForUI(); |
| + } |
| + |
| + protected: |
| + syncer_v2::ModelTypeChangeProcessor* CreateProcessorForTest( |
| + syncer::ModelType type, |
| + ModelTypeService* service) override { |
| + return test_processor_; |
| + } |
| + |
| + private: |
| + syncer_v2::ModelTypeChangeProcessor* test_processor_; |
| +}; |
| + |
| class NonUIModelTypeControllerTest : public testing::Test, |
| public sync_driver::FakeSyncClient { |
| public: |
| @@ -167,7 +189,6 @@ class NonUIModelTypeControllerTest : public testing::Test, |
| controller_ = new TestNonUIModelTypeController( |
| ui_loop_.task_runner(), model_thread_runner_, base::Closure(), |
| syncer::DICTIONARY, this); |
| - controller_->InitializeProcessorInTest(); |
| InitializeTypeProcessor(); |
| } |
| @@ -186,7 +207,9 @@ class NonUIModelTypeControllerTest : public testing::Test, |
| void InitializeTypeProcessor() { |
| if (!model_thread_runner_ || |
| model_thread_runner_->BelongsToCurrentThread()) { |
| - type_processor_ = controller_->get_type_processor(); |
| + type_processor_ = |
| + service_->SetUpProcessor(new syncer_v2::SharedModelTypeProcessor( |
| + syncer::DICTIONARY, service_.get())); |
| } else { |
| model_thread_runner_->PostTask( |
| FROM_HERE, |
| @@ -199,7 +222,7 @@ class NonUIModelTypeControllerTest : public testing::Test, |
| void InitializeModelTypeService() { |
| if (!model_thread_runner_ || |
| model_thread_runner_->BelongsToCurrentThread()) { |
| - service_.reset(new syncer_v2::FakeModelTypeService()); |
| + service_.reset(new FakeNonUIModelTypeService()); |
| } else { |
| model_thread_runner_->PostTask( |
| FROM_HERE, |
| @@ -349,7 +372,7 @@ class NonUIModelTypeControllerTest : public testing::Test, |
| scoped_refptr<base::TestSimpleTaskRunner> sync_thread_runner_; |
| MockSyncBackend backend_; |
| MockBackendDataTypeConfigurer configurer_; |
| - scoped_ptr<syncer_v2::FakeModelTypeService> service_; |
| + scoped_ptr<FakeNonUIModelTypeService> service_; |
| }; |
| TEST_F(NonUIModelTypeControllerTest, InitialState) { |