Chromium Code Reviews| Index: components/sync_driver/ui_model_type_controller_unittest.cc |
| diff --git a/components/sync_driver/ui_model_type_controller_unittest.cc b/components/sync_driver/ui_model_type_controller_unittest.cc |
| index 5a9878ba6c5d7acb7e569fc02baa3caf6b4f6b6a..e0c9508511cc2345a3d9cbb363df7a42038a8939 100644 |
| --- a/components/sync_driver/ui_model_type_controller_unittest.cc |
| +++ b/components/sync_driver/ui_model_type_controller_unittest.cc |
| @@ -39,8 +39,6 @@ class TestUIModelTypeController : public UIModelTypeController { |
| model_type, |
| sync_client) {} |
| - void InitializeProcessorInTest() { InitializeProcessor(); } |
| - |
| private: |
| ~TestUIModelTypeController() override {} |
| }; |
| @@ -130,6 +128,30 @@ class MockBackendDataTypeConfigurer |
| } // namespace |
| +class FakeUIModelTypeService : public syncer_v2::FakeModelTypeService { |
|
skym
2016/03/24 16:08:52
You wouldn't need this class if you used an overlo
Gang Wu
2016/03/24 22:18:11
This test uses direct access to processor for load
|
| + public: |
| + FakeUIModelTypeService() : test_processor_(nullptr){}; |
| + ~FakeUIModelTypeService() 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 UIModelTypeControllerTest : public testing::Test, |
| public sync_driver::FakeSyncClient { |
| public: |
| @@ -145,10 +167,9 @@ class UIModelTypeControllerTest : public testing::Test, |
| void SetUp() override { |
| controller_ = new TestUIModelTypeController( |
| ui_loop_.task_runner(), base::Closure(), syncer::DEVICE_INFO, this); |
| - controller_->InitializeProcessorInTest(); |
| type_processor_ = |
| - ((syncer_v2::SharedModelTypeProcessor*)service_.change_processor()) |
| - ->AsWeakPtrForUI(); |
| + service_.SetUpProcessor(new syncer_v2::SharedModelTypeProcessor( |
| + syncer::DEVICE_INFO, &service_)); |
| } |
| void TearDown() override { |
| @@ -248,7 +269,7 @@ class UIModelTypeControllerTest : public testing::Test, |
| scoped_refptr<base::TestSimpleTaskRunner> sync_thread_runner_; |
| MockSyncBackend backend_; |
| MockBackendDataTypeConfigurer configurer_; |
| - syncer_v2::FakeModelTypeService service_; |
| + FakeUIModelTypeService service_; |
| }; |
| TEST_F(UIModelTypeControllerTest, InitialState) { |