Make extensions DSE persistent in browser prefs

chrome/browser/search_engines/template_url_service_test_util.cc

 // Copyright (c) 2012 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 "chrome/browser/search_engines/template_url_service_test_util.h"
 
 #include "base/macros.h"
 #include "base/run_loop.h"
 #include "base/threading/thread_task_runner_handle.h"
 #include "chrome/browser/history/history_service_factory.h"
 #include "chrome/browser/search_engines/chrome_template_url_service_client.h"
 #include "chrome/test/base/testing_profile.h"
 #include "components/search_engines/keyword_table.h"
 #include "components/search_engines/keyword_web_data_service.h"
 #include "components/search_engines/template_url_data_util.h"
 #include "components/search_engines/template_url_service.h"
 #include "components/search_engines/testing_search_terms_data.h"
 #include "components/sync_preferences/testing_pref_service_syncable.h"
 #include "components/webdata/common/web_database_service.h"
+#include "extensions/browser/extension_pref_value_map.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace {
 
 class TestingTemplateURLServiceClient : public ChromeTemplateURLServiceClient {
  public:
   TestingTemplateURLServiceClient(history::HistoryService* history_service,
+                                  ExtensionPrefValueMap* ext_pref_map,
                                   base::string16* search_term)
-      : ChromeTemplateURLServiceClient(history_service),
+      : ChromeTemplateURLServiceClient(history_service, ext_pref_map),
         search_term_(search_term) {}
 
   void SetKeywordSearchTermsForURL(const GURL& url,
                                    TemplateURLID id,
                                    const base::string16& term) override {
     *search_term_ = term;
   }
 
  private:
   base::string16* search_term_;
@@ skipping 31 matching lines @@
                              base::ThreadTaskRunnerHandle::Get());
   web_database_service->AddTable(
       std::unique_ptr<WebDatabaseTable>(new KeywordTable()));
   web_database_service->LoadDatabase();
 
   web_data_service_ = new KeywordWebDataService(
       web_database_service.get(), base::ThreadTaskRunnerHandle::Get(),
       KeywordWebDataService::ProfileErrorCallback());
   web_data_service_->Init();
 
+  ext_pref_map_.reset(new ExtensionPrefValueMap());
+
   ResetModel(false);
 }
 
 TemplateURLServiceTestUtil::~TemplateURLServiceTestUtil() {
   ClearModel();
   profile_.reset();
 
   // Flush the message loop to make application verifiers happy.
   base::RunLoop().RunUntilIdle();
 }
@@ skipping 30 matching lines @@
 void TemplateURLServiceTestUtil::ClearModel() {
   model_->Shutdown();
   model_.reset();
   search_terms_data_ = NULL;
 }
 
 void TemplateURLServiceTestUtil::ResetModel(bool verify_load) {
   if (model_)
     ClearModel();
   search_terms_data_ = new TestingSearchTermsData("http://www.google.com/");
+
   model_.reset(new TemplateURLService(
       profile()->GetPrefs(),
       std::unique_ptr<SearchTermsData>(search_terms_data_),
       web_data_service_.get(),
       std::unique_ptr<TemplateURLServiceClient>(
           new TestingTemplateURLServiceClient(
               HistoryServiceFactory::GetForProfileIfExists(
                   profile(), ServiceAccessType::EXPLICIT_ACCESS),
-              &search_term_)),
+              ext_pref_map_.get(), &search_term_)),
       NULL, NULL, base::Closure()));
   model()->AddObserver(this);
   changed_count_ = 0;
   if (verify_load)
     VerifyLoad();
 }
 
 base::string16 TemplateURLServiceTestUtil::GetAndClearSearchTerm() {
   base::string16 search_term;
   search_term.swap(search_term_);
   return search_term;
 }
 
 void TemplateURLServiceTestUtil::SetGoogleBaseURL(const GURL& base_url) {
   DCHECK(base_url.is_valid());
   search_terms_data_->set_google_base_url(base_url.spec());
   model_->GoogleBaseURLChanged();
 }
+
+void TemplateURLServiceTestUtil::SetExtensionDefaultSearchInPrefs(
+    const std::string& ext_id,
+    const TemplateURLData& ext_data) {
+  auto dict = TemplateURLDataToDictionary(ext_data);

[vasilii, 2016/12/06 19:16:42]

Readabilty would improve with an explicit type..

[Alexander Yashkin, 2016/12/09 08:19:52]

Expanded.

+  ext_pref_map_->RegisterExtension(ext_id, base::Time::Now(), true, true);
+  // Set mapping between extension and pref.
+  ext_pref_map_->SetExtensionPref(
+      ext_id, DefaultSearchManager::kDefaultSearchProviderDataPrefName,
+      extensions::ExtensionPrefsScope::kExtensionPrefsScopeRegular,
+      dict->DeepCopy());
+  // Set extension overriden value in test pref service.
+  profile()->GetTestingPrefService()->SetExtensionPref(
+      DefaultSearchManager::kDefaultSearchProviderDataPrefName, dict.release());
+}
+
+void TemplateURLServiceTestUtil::RemoveExtensionDefaultSearchFromPrefs(
+    const std::string& ext_id) {
+  ext_pref_map_->UnregisterExtension(ext_id);
+  // If no more extensions control DSE pref, remove from test pref service.
+  auto pref_ext_id = ext_pref_map_->GetExtensionControllingPref(
+      DefaultSearchManager::kDefaultSearchProviderDataPrefName);
+  if (pref_ext_id.empty()) {
+    profile()->GetTestingPrefService()->RemoveExtensionPref(
+        DefaultSearchManager::kDefaultSearchProviderDataPrefName);
+  }
+}
+
+TemplateURL* TemplateURLServiceTestUtil::AddExtensionControlledTURL(
+    std::unique_ptr<TemplateURL> ext_dse,
+    std::unique_ptr<TemplateURL::AssociatedExtensionInfo> ext_info) {
+  TemplateURL::AssociatedExtensionInfo ext_info_copy = *ext_info;
+  TemplateURLData ext_data = ext_dse->data();
+  TemplateURL* result = model()->AddExtensionControlledTURL(
+      std::move(ext_dse), std::move(ext_info));
+  if (ext_info_copy.wants_to_be_default_engine) {
+    // Set extension overriden DSE value to prefs.
+    SetExtensionDefaultSearchInPrefs(ext_info_copy.extension_id, ext_data);
+  }
+  return result;
+}
+
+void TemplateURLServiceTestUtil::RemoveExtensionControlledTURL(
+    const std::string& ext_id) {
+  TemplateURL* ext_turl = model()->FindTemplateURLForExtension(
+      ext_id, TemplateURL::NORMAL_CONTROLLED_BY_EXTENSION);
+  ASSERT_TRUE(ext_turl);
+  ASSERT_TRUE(ext_turl->extension_info_for_tests());
+  if (ext_turl->extension_info_for_tests()->wants_to_be_default_engine) {
+    RemoveExtensionDefaultSearchFromPrefs(ext_id);
+  }
+  model()->RemoveExtensionControlledTURL(
+      ext_id, TemplateURL::NORMAL_CONTROLLED_BY_EXTENSION);
+}