Index: components/policy/core/common/cloud/policy_header_io_helper.cc |
diff --git a/components/policy/core/common/cloud/policy_header_io_helper.cc b/components/policy/core/common/cloud/policy_header_io_helper.cc |
index a749dda4dc8f07236b4cbcbec9c7d42bdac42ba0..a5d15a2c80de90e06663816fe7e29ea77b5b1ade 100644 |
--- a/components/policy/core/common/cloud/policy_header_io_helper.cc |
+++ b/components/policy/core/common/cloud/policy_header_io_helper.cc |
@@ -7,15 +7,9 @@ |
#include "base/bind.h" |
#include "base/location.h" |
#include "base/sequenced_task_runner.h" |
+#include "components/policy/core/common/cloud/cloud_policy_constants.h" |
#include "net/url_request/url_request.h" |
-namespace { |
- |
-// The name of the header containing the policy information. |
-const char kChromePolicyHeader[] = "Chrome-Policy-Posture"; |
- |
-} // namespace |
- |
namespace policy { |
PolicyHeaderIOHelper::PolicyHeaderIOHelper( |
@@ -31,9 +25,9 @@ PolicyHeaderIOHelper::~PolicyHeaderIOHelper() { |
} |
// Sets any necessary policy headers on the passed request. |
-void PolicyHeaderIOHelper::AddPolicyHeaders(net::URLRequest* request) const { |
+void PolicyHeaderIOHelper::AddPolicyHeaders(const GURL& url, |
+ net::URLRequest* request) const { |
DCHECK(io_task_runner_->RunsTasksOnCurrentThread()); |
- const GURL& url = request->url(); |
if (!policy_header_.empty() && |
url.spec().compare(0, server_url_.size(), server_url_) == 0) { |
request->SetExtraRequestHeaderByName(kChromePolicyHeader, |
@@ -56,4 +50,18 @@ void PolicyHeaderIOHelper::UpdateHeaderOnIOThread( |
policy_header_ = new_header; |
} |
+void PolicyHeaderIOHelper::SetServerURLForTest(const std::string& server_url) { |
+ io_task_runner_->PostTask( |
+ FROM_HERE, |
+ base::Bind(&PolicyHeaderIOHelper::SetServerURLOnIOThread, |
+ base::Unretained(this), server_url)); |
+ |
+} |
+ |
+void PolicyHeaderIOHelper::SetServerURLOnIOThread( |
+ const std::string& server_url) { |
+ DCHECK(io_task_runner_->RunsTasksOnCurrentThread()); |
+ server_url_ = server_url; |
+} |
+ |
} // namespace policy |