Communicate ExtensionSettings policy to renderers

chrome/browser/extensions/extension_with_management_policy_apitest.cc

+// Copyright (c) 2016 The Chromium Authors. All rights reserved.

[Devlin, 2017/02/14 23:17:09]

no (c)

[nrpeter, 2017/03/22 23:47:38]

Done.

+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "chrome/browser/extensions/extension_with_management_policy_apitest.h"
+#include "components/policy/core/browser/browser_policy_connector.h"
+#include "net/test/embedded_test_server/http_request.h"
+
+ExtensionApiTestWithManagementPolicy::ExtensionApiTestWithManagementPolicy()
+    : ExtensionApiTest() {}
+
+ExtensionApiTestWithManagementPolicy::~ExtensionApiTestWithManagementPolicy() {}
+
+void ExtensionApiTestWithManagementPolicy::SetUpInProcessBrowserTestFixture() {
+  ExtensionApiTest::SetUpInProcessBrowserTestFixture();
+  embedded_test_server()->RegisterRequestMonitor(
+      base::Bind(&ExtensionApiTestWithManagementPolicy::MonitorRequestHandler,
+                 base::Unretained(this)));
+  EXPECT_CALL(policy_provider_, IsInitializationComplete(testing::_))
+      .WillRepeatedly(testing::Return(true));
+  policy_provider_.SetAutoRefresh();
+  policy::BrowserPolicyConnector::SetPolicyProviderForTesting(
+      &policy_provider_);
+}
+
+void ExtensionApiTestWithManagementPolicy::MonitorRequestHandler(
+    const net::test_server::HttpRequest& request) {
+  auto host = request.headers.find("Host");
+  if (host != request.headers.end()) {
+    ManagementPolicyRequestLog log;
+    size_t delimiter_pos = host->second.find(":");
+    log.host = host->second.substr(0, delimiter_pos);
+    request_log_.push_back(log);
+  }
+}
+
+bool ExtensionApiTestWithManagementPolicy::BrowsedTo(const char* test_host) {
+  for (auto it : request_log_) {
+    if (it.host == test_host)
+      return true;
+  }
+  return false;
+}
+
+void ExtensionApiTestWithManagementPolicy::ClearRequestLog() {
+  request_log_.clear();
+}