Index: components/policy/core/common/cloud/policy_header_io_helper_unittest.cc |
diff --git a/components/policy/core/common/cloud/policy_header_io_helper_unittest.cc b/components/policy/core/common/cloud/policy_header_io_helper_unittest.cc |
index fdb19bdf5958e537278ecdf8add531465e3bd75f..862cb58be004c3d668a74c6e5f3b2bc3ded6c950 100644 |
--- a/components/policy/core/common/cloud/policy_header_io_helper_unittest.cc |
+++ b/components/policy/core/common/cloud/policy_header_io_helper_unittest.cc |
@@ -2,9 +2,11 @@ |
// Use of this source code is governed by a BSD-style license that can be |
// found in the LICENSE file. |
+#include "base/memory/scoped_ptr.h" |
#include "base/test/test_simple_task_runner.h" |
#include "components/policy/core/common/cloud/policy_header_io_helper.h" |
#include "net/http/http_request_headers.h" |
+#include "net/url_request/url_request.h" |
#include "net/url_request/url_request_test_util.h" |
#include "testing/gtest/include/gtest/gtest.h" |
@@ -49,36 +51,36 @@ class PolicyHeaderIOHelperTest : public testing::Test { |
} // namespace |
TEST_F(PolicyHeaderIOHelperTest, InitialHeader) { |
- net::TestURLRequest request( |
- GURL(kDMServerURL), net::DEFAULT_PRIORITY, NULL, &context_); |
- helper_->AddPolicyHeaders(request.url(), &request); |
- ValidateHeader(request.extra_request_headers(), kInitialPolicyHeader); |
+ scoped_ptr<net::URLRequest> request(context_.CreateRequest( |
+ GURL(kDMServerURL), net::DEFAULT_PRIORITY, NULL, NULL)); |
+ helper_->AddPolicyHeaders(request->url(), request.get()); |
+ ValidateHeader(request->extra_request_headers(), kInitialPolicyHeader); |
} |
TEST_F(PolicyHeaderIOHelperTest, NoHeaderOnNonMatchingURL) { |
- net::TestURLRequest request( |
- GURL("http://non-matching.com"), net::DEFAULT_PRIORITY, NULL, &context_); |
- helper_->AddPolicyHeaders(request.url(), &request); |
- EXPECT_TRUE(request.extra_request_headers().IsEmpty()); |
+ scoped_ptr<net::URLRequest> request(context_.CreateRequest( |
+ GURL("http://non-matching.com"), net::DEFAULT_PRIORITY, NULL, NULL)); |
+ helper_->AddPolicyHeaders(request->url(), request.get()); |
+ EXPECT_TRUE(request->extra_request_headers().IsEmpty()); |
} |
TEST_F(PolicyHeaderIOHelperTest, HeaderChange) { |
std::string new_header = "new_header"; |
helper_->UpdateHeader(new_header); |
task_runner_->RunUntilIdle(); |
- net::TestURLRequest request( |
- GURL(kDMServerURL), net::DEFAULT_PRIORITY, NULL, &context_); |
- helper_->AddPolicyHeaders(request.url(), &request); |
- ValidateHeader(request.extra_request_headers(), new_header); |
+ scoped_ptr<net::URLRequest> request(context_.CreateRequest( |
+ GURL(kDMServerURL), net::DEFAULT_PRIORITY, NULL, NULL)); |
+ helper_->AddPolicyHeaders(request->url(), request.get()); |
+ ValidateHeader(request->extra_request_headers(), new_header); |
} |
TEST_F(PolicyHeaderIOHelperTest, ChangeToNoHeader) { |
helper_->UpdateHeader(""); |
task_runner_->RunUntilIdle(); |
- net::TestURLRequest request( |
- GURL(kDMServerURL), net::DEFAULT_PRIORITY, NULL, &context_); |
- helper_->AddPolicyHeaders(request.url(), &request); |
- EXPECT_TRUE(request.extra_request_headers().IsEmpty()); |
+ scoped_ptr<net::URLRequest> request(context_.CreateRequest( |
+ GURL(kDMServerURL), net::DEFAULT_PRIORITY, NULL, NULL)); |
+ helper_->AddPolicyHeaders(request->url(), request.get()); |
+ EXPECT_TRUE(request->extra_request_headers().IsEmpty()); |
} |
} // namespace policy |