| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 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/common/features/complex_feature.h" | 5 #include "extensions/common/features/complex_feature.h" |
| 6 | 6 |
| 7 #include "chrome/common/extensions/features/chrome_channel_feature_filter.h" | 7 #include <string> |
| 8 #include "chrome/common/extensions/features/feature_channel.h" | 8 |
| 9 #include "extensions/common/features/simple_feature.h" | 9 #include "extensions/common/features/simple_feature.h" |
| 10 #include "extensions/common/manifest.h" |
| 10 #include "extensions/common/value_builder.h" | 11 #include "extensions/common/value_builder.h" |
| 11 #include "testing/gtest/include/gtest/gtest.h" | 12 #include "testing/gtest/include/gtest/gtest.h" |
| 12 | 13 |
| 13 using chrome::VersionInfo; | 14 namespace extensions { |
| 14 using extensions::ComplexFeature; | |
| 15 using extensions::DictionaryBuilder; | |
| 16 using extensions::Feature; | |
| 17 using extensions::ListBuilder; | |
| 18 using extensions::Manifest; | |
| 19 using extensions::ScopedCurrentChannel; | |
| 20 using extensions::SimpleFeature; | |
| 21 | 15 |
| 22 namespace { | 16 TEST(ComplexFeatureTest, MultipleRulesWhitelist) { |
| 23 | |
| 24 class ExtensionComplexFeatureTest : public testing::Test { | |
| 25 protected: | |
| 26 ExtensionComplexFeatureTest() | |
| 27 : current_channel_(VersionInfo::CHANNEL_UNKNOWN) {} | |
| 28 virtual ~ExtensionComplexFeatureTest() {} | |
| 29 | |
| 30 SimpleFeature* CreateFeature() { | |
| 31 SimpleFeature* feature = new SimpleFeature(); | |
| 32 feature->AddFilter(scoped_ptr<extensions::SimpleFeatureFilter>( | |
| 33 new extensions::ChromeChannelFeatureFilter(feature))); | |
| 34 return feature; | |
| 35 } | |
| 36 | |
| 37 private: | |
| 38 ScopedCurrentChannel current_channel_; | |
| 39 }; | |
| 40 | |
| 41 TEST_F(ExtensionComplexFeatureTest, MultipleRulesWhitelist) { | |
| 42 const std::string kIdFoo("fooabbbbccccddddeeeeffffgggghhhh"); | 17 const std::string kIdFoo("fooabbbbccccddddeeeeffffgggghhhh"); |
| 43 const std::string kIdBar("barabbbbccccddddeeeeffffgggghhhh"); | 18 const std::string kIdBar("barabbbbccccddddeeeeffffgggghhhh"); |
| 44 scoped_ptr<ComplexFeature::FeatureList> features( | 19 scoped_ptr<ComplexFeature::FeatureList> features( |
| 45 new ComplexFeature::FeatureList()); | 20 new ComplexFeature::FeatureList()); |
| 46 | 21 |
| 47 // Rule: "extension", whitelist "foo". | 22 // Rule: "extension", whitelist "foo". |
| 48 scoped_ptr<SimpleFeature> simple_feature(CreateFeature()); | 23 scoped_ptr<SimpleFeature> simple_feature(new SimpleFeature); |
| 49 scoped_ptr<base::DictionaryValue> rule( | 24 scoped_ptr<base::DictionaryValue> rule( |
| 50 DictionaryBuilder() | 25 DictionaryBuilder() |
| 51 .Set("whitelist", ListBuilder().Append(kIdFoo)) | 26 .Set("whitelist", ListBuilder().Append(kIdFoo)) |
| 52 .Set("extension_types", ListBuilder() | 27 .Set("extension_types", ListBuilder() |
| 53 .Append("extension")).Build()); | 28 .Append("extension")).Build()); |
| 54 simple_feature->Parse(rule.get()); | 29 simple_feature->Parse(rule.get()); |
| 55 features->push_back(simple_feature.release()); | 30 features->push_back(simple_feature.release()); |
| 56 | 31 |
| 57 // Rule: "legacy_packaged_app", whitelist "bar". | 32 // Rule: "legacy_packaged_app", whitelist "bar". |
| 58 simple_feature.reset(CreateFeature()); | 33 simple_feature.reset(new SimpleFeature); |
| 59 rule = DictionaryBuilder() | 34 rule = DictionaryBuilder() |
| 60 .Set("whitelist", ListBuilder().Append(kIdBar)) | 35 .Set("whitelist", ListBuilder().Append(kIdBar)) |
| 61 .Set("extension_types", ListBuilder() | 36 .Set("extension_types", ListBuilder() |
| 62 .Append("legacy_packaged_app")).Build(); | 37 .Append("legacy_packaged_app")).Build(); |
| 63 simple_feature->Parse(rule.get()); | 38 simple_feature->Parse(rule.get()); |
| 64 features->push_back(simple_feature.release()); | 39 features->push_back(simple_feature.release()); |
| 65 | 40 |
| 66 scoped_ptr<ComplexFeature> feature(new ComplexFeature(features.Pass())); | 41 scoped_ptr<ComplexFeature> feature(new ComplexFeature(features.Pass())); |
| 67 | 42 |
| 68 // Test match 1st rule. | 43 // Test match 1st rule. |
| (...skipping 24 matching lines...) Expand all Loading... |
| 93 Feature::GetCurrentPlatform()).result()); | 68 Feature::GetCurrentPlatform()).result()); |
| 94 EXPECT_NE( | 69 EXPECT_NE( |
| 95 Feature::IS_AVAILABLE, | 70 Feature::IS_AVAILABLE, |
| 96 feature->IsAvailableToManifest(kIdFoo, | 71 feature->IsAvailableToManifest(kIdFoo, |
| 97 Manifest::TYPE_LEGACY_PACKAGED_APP, | 72 Manifest::TYPE_LEGACY_PACKAGED_APP, |
| 98 Manifest::INVALID_LOCATION, | 73 Manifest::INVALID_LOCATION, |
| 99 Feature::UNSPECIFIED_PLATFORM, | 74 Feature::UNSPECIFIED_PLATFORM, |
| 100 Feature::GetCurrentPlatform()).result()); | 75 Feature::GetCurrentPlatform()).result()); |
| 101 } | 76 } |
| 102 | 77 |
| 103 TEST_F(ExtensionComplexFeatureTest, MultipleRulesChannels) { | 78 // Tests that dependencies are correctly checked. |
| 79 TEST(ComplexFeatureTest, Dependencies) { |
| 104 scoped_ptr<ComplexFeature::FeatureList> features( | 80 scoped_ptr<ComplexFeature::FeatureList> features( |
| 105 new ComplexFeature::FeatureList()); | 81 new ComplexFeature::FeatureList()); |
| 106 | 82 |
| 107 // Rule: "extension", channel trunk. | 83 // Rule which depends on an extension-only feature (content_security_policy). |
| 108 scoped_ptr<SimpleFeature> simple_feature(CreateFeature()); | 84 scoped_ptr<SimpleFeature> simple_feature(new SimpleFeature); |
| 109 scoped_ptr<base::DictionaryValue> rule( | |
| 110 DictionaryBuilder() | |
| 111 .Set("channel", "trunk") | |
| 112 .Set("extension_types", ListBuilder().Append("extension")).Build()); | |
| 113 simple_feature->Parse(rule.get()); | |
| 114 features->push_back(simple_feature.release()); | |
| 115 | |
| 116 // Rule: "legacy_packaged_app", channel stable. | |
| 117 simple_feature.reset(CreateFeature()); | |
| 118 rule = DictionaryBuilder() | |
| 119 .Set("channel", "stable") | |
| 120 .Set("extension_types", ListBuilder() | |
| 121 .Append("legacy_packaged_app")).Build(); | |
| 122 simple_feature->Parse(rule.get()); | |
| 123 features->push_back(simple_feature.release()); | |
| 124 | |
| 125 scoped_ptr<ComplexFeature> feature(new ComplexFeature(features.Pass())); | |
| 126 | |
| 127 // Test match 1st rule. | |
| 128 { | |
| 129 ScopedCurrentChannel current_channel(VersionInfo::CHANNEL_UNKNOWN); | |
| 130 EXPECT_EQ( | |
| 131 Feature::IS_AVAILABLE, | |
| 132 feature->IsAvailableToManifest("1", | |
| 133 Manifest::TYPE_EXTENSION, | |
| 134 Manifest::INVALID_LOCATION, | |
| 135 Feature::UNSPECIFIED_PLATFORM, | |
| 136 Feature::GetCurrentPlatform()).result()); | |
| 137 } | |
| 138 | |
| 139 // Test match 2nd rule. | |
| 140 { | |
| 141 ScopedCurrentChannel current_channel(VersionInfo::CHANNEL_BETA); | |
| 142 EXPECT_EQ( | |
| 143 Feature::IS_AVAILABLE, | |
| 144 feature->IsAvailableToManifest("2", | |
| 145 Manifest::TYPE_LEGACY_PACKAGED_APP, | |
| 146 Manifest::INVALID_LOCATION, | |
| 147 Feature::UNSPECIFIED_PLATFORM, | |
| 148 Feature::GetCurrentPlatform()).result()); | |
| 149 } | |
| 150 | |
| 151 // Test feature not available to extensions above channel unknown. | |
| 152 { | |
| 153 ScopedCurrentChannel current_channel(VersionInfo::CHANNEL_BETA); | |
| 154 EXPECT_NE( | |
| 155 Feature::IS_AVAILABLE, | |
| 156 feature->IsAvailableToManifest("1", | |
| 157 Manifest::TYPE_EXTENSION, | |
| 158 Manifest::INVALID_LOCATION, | |
| 159 Feature::UNSPECIFIED_PLATFORM, | |
| 160 Feature::GetCurrentPlatform()).result()); | |
| 161 } | |
| 162 } | |
| 163 | |
| 164 // Tests that dependencies are correctly checked. | |
| 165 TEST_F(ExtensionComplexFeatureTest, Dependencies) { | |
| 166 scoped_ptr<ComplexFeature::FeatureList> features( | |
| 167 new ComplexFeature::FeatureList()); | |
| 168 | |
| 169 // Rule which depends on an extension-only feature (omnibox). | |
| 170 scoped_ptr<SimpleFeature> simple_feature(CreateFeature()); | |
| 171 scoped_ptr<base::DictionaryValue> rule = | 85 scoped_ptr<base::DictionaryValue> rule = |
| 172 DictionaryBuilder() | 86 DictionaryBuilder() |
| 173 .Set("dependencies", ListBuilder().Append("manifest:omnibox")) | 87 .Set("dependencies", |
| 88 ListBuilder().Append("manifest:content_security_policy")) |
| 174 .Build(); | 89 .Build(); |
| 175 simple_feature->Parse(rule.get()); | 90 simple_feature->Parse(rule.get()); |
| 176 features->push_back(simple_feature.release()); | 91 features->push_back(simple_feature.release()); |
| 177 | 92 |
| 178 // Rule which depends on an platform-app-only feature (serial). | 93 // Rule which depends on an platform-app-only feature (serial). |
| 179 simple_feature.reset(CreateFeature()); | 94 simple_feature.reset(new SimpleFeature); |
| 180 rule = DictionaryBuilder() | 95 rule = DictionaryBuilder() |
| 181 .Set("dependencies", ListBuilder().Append("permission:serial")) | 96 .Set("dependencies", ListBuilder().Append("permission:serial")) |
| 182 .Build(); | 97 .Build(); |
| 183 simple_feature->Parse(rule.get()); | 98 simple_feature->Parse(rule.get()); |
| 184 features->push_back(simple_feature.release()); | 99 features->push_back(simple_feature.release()); |
| 185 | 100 |
| 186 scoped_ptr<ComplexFeature> feature(new ComplexFeature(features.Pass())); | 101 scoped_ptr<ComplexFeature> feature(new ComplexFeature(features.Pass())); |
| 187 | 102 |
| 188 // Available to extensions because of the omnibox rule. | 103 // Available to extensions because of the content_security_policy rule. |
| 189 EXPECT_EQ( | 104 EXPECT_EQ( |
| 190 Feature::IS_AVAILABLE, | 105 Feature::IS_AVAILABLE, |
| 191 feature->IsAvailableToManifest("extensionid", | 106 feature->IsAvailableToManifest("extensionid", |
| 192 Manifest::TYPE_EXTENSION, | 107 Manifest::TYPE_EXTENSION, |
| 193 Manifest::INVALID_LOCATION, | 108 Manifest::INVALID_LOCATION, |
| 194 Feature::UNSPECIFIED_PLATFORM, | 109 Feature::UNSPECIFIED_PLATFORM, |
| 195 Feature::GetCurrentPlatform()).result()); | 110 Feature::GetCurrentPlatform()).result()); |
| 196 | 111 |
| 197 // Available to platofrm apps because of the serial rule. | 112 // Available to platform apps because of the serial rule. |
| 198 EXPECT_EQ( | 113 EXPECT_EQ( |
| 199 Feature::IS_AVAILABLE, | 114 Feature::IS_AVAILABLE, |
| 200 feature->IsAvailableToManifest("platformappid", | 115 feature->IsAvailableToManifest("platformappid", |
| 201 Manifest::TYPE_PLATFORM_APP, | 116 Manifest::TYPE_PLATFORM_APP, |
| 202 Manifest::INVALID_LOCATION, | 117 Manifest::INVALID_LOCATION, |
| 203 Feature::UNSPECIFIED_PLATFORM, | 118 Feature::UNSPECIFIED_PLATFORM, |
| 204 Feature::GetCurrentPlatform()).result()); | 119 Feature::GetCurrentPlatform()).result()); |
| 205 | 120 |
| 206 // Not available to hosted apps. | 121 // Not available to hosted apps. |
| 207 EXPECT_EQ( | 122 EXPECT_EQ( |
| 208 Feature::INVALID_TYPE, | 123 Feature::INVALID_TYPE, |
| 209 feature->IsAvailableToManifest("hostedappid", | 124 feature->IsAvailableToManifest("hostedappid", |
| 210 Manifest::TYPE_HOSTED_APP, | 125 Manifest::TYPE_HOSTED_APP, |
| 211 Manifest::INVALID_LOCATION, | 126 Manifest::INVALID_LOCATION, |
| 212 Feature::UNSPECIFIED_PLATFORM, | 127 Feature::UNSPECIFIED_PLATFORM, |
| 213 Feature::GetCurrentPlatform()).result()); | 128 Feature::GetCurrentPlatform()).result()); |
| 214 } | 129 } |
| 215 | 130 |
| 216 } // namespace | 131 } // namespace extensions |
| OLD | NEW |