OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #ifndef CHROME_BROWSER_SEARCH_ENGINES_TEMPLATE_URL_SERVICE_TEST_UTIL_H_ | 5 #ifndef CHROME_BROWSER_SEARCH_ENGINES_TEMPLATE_URL_SERVICE_TEST_UTIL_H_ |
6 #define CHROME_BROWSER_SEARCH_ENGINES_TEMPLATE_URL_SERVICE_TEST_UTIL_H_ | 6 #define CHROME_BROWSER_SEARCH_ENGINES_TEMPLATE_URL_SERVICE_TEST_UTIL_H_ |
7 | 7 |
8 #include <memory> | 8 #include <memory> |
9 #include <string> | 9 #include <string> |
10 | 10 |
11 #include "base/files/scoped_temp_dir.h" | 11 #include "base/files/scoped_temp_dir.h" |
12 #include "base/macros.h" | 12 #include "base/macros.h" |
13 #include "base/memory/ref_counted.h" | 13 #include "base/memory/ref_counted.h" |
14 #include "base/strings/string16.h" | 14 #include "base/strings/string16.h" |
| 15 #include "components/search_engines/template_url.h" |
15 #include "components/search_engines/template_url_data.h" | 16 #include "components/search_engines/template_url_data.h" |
16 #include "components/search_engines/template_url_service_observer.h" | 17 #include "components/search_engines/template_url_service_observer.h" |
17 | 18 |
| 19 class ExtensionPrefValueMap; |
18 class GURL; | 20 class GURL; |
19 class KeywordWebDataService; | 21 class KeywordWebDataService; |
20 class TemplateURLService; | 22 class TemplateURLService; |
21 class TestingProfile; | 23 class TestingProfile; |
22 class TestingSearchTermsData; | 24 class TestingSearchTermsData; |
23 | 25 |
24 // Sets the managed preferences for the default search provider. | 26 // Sets the managed preferences for the default search provider. |
25 // enabled arg enables/disables use of managed engine by DefaultSearchManager. | 27 // enabled arg enables/disables use of managed engine by DefaultSearchManager. |
26 void SetManagedDefaultSearchPreferences(const TemplateURLData& managed_data, | 28 void SetManagedDefaultSearchPreferences(const TemplateURLData& managed_data, |
27 bool enabled, | 29 bool enabled, |
(...skipping 30 matching lines...) Expand all Loading... |
58 // Creates a new TemplateURLService. | 60 // Creates a new TemplateURLService. |
59 void ResetModel(bool verify_load); | 61 void ResetModel(bool verify_load); |
60 | 62 |
61 // Returns the search term from the last invocation of | 63 // Returns the search term from the last invocation of |
62 // TemplateURLService::SetKeywordSearchTermsForURL and clears the search term. | 64 // TemplateURLService::SetKeywordSearchTermsForURL and clears the search term. |
63 base::string16 GetAndClearSearchTerm(); | 65 base::string16 GetAndClearSearchTerm(); |
64 | 66 |
65 // Sets the google base url. |base_url| must be valid. | 67 // Sets the google base url. |base_url| must be valid. |
66 void SetGoogleBaseURL(const GURL& base_url); | 68 void SetGoogleBaseURL(const GURL& base_url); |
67 | 69 |
| 70 // Write default search engine into extension controlled preference in test |
| 71 // profile. |
| 72 void SetExtensionDefaultSearchInPrefs(const std::string& ext_id, |
| 73 const TemplateURLData& ext_data); |
| 74 // Remove extension controlled default search engine preference from test |
| 75 // profile. |
| 76 void RemoveExtensionDefaultSearchFromPrefs(const std::string& ext_id); |
| 77 |
| 78 // Add extension turl to model and override default search preference if |
| 79 // extension wants to be default. |
| 80 TemplateURL* AddExtensionControlledTURL( |
| 81 std::unique_ptr<TemplateURL> ext_dse, |
| 82 std::unique_ptr<TemplateURL::AssociatedExtensionInfo> ext_info); |
| 83 // Remove extension turl from preferences and model. |
| 84 void RemoveExtensionControlledTURL(const std::string& ext_id); |
| 85 |
68 KeywordWebDataService* web_data_service() { return web_data_service_.get(); } | 86 KeywordWebDataService* web_data_service() { return web_data_service_.get(); } |
69 TemplateURLService* model() { return model_.get(); } | 87 TemplateURLService* model() { return model_.get(); } |
70 TestingProfile* profile() { return profile_.get(); } | 88 TestingProfile* profile() { return profile_.get(); } |
71 | 89 |
72 private: | 90 private: |
73 std::unique_ptr<TestingProfile> profile_; | 91 std::unique_ptr<TestingProfile> profile_; |
74 base::ScopedTempDir temp_dir_; | 92 base::ScopedTempDir temp_dir_; |
75 int changed_count_; | 93 int changed_count_; |
76 base::string16 search_term_; | 94 base::string16 search_term_; |
77 scoped_refptr<KeywordWebDataService> web_data_service_; | 95 scoped_refptr<KeywordWebDataService> web_data_service_; |
78 TestingSearchTermsData* search_terms_data_; | 96 TestingSearchTermsData* search_terms_data_; |
79 std::unique_ptr<TemplateURLService> model_; | 97 std::unique_ptr<TemplateURLService> model_; |
| 98 std::unique_ptr<ExtensionPrefValueMap> ext_pref_map_; |
80 | 99 |
81 DISALLOW_COPY_AND_ASSIGN(TemplateURLServiceTestUtil); | 100 DISALLOW_COPY_AND_ASSIGN(TemplateURLServiceTestUtil); |
82 }; | 101 }; |
83 | 102 |
84 #endif // CHROME_BROWSER_SEARCH_ENGINES_TEMPLATE_URL_SERVICE_TEST_UTIL_H_ | 103 #endif // CHROME_BROWSER_SEARCH_ENGINES_TEMPLATE_URL_SERVICE_TEST_UTIL_H_ |
OLD | NEW |