OLD | NEW |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 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 #include "chrome/browser/extensions/extension_action_manager.h" | 5 #include "chrome/browser/extensions/extension_action_manager.h" |
6 | 6 |
7 #include "base/strings/string_number_conversions.h" | 7 #include "base/strings/string_number_conversions.h" |
8 #include "chrome/browser/extensions/extension_action.h" | 8 #include "chrome/browser/extensions/extension_action.h" |
9 #include "chrome/test/base/testing_profile.h" | 9 #include "chrome/test/base/testing_profile.h" |
10 #include "content/public/test/test_browser_thread_bundle.h" | 10 #include "content/public/test/test_browser_thread_bundle.h" |
(...skipping 12 matching lines...) Expand all Loading... |
23 | 23 |
24 } // namespace | 24 } // namespace |
25 | 25 |
26 class ExtensionActionManagerTest : public testing::Test { | 26 class ExtensionActionManagerTest : public testing::Test { |
27 public: | 27 public: |
28 ExtensionActionManagerTest(); | 28 ExtensionActionManagerTest(); |
29 | 29 |
30 protected: | 30 protected: |
31 // Build an extension, populating |action_type| key with |action|, and | 31 // Build an extension, populating |action_type| key with |action|, and |
32 // "icons" key with |extension_icons|. | 32 // "icons" key with |extension_icons|. |
33 scoped_refptr<Extension> BuildExtension(DictionaryBuilder extension_icons, | 33 scoped_refptr<Extension> BuildExtension( |
34 DictionaryBuilder action, | 34 scoped_ptr<base::DictionaryValue> extension_icons, |
35 const char* action_type); | 35 scoped_ptr<base::DictionaryValue> action, |
| 36 const char* action_type); |
36 | 37 |
37 // Returns true if |action|'s title matches |extension|'s name. | 38 // Returns true if |action|'s title matches |extension|'s name. |
38 bool TitlesMatch(const Extension& extension, const ExtensionAction& action); | 39 bool TitlesMatch(const Extension& extension, const ExtensionAction& action); |
39 | 40 |
40 // Returns true if |action|'s icon for size |action_key| matches | 41 // Returns true if |action|'s icon for size |action_key| matches |
41 // |extension|'s icon for size |extension_key|; | 42 // |extension|'s icon for size |extension_key|; |
42 bool IconsMatch(const Extension& extension, | 43 bool IconsMatch(const Extension& extension, |
43 int extension_key, | 44 int extension_key, |
44 const ExtensionAction& action, | 45 const ExtensionAction& action, |
45 int action_key); | 46 int action_key); |
(...skipping 18 matching lines...) Expand all Loading... |
64 }; | 65 }; |
65 | 66 |
66 ExtensionActionManagerTest::ExtensionActionManagerTest() | 67 ExtensionActionManagerTest::ExtensionActionManagerTest() |
67 : curr_id_(0), | 68 : curr_id_(0), |
68 profile_(new TestingProfile) { | 69 profile_(new TestingProfile) { |
69 registry_ = ExtensionRegistry::Get(profile_.get()); | 70 registry_ = ExtensionRegistry::Get(profile_.get()); |
70 manager_ = ExtensionActionManager::Get(profile_.get()); | 71 manager_ = ExtensionActionManager::Get(profile_.get()); |
71 } | 72 } |
72 | 73 |
73 scoped_refptr<Extension> ExtensionActionManagerTest::BuildExtension( | 74 scoped_refptr<Extension> ExtensionActionManagerTest::BuildExtension( |
74 DictionaryBuilder extension_icons, | 75 scoped_ptr<base::DictionaryValue> extension_icons, |
75 DictionaryBuilder action, | 76 scoped_ptr<base::DictionaryValue> action, |
76 const char* action_type) { | 77 const char* action_type) { |
77 std::string id = base::IntToString(curr_id_++); | 78 std::string id = base::IntToString(curr_id_++); |
78 scoped_refptr<Extension> extension = | 79 scoped_refptr<Extension> extension = |
79 ExtensionBuilder() | 80 ExtensionBuilder() |
80 .SetManifest(std::move( | 81 .SetManifest( |
81 DictionaryBuilder() | 82 DictionaryBuilder() |
82 .Set("version", "1") | 83 .Set("version", "1") |
83 .Set("manifest_version", 2) | 84 .Set("manifest_version", 2) |
84 .Set("icons", std::move(extension_icons)) | 85 .Set("icons", std::move(extension_icons)) |
85 .Set(action_type, std::move(action)) | 86 .Set(action_type, std::move(action)) |
86 .Set("name", std::string("Test Extension").append(id)))) | 87 .Set("name", std::string("Test Extension").append(id)) |
| 88 .Build()) |
87 .SetID(id) | 89 .SetID(id) |
88 .Build(); | 90 .Build(); |
89 registry_->AddEnabled(extension); | 91 registry_->AddEnabled(extension); |
90 return extension; | 92 return extension; |
91 } | 93 } |
92 | 94 |
93 bool ExtensionActionManagerTest::TitlesMatch(const Extension& extension, | 95 bool ExtensionActionManagerTest::TitlesMatch(const Extension& extension, |
94 const ExtensionAction& action) { | 96 const ExtensionAction& action) { |
95 return action.GetTitle(ExtensionAction::kDefaultTabId) == extension.name(); | 97 return action.GetTitle(ExtensionAction::kDefaultTabId) == extension.name(); |
96 } | 98 } |
(...skipping 15 matching lines...) Expand all Loading... |
112 manager_->GetBrowserAction(extension) : | 114 manager_->GetBrowserAction(extension) : |
113 manager_->GetPageAction(extension); | 115 manager_->GetPageAction(extension); |
114 } | 116 } |
115 | 117 |
116 void ExtensionActionManagerTest::TestPopulateMissingValues( | 118 void ExtensionActionManagerTest::TestPopulateMissingValues( |
117 const char* action_type) { | 119 const char* action_type) { |
118 // Test that the largest icon from the extension's "icons" key is chosen as a | 120 // Test that the largest icon from the extension's "icons" key is chosen as a |
119 // replacement for missing action default_icons keys. "19" should not be | 121 // replacement for missing action default_icons keys. "19" should not be |
120 // replaced because "38" can always be used in its place. | 122 // replaced because "38" can always be used in its place. |
121 scoped_refptr<Extension> extension = | 123 scoped_refptr<Extension> extension = |
122 BuildExtension(std::move(DictionaryBuilder() | 124 BuildExtension(DictionaryBuilder() |
123 .Set("48", "icon48.png") | 125 .Set("48", "icon48.png") |
124 .Set("128", "icon128.png")), | 126 .Set("128", "icon128.png") |
125 DictionaryBuilder(), action_type); | 127 .Build(), |
| 128 DictionaryBuilder().Build(), action_type); |
126 | 129 |
127 ASSERT_TRUE(extension.get()); | 130 ASSERT_TRUE(extension.get()); |
128 const ExtensionAction* action = GetAction(action_type, *extension.get()); | 131 const ExtensionAction* action = GetAction(action_type, *extension.get()); |
129 ASSERT_TRUE(action); | 132 ASSERT_TRUE(action); |
130 | 133 |
131 ASSERT_TRUE(TitlesMatch(*extension.get(), *action)); | 134 ASSERT_TRUE(TitlesMatch(*extension.get(), *action)); |
132 ASSERT_TRUE(IconsMatch(*extension.get(), 48, *action, 38)); | 135 ASSERT_TRUE(IconsMatch(*extension.get(), 48, *action, 38)); |
133 | 136 |
134 // Test that the action's missing default_icons are not replaced with smaller | 137 // Test that the action's missing default_icons are not replaced with smaller |
135 // icons. | 138 // icons. |
136 extension = | 139 extension = |
137 BuildExtension(std::move(DictionaryBuilder().Set("24", "icon24.png")), | 140 BuildExtension(DictionaryBuilder().Set("24", "icon24.png").Build(), |
138 DictionaryBuilder(), action_type); | 141 DictionaryBuilder().Build(), action_type); |
139 | 142 |
140 ASSERT_TRUE(extension.get()); | 143 ASSERT_TRUE(extension.get()); |
141 action = GetAction(action_type, *extension.get()); | 144 action = GetAction(action_type, *extension.get()); |
142 ASSERT_TRUE(action); | 145 ASSERT_TRUE(action); |
143 | 146 |
144 ASSERT_TRUE(IconsMatch(*extension.get(), 24, *action, 19)); | 147 ASSERT_TRUE(IconsMatch(*extension.get(), 24, *action, 19)); |
145 ASSERT_FALSE(IconsMatch(*extension.get(), 24, *action, 38)); | 148 ASSERT_FALSE(IconsMatch(*extension.get(), 24, *action, 38)); |
146 | 149 |
147 // Test that an action's 19px icon is not replaced if a 38px action icon | 150 // Test that an action's 19px icon is not replaced if a 38px action icon |
148 // exists. | 151 // exists. |
149 extension = BuildExtension( | 152 extension = BuildExtension( |
150 std::move(DictionaryBuilder().Set("128", "icon128.png")), | 153 DictionaryBuilder().Set("128", "icon128.png").Build(), |
151 std::move(DictionaryBuilder().Set( | 154 DictionaryBuilder() |
152 "default_icon", | 155 .Set("default_icon", |
153 std::move(DictionaryBuilder().Set("38", "action38.png")))), | 156 DictionaryBuilder().Set("38", "action38.png").Build()) |
| 157 .Build(), |
154 action_type); | 158 action_type); |
155 | 159 |
156 ASSERT_TRUE(extension.get()); | 160 ASSERT_TRUE(extension.get()); |
157 action = GetAction(action_type, *extension.get()); | 161 action = GetAction(action_type, *extension.get()); |
158 ASSERT_TRUE(action); | 162 ASSERT_TRUE(action); |
159 | 163 |
160 ASSERT_FALSE(IconsMatch(*extension.get(), 128, *action, 19)); | 164 ASSERT_FALSE(IconsMatch(*extension.get(), 128, *action, 19)); |
161 | 165 |
162 // Test that existing default_icons and default_title are not replaced. | 166 // Test that existing default_icons and default_title are not replaced. |
163 extension = BuildExtension( | 167 extension = |
164 std::move(DictionaryBuilder().Set("128", "icon128.png")), | 168 BuildExtension(DictionaryBuilder().Set("128", "icon128.png").Build(), |
165 std::move( | 169 DictionaryBuilder() |
166 DictionaryBuilder() | 170 .Set("default_title", "Action!") |
167 .Set("default_title", "Action!") | 171 .Set("default_icon", DictionaryBuilder() |
168 .Set("default_icon", std::move(DictionaryBuilder() | 172 .Set("19", "action19.png") |
169 .Set("19", "action19.png") | 173 .Set("38", "action38.png") |
170 .Set("38", "action38.png")))), | 174 .Build()) |
171 action_type); | 175 .Build(), |
| 176 action_type); |
172 | 177 |
173 ASSERT_TRUE(extension.get()); | 178 ASSERT_TRUE(extension.get()); |
174 action = GetAction(action_type, *extension.get()); | 179 action = GetAction(action_type, *extension.get()); |
175 ASSERT_TRUE(action); | 180 ASSERT_TRUE(action); |
176 | 181 |
177 ASSERT_FALSE(TitlesMatch(*extension.get(), *action)); | 182 ASSERT_FALSE(TitlesMatch(*extension.get(), *action)); |
178 ASSERT_FALSE(IconsMatch(*extension.get(), 128, *action, 19)); | 183 ASSERT_FALSE(IconsMatch(*extension.get(), 128, *action, 19)); |
179 ASSERT_FALSE(IconsMatch(*extension.get(), 128, *action, 38)); | 184 ASSERT_FALSE(IconsMatch(*extension.get(), 128, *action, 38)); |
180 } | 185 } |
181 | 186 |
182 namespace { | 187 namespace { |
183 | 188 |
184 TEST_F(ExtensionActionManagerTest, PopulateBrowserAction) { | 189 TEST_F(ExtensionActionManagerTest, PopulateBrowserAction) { |
185 TestPopulateMissingValues(kBrowserAction); | 190 TestPopulateMissingValues(kBrowserAction); |
186 } | 191 } |
187 | 192 |
188 TEST_F(ExtensionActionManagerTest, PopulatePageAction) { | 193 TEST_F(ExtensionActionManagerTest, PopulatePageAction) { |
189 TestPopulateMissingValues(kPageAction); | 194 TestPopulateMissingValues(kPageAction); |
190 } | 195 } |
191 | 196 |
192 TEST_F(ExtensionActionManagerTest, GetBestFitActionTest) { | 197 TEST_F(ExtensionActionManagerTest, GetBestFitActionTest) { |
193 // Create an extension with page action defaults. | 198 // Create an extension with page action defaults. |
194 scoped_refptr<Extension> extension = BuildExtension( | 199 scoped_refptr<Extension> extension = BuildExtension( |
195 std::move(DictionaryBuilder().Set("48", "icon48.png")), | 200 DictionaryBuilder().Set("48", "icon48.png").Build(), |
196 std::move(DictionaryBuilder() | 201 DictionaryBuilder() |
197 .Set("default_title", "Action!") | 202 .Set("default_title", "Action!") |
198 .Set("default_icon", std::move(DictionaryBuilder().Set( | 203 .Set("default_icon", |
199 "38", "action38.png")))), | 204 DictionaryBuilder().Set("38", "action38.png").Build()) |
| 205 .Build(), |
200 kPageAction); | 206 kPageAction); |
201 ASSERT_TRUE(extension.get()); | 207 ASSERT_TRUE(extension.get()); |
202 | 208 |
203 // Get a "best fit" browser action for |extension|. | 209 // Get a "best fit" browser action for |extension|. |
204 scoped_ptr<ExtensionAction> action = | 210 scoped_ptr<ExtensionAction> action = |
205 manager()->GetBestFitAction(*extension.get(), ActionInfo::TYPE_BROWSER); | 211 manager()->GetBestFitAction(*extension.get(), ActionInfo::TYPE_BROWSER); |
206 ASSERT_TRUE(action.get()); | 212 ASSERT_TRUE(action.get()); |
207 ASSERT_EQ(action->action_type(), ActionInfo::TYPE_BROWSER); | 213 ASSERT_EQ(action->action_type(), ActionInfo::TYPE_BROWSER); |
208 | 214 |
209 // |action|'s title and default icon should match |extension|'s page action's. | 215 // |action|'s title and default icon should match |extension|'s page action's. |
210 ASSERT_EQ(action->GetTitle(ExtensionAction::kDefaultTabId), "Action!"); | 216 ASSERT_EQ(action->GetTitle(ExtensionAction::kDefaultTabId), "Action!"); |
211 ASSERT_EQ(action->default_icon()->Get(38, ExtensionIconSet::MATCH_EXACTLY), | 217 ASSERT_EQ(action->default_icon()->Get(38, ExtensionIconSet::MATCH_EXACTLY), |
212 "action38.png"); | 218 "action38.png"); |
213 | 219 |
214 // Create a new extension without page action defaults. | 220 // Create a new extension without page action defaults. |
215 extension = | 221 extension = |
216 BuildExtension(std::move(DictionaryBuilder().Set("48", "icon48.png")), | 222 BuildExtension(DictionaryBuilder().Set("48", "icon48.png").Build(), |
217 DictionaryBuilder(), kPageAction); | 223 DictionaryBuilder().Build(), kPageAction); |
218 ASSERT_TRUE(extension.get()); | 224 ASSERT_TRUE(extension.get()); |
219 | 225 |
220 action = | 226 action = |
221 manager()->GetBestFitAction(*extension.get(), ActionInfo::TYPE_BROWSER); | 227 manager()->GetBestFitAction(*extension.get(), ActionInfo::TYPE_BROWSER); |
222 | 228 |
223 // Now these values match because |extension| does not have page action | 229 // Now these values match because |extension| does not have page action |
224 // defaults. | 230 // defaults. |
225 ASSERT_TRUE(TitlesMatch(*extension.get(), *action)); | 231 ASSERT_TRUE(TitlesMatch(*extension.get(), *action)); |
226 ASSERT_TRUE(IconsMatch(*extension.get(), 48, *action, 38)); | 232 ASSERT_TRUE(IconsMatch(*extension.get(), 48, *action, 38)); |
227 } | 233 } |
228 | 234 |
229 } // namespace | 235 } // namespace |
230 } // namespace extensions | 236 } // namespace extensions |
OLD | NEW |