Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(772)

Unified Diff: components/policy/core/common/cloud/policy_header_io_helper.cc

Issue 303993005: Add browser test for ChromeResourceDispatcherHostDelegate. (Closed) Base URL: svn://svn.chromium.org/chrome/trunk/src
Patch Set: Fix use-after-free of ResourceDispatcherHostDelegate in test. Created 6 years, 6 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
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

Powered by Google App Engine
This is Rietveld 408576698