Implement feature policy checks in the browser process

content/public/test/test_renderer_host.h

 // Copyright (c) 2012 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.
 
 #ifndef CONTENT_PUBLIC_TEST_TEST_RENDERER_HOST_H_
 #define CONTENT_PUBLIC_TEST_TEST_RENDERER_HOST_H_
 
 #include <stdint.h>
 
 #include <memory>
@@ skipping 121 matching lines @@
   // Calls OnBeforeUnloadACK on this RenderFrameHost with the given parameter.
   virtual void SendBeforeUnloadACK(bool proceed) = 0;
 
   // Simulates the SwapOut_ACK that fires if you commit a cross-site
   // navigation without making any network requests.
   virtual void SimulateSwapOutACK() = 0;
 
   // Simulate a renderer-initiated navigation up until commit.
   virtual void NavigateAndCommitRendererInitiated(bool did_create_new_entry,
                                                   const GURL& url) = 0;
+
+  // Set the feature policy header for the RenderFrameHost for test. Currently
+  // this is limited to setting a whitelist for a single feature. This function
+  // can be generalized as needed. Setting a header policy should only be done
+  // once per navigation of the RFH.
+  virtual void SetFeaturePolicyHeader(

[alexmos, 2017/05/30 20:37:57]

nit: SimulateFeaturePolicyHeader might be more consistent with other function
names here.

[raymes, 2017/05/31 07:07:54]

Done.

+      blink::WebFeaturePolicyFeature feature,
+      const std::vector<url::Origin>& whitelist) = 0;
 };
 
 // An interface and utility for driving tests of RenderViewHost.
 class RenderViewHostTester {
  public:
   // Retrieves the RenderViewHostTester that drives the specified
   // RenderViewHost.  The RenderViewHost must have been created while
   // RenderViewHost testing was enabled; use a
   // RenderViewHostTestEnabler instance (see below) to do this.
   static RenderViewHostTester* For(RenderViewHost* host);
@@ skipping 141 matching lines @@
 #endif
   std::unique_ptr<RenderViewHostTestEnabler> rvh_test_enabler_;
   RenderProcessHostFactory* factory_ = nullptr;
 
   DISALLOW_COPY_AND_ASSIGN(RenderViewHostTestHarness);
 };
 
 }  // namespace content
 
 #endif  // CONTENT_PUBLIC_TEST_TEST_RENDERER_HOST_H_