Initial Implementation of Iframe Attribute for Feature Policy (Part 4)

third_party/WebKit/Source/platform/feature_policy/FeaturePolicyTest.cpp

 // Copyright 2016 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 "platform/feature_policy/FeaturePolicy.h"
 
 #include "platform/RuntimeEnabledFeatures.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 // Origin strings used for tests
@@ skipping 109 matching lines @@
   EXPECT_EQ(WebFeaturePolicyFeature::kPayment, parsed_policy[2].feature);
   EXPECT_FALSE(parsed_policy[2].matches_all_origins);
   EXPECT_EQ(1UL, parsed_policy[2].origins.size());
   EXPECT_TRUE(origin_a_->IsSameSchemeHostPortAndSuborigin(
       parsed_policy[2].origins[0].Get()));
 }
 
 TEST_F(FeaturePolicyTest, ParseEmptyContainerPolicy) {
   WebParsedFeaturePolicy container_policy =
       GetContainerPolicyFromAllowedFeatures(
-          std::vector<WebFeaturePolicyFeature>({}), origin_a_.Get());
+          std::vector<WebFeaturePolicyFeature>({}), false, false,
+          origin_a_.Get());
   EXPECT_EQ(0UL, container_policy.size());
 }
 
 TEST_F(FeaturePolicyTest, ParseContainerPolicy) {
   WebParsedFeaturePolicy container_policy =
       GetContainerPolicyFromAllowedFeatures(
           std::vector<WebFeaturePolicyFeature>(
               {WebFeaturePolicyFeature::kVibrate,
                WebFeaturePolicyFeature::kPayment}),
-          origin_a_.Get());
+          false, false, origin_a_.Get());
   EXPECT_EQ(2UL, container_policy.size());
   EXPECT_EQ(WebFeaturePolicyFeature::kVibrate, container_policy[0].feature);
   EXPECT_FALSE(container_policy[0].matches_all_origins);
+  EXPECT_EQ(1UL, container_policy[0].origins.size());
+  EXPECT_TRUE(origin_a_->IsSameSchemeHostPortAndSuborigin(
+      container_policy[0].origins[0].Get()));
+  EXPECT_EQ(WebFeaturePolicyFeature::kPayment, container_policy[1].feature);
+  EXPECT_FALSE(container_policy[1].matches_all_origins);
+  EXPECT_EQ(1UL, container_policy[1].origins.size());
+  EXPECT_TRUE(origin_a_->IsSameSchemeHostPortAndSuborigin(
+      container_policy[1].origins[0].Get()));
+}
+
+TEST_F(FeaturePolicyTest, ParseContainerPolicyWithAllowFullscreen) {
+  WebParsedFeaturePolicy container_policy =
+      GetContainerPolicyFromAllowedFeatures(
+          std::vector<WebFeaturePolicyFeature>({}), true, false,
+          origin_a_.Get());
+  EXPECT_EQ(1UL, container_policy.size());
+  EXPECT_EQ(WebFeaturePolicyFeature::kFullscreen, container_policy[0].feature);
+  EXPECT_TRUE(container_policy[0].matches_all_origins);
+}
+
+TEST_F(FeaturePolicyTest, ParseContainerPolicyWithAllowPaymentRequest) {
+  WebParsedFeaturePolicy container_policy =
+      GetContainerPolicyFromAllowedFeatures(
+          std::vector<WebFeaturePolicyFeature>(
+              {WebFeaturePolicyFeature::kVibrate}),
+          false, true, origin_a_.Get());
+  EXPECT_EQ(2UL, container_policy.size());
+  EXPECT_EQ(WebFeaturePolicyFeature::kVibrate, container_policy[0].feature);
+  EXPECT_FALSE(container_policy[0].matches_all_origins);
   EXPECT_EQ(1UL, container_policy[0].origins.size());
   EXPECT_TRUE(origin_a_->IsSameSchemeHostPortAndSuborigin(
       container_policy[0].origins[0].Get()));
   EXPECT_EQ(WebFeaturePolicyFeature::kPayment, container_policy[1].feature);
+  EXPECT_TRUE(container_policy[1].matches_all_origins);
+}
+
+TEST_F(FeaturePolicyTest, ParseContainerPolicyWithAllowAttributes) {
+  WebParsedFeaturePolicy container_policy =
+      GetContainerPolicyFromAllowedFeatures(
+          std::vector<WebFeaturePolicyFeature>(
+              {WebFeaturePolicyFeature::kVibrate,
+               WebFeaturePolicyFeature::kPayment}),
+          true, true, origin_a_.Get());
+  EXPECT_EQ(3UL, container_policy.size());
+  EXPECT_EQ(WebFeaturePolicyFeature::kVibrate, container_policy[0].feature);
+  EXPECT_FALSE(container_policy[0].matches_all_origins);
+  EXPECT_EQ(1UL, container_policy[0].origins.size());
+  EXPECT_TRUE(origin_a_->IsSameSchemeHostPortAndSuborigin(
+      container_policy[0].origins[0].Get()));
+  EXPECT_EQ(WebFeaturePolicyFeature::kPayment, container_policy[1].feature);
   EXPECT_FALSE(container_policy[1].matches_all_origins);
   EXPECT_EQ(1UL, container_policy[1].origins.size());
   EXPECT_TRUE(origin_a_->IsSameSchemeHostPortAndSuborigin(
       container_policy[1].origins[0].Get()));
+  EXPECT_EQ(WebFeaturePolicyFeature::kFullscreen, container_policy[2].feature);
+  EXPECT_TRUE(container_policy[2].matches_all_origins);
 }
 
 }  // namespace blink