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());
+
+  EXPECT_TRUE(ComplexFeature(features.Pass()).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());
+
+  EXPECT_FALSE(ComplexFeature(features.Pass()).IsBlockedInServiceWorker());
+}
+
 }  // namespace