Add support for command line switches to Features, and as proof that it works,

extensions/common/features/simple_feature_unittest.cc

 // Copyright 2014 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 "extensions/common/features/simple_feature.h"
 
 #include <string>
 
+#include "base/command_line.h"
 #include "base/values.h"
 #include "extensions/common/manifest.h"
 #include "extensions/common/value_builder.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace extensions {
 
 namespace {
 
 struct IsAvailableTestData {
@@ skipping 11 matching lines @@
   feature.set_location(feature_location);
   Feature::AvailabilityResult availability_result =
       feature.IsAvailableToManifest(std::string(),
                                     Manifest::TYPE_UNKNOWN,
                                     manifest_location,
                                     -1,
                                     Feature::UNSPECIFIED_PLATFORM).result();
   return availability_result == Feature::IS_AVAILABLE;
 }
 
+class ScopedCommandLineSwitch {
+ public:
+  explicit ScopedCommandLineSwitch(const std::string& arg)
+      : original_command_line_(*base::CommandLine::ForCurrentProcess()) {
+    base::CommandLine::ForCurrentProcess()->AppendSwitch(arg);
+  }
+
+  ~ScopedCommandLineSwitch() {
+    *base::CommandLine::ForCurrentProcess() = original_command_line_;
+  }
+
+ private:
+  base::CommandLine original_command_line_;
+};
+
 }  // namespace
 
 TEST(SimpleFeatureTest, IsAvailableNullCase) {
   const IsAvailableTestData tests[] = {
       {"", Manifest::TYPE_UNKNOWN, Manifest::INVALID_LOCATION,
        Feature::UNSPECIFIED_PLATFORM, -1, Feature::IS_AVAILABLE},
       {"random-extension", Manifest::TYPE_UNKNOWN, Manifest::INVALID_LOCATION,
        Feature::UNSPECIFIED_PLATFORM, -1, Feature::IS_AVAILABLE},
       {"", Manifest::TYPE_LEGACY_PACKAGED_APP, Manifest::INVALID_LOCATION,
        Feature::UNSPECIFIED_PLATFORM, -1, Feature::IS_AVAILABLE},
@@ skipping 592 matching lines @@
   EXPECT_EQ(1u, feature.extension_types()->size());
   EXPECT_EQ(1u, feature.contexts()->size());
   EXPECT_EQ(1u, feature.whitelist()->count("bar"));
   EXPECT_EQ(1u, feature.extension_types()->count(Manifest::TYPE_EXTENSION));
   EXPECT_EQ(1u,
             feature.contexts()->count(Feature::UNBLESSED_EXTENSION_CONTEXT));
   EXPECT_EQ(2, feature.min_manifest_version());
   EXPECT_EQ(3, feature.max_manifest_version());
 }
 
+TEST(SimpleFeatureTest, CommandLineSwitch) {
+  SimpleFeature feature;
+  feature.set_command_line_switch("laser-beams");
+  {
+    EXPECT_EQ(Feature::MISSING_COMMAND_LINE_SWITCH,
+              feature.IsAvailableToEnvironment().result());
+  }
+  {
+    ScopedCommandLineSwitch scoped_switch("laser-beams");
+    EXPECT_EQ(Feature::MISSING_COMMAND_LINE_SWITCH,
+              feature.IsAvailableToEnvironment().result());
+  }
+  {
+    ScopedCommandLineSwitch scoped_switch("enable-laser-beams");
+    EXPECT_EQ(Feature::IS_AVAILABLE,
+              feature.IsAvailableToEnvironment().result());
+  }
+  {
+    ScopedCommandLineSwitch scoped_switch("disable-laser-beams");
+    EXPECT_EQ(Feature::MISSING_COMMAND_LINE_SWITCH,
+              feature.IsAvailableToEnvironment().result());
+  }
+  {
+    ScopedCommandLineSwitch scoped_switch("laser-beams=1");
+    EXPECT_EQ(Feature::IS_AVAILABLE,
+              feature.IsAvailableToEnvironment().result());
+  }
+  {
+    ScopedCommandLineSwitch scoped_switch("laser-beams=0");
+    EXPECT_EQ(Feature::MISSING_COMMAND_LINE_SWITCH,
+              feature.IsAvailableToEnvironment().result());
+  }
+}
+
 }  // namespace extensions