app_shell: Extract extension runtime data from ExtensionService

extensions/browser/extension_registry_unittest.cc

 // Copyright 2013 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "extensions/browser/extension_registry.h"
 
 #include <string>
 
 #include "base/memory/ref_counted.h"
 #include "extensions/common/extension.h"
@@ skipping 32 matching lines @@
   // Clearing the registry clears all sets.
   registry.ClearAll();
 
   EXPECT_EQ(0u, 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());
 }
 
 // A simple test of adding and removing things from sets.
-TEST_F(ExtensionRegistryTest, AddAndRemoveExtensionFromRegistry) {
+TEST_F(ExtensionRegistryTest, AddExtensionToRegistry) {
   ExtensionRegistry registry;
 
   // Adding an extension works.
   scoped_refptr<Extension> extension = test_util::CreateExtensionWithID("id");
   EXPECT_TRUE(registry.AddEnabled(extension));
   EXPECT_EQ(1u, registry.enabled_extensions().size());
 
   // The extension was only added to one set.
   EXPECT_EQ(0u, registry.disabled_extensions().size());
   EXPECT_EQ(0u, registry.terminated_extensions().size());
   EXPECT_EQ(0u, registry.blacklisted_extensions().size());
-
-  // Removing an extension works.
-  EXPECT_TRUE(registry.RemoveEnabled(extension->id()));
-  EXPECT_EQ(0u, registry.enabled_extensions().size());
-
-  // Trying to remove an extension that isn't in the set fails cleanly.
-  EXPECT_FALSE(registry.RemoveEnabled(extension->id()));
 }
 
 TEST_F(ExtensionRegistryTest, AddExtensionToRegistryTwice) {
   ExtensionRegistry registry;
   scoped_refptr<Extension> extension = test_util::CreateExtensionWithID("id");
 
   // An extension can exist in two sets at once. It would be nice to eliminate
   // this functionality, but some users of ExtensionRegistry need it.
   EXPECT_TRUE(registry.AddEnabled(extension));
   EXPECT_TRUE(registry.AddDisabled(extension));
@@ skipping 76 matching lines @@
 
   // Enabled can be found with multiple flags set.
   EXPECT_TRUE(registry.GetExtensionById(
       "enabled", ExtensionRegistry::ENABLED | ExtensionRegistry::TERMINATED));
 
   // Enabled isn't found if the wrong flags are set.
   EXPECT_FALSE(registry.GetExtensionById(
       "enabled", ExtensionRegistry::DISABLED | ExtensionRegistry::BLACKLISTED));
 }
 
+TEST_F(ExtensionRegistryTest, UnloadExtension) {
+  ExtensionRegistry registry;
+  scoped_refptr<Extension> extension = test_util::CreateExtensionWithID("id");
+
+  // Unloading an extension removes it from the enabled set.
+  registry.AddEnabled(extension);
+  ASSERT_EQ(1u, registry.enabled_extensions().size());
+  registry.UnloadExtension(extension);
+  EXPECT_EQ(0u, registry.enabled_extensions().size());
+
+  // Unloading an extension removes it from the disabled set.
+  registry.AddDisabled(extension);
+  ASSERT_EQ(1u, registry.disabled_extensions().size());
+  registry.UnloadExtension(extension);
+  EXPECT_EQ(0u, registry.disabled_extensions().size());
+}
+
+TEST_F(ExtensionRegistryTest, EnableExtension) {
+  ExtensionRegistry registry;
+  scoped_refptr<Extension> extension = test_util::CreateExtensionWithID("id");
+
+  // Enabling an extension moves it from disabled to enabled.
+  registry.AddDisabled(extension);
+  ASSERT_EQ(1u, registry.disabled_extensions().size());
+  registry.EnableExtension(extension);
+  EXPECT_EQ(1u, registry.enabled_extensions().size());
+  EXPECT_EQ(0u, registry.disabled_extensions().size());
+}
+
+TEST_F(ExtensionRegistryTest, DisableExtension) {
+  ExtensionRegistry registry;
+  scoped_refptr<Extension> extension = test_util::CreateExtensionWithID("id");
+
+  // Disabling an extension moves it from enabled to disabled.
+  registry.AddEnabled(extension);
+  ASSERT_EQ(1u, registry.enabled_extensions().size());
+  registry.DisableExtension(extension);
+  EXPECT_EQ(0u, registry.enabled_extensions().size());
+  EXPECT_EQ(1u, registry.disabled_extensions().size());
+
+  registry.ClearAll();
+
+  // Disabling an extension moves it from terminated to disabled.
+  registry.AddTerminated(extension);
+  ASSERT_EQ(1u, registry.terminated_extensions().size());
+  registry.DisableExtension(extension);
+  EXPECT_EQ(0u, registry.terminated_extensions().size());
+  EXPECT_EQ(1u, registry.disabled_extensions().size());
+}
+
 }  // namespace
 }  // namespace extensions