components/browser_sync/profile_sync_test_util.cc - chromium/src - Git at Google

 // Copyright 2016 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 "components/browser_sync/profile_sync_test_util.h"

 #include <utility>

 #include "base/bind.h"
 #include "base/single_thread_task_runner.h"
 #include "base/threading/thread_task_runner_handle.h"
 #include "components/bookmarks/browser/bookmark_model.h"
 #include "components/history/core/browser/history_model_worker.h"
 #include "components/history/core/browser/history_service.h"
 #include "components/pref_registry/pref_registry_syncable.h"
 #include "components/signin/core/browser/profile_management_switches.h"
 #include "components/signin/core/browser/signin_manager_base.h"
 #include "components/sync/base/sync_prefs.h"
 #include "components/sync/driver/signin_manager_wrapper.h"
 #include "components/sync/engine/passive_model_worker.h"
 #include "components/sync/engine/sequenced_model_worker.h"
 #include "components/sync/engine/ui_model_worker.h"
 #include "components/sync/model/model_type_store_test_util.h"
 #include "net/url_request/url_request_test_util.h"

 namespace browser_sync {

 namespace {

 class BundleSyncClient : public syncer::FakeSyncClient {
  public:
   BundleSyncClient(syncer::SyncApiComponentFactory* factory,
                    PrefService* pref_service,
                    sync_sessions::SyncSessionsClient* sync_sessions_client,
                    autofill::PersonalDataManager* personal_data_manager,
                    const base::Callback<base::WeakPtr<syncer::SyncableService>(
                        syncer::ModelType type)>& get_syncable_service_callback,
                    const base::Callback<syncer::SyncService*(void)>&
                        get_sync_service_callback,
                    const base::Callback<bookmarks::BookmarkModel*(void)>&
                        get_bookmark_model_callback,
                    scoped_refptr<base::SingleThreadTaskRunner> db_thread,
                    scoped_refptr<base::SingleThreadTaskRunner> file_thread,
                    history::HistoryService* history_service);

   ~BundleSyncClient() override;

   PrefService* GetPrefService() override;
   sync_sessions::SyncSessionsClient* GetSyncSessionsClient() override;
   autofill::PersonalDataManager* GetPersonalDataManager() override;
   base::WeakPtr<syncer::SyncableService> GetSyncableServiceForType(
       syncer::ModelType type) override;
   syncer::SyncService* GetSyncService() override;
   scoped_refptr<syncer::ModelSafeWorker> CreateModelWorkerForGroup(
       syncer::ModelSafeGroup group) override;
   history::HistoryService* GetHistoryService() override;
   bookmarks::BookmarkModel* GetBookmarkModel() override;

  private:
   PrefService* const pref_service_;
   sync_sessions::SyncSessionsClient* const sync_sessions_client_;
   autofill::PersonalDataManager* const personal_data_manager_;
   const base::Callback<base::WeakPtr<syncer::SyncableService>(
       syncer::ModelType type)>
       get_syncable_service_callback_;
   const base::Callback<syncer::SyncService*(void)> get_sync_service_callback_;
   const base::Callback<bookmarks::BookmarkModel*(void)>
       get_bookmark_model_callback_;
   // These task runners, if not null, are used in CreateModelWorkerForGroup.
   const scoped_refptr<base::SingleThreadTaskRunner> db_thread_;
   const scoped_refptr<base::SingleThreadTaskRunner> file_thread_;
   history::HistoryService* history_service_;
 };

 BundleSyncClient::BundleSyncClient(
     syncer::SyncApiComponentFactory* factory,
     PrefService* pref_service,
     sync_sessions::SyncSessionsClient* sync_sessions_client,
     autofill::PersonalDataManager* personal_data_manager,
     const base::Callback<base::WeakPtr<syncer::SyncableService>(
         syncer::ModelType type)>& get_syncable_service_callback,
     const base::Callback<syncer::SyncService*(void)>& get_sync_service_callback,
     const base::Callback<bookmarks::BookmarkModel*(void)>&
         get_bookmark_model_callback,
     scoped_refptr<base::SingleThreadTaskRunner> db_thread,
     scoped_refptr<base::SingleThreadTaskRunner> file_thread,
     history::HistoryService* history_service)
     : syncer::FakeSyncClient(factory),
       pref_service_(pref_service),
       sync_sessions_client_(sync_sessions_client),
       personal_data_manager_(personal_data_manager),
       get_syncable_service_callback_(get_syncable_service_callback),
       get_sync_service_callback_(get_sync_service_callback),
       get_bookmark_model_callback_(get_bookmark_model_callback),
       db_thread_(db_thread),
       file_thread_(file_thread),
       history_service_(history_service) {
   EXPECT_EQ(!!db_thread_, !!file_thread_);
 }

 BundleSyncClient::~BundleSyncClient() = default;

 PrefService* BundleSyncClient::GetPrefService() {
   return pref_service_;
 }

 sync_sessions::SyncSessionsClient* BundleSyncClient::GetSyncSessionsClient() {
   return sync_sessions_client_;
 }

 autofill::PersonalDataManager* BundleSyncClient::GetPersonalDataManager() {
   return personal_data_manager_;
 }

 base::WeakPtr<syncer::SyncableService>
 BundleSyncClient::GetSyncableServiceForType(syncer::ModelType type) {
   if (get_syncable_service_callback_.is_null())
     return syncer::FakeSyncClient::GetSyncableServiceForType(type);
   return get_syncable_service_callback_.Run(type);
 }

 syncer::SyncService* BundleSyncClient::GetSyncService() {
   if (get_sync_service_callback_.is_null())
     return syncer::FakeSyncClient::GetSyncService();
   return get_sync_service_callback_.Run();
 }

 scoped_refptr<syncer::ModelSafeWorker>
 BundleSyncClient::CreateModelWorkerForGroup(syncer::ModelSafeGroup group) {
   if (!db_thread_)
     return FakeSyncClient::CreateModelWorkerForGroup(group);
   EXPECT_TRUE(file_thread_)
       << "DB thread was specified but FILE thread was not.";
   switch (group) {
     case syncer::GROUP_DB:
       return new syncer::SequencedModelWorker(db_thread_, syncer::GROUP_DB);
     case syncer::GROUP_FILE:
       return new syncer::SequencedModelWorker(file_thread_, syncer::GROUP_FILE);
     case syncer::GROUP_UI:
       return new syncer::UIModelWorker(base::ThreadTaskRunnerHandle::Get());
     case syncer::GROUP_PASSIVE:
       return new syncer::PassiveModelWorker();
     case syncer::GROUP_HISTORY: {
       history::HistoryService* history_service = GetHistoryService();
       if (!history_service)
         return nullptr;
       return new HistoryModelWorker(history_service->AsWeakPtr(),
                                     base::ThreadTaskRunnerHandle::Get());
     }
     default:
       return nullptr;
   }
 }

 history::HistoryService* BundleSyncClient::GetHistoryService() {
   if (history_service_)
     return history_service_;
   return FakeSyncClient::GetHistoryService();
 }

 bookmarks::BookmarkModel* BundleSyncClient::GetBookmarkModel() {
   if (get_bookmark_model_callback_.is_null())
     return FakeSyncClient::GetBookmarkModel();
   return get_bookmark_model_callback_.Run();
 }

 }  // namespace

 void EmptyNetworkTimeUpdate(const base::Time&,
                             const base::TimeDelta&,
                             const base::TimeDelta&) {}

 void RegisterPrefsForProfileSyncService(
     user_prefs::PrefRegistrySyncable* registry) {
   syncer::SyncPrefs::RegisterProfilePrefs(registry);
   AccountTrackerService::RegisterPrefs(registry);
   SigninManagerBase::RegisterProfilePrefs(registry);
   SigninManagerBase::RegisterPrefs(registry);
   signin::RegisterAccountConsistencyProfilePrefs(registry);
 }

 ProfileSyncServiceBundle::SyncClientBuilder::~SyncClientBuilder() = default;

 ProfileSyncServiceBundle::SyncClientBuilder::SyncClientBuilder(
     ProfileSyncServiceBundle* bundle)
     : bundle_(bundle) {}

 void ProfileSyncServiceBundle::SyncClientBuilder::SetPersonalDataManager(
     autofill::PersonalDataManager* personal_data_manager) {
   personal_data_manager_ = personal_data_manager;
 }

 // The client will call this callback to produce the service.
 void ProfileSyncServiceBundle::SyncClientBuilder::SetSyncableServiceCallback(
     const base::Callback<base::WeakPtr<syncer::SyncableService>(
         syncer::ModelType type)>& get_syncable_service_callback) {
   get_syncable_service_callback_ = get_syncable_service_callback;
 }

 // The client will call this callback to produce the service.
 void ProfileSyncServiceBundle::SyncClientBuilder::SetSyncServiceCallback(
     const base::Callback<syncer::SyncService*(void)>&
         get_sync_service_callback) {
   get_sync_service_callback_ = get_sync_service_callback;
 }

 void ProfileSyncServiceBundle::SyncClientBuilder::SetHistoryService(
     history::HistoryService* history_service) {
   history_service_ = history_service;
 }

 void ProfileSyncServiceBundle::SyncClientBuilder::SetBookmarkModelCallback(
     const base::Callback<bookmarks::BookmarkModel*(void)>&
         get_bookmark_model_callback) {
   get_bookmark_model_callback_ = get_bookmark_model_callback;
 }

 std::unique_ptr<syncer::FakeSyncClient>
 ProfileSyncServiceBundle::SyncClientBuilder::Build() {
   return std::make_unique<BundleSyncClient>(
       bundle_->component_factory(), bundle_->pref_service(),
       bundle_->sync_sessions_client(), personal_data_manager_,
       get_syncable_service_callback_, get_sync_service_callback_,
       get_bookmark_model_callback_,
       activate_model_creation_ ? bundle_->db_thread() : nullptr,
       activate_model_creation_ ? base::ThreadTaskRunnerHandle::Get() : nullptr,
       history_service_);
 }

 ProfileSyncServiceBundle::ProfileSyncServiceBundle()
     : db_thread_(base::ThreadTaskRunnerHandle::Get()),
       signin_client_(&pref_service_),
 #if defined(OS_CHROMEOS)
       signin_manager_(&signin_client_, &account_tracker_),
 #else
       signin_manager_(&signin_client_,
                       &auth_service_,
                       &account_tracker_,
                       nullptr),
 #endif
       url_request_context_(new net::TestURLRequestContextGetter(
           base::ThreadTaskRunnerHandle::Get())) {
   RegisterPrefsForProfileSyncService(pref_service_.registry());
   signin::SetGaiaOriginIsolatedCallback(base::Bind([] { return true; }));
   auth_service_.set_auto_post_fetch_response_on_message_loop(true);
   account_tracker_.Initialize(&signin_client_);
   signin_manager_.Initialize(&pref_service_);
 }

 ProfileSyncServiceBundle::~ProfileSyncServiceBundle() {}

 ProfileSyncService::InitParams ProfileSyncServiceBundle::CreateBasicInitParams(
     ProfileSyncService::StartBehavior start_behavior,
     std::unique_ptr<syncer::SyncClient> sync_client) {
   ProfileSyncService::InitParams init_params;

   init_params.start_behavior = start_behavior;
   init_params.sync_client = std::move(sync_client);
   init_params.signin_wrapper =
       std::make_unique<SigninManagerWrapper>(signin_manager());
   init_params.oauth2_token_service = auth_service();
   init_params.network_time_update_callback =
       base::Bind(&EmptyNetworkTimeUpdate);
   EXPECT_TRUE(base_directory_.CreateUniqueTempDir());
   init_params.base_directory = base_directory_.GetPath();
   init_params.url_request_context = url_request_context();
   init_params.debug_identifier = "dummyDebugName";
   init_params.channel = version_info::Channel::UNKNOWN;
   init_params.model_type_store_factory =
       syncer::ModelTypeStoreTestUtil::FactoryForInMemoryStoreForTest();

   return init_params;
 }

 }  // namespace browser_sync
	// Copyright 2016 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 "components/browser_sync/profile_sync_test_util.h"

	#include <utility>

	#include "base/bind.h"
	#include "base/single_thread_task_runner.h"
	#include "base/threading/thread_task_runner_handle.h"
	#include "components/bookmarks/browser/bookmark_model.h"
	#include "components/history/core/browser/history_model_worker.h"
	#include "components/history/core/browser/history_service.h"
	#include "components/pref_registry/pref_registry_syncable.h"
	#include "components/signin/core/browser/profile_management_switches.h"
	#include "components/signin/core/browser/signin_manager_base.h"
	#include "components/sync/base/sync_prefs.h"
	#include "components/sync/driver/signin_manager_wrapper.h"
	#include "components/sync/engine/passive_model_worker.h"
	#include "components/sync/engine/sequenced_model_worker.h"
	#include "components/sync/engine/ui_model_worker.h"
	#include "components/sync/model/model_type_store_test_util.h"
	#include "net/url_request/url_request_test_util.h"

	namespace browser_sync {

	namespace {

	class BundleSyncClient : public syncer::FakeSyncClient {
	public:
	BundleSyncClient(syncer::SyncApiComponentFactory* factory,
	PrefService* pref_service,
	sync_sessions::SyncSessionsClient* sync_sessions_client,
	autofill::PersonalDataManager* personal_data_manager,
	const base::Callback<base::WeakPtr<syncer::SyncableService>(
	syncer::ModelType type)>& get_syncable_service_callback,
	const base::Callback<syncer::SyncService*(void)>&
	get_sync_service_callback,
	const base::Callback<bookmarks::BookmarkModel*(void)>&
	get_bookmark_model_callback,
	scoped_refptr<base::SingleThreadTaskRunner> db_thread,
	scoped_refptr<base::SingleThreadTaskRunner> file_thread,
	history::HistoryService* history_service);

	~BundleSyncClient() override;

	PrefService* GetPrefService() override;
	sync_sessions::SyncSessionsClient* GetSyncSessionsClient() override;
	autofill::PersonalDataManager* GetPersonalDataManager() override;
	base::WeakPtr<syncer::SyncableService> GetSyncableServiceForType(
	syncer::ModelType type) override;
	syncer::SyncService* GetSyncService() override;
	scoped_refptr<syncer::ModelSafeWorker> CreateModelWorkerForGroup(
	syncer::ModelSafeGroup group) override;
	history::HistoryService* GetHistoryService() override;
	bookmarks::BookmarkModel* GetBookmarkModel() override;

	private:
	PrefService* const pref_service_;
	sync_sessions::SyncSessionsClient* const sync_sessions_client_;
	autofill::PersonalDataManager* const personal_data_manager_;
	const base::Callback<base::WeakPtr<syncer::SyncableService>(
	syncer::ModelType type)>
	get_syncable_service_callback_;
	const base::Callback<syncer::SyncService*(void)> get_sync_service_callback_;
	const base::Callback<bookmarks::BookmarkModel*(void)>
	get_bookmark_model_callback_;
	// These task runners, if not null, are used in CreateModelWorkerForGroup.
	const scoped_refptr<base::SingleThreadTaskRunner> db_thread_;
	const scoped_refptr<base::SingleThreadTaskRunner> file_thread_;
	history::HistoryService* history_service_;
	};

	BundleSyncClient::BundleSyncClient(
	syncer::SyncApiComponentFactory* factory,
	PrefService* pref_service,
	sync_sessions::SyncSessionsClient* sync_sessions_client,
	autofill::PersonalDataManager* personal_data_manager,
	const base::Callback<base::WeakPtr<syncer::SyncableService>(
	syncer::ModelType type)>& get_syncable_service_callback,
	const base::Callback<syncer::SyncService*(void)>& get_sync_service_callback,
	const base::Callback<bookmarks::BookmarkModel*(void)>&
	get_bookmark_model_callback,
	scoped_refptr<base::SingleThreadTaskRunner> db_thread,
	scoped_refptr<base::SingleThreadTaskRunner> file_thread,
	history::HistoryService* history_service)
	: syncer::FakeSyncClient(factory),
	pref_service_(pref_service),
	sync_sessions_client_(sync_sessions_client),
	personal_data_manager_(personal_data_manager),
	get_syncable_service_callback_(get_syncable_service_callback),
	get_sync_service_callback_(get_sync_service_callback),
	get_bookmark_model_callback_(get_bookmark_model_callback),
	db_thread_(db_thread),
	file_thread_(file_thread),
	history_service_(history_service) {
	EXPECT_EQ(!!db_thread_, !!file_thread_);
	}

	BundleSyncClient::~BundleSyncClient() = default;

	PrefService* BundleSyncClient::GetPrefService() {
	return pref_service_;
	}

	sync_sessions::SyncSessionsClient* BundleSyncClient::GetSyncSessionsClient() {
	return sync_sessions_client_;
	}

	autofill::PersonalDataManager* BundleSyncClient::GetPersonalDataManager() {
	return personal_data_manager_;
	}

	base::WeakPtr<syncer::SyncableService>
	BundleSyncClient::GetSyncableServiceForType(syncer::ModelType type) {
	if (get_syncable_service_callback_.is_null())
	return syncer::FakeSyncClient::GetSyncableServiceForType(type);
	return get_syncable_service_callback_.Run(type);
	}

	syncer::SyncService* BundleSyncClient::GetSyncService() {
	if (get_sync_service_callback_.is_null())
	return syncer::FakeSyncClient::GetSyncService();
	return get_sync_service_callback_.Run();
	}

	scoped_refptr<syncer::ModelSafeWorker>
	BundleSyncClient::CreateModelWorkerForGroup(syncer::ModelSafeGroup group) {
	if (!db_thread_)
	return FakeSyncClient::CreateModelWorkerForGroup(group);
	EXPECT_TRUE(file_thread_)
	<< "DB thread was specified but FILE thread was not.";
	switch (group) {
	case syncer::GROUP_DB:
	return new syncer::SequencedModelWorker(db_thread_, syncer::GROUP_DB);
	case syncer::GROUP_FILE:
	return new syncer::SequencedModelWorker(file_thread_, syncer::GROUP_FILE);
	case syncer::GROUP_UI:
	return new syncer::UIModelWorker(base::ThreadTaskRunnerHandle::Get());
	case syncer::GROUP_PASSIVE:
	return new syncer::PassiveModelWorker();
	case syncer::GROUP_HISTORY: {
	history::HistoryService* history_service = GetHistoryService();
	if (!history_service)
	return nullptr;
	return new HistoryModelWorker(history_service->AsWeakPtr(),
	base::ThreadTaskRunnerHandle::Get());
	}
	default:
	return nullptr;
	}
	}

	history::HistoryService* BundleSyncClient::GetHistoryService() {
	if (history_service_)
	return history_service_;
	return FakeSyncClient::GetHistoryService();
	}

	bookmarks::BookmarkModel* BundleSyncClient::GetBookmarkModel() {
	if (get_bookmark_model_callback_.is_null())
	return FakeSyncClient::GetBookmarkModel();
	return get_bookmark_model_callback_.Run();
	}

	} // namespace

	void EmptyNetworkTimeUpdate(const base::Time&,
	const base::TimeDelta&,
	const base::TimeDelta&) {}

	void RegisterPrefsForProfileSyncService(
	user_prefs::PrefRegistrySyncable* registry) {
	syncer::SyncPrefs::RegisterProfilePrefs(registry);
	AccountTrackerService::RegisterPrefs(registry);
	SigninManagerBase::RegisterProfilePrefs(registry);
	SigninManagerBase::RegisterPrefs(registry);
	signin::RegisterAccountConsistencyProfilePrefs(registry);
	}

	ProfileSyncServiceBundle::SyncClientBuilder::~SyncClientBuilder() = default;

	ProfileSyncServiceBundle::SyncClientBuilder::SyncClientBuilder(
	ProfileSyncServiceBundle* bundle)
	: bundle_(bundle) {}

	void ProfileSyncServiceBundle::SyncClientBuilder::SetPersonalDataManager(
	autofill::PersonalDataManager* personal_data_manager) {
	personal_data_manager_ = personal_data_manager;
	}

	// The client will call this callback to produce the service.
	void ProfileSyncServiceBundle::SyncClientBuilder::SetSyncableServiceCallback(
	const base::Callback<base::WeakPtr<syncer::SyncableService>(
	syncer::ModelType type)>& get_syncable_service_callback) {
	get_syncable_service_callback_ = get_syncable_service_callback;
	}

	// The client will call this callback to produce the service.
	void ProfileSyncServiceBundle::SyncClientBuilder::SetSyncServiceCallback(
	const base::Callback<syncer::SyncService*(void)>&
	get_sync_service_callback) {
	get_sync_service_callback_ = get_sync_service_callback;
	}

	void ProfileSyncServiceBundle::SyncClientBuilder::SetHistoryService(
	history::HistoryService* history_service) {
	history_service_ = history_service;
	}

	void ProfileSyncServiceBundle::SyncClientBuilder::SetBookmarkModelCallback(
	const base::Callback<bookmarks::BookmarkModel*(void)>&
	get_bookmark_model_callback) {
	get_bookmark_model_callback_ = get_bookmark_model_callback;
	}

	std::unique_ptr<syncer::FakeSyncClient>
	ProfileSyncServiceBundle::SyncClientBuilder::Build() {
	return std::make_unique<BundleSyncClient>(
	bundle_->component_factory(), bundle_->pref_service(),
	bundle_->sync_sessions_client(), personal_data_manager_,
	get_syncable_service_callback_, get_sync_service_callback_,
	get_bookmark_model_callback_,
	activate_model_creation_ ? bundle_->db_thread() : nullptr,
	activate_model_creation_ ? base::ThreadTaskRunnerHandle::Get() : nullptr,
	history_service_);
	}

	ProfileSyncServiceBundle::ProfileSyncServiceBundle()
	: db_thread_(base::ThreadTaskRunnerHandle::Get()),
	signin_client_(&pref_service_),
	#if defined(OS_CHROMEOS)
	signin_manager_(&signin_client_, &account_tracker_),
	#else
	signin_manager_(&signin_client_,
	&auth_service_,
	&account_tracker_,
	nullptr),
	#endif
	url_request_context_(new net::TestURLRequestContextGetter(
	base::ThreadTaskRunnerHandle::Get())) {
	RegisterPrefsForProfileSyncService(pref_service_.registry());
	signin::SetGaiaOriginIsolatedCallback(base::Bind([] { return true; }));
	auth_service_.set_auto_post_fetch_response_on_message_loop(true);
	account_tracker_.Initialize(&signin_client_);
	signin_manager_.Initialize(&pref_service_);
	}

	ProfileSyncServiceBundle::~ProfileSyncServiceBundle() {}

	ProfileSyncService::InitParams ProfileSyncServiceBundle::CreateBasicInitParams(
	ProfileSyncService::StartBehavior start_behavior,
	std::unique_ptr<syncer::SyncClient> sync_client) {
	ProfileSyncService::InitParams init_params;

	init_params.start_behavior = start_behavior;
	init_params.sync_client = std::move(sync_client);
	init_params.signin_wrapper =
	std::make_unique<SigninManagerWrapper>(signin_manager());
	init_params.oauth2_token_service = auth_service();
	init_params.network_time_update_callback =
	base::Bind(&EmptyNetworkTimeUpdate);
	EXPECT_TRUE(base_directory_.CreateUniqueTempDir());
	init_params.base_directory = base_directory_.GetPath();
	init_params.url_request_context = url_request_context();
	init_params.debug_identifier = "dummyDebugName";
	init_params.channel = version_info::Channel::UNKNOWN;
	init_params.model_type_store_factory =
	syncer::ModelTypeStoreTestUtil::FactoryForInMemoryStoreForTest();

	return init_params;
	}

	} // namespace browser_sync