| 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 175 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 186 EXPECT_TRUE(extension_service_.extensions()->GetByID(extension_id)); | 186 EXPECT_TRUE(extension_service_.extensions()->GetByID(extension_id)); |
| 187 } | 187 } |
| 188 | 188 |
| 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 BulkLoadAll() 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_.BulkLoadAll(); |
| 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 BulkLoadAll(), it shouldn't get loaded. |
| 211 component_loader_.LoadAll(); | 211 component_loader_.BulkLoadAll(); |
| 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_.BulkLoadAll(); |
| 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 BulkLoadAll() to load the default extensions. |
| 223 component_loader_.AddDefaultComponentExtensions(false); | 223 component_loader_.AddDefaultComponentExtensions(false); |
| 224 component_loader_.LoadAll(); | 224 component_loader_.BulkLoadAll(); |
| 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_.BulkLoadAll(); |
| 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, BulkLoadDeferred) { |
| 236 extension_service_.set_ready(false); |
| 237 |
| 238 // Use BulkLoadDeferBackgroundPages to get a baseline of extensions with |
| 239 // background pages. |
| 240 component_loader_.AddDefaultComponentExtensions(false); |
| 241 component_loader_.BulkLoadDeferBackgroundPages(); |
| 242 unsigned int default_count_no_bg = extension_service_.extensions()->size(); |
| 243 component_loader_.BulkLoadDeferred(); |
| 244 unsigned int default_count_bg = extension_service_.extensions()->size(); |
| 245 |
| 246 // Clear and reload with one more, which is known to have a background page. |
| 247 extension_service_.clear_extensions(); |
| 248 component_loader_.Add(manifest_contents_, extension_path_); |
| 249 component_loader_.BulkLoadDeferBackgroundPages(); |
| 250 |
| 251 // Count should not change, since the load will be deferred. |
| 252 EXPECT_EQ(default_count_no_bg, extension_service_.extensions()->size()); |
| 253 |
| 254 // After loading deferred there should be one more than previously. |
| 255 component_loader_.BulkLoadDeferred(); |
| 256 EXPECT_EQ(default_count_bg + 1, extension_service_.extensions()->size()); |
| 257 } |
| 258 |
| 235 TEST_F(ComponentLoaderTest, RemoveAll) { | 259 TEST_F(ComponentLoaderTest, RemoveAll) { |
| 236 extension_service_.set_ready(true); | 260 extension_service_.set_ready(true); |
| 237 EXPECT_EQ(0u, extension_service_.extensions()->size()); | 261 EXPECT_EQ(0u, extension_service_.extensions()->size()); |
| 238 // Use LoadAll() to load the default extensions. | 262 // Add all the default extensions. Since the extension service is ready, they |
| 263 // will be loaded immediately. |
| 239 component_loader_.AddDefaultComponentExtensions(false); | 264 component_loader_.AddDefaultComponentExtensions(false); |
| 240 unsigned int default_count = extension_service_.extensions()->size(); | 265 unsigned int default_count = extension_service_.extensions()->size(); |
| 241 | 266 |
| 242 // And add one more just to make sure there is anything in there in case | 267 // And add one more just to make sure there is anything in there in case |
| 243 // there are no defaults for this platform. | 268 // there are no defaults for this platform. |
| 244 component_loader_.Add(manifest_contents_, extension_path_); | 269 component_loader_.Add(manifest_contents_, extension_path_); |
| 245 EXPECT_EQ(default_count + 1, extension_service_.extensions()->size()); | 270 EXPECT_EQ(default_count + 1, extension_service_.extensions()->size()); |
| 246 | 271 |
| 247 // Remove all default extensions. | 272 // Remove all default extensions. |
| 248 component_loader_.RemoveAll(); | 273 component_loader_.RemoveAll(); |
| 249 EXPECT_EQ(0u, extension_service_.extensions()->size()); | 274 EXPECT_EQ(0u, extension_service_.extensions()->size()); |
| 250 } | 275 } |
| 251 | 276 |
| 252 TEST_F(ComponentLoaderTest, EnterpriseWebStore) { | 277 TEST_F(ComponentLoaderTest, EnterpriseWebStore) { |
| 253 component_loader_.AddDefaultComponentExtensions(false); | 278 component_loader_.AddDefaultComponentExtensions(false); |
| 254 component_loader_.LoadAll(); | 279 component_loader_.BulkLoadAll(); |
| 255 unsigned int default_count = extension_service_.extensions()->size(); | 280 unsigned int default_count = extension_service_.extensions()->size(); |
| 256 | 281 |
| 257 // Set the pref, and it should get loaded automatically. | 282 // Set the pref, and it should get loaded automatically. |
| 258 extension_service_.set_ready(true); | 283 extension_service_.set_ready(true); |
| 259 prefs_.SetUserPref(prefs::kEnterpriseWebStoreURL, | 284 prefs_.SetUserPref(prefs::kEnterpriseWebStoreURL, |
| 260 Value::CreateStringValue("http://www.google.com")); | 285 Value::CreateStringValue("http://www.google.com")); |
| 261 EXPECT_EQ(default_count + 1, extension_service_.extensions()->size()); | 286 EXPECT_EQ(default_count + 1, extension_service_.extensions()->size()); |
| 262 | 287 |
| 263 // Now that the pref is set, check if it's added by default. | 288 // Now that the pref is set, check if it's added by default. |
| 264 extension_service_.set_ready(false); | 289 extension_service_.set_ready(false); |
| 265 extension_service_.clear_extensions(); | 290 extension_service_.clear_extensions(); |
| 266 component_loader_.ClearAllRegistered(); | 291 component_loader_.ClearAllRegistered(); |
| 267 component_loader_.AddDefaultComponentExtensions(false); | 292 component_loader_.AddDefaultComponentExtensions(false); |
| 268 component_loader_.LoadAll(); | 293 component_loader_.BulkLoadAll(); |
| 269 EXPECT_EQ(default_count + 1, extension_service_.extensions()->size()); | 294 EXPECT_EQ(default_count + 1, extension_service_.extensions()->size()); |
| 270 | 295 |
| 271 // Number of loaded extensions should be the same after changing the pref. | 296 // Number of loaded extensions should be the same after changing the pref. |
| 272 prefs_.SetUserPref(prefs::kEnterpriseWebStoreURL, | 297 prefs_.SetUserPref(prefs::kEnterpriseWebStoreURL, |
| 273 Value::CreateStringValue("http://www.google.de")); | 298 Value::CreateStringValue("http://www.google.de")); |
| 274 EXPECT_EQ(default_count + 1, extension_service_.extensions()->size()); | 299 EXPECT_EQ(default_count + 1, extension_service_.extensions()->size()); |
| 275 } | 300 } |
| 276 | 301 |
| 277 TEST_F(ComponentLoaderTest, AddOrReplace) { | 302 TEST_F(ComponentLoaderTest, AddOrReplace) { |
| 278 EXPECT_EQ(0u, component_loader_.registered_extensions_count()); | 303 EXPECT_EQ(0u, component_loader_.registered_extensions_count()); |
| 279 component_loader_.AddDefaultComponentExtensions(false); | 304 component_loader_.AddDefaultComponentExtensions(false); |
| 280 size_t const default_count = component_loader_.registered_extensions_count(); | 305 size_t const default_count = component_loader_.registered_extensions_count(); |
| 281 FilePath known_extension = GetBasePath() | 306 FilePath known_extension = GetBasePath() |
| 282 .AppendASCII("override_component_extension"); | 307 .AppendASCII("override_component_extension"); |
| 283 FilePath unknow_extension = extension_path_; | 308 FilePath unknow_extension = extension_path_; |
| 284 | 309 |
| 285 // Replace a default component extension. | 310 // Replace a default component extension. |
| 286 component_loader_.AddOrReplace(known_extension); | 311 component_loader_.AddOrReplace(known_extension); |
| 287 EXPECT_EQ(default_count, | 312 EXPECT_EQ(default_count, |
| 288 component_loader_.registered_extensions_count()); | 313 component_loader_.registered_extensions_count()); |
| 289 | 314 |
| 290 // Add a new component extension. | 315 // Add a new component extension. |
| 291 component_loader_.AddOrReplace(unknow_extension); | 316 component_loader_.AddOrReplace(unknow_extension); |
| 292 EXPECT_EQ(default_count + 1, | 317 EXPECT_EQ(default_count + 1, |
| 293 component_loader_.registered_extensions_count()); | 318 component_loader_.registered_extensions_count()); |
| 294 | 319 |
| 295 extension_service_.set_ready(true); | 320 extension_service_.set_ready(true); |
| 296 component_loader_.LoadAll(); | 321 component_loader_.BulkLoadAll(); |
| 297 | 322 |
| 298 EXPECT_EQ(default_count + 1, extension_service_.extensions()->size()); | 323 EXPECT_EQ(default_count + 1, extension_service_.extensions()->size()); |
| 299 EXPECT_EQ(0u, extension_service_.unloaded_count()); | 324 EXPECT_EQ(0u, extension_service_.unloaded_count()); |
| 300 | 325 |
| 301 // replace loaded component extension. | 326 // replace loaded component extension. |
| 302 component_loader_.AddOrReplace(known_extension); | 327 component_loader_.AddOrReplace(known_extension); |
| 303 EXPECT_EQ(default_count + 1, extension_service_.extensions()->size()); | 328 EXPECT_EQ(default_count + 1, extension_service_.extensions()->size()); |
| 304 EXPECT_EQ(1u, extension_service_.unloaded_count()); | 329 EXPECT_EQ(1u, extension_service_.unloaded_count()); |
| 305 } | 330 } |
| 306 | 331 |
| 307 } // namespace extensions | 332 } // namespace extensions |
| OLD | NEW |