Index: chrome/browser/extensions/extension_service_unittest.cc |
diff --git a/chrome/browser/extensions/extension_service_unittest.cc b/chrome/browser/extensions/extension_service_unittest.cc |
index bf5bed9a8dfd56aa897f786222374e079356c28f..6dae5fc968dd1f72e28ce001abc732e0370e49d1 100644 |
--- a/chrome/browser/extensions/extension_service_unittest.cc |
+++ b/chrome/browser/extensions/extension_service_unittest.cc |
@@ -87,6 +87,7 @@ |
#include "content/public/browser/storage_partition.h" |
#include "content/public/common/content_constants.h" |
#include "content/public/test/test_utils.h" |
+#include "extensions/browser/extension_registry.h" |
#include "extensions/browser/external_provider_interface.h" |
#include "extensions/browser/management_policy.h" |
#include "extensions/browser/pending_extension_info.h" |
@@ -151,6 +152,7 @@ using extensions::CrxInstaller; |
using extensions::Extension; |
using extensions::ExtensionCreator; |
using extensions::ExtensionPrefs; |
+using extensions::ExtensionRegistry; |
using extensions::ExtensionResource; |
using extensions::ExtensionSystem; |
using extensions::FakeSafeBrowsingDatabaseManager; |
@@ -3369,16 +3371,18 @@ TEST_F(ExtensionServiceTest, SetUnsetBlacklistInPrefs) { |
InitializeGoodInstalledExtensionService(); |
service_->Init(); |
- const extensions::ExtensionSet* extensions = service_->extensions(); |
- const extensions::ExtensionSet* blacklisted_extensions = |
- service_->blacklisted_extensions(); |
+ ExtensionRegistry* registry = ExtensionRegistry::Get(profile_.get()); |
+ const extensions::ExtensionSet& enabled_extensions = |
+ registry->enabled_extensions(); |
+ const extensions::ExtensionSet& blacklisted_extensions = |
+ registry->blacklisted_extensions(); |
- EXPECT_TRUE( extensions->Contains(good0) && |
- !blacklisted_extensions->Contains(good0)); |
- EXPECT_TRUE( extensions->Contains(good1) && |
- !blacklisted_extensions->Contains(good1)); |
- EXPECT_TRUE( extensions->Contains(good2) && |
- !blacklisted_extensions->Contains(good2)); |
+ EXPECT_TRUE(enabled_extensions.Contains(good0) && |
+ !blacklisted_extensions.Contains(good0)); |
+ EXPECT_TRUE(enabled_extensions.Contains(good1) && |
+ !blacklisted_extensions.Contains(good1)); |
+ EXPECT_TRUE(enabled_extensions.Contains(good2) && |
+ !blacklisted_extensions.Contains(good2)); |
EXPECT_FALSE(IsPrefExist(good0, "blacklist")); |
EXPECT_FALSE(IsPrefExist(good1, "blacklist")); |
@@ -3389,12 +3393,12 @@ TEST_F(ExtensionServiceTest, SetUnsetBlacklistInPrefs) { |
blacklist_db->SetUnsafe(good0, good1, "invalid_id").NotifyUpdate(); |
base::RunLoop().RunUntilIdle(); |
- EXPECT_TRUE(!extensions->Contains(good0) && |
- blacklisted_extensions->Contains(good0)); |
- EXPECT_TRUE(!extensions->Contains(good1) && |
- blacklisted_extensions->Contains(good1)); |
- EXPECT_TRUE( extensions->Contains(good2) && |
- !blacklisted_extensions->Contains(good2)); |
+ EXPECT_TRUE(!enabled_extensions.Contains(good0) && |
+ blacklisted_extensions.Contains(good0)); |
+ EXPECT_TRUE(!enabled_extensions.Contains(good1) && |
+ blacklisted_extensions.Contains(good1)); |
+ EXPECT_TRUE(enabled_extensions.Contains(good2) && |
+ !blacklisted_extensions.Contains(good2)); |
EXPECT_TRUE(ValidateBooleanPref(good0, "blacklist", true)); |
EXPECT_TRUE(ValidateBooleanPref(good1, "blacklist", true)); |
@@ -3405,12 +3409,12 @@ TEST_F(ExtensionServiceTest, SetUnsetBlacklistInPrefs) { |
blacklist_db->SetUnsafe(good0, good2, "invalid_id").NotifyUpdate(); |
base::RunLoop().RunUntilIdle(); |
- EXPECT_TRUE(!extensions->Contains(good0) && |
- blacklisted_extensions->Contains(good0)); |
- EXPECT_TRUE( extensions->Contains(good1) && |
- !blacklisted_extensions->Contains(good1)); |
- EXPECT_TRUE(!extensions->Contains(good2) && |
- blacklisted_extensions->Contains(good2)); |
+ EXPECT_TRUE(!enabled_extensions.Contains(good0) && |
+ blacklisted_extensions.Contains(good0)); |
+ EXPECT_TRUE(enabled_extensions.Contains(good1) && |
+ !blacklisted_extensions.Contains(good1)); |
+ EXPECT_TRUE(!enabled_extensions.Contains(good2) && |
+ blacklisted_extensions.Contains(good2)); |
EXPECT_TRUE(ValidateBooleanPref(good0, "blacklist", true)); |
EXPECT_FALSE(IsPrefExist(good1, "blacklist")); |
@@ -3492,12 +3496,14 @@ TEST_F(ExtensionServiceTest, WillNotLoadBlacklistedExtensionsFromDirectory) { |
ASSERT_EQ(3u, loaded_.size()); // hasn't had time to blacklist yet |
base::RunLoop().RunUntilIdle(); |
- ASSERT_EQ(1u, service_->blacklisted_extensions()->size()); |
- ASSERT_EQ(2u, service_->extensions()->size()); |
- ASSERT_TRUE(service_->extensions()->Contains(good0)); |
- ASSERT_TRUE(service_->blacklisted_extensions()->Contains(good1)); |
- ASSERT_TRUE(service_->extensions()->Contains(good2)); |
+ ExtensionRegistry* registry = ExtensionRegistry::Get(profile_.get()); |
+ ASSERT_EQ(1u, registry->blacklisted_extensions().size()); |
+ ASSERT_EQ(2u, registry->enabled_extensions().size()); |
+ |
+ ASSERT_TRUE(registry->enabled_extensions().Contains(good0)); |
+ ASSERT_TRUE(registry->blacklisted_extensions().Contains(good1)); |
+ ASSERT_TRUE(registry->enabled_extensions().Contains(good2)); |
} |
#endif // defined(ENABLE_BLACKLIST_TESTS) |
@@ -3517,22 +3523,23 @@ TEST_F(ExtensionServiceTest, BlacklistedInPrefsFromStartup) { |
service_->Init(); |
- ASSERT_EQ(2u, service_->blacklisted_extensions()->size()); |
- ASSERT_EQ(1u, service_->extensions()->size()); |
+ ExtensionRegistry* registry = ExtensionRegistry::Get(profile_.get()); |
+ ASSERT_EQ(2u, registry->blacklisted_extensions().size()); |
+ ASSERT_EQ(1u, registry->enabled_extensions().size()); |
- ASSERT_TRUE(service_->blacklisted_extensions()->Contains(good0)); |
- ASSERT_TRUE(service_->blacklisted_extensions()->Contains(good1)); |
- ASSERT_TRUE(service_->extensions()->Contains(good2)); |
+ ASSERT_TRUE(registry->blacklisted_extensions().Contains(good0)); |
+ ASSERT_TRUE(registry->blacklisted_extensions().Contains(good1)); |
+ ASSERT_TRUE(registry->enabled_extensions().Contains(good2)); |
// Give time for the blacklist to update. |
base::RunLoop().RunUntilIdle(); |
- ASSERT_EQ(1u, service_->blacklisted_extensions()->size()); |
- ASSERT_EQ(2u, service_->extensions()->size()); |
+ ASSERT_EQ(1u, registry->blacklisted_extensions().size()); |
+ ASSERT_EQ(2u, registry->enabled_extensions().size()); |
- ASSERT_TRUE(service_->extensions()->Contains(good0)); |
- ASSERT_TRUE(service_->blacklisted_extensions()->Contains(good1)); |
- ASSERT_TRUE(service_->extensions()->Contains(good2)); |
+ ASSERT_TRUE(registry->enabled_extensions().Contains(good0)); |
+ ASSERT_TRUE(registry->blacklisted_extensions().Contains(good1)); |
+ ASSERT_TRUE(registry->enabled_extensions().Contains(good2)); |
} |
#endif // defined(ENABLE_BLACKLIST_TESTS) |
@@ -3915,37 +3922,40 @@ TEST_F(ExtensionServiceTest, DisableExtension) { |
InstallCRX(data_dir_.AppendASCII("good.crx"), INSTALL_NEW); |
EXPECT_TRUE(service_->GetExtensionById(good_crx, true)); |
EXPECT_TRUE(service_->GetExtensionById(good_crx, false)); |
- EXPECT_EQ(1u, service_->extensions()->size()); |
- EXPECT_EQ(0u, service_->disabled_extensions()->size()); |
- EXPECT_EQ(0u, service_->terminated_extensions()->size()); |
- EXPECT_EQ(0u, service_->blacklisted_extensions()->size()); |
+ |
+ ExtensionRegistry* registry = ExtensionRegistry::Get(profile_.get()); |
+ EXPECT_EQ(1u, registry->enabled_extensions().size()); |
+ EXPECT_EQ(0u, registry->disabled_extensions().size()); |
+ EXPECT_EQ(0u, registry->terminated_extensions().size()); |
+ EXPECT_EQ(0u, registry->blacklisted_extensions().size()); |
// Disable it. |
service_->DisableExtension(good_crx, Extension::DISABLE_USER_ACTION); |
EXPECT_TRUE(service_->GetExtensionById(good_crx, true)); |
EXPECT_FALSE(service_->GetExtensionById(good_crx, false)); |
- EXPECT_EQ(0u, service_->extensions()->size()); |
- EXPECT_EQ(1u, service_->disabled_extensions()->size()); |
- EXPECT_EQ(0u, service_->terminated_extensions()->size()); |
- EXPECT_EQ(0u, service_->blacklisted_extensions()->size()); |
+ EXPECT_EQ(0u, registry->enabled_extensions().size()); |
+ EXPECT_EQ(1u, registry->disabled_extensions().size()); |
+ EXPECT_EQ(0u, registry->terminated_extensions().size()); |
+ EXPECT_EQ(0u, registry->blacklisted_extensions().size()); |
} |
TEST_F(ExtensionServiceTest, TerminateExtension) { |
InitializeEmptyExtensionService(); |
InstallCRX(data_dir_.AppendASCII("good.crx"), INSTALL_NEW); |
- EXPECT_EQ(1u, service_->extensions()->size()); |
- EXPECT_EQ(0u, service_->disabled_extensions()->size()); |
- EXPECT_EQ(0u, service_->terminated_extensions()->size()); |
- EXPECT_EQ(0u, service_->blacklisted_extensions()->size()); |
+ ExtensionRegistry* registry = ExtensionRegistry::Get(profile_.get()); |
+ EXPECT_EQ(1u, registry->enabled_extensions().size()); |
+ EXPECT_EQ(0u, registry->disabled_extensions().size()); |
+ EXPECT_EQ(0u, registry->terminated_extensions().size()); |
+ EXPECT_EQ(0u, registry->blacklisted_extensions().size()); |
TerminateExtension(good_crx); |
- EXPECT_EQ(0u, service_->extensions()->size()); |
- EXPECT_EQ(0u, service_->disabled_extensions()->size()); |
- EXPECT_EQ(1u, service_->terminated_extensions()->size()); |
- EXPECT_EQ(0u, service_->blacklisted_extensions()->size()); |
+ EXPECT_EQ(0u, registry->enabled_extensions().size()); |
+ EXPECT_EQ(0u, registry->disabled_extensions().size()); |
+ EXPECT_EQ(1u, registry->terminated_extensions().size()); |
+ EXPECT_EQ(0u, registry->blacklisted_extensions().size()); |
} |
TEST_F(ExtensionServiceTest, DisableTerminatedExtension) { |
@@ -3961,10 +3971,11 @@ TEST_F(ExtensionServiceTest, DisableTerminatedExtension) { |
EXPECT_FALSE(service_->GetTerminatedExtension(good_crx)); |
EXPECT_TRUE(service_->GetExtensionById(good_crx, true)); |
- EXPECT_EQ(0u, service_->extensions()->size()); |
- EXPECT_EQ(1u, service_->disabled_extensions()->size()); |
- EXPECT_EQ(0u, service_->terminated_extensions()->size()); |
- EXPECT_EQ(0u, service_->blacklisted_extensions()->size()); |
+ ExtensionRegistry* registry = ExtensionRegistry::Get(profile_.get()); |
+ EXPECT_EQ(0u, registry->enabled_extensions().size()); |
+ EXPECT_EQ(1u, registry->disabled_extensions().size()); |
+ EXPECT_EQ(0u, registry->terminated_extensions().size()); |
+ EXPECT_EQ(0u, registry->blacklisted_extensions().size()); |
} |
// Tests disabling all extensions (simulating --disable-extensions flag). |
@@ -6703,10 +6714,12 @@ TEST_F(ExtensionServiceTest, InstallBlacklistedExtension) { |
// Extension was installed but not loaded. |
EXPECT_TRUE(notifications.CheckNotifications( |
chrome::NOTIFICATION_EXTENSION_INSTALLED)); |
- |
EXPECT_TRUE(service_->GetInstalledExtension(id)); |
- EXPECT_FALSE(service_->extensions()->Contains(id)); |
- EXPECT_TRUE(service_->blacklisted_extensions()->Contains(id)); |
+ |
+ ExtensionRegistry* registry = ExtensionRegistry::Get(profile_.get()); |
+ EXPECT_FALSE(registry->enabled_extensions().Contains(id)); |
+ EXPECT_TRUE(registry->blacklisted_extensions().Contains(id)); |
+ |
EXPECT_TRUE(service_->extension_prefs()->IsExtensionBlacklisted(id)); |
EXPECT_TRUE( |
service_->extension_prefs()->IsBlacklistedExtensionAcknowledged(id)); |