| OLD | NEW |
| (Empty) | |
| 1 // Copyright 2017 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. |
| 4 |
| 5 #include "chrome/browser/extensions/extension_with_management_policy_apitest.h" |
| 6 #include "components/policy/core/browser/browser_policy_connector.h" |
| 7 #include "net/test/embedded_test_server/http_request.h" |
| 8 |
| 9 ExtensionApiTestWithManagementPolicy::ExtensionApiTestWithManagementPolicy() |
| 10 : ExtensionApiTest() {} |
| 11 |
| 12 ExtensionApiTestWithManagementPolicy::~ExtensionApiTestWithManagementPolicy() {} |
| 13 |
| 14 void ExtensionApiTestWithManagementPolicy::SetUpInProcessBrowserTestFixture() { |
| 15 ExtensionApiTest::SetUpInProcessBrowserTestFixture(); |
| 16 embedded_test_server()->RegisterRequestMonitor( |
| 17 base::Bind(&ExtensionApiTestWithManagementPolicy::MonitorRequestHandler, |
| 18 base::Unretained(this))); |
| 19 EXPECT_CALL(policy_provider_, IsInitializationComplete(testing::_)) |
| 20 .WillRepeatedly(testing::Return(true)); |
| 21 policy_provider_.SetAutoRefresh(); |
| 22 policy::BrowserPolicyConnector::SetPolicyProviderForTesting( |
| 23 &policy_provider_); |
| 24 } |
| 25 |
| 26 void ExtensionApiTestWithManagementPolicy::MonitorRequestHandler( |
| 27 const net::test_server::HttpRequest& request) { |
| 28 auto host = request.headers.find("Host"); |
| 29 if (host != request.headers.end()) { |
| 30 ManagementPolicyRequestLog log; |
| 31 size_t delimiter_pos = host->second.find(":"); |
| 32 log.host = host->second.substr(0, delimiter_pos); |
| 33 request_log_.push_back(log); |
| 34 } |
| 35 } |
| 36 |
| 37 bool ExtensionApiTestWithManagementPolicy::BrowsedTo(const char* test_host) { |
| 38 return std::find_if(request_log_.begin(), request_log_.end(), |
| 39 [test_host](const ManagementPolicyRequestLog& log) { |
| 40 return log.host == test_host; |
| 41 }) != request_log_.end(); |
| 42 } |
| 43 |
| 44 void ExtensionApiTestWithManagementPolicy::ClearRequestLog() { |
| 45 request_log_.clear(); |
| 46 } |
| OLD | NEW |