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 ee504f3ca03e7805d4ed129166419237b7c4f2ff..35c31efc23a5713aed08d52d973e0e868ee6aca0 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 |
@@ -10,6 +10,7 @@ |
#include "base/message_loop/message_loop.h" |
#include "base/test/test_simple_task_runner.h" |
#include "net/http/http_request_headers.h" |
+#include "net/traffic_annotation/network_traffic_annotation_test_helper.h" |
#include "net/url_request/url_request.h" |
#include "net/url_request/url_request_test_util.h" |
#include "testing/gtest/include/gtest/gtest.h" |
@@ -55,14 +56,16 @@ class PolicyHeaderIOHelperTest : public testing::Test { |
TEST_F(PolicyHeaderIOHelperTest, InitialHeader) { |
std::unique_ptr<net::URLRequest> request( |
- context_.CreateRequest(GURL(kDMServerURL), net::DEFAULT_PRIORITY, NULL)); |
+ context_.CreateRequest(GURL(kDMServerURL), net::DEFAULT_PRIORITY, NULL, |
+ TRAFFIC_ANNOTATION_FOR_TESTS)); |
helper_->AddPolicyHeaders(request->url(), request.get()); |
ValidateHeader(request->extra_request_headers(), kInitialPolicyHeader); |
} |
TEST_F(PolicyHeaderIOHelperTest, NoHeaderOnNonMatchingURL) { |
std::unique_ptr<net::URLRequest> request(context_.CreateRequest( |
- GURL("http://non-matching.com"), net::DEFAULT_PRIORITY, NULL)); |
+ GURL("http://non-matching.com"), net::DEFAULT_PRIORITY, NULL, |
+ TRAFFIC_ANNOTATION_FOR_TESTS)); |
helper_->AddPolicyHeaders(request->url(), request.get()); |
EXPECT_TRUE(request->extra_request_headers().IsEmpty()); |
} |
@@ -72,7 +75,8 @@ TEST_F(PolicyHeaderIOHelperTest, HeaderChange) { |
helper_->UpdateHeader(new_header); |
task_runner_->RunUntilIdle(); |
std::unique_ptr<net::URLRequest> request( |
- context_.CreateRequest(GURL(kDMServerURL), net::DEFAULT_PRIORITY, NULL)); |
+ context_.CreateRequest(GURL(kDMServerURL), net::DEFAULT_PRIORITY, NULL, |
+ TRAFFIC_ANNOTATION_FOR_TESTS)); |
helper_->AddPolicyHeaders(request->url(), request.get()); |
ValidateHeader(request->extra_request_headers(), new_header); |
} |
@@ -81,7 +85,8 @@ TEST_F(PolicyHeaderIOHelperTest, ChangeToNoHeader) { |
helper_->UpdateHeader(""); |
task_runner_->RunUntilIdle(); |
std::unique_ptr<net::URLRequest> request( |
- context_.CreateRequest(GURL(kDMServerURL), net::DEFAULT_PRIORITY, NULL)); |
+ context_.CreateRequest(GURL(kDMServerURL), net::DEFAULT_PRIORITY, NULL, |
+ TRAFFIC_ANNOTATION_FOR_TESTS)); |
helper_->AddPolicyHeaders(request->url(), request.get()); |
EXPECT_TRUE(request->extra_request_headers().IsEmpty()); |
} |