| 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 #include <string> | 5 #include <string> |
| 6 | 6 |
| 7 #include "chrome/browser/extensions/component_loader.h" | 7 #include "chrome/browser/extensions/component_loader.h" |
| 8 | 8 |
| 9 #include "base/file_util.h" | 9 #include "base/file_util.h" |
| 10 #include "base/path_service.h" | 10 #include "base/path_service.h" |
| (...skipping 178 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 189 TEST_F(ComponentLoaderTest, Remove) { | 189 TEST_F(ComponentLoaderTest, Remove) { |
| 190 extension_service_.set_ready(false); | 190 extension_service_.set_ready(false); |
| 191 | 191 |
| 192 // Removing an extension that was never added should be ok. | 192 // Removing an extension that was never added should be ok. |
| 193 component_loader_.Remove(extension_path_); | 193 component_loader_.Remove(extension_path_); |
| 194 EXPECT_EQ(0u, extension_service_.extensions()->size()); | 194 EXPECT_EQ(0u, extension_service_.extensions()->size()); |
| 195 | 195 |
| 196 // Try adding and removing before LoadAll() is called. | 196 // Try adding and removing before LoadAll() is called. |
| 197 component_loader_.Add(manifest_contents_, extension_path_); | 197 component_loader_.Add(manifest_contents_, extension_path_); |
| 198 component_loader_.Remove(extension_path_); | 198 component_loader_.Remove(extension_path_); |
| 199 component_loader_.LoadAll(); | 199 component_loader_.LoadAll(false); |
| 200 EXPECT_EQ(0u, extension_service_.extensions()->size()); | 200 EXPECT_EQ(0u, extension_service_.extensions()->size()); |
| 201 | 201 |
| 202 // Load an extension, and check that it's unloaded when Remove() is called. | 202 // Load an extension, and check that it's unloaded when Remove() is called. |
| 203 extension_service_.set_ready(true); | 203 extension_service_.set_ready(true); |
| 204 std::string extension_id = | 204 std::string extension_id = |
| 205 component_loader_.Add(manifest_contents_, extension_path_); | 205 component_loader_.Add(manifest_contents_, extension_path_); |
| 206 EXPECT_EQ(1u, extension_service_.extensions()->size()); | 206 EXPECT_EQ(1u, extension_service_.extensions()->size()); |
| 207 component_loader_.Remove(extension_path_); | 207 component_loader_.Remove(extension_path_); |
| 208 EXPECT_EQ(0u, extension_service_.extensions()->size()); | 208 EXPECT_EQ(0u, extension_service_.extensions()->size()); |
| 209 | 209 |
| 210 // And after calling LoadAll(), it shouldn't get loaded. | 210 // And after calling LoadAll(), it shouldn't get loaded. |
| 211 component_loader_.LoadAll(); | 211 component_loader_.LoadAll(false); |
| 212 EXPECT_EQ(0u, extension_service_.extensions()->size()); | 212 EXPECT_EQ(0u, extension_service_.extensions()->size()); |
| 213 } | 213 } |
| 214 | 214 |
| 215 TEST_F(ComponentLoaderTest, LoadAll) { | 215 TEST_F(ComponentLoaderTest, LoadAll) { |
| 216 extension_service_.set_ready(false); | 216 extension_service_.set_ready(false); |
| 217 | 217 |
| 218 // No extensions should be loaded if none were added. | 218 // No extensions should be loaded if none were added. |
| 219 component_loader_.LoadAll(); | 219 component_loader_.LoadAll(false); |
| 220 EXPECT_EQ(0u, extension_service_.extensions()->size()); | 220 EXPECT_EQ(0u, extension_service_.extensions()->size()); |
| 221 | 221 |
| 222 // Use LoadAll() to load the default extensions. | 222 // Use LoadAll() to load the default extensions. |
| 223 component_loader_.AddDefaultComponentExtensions(false); | 223 component_loader_.AddDefaultComponentExtensions(false); |
| 224 component_loader_.LoadAll(); | 224 component_loader_.LoadAll(false); |
| 225 unsigned int default_count = extension_service_.extensions()->size(); | 225 unsigned int default_count = extension_service_.extensions()->size(); |
| 226 | 226 |
| 227 // Clear the list of loaded extensions, and reload with one more. | 227 // Clear the list of loaded extensions, and reload with one more. |
| 228 extension_service_.clear_extensions(); | 228 extension_service_.clear_extensions(); |
| 229 component_loader_.Add(manifest_contents_, extension_path_); | 229 component_loader_.Add(manifest_contents_, extension_path_); |
| 230 component_loader_.LoadAll(); | 230 component_loader_.LoadAll(false); |
| 231 | 231 |
| 232 EXPECT_EQ(default_count + 1, extension_service_.extensions()->size()); | 232 EXPECT_EQ(default_count + 1, extension_service_.extensions()->size()); |
| 233 } | 233 } |
| 234 | 234 |
| 235 TEST_F(ComponentLoaderTest, LoadAllDeferred) { |
| 236 extension_service_.set_ready(false); |
| 237 |
| 238 // Use LoadAll(true) to get a baseline of extensions with background pages. |
| 239 component_loader_.AddDefaultComponentExtensions(false); |
| 240 component_loader_.LoadAll(true); |
| 241 unsigned int default_count_no_bg = extension_service_.extensions()->size(); |
| 242 component_loader_.LoadAllDeferred(); |
| 243 unsigned int default_count_bg = extension_service_.extensions()->size(); |
| 244 |
| 245 // Clear and reload with one more, which is known to have a background page. |
| 246 extension_service_.clear_extensions(); |
| 247 component_loader_.Add(manifest_contents_, extension_path_); |
| 248 component_loader_.LoadAll(true); |
| 249 |
| 250 // Count should not change, since the load will be deferred. |
| 251 EXPECT_EQ(default_count_no_bg, extension_service_.extensions()->size()); |
| 252 |
| 253 // After loading deferred there should be one more than previously. |
| 254 component_loader_.LoadAllDeferred(); |
| 255 EXPECT_EQ(default_count_bg + 1, extension_service_.extensions()->size()); |
| 256 } |
| 257 |
| 235 TEST_F(ComponentLoaderTest, RemoveAll) { | 258 TEST_F(ComponentLoaderTest, RemoveAll) { |
| 236 extension_service_.set_ready(true); | 259 extension_service_.set_ready(true); |
| 237 EXPECT_EQ(0u, extension_service_.extensions()->size()); | 260 EXPECT_EQ(0u, extension_service_.extensions()->size()); |
| 238 // Use LoadAll() to load the default extensions. | 261 // Use LoadAll() to load the default extensions. |
| 239 component_loader_.AddDefaultComponentExtensions(false); | 262 component_loader_.AddDefaultComponentExtensions(false); |
| 240 unsigned int default_count = extension_service_.extensions()->size(); | 263 unsigned int default_count = extension_service_.extensions()->size(); |
| 241 | 264 |
| 242 // And add one more just to make sure there is anything in there in case | 265 // And add one more just to make sure there is anything in there in case |
| 243 // there are no defaults for this platform. | 266 // there are no defaults for this platform. |
| 244 component_loader_.Add(manifest_contents_, extension_path_); | 267 component_loader_.Add(manifest_contents_, extension_path_); |
| 245 EXPECT_EQ(default_count + 1, extension_service_.extensions()->size()); | 268 EXPECT_EQ(default_count + 1, extension_service_.extensions()->size()); |
| 246 | 269 |
| 247 // Remove all default extensions. | 270 // Remove all default extensions. |
| 248 component_loader_.RemoveAll(); | 271 component_loader_.RemoveAll(); |
| 249 EXPECT_EQ(0u, extension_service_.extensions()->size()); | 272 EXPECT_EQ(0u, extension_service_.extensions()->size()); |
| 250 } | 273 } |
| 251 | 274 |
| 252 TEST_F(ComponentLoaderTest, EnterpriseWebStore) { | 275 TEST_F(ComponentLoaderTest, EnterpriseWebStore) { |
| 253 component_loader_.AddDefaultComponentExtensions(false); | 276 component_loader_.AddDefaultComponentExtensions(false); |
| 254 component_loader_.LoadAll(); | 277 component_loader_.LoadAll(false); |
| 255 unsigned int default_count = extension_service_.extensions()->size(); | 278 unsigned int default_count = extension_service_.extensions()->size(); |
| 256 | 279 |
| 257 // Set the pref, and it should get loaded automatically. | 280 // Set the pref, and it should get loaded automatically. |
| 258 extension_service_.set_ready(true); | 281 extension_service_.set_ready(true); |
| 259 prefs_.SetUserPref(prefs::kEnterpriseWebStoreURL, | 282 prefs_.SetUserPref(prefs::kEnterpriseWebStoreURL, |
| 260 Value::CreateStringValue("http://www.google.com")); | 283 Value::CreateStringValue("http://www.google.com")); |
| 261 EXPECT_EQ(default_count + 1, extension_service_.extensions()->size()); | 284 EXPECT_EQ(default_count + 1, extension_service_.extensions()->size()); |
| 262 | 285 |
| 263 // Now that the pref is set, check if it's added by default. | 286 // Now that the pref is set, check if it's added by default. |
| 264 extension_service_.set_ready(false); | 287 extension_service_.set_ready(false); |
| 265 extension_service_.clear_extensions(); | 288 extension_service_.clear_extensions(); |
| 266 component_loader_.ClearAllRegistered(); | 289 component_loader_.ClearAllRegistered(); |
| 267 component_loader_.AddDefaultComponentExtensions(false); | 290 component_loader_.AddDefaultComponentExtensions(false); |
| 268 component_loader_.LoadAll(); | 291 component_loader_.LoadAll(false); |
| 269 EXPECT_EQ(default_count + 1, extension_service_.extensions()->size()); | 292 EXPECT_EQ(default_count + 1, extension_service_.extensions()->size()); |
| 270 | 293 |
| 271 // Number of loaded extensions should be the same after changing the pref. | 294 // Number of loaded extensions should be the same after changing the pref. |
| 272 prefs_.SetUserPref(prefs::kEnterpriseWebStoreURL, | 295 prefs_.SetUserPref(prefs::kEnterpriseWebStoreURL, |
| 273 Value::CreateStringValue("http://www.google.de")); | 296 Value::CreateStringValue("http://www.google.de")); |
| 274 EXPECT_EQ(default_count + 1, extension_service_.extensions()->size()); | 297 EXPECT_EQ(default_count + 1, extension_service_.extensions()->size()); |
| 275 } | 298 } |
| 276 | 299 |
| 277 TEST_F(ComponentLoaderTest, AddOrReplace) { | 300 TEST_F(ComponentLoaderTest, AddOrReplace) { |
| 278 EXPECT_EQ(0u, component_loader_.registered_extensions_count()); | 301 EXPECT_EQ(0u, component_loader_.registered_extensions_count()); |
| 279 component_loader_.AddDefaultComponentExtensions(false); | 302 component_loader_.AddDefaultComponentExtensions(false); |
| 280 size_t const default_count = component_loader_.registered_extensions_count(); | 303 size_t const default_count = component_loader_.registered_extensions_count(); |
| 281 FilePath known_extension = GetBasePath() | 304 FilePath known_extension = GetBasePath() |
| 282 .AppendASCII("override_component_extension"); | 305 .AppendASCII("override_component_extension"); |
| 283 FilePath unknow_extension = extension_path_; | 306 FilePath unknow_extension = extension_path_; |
| 284 | 307 |
| 285 // Replace a default component extension. | 308 // Replace a default component extension. |
| 286 component_loader_.AddOrReplace(known_extension); | 309 component_loader_.AddOrReplace(known_extension); |
| 287 EXPECT_EQ(default_count, | 310 EXPECT_EQ(default_count, |
| 288 component_loader_.registered_extensions_count()); | 311 component_loader_.registered_extensions_count()); |
| 289 | 312 |
| 290 // Add a new component extension. | 313 // Add a new component extension. |
| 291 component_loader_.AddOrReplace(unknow_extension); | 314 component_loader_.AddOrReplace(unknow_extension); |
| 292 EXPECT_EQ(default_count + 1, | 315 EXPECT_EQ(default_count + 1, |
| 293 component_loader_.registered_extensions_count()); | 316 component_loader_.registered_extensions_count()); |
| 294 | 317 |
| 295 extension_service_.set_ready(true); | 318 extension_service_.set_ready(true); |
| 296 component_loader_.LoadAll(); | 319 component_loader_.LoadAll(false); |
| 297 | 320 |
| 298 EXPECT_EQ(default_count + 1, extension_service_.extensions()->size()); | 321 EXPECT_EQ(default_count + 1, extension_service_.extensions()->size()); |
| 299 EXPECT_EQ(0u, extension_service_.unloaded_count()); | 322 EXPECT_EQ(0u, extension_service_.unloaded_count()); |
| 300 | 323 |
| 301 // replace loaded component extension. | 324 // replace loaded component extension. |
| 302 component_loader_.AddOrReplace(known_extension); | 325 component_loader_.AddOrReplace(known_extension); |
| 303 EXPECT_EQ(default_count + 1, extension_service_.extensions()->size()); | 326 EXPECT_EQ(default_count + 1, extension_service_.extensions()->size()); |
| 304 EXPECT_EQ(1u, extension_service_.unloaded_count()); | 327 EXPECT_EQ(1u, extension_service_.unloaded_count()); |
| 305 } | 328 } |
| 306 | 329 |
| 307 } // namespace extensions | 330 } // namespace extensions |
| OLD | NEW |