| OLD | NEW |
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 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 "extensions/browser/extension_registry.h" | 5 #include "extensions/browser/extension_registry.h" |
| 6 | 6 |
| 7 #include <string> | 7 #include <string> |
| 8 | 8 |
| 9 #include "base/memory/ref_counted.h" | 9 #include "base/memory/ref_counted.h" |
| 10 #include "base/strings/string_util.h" | 10 #include "base/strings/string_util.h" |
| (...skipping 70 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 81 } | 81 } |
| 82 | 82 |
| 83 ExtensionList loaded_; | 83 ExtensionList loaded_; |
| 84 ExtensionList unloaded_; | 84 ExtensionList unloaded_; |
| 85 ExtensionList installed_; | 85 ExtensionList installed_; |
| 86 ExtensionList uninstalled_; | 86 ExtensionList uninstalled_; |
| 87 }; | 87 }; |
| 88 | 88 |
| 89 TEST_F(ExtensionRegistryTest, FillAndClearRegistry) { | 89 TEST_F(ExtensionRegistryTest, FillAndClearRegistry) { |
| 90 ExtensionRegistry registry(NULL); | 90 ExtensionRegistry registry(NULL); |
| 91 scoped_refptr<Extension> extension1 = test_util::CreateExtensionWithID("id1"); | 91 scoped_refptr<Extension> extension1 = test_util::CreateEmptyExtension("id1"); |
| 92 scoped_refptr<Extension> extension2 = test_util::CreateExtensionWithID("id2"); | 92 scoped_refptr<Extension> extension2 = test_util::CreateEmptyExtension("id2"); |
| 93 scoped_refptr<Extension> extension3 = test_util::CreateExtensionWithID("id3"); | 93 scoped_refptr<Extension> extension3 = test_util::CreateEmptyExtension("id3"); |
| 94 scoped_refptr<Extension> extension4 = test_util::CreateExtensionWithID("id4"); | 94 scoped_refptr<Extension> extension4 = test_util::CreateEmptyExtension("id4"); |
| 95 | 95 |
| 96 // All the sets start empty. | 96 // All the sets start empty. |
| 97 EXPECT_EQ(0u, registry.enabled_extensions().size()); | 97 EXPECT_EQ(0u, registry.enabled_extensions().size()); |
| 98 EXPECT_EQ(0u, registry.disabled_extensions().size()); | 98 EXPECT_EQ(0u, registry.disabled_extensions().size()); |
| 99 EXPECT_EQ(0u, registry.terminated_extensions().size()); | 99 EXPECT_EQ(0u, registry.terminated_extensions().size()); |
| 100 EXPECT_EQ(0u, registry.blacklisted_extensions().size()); | 100 EXPECT_EQ(0u, registry.blacklisted_extensions().size()); |
| 101 | 101 |
| 102 // Extensions can be added to each set. | 102 // Extensions can be added to each set. |
| 103 registry.AddEnabled(extension1); | 103 registry.AddEnabled(extension1); |
| 104 registry.AddDisabled(extension2); | 104 registry.AddDisabled(extension2); |
| (...skipping 12 matching lines...) Expand all Loading... |
| 117 EXPECT_EQ(0u, registry.disabled_extensions().size()); | 117 EXPECT_EQ(0u, registry.disabled_extensions().size()); |
| 118 EXPECT_EQ(0u, registry.terminated_extensions().size()); | 118 EXPECT_EQ(0u, registry.terminated_extensions().size()); |
| 119 EXPECT_EQ(0u, registry.blacklisted_extensions().size()); | 119 EXPECT_EQ(0u, registry.blacklisted_extensions().size()); |
| 120 } | 120 } |
| 121 | 121 |
| 122 // A simple test of adding and removing things from sets. | 122 // A simple test of adding and removing things from sets. |
| 123 TEST_F(ExtensionRegistryTest, AddAndRemoveExtensionFromRegistry) { | 123 TEST_F(ExtensionRegistryTest, AddAndRemoveExtensionFromRegistry) { |
| 124 ExtensionRegistry registry(NULL); | 124 ExtensionRegistry registry(NULL); |
| 125 | 125 |
| 126 // Adding an extension works. | 126 // Adding an extension works. |
| 127 scoped_refptr<Extension> extension = test_util::CreateExtensionWithID("id"); | 127 scoped_refptr<Extension> extension = test_util::CreateEmptyExtension("id"); |
| 128 EXPECT_TRUE(registry.AddEnabled(extension)); | 128 EXPECT_TRUE(registry.AddEnabled(extension)); |
| 129 EXPECT_EQ(1u, registry.enabled_extensions().size()); | 129 EXPECT_EQ(1u, registry.enabled_extensions().size()); |
| 130 | 130 |
| 131 // The extension was only added to one set. | 131 // The extension was only added to one set. |
| 132 EXPECT_EQ(0u, registry.disabled_extensions().size()); | 132 EXPECT_EQ(0u, registry.disabled_extensions().size()); |
| 133 EXPECT_EQ(0u, registry.terminated_extensions().size()); | 133 EXPECT_EQ(0u, registry.terminated_extensions().size()); |
| 134 EXPECT_EQ(0u, registry.blacklisted_extensions().size()); | 134 EXPECT_EQ(0u, registry.blacklisted_extensions().size()); |
| 135 | 135 |
| 136 // Removing an extension works. | 136 // Removing an extension works. |
| 137 EXPECT_TRUE(registry.RemoveEnabled(extension->id())); | 137 EXPECT_TRUE(registry.RemoveEnabled(extension->id())); |
| 138 EXPECT_EQ(0u, registry.enabled_extensions().size()); | 138 EXPECT_EQ(0u, registry.enabled_extensions().size()); |
| 139 | 139 |
| 140 // Trying to remove an extension that isn't in the set fails cleanly. | 140 // Trying to remove an extension that isn't in the set fails cleanly. |
| 141 EXPECT_FALSE(registry.RemoveEnabled(extension->id())); | 141 EXPECT_FALSE(registry.RemoveEnabled(extension->id())); |
| 142 } | 142 } |
| 143 | 143 |
| 144 TEST_F(ExtensionRegistryTest, AddExtensionToRegistryTwice) { | 144 TEST_F(ExtensionRegistryTest, AddExtensionToRegistryTwice) { |
| 145 ExtensionRegistry registry(NULL); | 145 ExtensionRegistry registry(NULL); |
| 146 scoped_refptr<Extension> extension = test_util::CreateExtensionWithID("id"); | 146 scoped_refptr<Extension> extension = test_util::CreateEmptyExtension("id"); |
| 147 | 147 |
| 148 // An extension can exist in two sets at once. It would be nice to eliminate | 148 // An extension can exist in two sets at once. It would be nice to eliminate |
| 149 // this functionality, but some users of ExtensionRegistry need it. | 149 // this functionality, but some users of ExtensionRegistry need it. |
| 150 EXPECT_TRUE(registry.AddEnabled(extension)); | 150 EXPECT_TRUE(registry.AddEnabled(extension)); |
| 151 EXPECT_TRUE(registry.AddDisabled(extension)); | 151 EXPECT_TRUE(registry.AddDisabled(extension)); |
| 152 | 152 |
| 153 EXPECT_EQ(1u, registry.enabled_extensions().size()); | 153 EXPECT_EQ(1u, registry.enabled_extensions().size()); |
| 154 EXPECT_EQ(1u, registry.disabled_extensions().size()); | 154 EXPECT_EQ(1u, registry.disabled_extensions().size()); |
| 155 EXPECT_EQ(0u, registry.terminated_extensions().size()); | 155 EXPECT_EQ(0u, registry.terminated_extensions().size()); |
| 156 EXPECT_EQ(0u, registry.blacklisted_extensions().size()); | 156 EXPECT_EQ(0u, registry.blacklisted_extensions().size()); |
| 157 } | 157 } |
| 158 | 158 |
| 159 TEST_F(ExtensionRegistryTest, GetExtensionById) { | 159 TEST_F(ExtensionRegistryTest, GetExtensionById) { |
| 160 ExtensionRegistry registry(NULL); | 160 ExtensionRegistry registry(NULL); |
| 161 | 161 |
| 162 // Trying to get an extension fails cleanly when the sets are empty. | 162 // Trying to get an extension fails cleanly when the sets are empty. |
| 163 EXPECT_FALSE( | 163 EXPECT_FALSE( |
| 164 registry.GetExtensionById("id", ExtensionRegistry::EVERYTHING)); | 164 registry.GetExtensionById("id", ExtensionRegistry::EVERYTHING)); |
| 165 | 165 |
| 166 scoped_refptr<Extension> enabled = | 166 scoped_refptr<Extension> enabled = test_util::CreateEmptyExtension("enabled"); |
| 167 test_util::CreateExtensionWithID("enabled"); | |
| 168 scoped_refptr<Extension> disabled = | 167 scoped_refptr<Extension> disabled = |
| 169 test_util::CreateExtensionWithID("disabled"); | 168 test_util::CreateEmptyExtension("disabled"); |
| 170 scoped_refptr<Extension> terminated = | 169 scoped_refptr<Extension> terminated = |
| 171 test_util::CreateExtensionWithID("terminated"); | 170 test_util::CreateEmptyExtension("terminated"); |
| 172 scoped_refptr<Extension> blacklisted = | 171 scoped_refptr<Extension> blacklisted = |
| 173 test_util::CreateExtensionWithID("blacklisted"); | 172 test_util::CreateEmptyExtension("blacklisted"); |
| 174 | 173 |
| 175 // Add an extension to each set. | 174 // Add an extension to each set. |
| 176 registry.AddEnabled(enabled); | 175 registry.AddEnabled(enabled); |
| 177 registry.AddDisabled(disabled); | 176 registry.AddDisabled(disabled); |
| 178 registry.AddTerminated(terminated); | 177 registry.AddTerminated(terminated); |
| 179 registry.AddBlacklisted(blacklisted); | 178 registry.AddBlacklisted(blacklisted); |
| 180 | 179 |
| 181 // Enabled is part of everything and the enabled list. | 180 // Enabled is part of everything and the enabled list. |
| 182 EXPECT_TRUE( | 181 EXPECT_TRUE( |
| 183 registry.GetExtensionById("enabled", ExtensionRegistry::EVERYTHING)); | 182 registry.GetExtensionById("enabled", ExtensionRegistry::EVERYTHING)); |
| (...skipping 54 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 238 TEST_F(ExtensionRegistryTest, Observer) { | 237 TEST_F(ExtensionRegistryTest, Observer) { |
| 239 ExtensionRegistry registry(NULL); | 238 ExtensionRegistry registry(NULL); |
| 240 TestObserver observer; | 239 TestObserver observer; |
| 241 registry.AddObserver(&observer); | 240 registry.AddObserver(&observer); |
| 242 | 241 |
| 243 EXPECT_TRUE(observer.loaded().empty()); | 242 EXPECT_TRUE(observer.loaded().empty()); |
| 244 EXPECT_TRUE(observer.unloaded().empty()); | 243 EXPECT_TRUE(observer.unloaded().empty()); |
| 245 EXPECT_TRUE(observer.installed().empty()); | 244 EXPECT_TRUE(observer.installed().empty()); |
| 246 | 245 |
| 247 scoped_refptr<const Extension> extension = | 246 scoped_refptr<const Extension> extension = |
| 248 test_util::CreateExtensionWithID("id"); | 247 test_util::CreateEmptyExtension("id"); |
| 249 | 248 |
| 250 registry.TriggerOnWillBeInstalled( | 249 registry.TriggerOnWillBeInstalled( |
| 251 extension.get(), false, false, base::EmptyString()); | 250 extension.get(), false, false, base::EmptyString()); |
| 252 EXPECT_TRUE(HasSingleExtension(observer.installed(), extension.get())); | 251 EXPECT_TRUE(HasSingleExtension(observer.installed(), extension.get())); |
| 253 | 252 |
| 254 registry.AddEnabled(extension); | 253 registry.AddEnabled(extension); |
| 255 registry.TriggerOnLoaded(extension.get()); | 254 registry.TriggerOnLoaded(extension.get()); |
| 256 | 255 |
| 257 registry.TriggerOnWillBeInstalled(extension.get(), true, false, "foo"); | 256 registry.TriggerOnWillBeInstalled(extension.get(), true, false, "foo"); |
| 258 | 257 |
| (...skipping 12 matching lines...) Expand all Loading... |
| 271 registry.TriggerOnUninstalled(extension.get(), | 270 registry.TriggerOnUninstalled(extension.get(), |
| 272 extensions::UNINSTALL_REASON_FOR_TESTING); | 271 extensions::UNINSTALL_REASON_FOR_TESTING); |
| 273 EXPECT_TRUE(observer.installed().empty()); | 272 EXPECT_TRUE(observer.installed().empty()); |
| 274 EXPECT_TRUE(HasSingleExtension(observer.uninstalled(), extension.get())); | 273 EXPECT_TRUE(HasSingleExtension(observer.uninstalled(), extension.get())); |
| 275 | 274 |
| 276 registry.RemoveObserver(&observer); | 275 registry.RemoveObserver(&observer); |
| 277 } | 276 } |
| 278 | 277 |
| 279 } // namespace | 278 } // namespace |
| 280 } // namespace extensions | 279 } // namespace extensions |
| OLD | NEW |