Enable _api_features.json to block APIs from service worker contexts.

chrome/common/extensions/features/complex_feature_unittest.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "chrome/common/extensions/features/complex_feature.h"
 
+#include "chrome/common/extensions/features/api_feature.h"
 #include "chrome/common/extensions/features/feature_channel.h"
 #include "chrome/common/extensions/features/simple_feature.h"
+#include "extensions/common/test_util.h"
 #include "extensions/common/value_builder.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 using chrome::VersionInfo;
+using extensions::APIFeature;
 using extensions::ComplexFeature;
 using extensions::DictionaryBuilder;
 using extensions::Feature;
 using extensions::ListBuilder;
 using extensions::Manifest;
 using extensions::ScopedCurrentChannel;
 using extensions::SimpleFeature;
+using extensions::test_util::ParseJsonDictionaryWithSingleQuotes;
 
 namespace {
 
 class ExtensionComplexFeatureTest : public testing::Test {
  protected:
   ExtensionComplexFeatureTest()
       : current_channel_(VersionInfo::CHANNEL_UNKNOWN) {}
   virtual ~ExtensionComplexFeatureTest() {}
 
  private:
@@ skipping 108 matching lines @@
     ScopedCurrentChannel current_channel(VersionInfo::CHANNEL_BETA);
     EXPECT_NE(Feature::IS_AVAILABLE, feature->IsAvailableToManifest(
         "1",
         Manifest::TYPE_EXTENSION,
         Feature::UNSPECIFIED_LOCATION,
         Feature::UNSPECIFIED_PLATFORM,
         Feature::GetCurrentPlatform()).result());
   }
 }
 
+// Tests a complex feature with blocked_in_service_worker returns true for
+// IsBlockedInServiceWorker().
+TEST_F(ExtensionComplexFeatureTest, BlockedInServiceWorker) {
+  scoped_ptr<ComplexFeature::FeatureList> features(
+      new ComplexFeature::FeatureList());
+
+  // Two feature rules, both with blocked_in_service_worker: true.
+  scoped_ptr<SimpleFeature> api_feature(new APIFeature());
+  api_feature->Parse(ParseJsonDictionaryWithSingleQuotes(
+                         "{"
+                         "  'channel': 'trunk',"
+                         "  'blocked_in_service_worker': true"
+                         "}").get());
+  features->push_back(api_feature.release());
+
+  api_feature.reset(new APIFeature());
+  api_feature->Parse(ParseJsonDictionaryWithSingleQuotes(
+                         "{"
+                         "  'channel': 'stable',"
+                         "  'blocked_in_service_worker': true"
+                         "}").get());
+  features->push_back(api_feature.release());
+
+  scoped_ptr<ComplexFeature> feature(new ComplexFeature(features.Pass()));

[not at google - send to devlin, 2014/03/28 17:30:49]

allocate on stack?

[scheib, 2014/03/28 18:15:21]

Done.

+
+  EXPECT_TRUE(feature->IsBlockedInServiceWorker());
+}
+
+// Tests a complex feature without blocked_in_service_worker returns false for
+// IsBlockedInServiceWorker().
+TEST_F(ExtensionComplexFeatureTest, NotBlockedInServiceWorker) {
+  scoped_ptr<ComplexFeature::FeatureList> features(
+      new ComplexFeature::FeatureList());
+
+  // Two feature rules without blocked_in_service_worker.
+  scoped_ptr<SimpleFeature> api_feature(new APIFeature());
+  api_feature->Parse(ParseJsonDictionaryWithSingleQuotes(
+                         "{"
+                         "  'channel': 'trunk'"
+                         "}").get());
+  features->push_back(api_feature.release());
+
+  api_feature.reset(new APIFeature());
+  api_feature->Parse(ParseJsonDictionaryWithSingleQuotes(
+                         "{"
+                         "  'channel': 'stable'"
+                         "}").get());
+  features->push_back(api_feature.release());
+
+  scoped_ptr<ComplexFeature> feature(new ComplexFeature(features.Pass()));

[not at google - send to devlin, 2014/03/28 17:30:49]

ditto

[scheib, 2014/03/28 18:15:21]

Done.

+
+  EXPECT_FALSE(feature->IsBlockedInServiceWorker());
+}
+
+// Tests that a complex feature composing different values for
+// blocked_in_service_worker will DCHECK.

[not at google - send to devlin, 2014/03/28 17:30:49]

testing DCHECKs isn't that useful. DCHECK means that a fundamental assumption
has been broken somewhere. if it's something that worth dealing with then that's
a different question but I don't think so.

[scheib, 2014/03/28 18:15:21]

Done.

+#if (!defined(NDEBUG) || defined(DCHECK_ALWAYS_ON)) && GTEST_HAS_DEATH_TEST
+TEST_F(ExtensionComplexFeatureTest, MixedBlockedInServiceWorker) {
+  scoped_ptr<ComplexFeature::FeatureList> features(
+      new ComplexFeature::FeatureList());
+
+  // mixed blocked_in_service_worker.
+  scoped_ptr<SimpleFeature> api_feature(new APIFeature());
+  api_feature->Parse(ParseJsonDictionaryWithSingleQuotes(
+                         "{"
+                         "  'channel': 'trunk',"
+                         "  'blocked_in_service_worker': true"
+                         "}").get());
+  features->push_back(api_feature.release());
+
+  api_feature.reset(new APIFeature());
+  api_feature->Parse(ParseJsonDictionaryWithSingleQuotes(
+                         "{"
+                         "  'channel': 'stable'"
+                         "}").get());
+  features->push_back(api_feature.release());
+
+  ASSERT_DEATH(
+      scoped_ptr<ComplexFeature> feature(new ComplexFeature(features.Pass())),
+      "");
+}
+#endif
+
 }  // namespace