Replicate feature policy headers to remote frames

third_party/WebKit/Source/core/frame/FeaturePolicyInFrameTest.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 "bindings/core/v8/ConditionalFeatures.h"
 #include "core/frame/LocalFrame.h"
 #include "core/testing/DummyPageHolder.h"
 #include "platform/RuntimeEnabledFeatures.h"
@@ skipping 59 matching lines @@
   std::unique_ptr<DummyPageHolder> m_dummyPageHolder;
 
   // Contains the list of controlled features, so that we are guaranteed to
   // have at least one of each kind of default behaviour represented.
   FeaturePolicy::FeatureList m_featureList;
 };
 
 TEST_F(FeaturePolicyInFrameTest, TestFeatureDefaultsInTopLevelFrame) {
   std::unique_ptr<FeaturePolicy> policy1 =
       createFromParentPolicy(nullptr, m_originA);
-  document().setFeaturePolicy(std::move(policy1));
+  document().setFeaturePolicyForTesting(std::move(policy1));
   EXPECT_TRUE(isFeatureEnabledInFrame(kDefaultOnFeature, frame()));
   EXPECT_TRUE(isFeatureEnabledInFrame(kDefaultSelfFeature, frame()));
   EXPECT_FALSE(isFeatureEnabledInFrame(kDefaultOffFeature, frame()));
 }
 
 TEST_F(FeaturePolicyInFrameTest, TestFeatureDefaultsInCrossOriginFrame) {
   std::unique_ptr<FeaturePolicy> policy1 =
       createFromParentPolicy(nullptr, m_originB);
   std::unique_ptr<FeaturePolicy> policy2 =
       createFromParentPolicy(policy1.get(), m_originA);
-  document().setFeaturePolicy(std::move(policy2));
+  document().setFeaturePolicyForTesting(std::move(policy2));
   EXPECT_TRUE(isFeatureEnabledInFrame(kDefaultOnFeature, frame()));
   EXPECT_FALSE(isFeatureEnabledInFrame(kDefaultSelfFeature, frame()));
   EXPECT_FALSE(isFeatureEnabledInFrame(kDefaultOffFeature, frame()));
 }
 
 TEST_F(FeaturePolicyInFrameTest, TestFeatureOverriddenInFrame) {
   Vector<String> messages;
   std::unique_ptr<FeaturePolicy> policy1 =
       createFromParentPolicy(nullptr, m_originA);
   policy1->setHeaderPolicy("{\"default-off\": [\"self\"], \"default-on\": []}",
-                           messages);
+                           &messages);
   EXPECT_EQ(0UL, messages.size());
-  document().setFeaturePolicy(std::move(policy1));
+  document().setFeaturePolicyForTesting(std::move(policy1));
   EXPECT_TRUE(isFeatureEnabledInFrame(kDefaultOffFeature, frame()));
   EXPECT_FALSE(isFeatureEnabledInFrame(kDefaultOnFeature, frame()));
 }
 
 // Ensure that features are enabled / disabled correctly when the FP runtime
 // flag is turned off.
 TEST_F(FeaturePolicyInFrameTest, TestFeatureDefaultsWhenFPDisabled) {
   RuntimeEnabledFeatures::setFeaturePolicyEnabled(false);
   EXPECT_TRUE(isFeatureEnabledInFrame(kDefaultOnFeature, frame()));
   EXPECT_TRUE(isFeatureEnabledInFrame(kDefaultSelfFeature, frame()));
   EXPECT_FALSE(isFeatureEnabledInFrame(kDefaultOffFeature, frame()));
 }
 
 // Ensure that feature policy has no effect when the FP runtime flag is turned
 // off.
 TEST_F(FeaturePolicyInFrameTest, TestPolicyInactiveWhenFPDisabled) {
   RuntimeEnabledFeatures::setFeaturePolicyEnabled(false);
   Vector<String> messages;
   std::unique_ptr<FeaturePolicy> policy1 =
       createFromParentPolicy(nullptr, m_originA);
-  policy1->setHeaderPolicy("{\"default-on\": []}", messages);
+  policy1->setHeaderPolicy("{\"default-on\": []}", &messages);
   EXPECT_EQ(0UL, messages.size());
-  document().setFeaturePolicy(std::move(policy1));
+  document().setFeaturePolicyForTesting(std::move(policy1));
   EXPECT_TRUE(isFeatureEnabledInFrame(kDefaultOnFeature, frame()));
 }
 
 }  // namespace blink