OLD | NEW |
1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "base/memory/scoped_ptr.h" | 5 #include "base/memory/scoped_ptr.h" |
6 #include "base/test/test_simple_task_runner.h" | 6 #include "base/test/test_simple_task_runner.h" |
7 #include "components/policy/core/common/cloud/policy_header_io_helper.h" | 7 #include "components/policy/core/common/cloud/policy_header_io_helper.h" |
8 #include "net/http/http_request_headers.h" | 8 #include "net/http/http_request_headers.h" |
9 #include "net/url_request/url_request.h" | 9 #include "net/url_request/url_request.h" |
10 #include "net/url_request/url_request_test_util.h" | 10 #include "net/url_request/url_request_test_util.h" |
11 #include "testing/gtest/include/gtest/gtest.h" | 11 #include "testing/gtest/include/gtest/gtest.h" |
12 | 12 |
13 namespace policy { | 13 namespace policy { |
14 | 14 |
15 namespace { | 15 namespace { |
16 const char kDMServerURL[] = "http://server_url"; | 16 const char kDMServerURL[] = "http://server_url"; |
17 const char kPolicyHeaderName[] = "Chrome-Policy-Posture"; | 17 const char kPolicyHeaderName[] = "Chrome-Policy-Posture"; |
18 const char kInitialPolicyHeader[] = "initial_header"; | 18 const char kInitialPolicyHeader[] = "initial_header"; |
19 | 19 |
20 class PolicyHeaderIOHelperTest : public testing::Test { | 20 class PolicyHeaderIOHelperTest : public testing::Test { |
21 public: | 21 public: |
22 PolicyHeaderIOHelperTest() { | 22 PolicyHeaderIOHelperTest() { |
23 task_runner_ = make_scoped_refptr(new base::TestSimpleTaskRunner()); | 23 task_runner_ = make_scoped_refptr(new base::TestSimpleTaskRunner()); |
24 } | 24 } |
25 virtual ~PolicyHeaderIOHelperTest() {} | 25 ~PolicyHeaderIOHelperTest() override {} |
26 | 26 |
27 virtual void SetUp() override { | 27 void SetUp() override { |
28 helper_ = make_scoped_ptr(new PolicyHeaderIOHelper(kDMServerURL, | 28 helper_ = make_scoped_ptr(new PolicyHeaderIOHelper(kDMServerURL, |
29 kInitialPolicyHeader, | 29 kInitialPolicyHeader, |
30 task_runner_)); | 30 task_runner_)); |
31 task_runner_->RunUntilIdle(); | 31 task_runner_->RunUntilIdle(); |
32 } | 32 } |
33 virtual void TearDown() override { | 33 void TearDown() override { |
34 task_runner_->RunUntilIdle(); | 34 task_runner_->RunUntilIdle(); |
35 helper_.reset(); | 35 helper_.reset(); |
36 } | 36 } |
37 | 37 |
38 void ValidateHeader(const net::HttpRequestHeaders& headers, | 38 void ValidateHeader(const net::HttpRequestHeaders& headers, |
39 const std::string& expected) { | 39 const std::string& expected) { |
40 std::string header; | 40 std::string header; |
41 EXPECT_TRUE(headers.GetHeader(kPolicyHeaderName, &header)); | 41 EXPECT_TRUE(headers.GetHeader(kPolicyHeaderName, &header)); |
42 EXPECT_EQ(header, expected); | 42 EXPECT_EQ(header, expected); |
43 } | 43 } |
(...skipping 33 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
77 TEST_F(PolicyHeaderIOHelperTest, ChangeToNoHeader) { | 77 TEST_F(PolicyHeaderIOHelperTest, ChangeToNoHeader) { |
78 helper_->UpdateHeader(""); | 78 helper_->UpdateHeader(""); |
79 task_runner_->RunUntilIdle(); | 79 task_runner_->RunUntilIdle(); |
80 scoped_ptr<net::URLRequest> request(context_.CreateRequest( | 80 scoped_ptr<net::URLRequest> request(context_.CreateRequest( |
81 GURL(kDMServerURL), net::DEFAULT_PRIORITY, NULL, NULL)); | 81 GURL(kDMServerURL), net::DEFAULT_PRIORITY, NULL, NULL)); |
82 helper_->AddPolicyHeaders(request->url(), request.get()); | 82 helper_->AddPolicyHeaders(request->url(), request.get()); |
83 EXPECT_TRUE(request->extra_request_headers().IsEmpty()); | 83 EXPECT_TRUE(request->extra_request_headers().IsEmpty()); |
84 } | 84 } |
85 | 85 |
86 } // namespace policy | 86 } // namespace policy |
OLD | NEW |