OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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 "chrome/common/extensions/features/base_feature_provider.h" | 5 #include "chrome/common/extensions/features/base_feature_provider.h" |
6 | 6 |
7 #include "chrome/common/extensions/value_builder.h" | 7 #include "chrome/common/extensions/value_builder.h" |
8 #include "testing/gtest/include/gtest/gtest.h" | 8 #include "testing/gtest/include/gtest/gtest.h" |
9 | 9 |
10 using chrome::VersionInfo; | 10 using chrome::VersionInfo; |
| 11 using extensions::BaseFeatureProvider; |
11 using extensions::DictionaryBuilder; | 12 using extensions::DictionaryBuilder; |
12 using extensions::Extension; | 13 using extensions::Extension; |
13 using extensions::Feature; | 14 using extensions::Feature; |
14 using extensions::ListBuilder; | 15 using extensions::ListBuilder; |
| 16 using extensions::Manifest; |
15 using extensions::SimpleFeature; | 17 using extensions::SimpleFeature; |
16 using extensions::BaseFeatureProvider; | |
17 | 18 |
18 TEST(BaseFeatureProvider, ManifestFeatures) { | 19 TEST(BaseFeatureProvider, ManifestFeatures) { |
19 BaseFeatureProvider* provider = | 20 BaseFeatureProvider* provider = |
20 BaseFeatureProvider::GetManifestFeatures(); | 21 BaseFeatureProvider::GetManifestFeatures(); |
21 SimpleFeature* feature = | 22 SimpleFeature* feature = |
22 static_cast<SimpleFeature*>(provider->GetFeature("description")); | 23 static_cast<SimpleFeature*>(provider->GetFeature("description")); |
23 ASSERT_TRUE(feature); | 24 ASSERT_TRUE(feature); |
24 EXPECT_EQ(5u, feature->extension_types()->size()); | 25 EXPECT_EQ(5u, feature->extension_types()->size()); |
25 EXPECT_EQ(1u, feature->extension_types()->count(Extension::TYPE_EXTENSION)); | 26 EXPECT_EQ(1u, feature->extension_types()->count(Manifest::TYPE_EXTENSION)); |
26 EXPECT_EQ(1u, | 27 EXPECT_EQ(1u, |
27 feature->extension_types()->count(Extension::TYPE_LEGACY_PACKAGED_APP)); | 28 feature->extension_types()->count(Manifest::TYPE_LEGACY_PACKAGED_APP)); |
28 EXPECT_EQ(1u, | 29 EXPECT_EQ(1u, |
29 feature->extension_types()->count(Extension::TYPE_PLATFORM_APP)); | 30 feature->extension_types()->count(Manifest::TYPE_PLATFORM_APP)); |
30 EXPECT_EQ(1u, feature->extension_types()->count(Extension::TYPE_HOSTED_APP)); | 31 EXPECT_EQ(1u, feature->extension_types()->count(Manifest::TYPE_HOSTED_APP)); |
31 EXPECT_EQ(1u, feature->extension_types()->count(Extension::TYPE_THEME)); | 32 EXPECT_EQ(1u, feature->extension_types()->count(Manifest::TYPE_THEME)); |
32 | 33 |
33 DictionaryValue manifest; | 34 DictionaryValue manifest; |
34 manifest.SetString("name", "test extension"); | 35 manifest.SetString("name", "test extension"); |
35 manifest.SetString("version", "1"); | 36 manifest.SetString("version", "1"); |
36 manifest.SetString("description", "hello there"); | 37 manifest.SetString("description", "hello there"); |
37 | 38 |
38 std::string error; | 39 std::string error; |
39 scoped_refptr<const Extension> extension(Extension::Create( | 40 scoped_refptr<const Extension> extension(Extension::Create( |
40 FilePath(), Extension::INTERNAL, manifest, Extension::NO_FLAGS, | 41 FilePath(), Manifest::INTERNAL, manifest, Extension::NO_FLAGS, |
41 &error)); | 42 &error)); |
42 | 43 |
43 ASSERT_TRUE(extension.get()); | 44 ASSERT_TRUE(extension.get()); |
44 EXPECT_EQ(Feature::IS_AVAILABLE, feature->IsAvailableToContext( | 45 EXPECT_EQ(Feature::IS_AVAILABLE, feature->IsAvailableToContext( |
45 extension.get(), Feature::UNSPECIFIED_CONTEXT).result()); | 46 extension.get(), Feature::UNSPECIFIED_CONTEXT).result()); |
46 | 47 |
47 feature = | 48 feature = |
48 static_cast<SimpleFeature*>(provider->GetFeature("theme")); | 49 static_cast<SimpleFeature*>(provider->GetFeature("theme")); |
49 ASSERT_TRUE(feature); | 50 ASSERT_TRUE(feature); |
50 EXPECT_EQ(Feature::INVALID_TYPE, feature->IsAvailableToContext( | 51 EXPECT_EQ(Feature::INVALID_TYPE, feature->IsAvailableToContext( |
51 extension.get(), Feature::UNSPECIFIED_CONTEXT).result()); | 52 extension.get(), Feature::UNSPECIFIED_CONTEXT).result()); |
52 | 53 |
53 feature = | 54 feature = |
54 static_cast<SimpleFeature*>(provider->GetFeature("devtools_page")); | 55 static_cast<SimpleFeature*>(provider->GetFeature("devtools_page")); |
55 ASSERT_TRUE(feature); | 56 ASSERT_TRUE(feature); |
56 EXPECT_EQ(Feature::NOT_PRESENT, feature->IsAvailableToContext( | 57 EXPECT_EQ(Feature::NOT_PRESENT, feature->IsAvailableToContext( |
57 extension.get(), Feature::UNSPECIFIED_CONTEXT).result()); | 58 extension.get(), Feature::UNSPECIFIED_CONTEXT).result()); |
58 } | 59 } |
59 | 60 |
60 TEST(BaseFeatureProvider, PermissionFeatures) { | 61 TEST(BaseFeatureProvider, PermissionFeatures) { |
61 BaseFeatureProvider* provider = | 62 BaseFeatureProvider* provider = |
62 BaseFeatureProvider::GetPermissionFeatures(); | 63 BaseFeatureProvider::GetPermissionFeatures(); |
63 SimpleFeature* feature = | 64 SimpleFeature* feature = |
64 static_cast<SimpleFeature*>(provider->GetFeature("contextMenus")); | 65 static_cast<SimpleFeature*>(provider->GetFeature("contextMenus")); |
65 ASSERT_TRUE(feature); | 66 ASSERT_TRUE(feature); |
66 EXPECT_EQ(3u, feature->extension_types()->size()); | 67 EXPECT_EQ(3u, feature->extension_types()->size()); |
67 EXPECT_EQ(1u, feature->extension_types()->count(Extension::TYPE_EXTENSION)); | 68 EXPECT_EQ(1u, feature->extension_types()->count(Manifest::TYPE_EXTENSION)); |
68 EXPECT_EQ(1u, | 69 EXPECT_EQ(1u, |
69 feature->extension_types()->count(Extension::TYPE_LEGACY_PACKAGED_APP)); | 70 feature->extension_types()->count(Manifest::TYPE_LEGACY_PACKAGED_APP)); |
70 EXPECT_EQ(1u, | 71 EXPECT_EQ(1u, |
71 feature->extension_types()->count(Extension::TYPE_PLATFORM_APP)); | 72 feature->extension_types()->count(Manifest::TYPE_PLATFORM_APP)); |
72 | 73 |
73 DictionaryValue manifest; | 74 DictionaryValue manifest; |
74 manifest.SetString("name", "test extension"); | 75 manifest.SetString("name", "test extension"); |
75 manifest.SetString("version", "1"); | 76 manifest.SetString("version", "1"); |
76 ListValue* permissions = new ListValue(); | 77 ListValue* permissions = new ListValue(); |
77 manifest.Set("permissions", permissions); | 78 manifest.Set("permissions", permissions); |
78 permissions->Append(Value::CreateStringValue("contextMenus")); | 79 permissions->Append(Value::CreateStringValue("contextMenus")); |
79 | 80 |
80 std::string error; | 81 std::string error; |
81 scoped_refptr<const Extension> extension(Extension::Create( | 82 scoped_refptr<const Extension> extension(Extension::Create( |
82 FilePath(), Extension::INTERNAL, manifest, Extension::NO_FLAGS, | 83 FilePath(), Manifest::INTERNAL, manifest, Extension::NO_FLAGS, |
83 &error)); | 84 &error)); |
84 | 85 |
85 ASSERT_TRUE(extension.get()); | 86 ASSERT_TRUE(extension.get()); |
86 EXPECT_EQ(Feature::IS_AVAILABLE, feature->IsAvailableToContext( | 87 EXPECT_EQ(Feature::IS_AVAILABLE, feature->IsAvailableToContext( |
87 extension.get(), Feature::UNSPECIFIED_CONTEXT).result()); | 88 extension.get(), Feature::UNSPECIFIED_CONTEXT).result()); |
88 | 89 |
89 feature = | 90 feature = |
90 static_cast<SimpleFeature*>(provider->GetFeature("chromePrivate")); | 91 static_cast<SimpleFeature*>(provider->GetFeature("chromePrivate")); |
91 ASSERT_TRUE(feature); | 92 ASSERT_TRUE(feature); |
92 EXPECT_EQ(Feature::NOT_FOUND_IN_WHITELIST, feature->IsAvailableToContext( | 93 EXPECT_EQ(Feature::NOT_FOUND_IN_WHITELIST, feature->IsAvailableToContext( |
(...skipping 59 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
152 new BaseFeatureProvider(*rule, NULL)); | 153 new BaseFeatureProvider(*rule, NULL)); |
153 | 154 |
154 Feature *feature = provider->GetFeature("feature1"); | 155 Feature *feature = provider->GetFeature("feature1"); |
155 EXPECT_TRUE(feature); | 156 EXPECT_TRUE(feature); |
156 | 157 |
157 // Make sure both rules are applied correctly. | 158 // Make sure both rules are applied correctly. |
158 { | 159 { |
159 Feature::ScopedCurrentChannel current_channel(VersionInfo::CHANNEL_BETA); | 160 Feature::ScopedCurrentChannel current_channel(VersionInfo::CHANNEL_BETA); |
160 EXPECT_EQ(Feature::IS_AVAILABLE, feature->IsAvailableToManifest( | 161 EXPECT_EQ(Feature::IS_AVAILABLE, feature->IsAvailableToManifest( |
161 "1", | 162 "1", |
162 Extension::TYPE_EXTENSION, | 163 Manifest::TYPE_EXTENSION, |
163 Feature::UNSPECIFIED_LOCATION, | 164 Feature::UNSPECIFIED_LOCATION, |
164 Feature::UNSPECIFIED_PLATFORM).result()); | 165 Feature::UNSPECIFIED_PLATFORM).result()); |
165 EXPECT_EQ(Feature::IS_AVAILABLE, feature->IsAvailableToManifest( | 166 EXPECT_EQ(Feature::IS_AVAILABLE, feature->IsAvailableToManifest( |
166 "2", | 167 "2", |
167 Extension::TYPE_LEGACY_PACKAGED_APP, | 168 Manifest::TYPE_LEGACY_PACKAGED_APP, |
168 Feature::UNSPECIFIED_LOCATION, | 169 Feature::UNSPECIFIED_LOCATION, |
169 Feature::UNSPECIFIED_PLATFORM).result()); | 170 Feature::UNSPECIFIED_PLATFORM).result()); |
170 } | 171 } |
171 { | 172 { |
172 Feature::ScopedCurrentChannel current_channel(VersionInfo::CHANNEL_STABLE); | 173 Feature::ScopedCurrentChannel current_channel(VersionInfo::CHANNEL_STABLE); |
173 EXPECT_NE(Feature::IS_AVAILABLE, feature->IsAvailableToManifest( | 174 EXPECT_NE(Feature::IS_AVAILABLE, feature->IsAvailableToManifest( |
174 "1", | 175 "1", |
175 Extension::TYPE_EXTENSION, | 176 Manifest::TYPE_EXTENSION, |
176 Feature::UNSPECIFIED_LOCATION, | 177 Feature::UNSPECIFIED_LOCATION, |
177 Feature::UNSPECIFIED_PLATFORM).result()); | 178 Feature::UNSPECIFIED_PLATFORM).result()); |
178 EXPECT_NE(Feature::IS_AVAILABLE, feature->IsAvailableToManifest( | 179 EXPECT_NE(Feature::IS_AVAILABLE, feature->IsAvailableToManifest( |
179 "2", | 180 "2", |
180 Extension::TYPE_LEGACY_PACKAGED_APP, | 181 Manifest::TYPE_LEGACY_PACKAGED_APP, |
181 Feature::UNSPECIFIED_LOCATION, | 182 Feature::UNSPECIFIED_LOCATION, |
182 Feature::UNSPECIFIED_PLATFORM).result()); | 183 Feature::UNSPECIFIED_PLATFORM).result()); |
183 } | 184 } |
184 } | 185 } |
OLD | NEW |