| 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 |