Refactor search_engines tests to better reuse code

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/default_search_pref_test_util.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 "testing/gtest/include/gtest/gtest.h"
 
 namespace {
 
 class TestingTemplateURLServiceClient : public ChromeTemplateURLServiceClient {
  public:
   TestingTemplateURLServiceClient(history::HistoryService* history_service,
                                   base::string16* search_term)
       : ChromeTemplateURLServiceClient(history_service),
         search_term_(search_term) {}
 
   void SetKeywordSearchTermsForURL(const GURL& url,
                                    TemplateURLID id,
                                    const base::string16& term) override {
     *search_term_ = term;
   }
 
  private:
   base::string16* search_term_;
 
   DISALLOW_COPY_AND_ASSIGN(TestingTemplateURLServiceClient);
 };
 
 }  // namespace
 
+void SetManagedDefaultSearchPreferences(const TemplateURLData& managed_tdata,
+                                        bool enabled,
+                                        TestingProfile* profile) {
+  auto dict = TemplateURLDataToDictionary(managed_tdata);
+  dict->SetBoolean(DefaultSearchManager::kDisabledByPolicy, !enabled);
+
+  profile->GetTestingPrefService()->SetManagedPref(
+      DefaultSearchManager::kDefaultSearchProviderDataPrefName, dict.release());
+}
+
+void RemoveManagedDefaultSearchPreferences(TestingProfile* profile) {
+  profile->GetTestingPrefService()->RemoveManagedPref(
+      DefaultSearchManager::kDefaultSearchProviderDataPrefName);
+}
+
 TemplateURLServiceTestUtil::TemplateURLServiceTestUtil()
     : changed_count_(0),
       search_terms_data_(NULL) {
   // Make unique temp directory.
   EXPECT_TRUE(temp_dir_.CreateUniqueTempDir());
   profile_.reset(new TestingProfile(temp_dir_.GetPath()));
 
   scoped_refptr<WebDatabaseService> web_database_service =
       new WebDatabaseService(temp_dir_.GetPath().AppendASCII("webdata"),
                              base::ThreadTaskRunnerHandle::Get(),
@@ skipping 77 matching lines @@
   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::SetManagedDefaultSearchPreferences(
-    bool enabled,
-    const std::string& name,
-    const std::string& keyword,
-    const std::string& search_url,
-    const std::string& suggest_url,
-    const std::string& icon_url,
-    const std::string& encodings,
-    const std::string& alternate_url,
-    const std::string& search_terms_replacement_key) {
-  DefaultSearchPrefTestUtil::SetManagedPref(
-      profile()->GetTestingPrefService(),
-      enabled, name, keyword, search_url, suggest_url, icon_url, encodings,
-      alternate_url, search_terms_replacement_key);
-}
-
-void TemplateURLServiceTestUtil::RemoveManagedDefaultSearchPreferences() {
-  DefaultSearchPrefTestUtil::RemoveManagedPref(
-      profile()->GetTestingPrefService());
-}