Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(6012)

Unified Diff: chrome/browser/extensions/extension_service_unittest.cc

Issue 120863003: Convert ExtensionRegistry to a BrowserContextKeyedService (Closed) Base URL: svn://svn.chromium.org/chrome/trunk/src
Patch Set: Created 7 years ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
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));

Powered by Google App Engine
This is Rietveld 408576698