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 |