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

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 98 matching lines @@
       parsedPolicy[1].origins[0].get()));
   EXPECT_TRUE(m_originC->isSameSchemeHostPortAndSuborigin(
       parsedPolicy[1].origins[1].get()));
   EXPECT_EQ(WebFeaturePolicyFeature::Payment, parsedPolicy[2].feature);
   EXPECT_FALSE(parsedPolicy[2].matchesAllOrigins);
   EXPECT_EQ(1UL, parsedPolicy[2].origins.size());
   EXPECT_TRUE(m_originA->isSameSchemeHostPortAndSuborigin(
       parsedPolicy[2].origins[0].get()));
 }
 
+TEST_F(FeaturePolicyTest, ParseEmptyContainerPolicy) {
+  WebParsedFeaturePolicy containerPolicy =
+      getContainerPolicyFromAllowedFeatures(
+          std::vector<WebFeaturePolicyFeature>({}), m_originA.get());
+  EXPECT_EQ(0UL, containerPolicy.size());
+}
+
+TEST_F(FeaturePolicyTest, ParseContainerPolicy) {
+  WebParsedFeaturePolicy containerPolicy =
+      getContainerPolicyFromAllowedFeatures(
+          std::vector<WebFeaturePolicyFeature>(
+              {WebFeaturePolicyFeature::Vibrate,
+               WebFeaturePolicyFeature::Payment}),
+          m_originA.get());
+  EXPECT_EQ(2UL, containerPolicy.size());
+  EXPECT_EQ(WebFeaturePolicyFeature::Vibrate, containerPolicy[0].feature);
+  EXPECT_FALSE(containerPolicy[0].matchesAllOrigins);
+  EXPECT_EQ(1UL, containerPolicy[0].origins.size());
+  EXPECT_TRUE(m_originA->isSameSchemeHostPortAndSuborigin(
+      containerPolicy[0].origins[0].get()));
+  EXPECT_EQ(WebFeaturePolicyFeature::Payment, containerPolicy[1].feature);
+  EXPECT_FALSE(containerPolicy[1].matchesAllOrigins);
+  EXPECT_EQ(1UL, containerPolicy[1].origins.size());
+  EXPECT_TRUE(m_originA->isSameSchemeHostPortAndSuborigin(
+      containerPolicy[1].origins[0].get()));
+}
+
 }  // namespace blink