Make extensions DSE persistent in browser prefs

chrome/browser/search_engines/template_url_service_test_util.h

 // 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.
 
 #ifndef CHROME_BROWSER_SEARCH_ENGINES_TEMPLATE_URL_SERVICE_TEST_UTIL_H_
 #define CHROME_BROWSER_SEARCH_ENGINES_TEMPLATE_URL_SERVICE_TEST_UTIL_H_
 
 #include <memory>
 #include <string>
 
 #include "base/files/scoped_temp_dir.h"
 #include "base/macros.h"
 #include "base/memory/ref_counted.h"
 #include "base/strings/string16.h"
+#include "components/search_engines/template_url.h"
 #include "components/search_engines/template_url_data.h"
 #include "components/search_engines/template_url_service_observer.h"
 
+class ExtensionPrefValueMap;
 class GURL;
 class KeywordWebDataService;
 class TemplateURLService;
 class TestingProfile;
 class TestingSearchTermsData;
 
 // Sets the managed preferences for the default search provider.
 // enabled arg enables/disables use of managed engine by DefaultSearchManager.
 void SetManagedDefaultSearchPreferences(const TemplateURLData& managed_data,
                                         bool enabled,
@@ skipping 30 matching lines @@
   // Creates a new TemplateURLService.
   void ResetModel(bool verify_load);
 
   // Returns the search term from the last invocation of
   // TemplateURLService::SetKeywordSearchTermsForURL and clears the search term.
   base::string16 GetAndClearSearchTerm();
 
   // Sets the google base url.  |base_url| must be valid.
   void SetGoogleBaseURL(const GURL& base_url);
 
+  // Write default search engine into extension controlled preference in test
+  // profile.
+  void SetExtensionDefaultSearchInPrefs(const std::string& ext_id,
+                                        const TemplateURLData& ext_data);
+  // Remove extension controlled default search engine preference from test
+  // profile.
+  void RemoveExtensionDefaultSearchFromPrefs(const std::string& ext_id);
+
+  // Add extension turl to model and override default search preference if
+  // extension wants to be default.
+  TemplateURL* AddExtensionControlledTURL(
+      std::unique_ptr<TemplateURL> ext_dse,
+      std::unique_ptr<TemplateURL::AssociatedExtensionInfo> ext_info);
+  // Remove extension turl from preferences and model.
+  void RemoveExtensionControlledTURL(const std::string& ext_id);
+
   KeywordWebDataService* web_data_service() { return web_data_service_.get(); }
   TemplateURLService* model() { return model_.get(); }
   TestingProfile* profile() { return profile_.get(); }
 
  private:
   std::unique_ptr<TestingProfile> profile_;
   base::ScopedTempDir temp_dir_;
   int changed_count_;
   base::string16 search_term_;
   scoped_refptr<KeywordWebDataService> web_data_service_;
   TestingSearchTermsData* search_terms_data_;
   std::unique_ptr<TemplateURLService> model_;
+  std::unique_ptr<ExtensionPrefValueMap> ext_pref_map_;
 
   DISALLOW_COPY_AND_ASSIGN(TemplateURLServiceTestUtil);
 };
 
 #endif  // CHROME_BROWSER_SEARCH_ENGINES_TEMPLATE_URL_SERVICE_TEST_UTIL_H_