Introduce session type parameter to extension features

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 <stddef.h>
 
 #include <string>
 
@@ skipping 404 matching lines @@
   EXPECT_EQ(Feature::INVALID_MAX_MANIFEST_VERSION,
             feature
                 .IsAvailableToContext(extension.get(),
                                       Feature::BLESSED_EXTENSION_CONTEXT,
                                       Feature::SESSION_TYPE_UNSPECIFIED,
                                       Feature::CHROMEOS_PLATFORM)
                 .result());
   feature.set_max_manifest_version(25);
 }
 
+TEST_F(SimpleFeatureTest, SessionType) {
+  base::DictionaryValue manifest;
+  manifest.SetString("name", "test");
+  manifest.SetString("version", "1");
+  manifest.SetInteger("manifest_version", 21);
+  manifest.SetString("app.launch.local_path", "foo.html");
+
+  std::string error;
+  scoped_refptr<const Extension> extension(
+      Extension::Create(base::FilePath(), Manifest::INTERNAL, manifest,
+                        Extension::NO_FLAGS, &error));
+  EXPECT_EQ("", error);
+  ASSERT_TRUE(extension.get());
+
+  SimpleFeature kiosk_feature;
+  kiosk_feature.session_types_.push_back(Feature::SESSION_TYPE_KIOSK);
+  EXPECT_EQ(Feature::IS_AVAILABLE,
+            kiosk_feature
+                .IsAvailableToContext(
+                    extension.get(), Feature::BLESSED_EXTENSION_CONTEXT,
+                    Feature::SESSION_TYPE_KIOSK, Feature::CHROMEOS_PLATFORM)
+                .result());
+  EXPECT_EQ(Feature::INVALID_SESSION_TYPE,
+            kiosk_feature
+                .IsAvailableToContext(
+                    extension.get(), Feature::BLESSED_EXTENSION_CONTEXT,
+                    Feature::SESSION_TYPE_REGULAR, Feature::CHROMEOS_PLATFORM)
+                .result());
+  EXPECT_EQ(Feature::INVALID_SESSION_TYPE,
+            kiosk_feature
+                .IsAvailableToContext(extension.get(),
+                                      Feature::BLESSED_EXTENSION_CONTEXT,
+                                      Feature::SESSION_TYPE_UNSPECIFIED,
+                                      Feature::CHROMEOS_PLATFORM)
+                .result());
+
+  SimpleFeature non_kiosk_feature;
+  non_kiosk_feature.session_types_.push_back(Feature::SESSION_TYPE_REGULAR);
+  EXPECT_EQ(Feature::INVALID_SESSION_TYPE,
+            non_kiosk_feature
+                .IsAvailableToContext(
+                    extension.get(), Feature::BLESSED_EXTENSION_CONTEXT,
+                    Feature::SESSION_TYPE_KIOSK, Feature::CHROMEOS_PLATFORM)
+                .result());
+  EXPECT_EQ(Feature::IS_AVAILABLE,
+            non_kiosk_feature
+                .IsAvailableToContext(
+                    extension.get(), Feature::BLESSED_EXTENSION_CONTEXT,
+                    Feature::SESSION_TYPE_REGULAR, Feature::CHROMEOS_PLATFORM)
+                .result());
+  EXPECT_EQ(Feature::INVALID_SESSION_TYPE,
+            non_kiosk_feature
+                .IsAvailableToContext(extension.get(),
+                                      Feature::BLESSED_EXTENSION_CONTEXT,
+                                      Feature::SESSION_TYPE_UNSPECIFIED,
+                                      Feature::CHROMEOS_PLATFORM)
+                .result());
+
+  SimpleFeature session_agnostic_feature;
+  EXPECT_EQ(Feature::IS_AVAILABLE,
+            session_agnostic_feature
+                .IsAvailableToContext(
+                    extension.get(), Feature::BLESSED_EXTENSION_CONTEXT,
+                    Feature::SESSION_TYPE_KIOSK, Feature::CHROMEOS_PLATFORM)
+                .result());
+  EXPECT_EQ(Feature::IS_AVAILABLE,
+            session_agnostic_feature
+                .IsAvailableToContext(
+                    extension.get(), Feature::BLESSED_EXTENSION_CONTEXT,
+                    Feature::SESSION_TYPE_REGULAR, Feature::CHROMEOS_PLATFORM)
+                .result());
+  EXPECT_EQ(Feature::IS_AVAILABLE,
+            session_agnostic_feature
+                .IsAvailableToContext(extension.get(),
+                                      Feature::BLESSED_EXTENSION_CONTEXT,
+                                      Feature::SESSION_TYPE_UNSPECIFIED,
+                                      Feature::CHROMEOS_PLATFORM)
+                .result());
+}
+
 TEST_F(SimpleFeatureTest, Location) {
   // Component extensions can access any location.
   EXPECT_TRUE(LocationIsAvailable(SimpleFeature::COMPONENT_LOCATION,
                                   Manifest::COMPONENT));
   EXPECT_TRUE(LocationIsAvailable(SimpleFeature::EXTERNAL_COMPONENT_LOCATION,
                                   Manifest::COMPONENT));
   EXPECT_TRUE(
       LocationIsAvailable(SimpleFeature::POLICY_LOCATION, Manifest::COMPONENT));
   EXPECT_TRUE(LocationIsAvailable(SimpleFeature::UNSPECIFIED_LOCATION,
                                   Manifest::COMPONENT));
@@ skipping 555 matching lines @@
     EXPECT_NE(Feature::IS_AVAILABLE,
               feature
                   ->IsAvailableToManifest("1", Manifest::TYPE_EXTENSION,
                                           Manifest::INVALID_LOCATION,
                                           Feature::UNSPECIFIED_PLATFORM)
                   .result());
   }
 }
 
 }  // namespace extensions