OLD | NEW |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 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 "content/browser/webui/url_data_manager_backend.h" |
| 6 |
| 7 #include <memory> |
| 8 |
5 #include "base/macros.h" | 9 #include "base/macros.h" |
6 #include "base/memory/scoped_ptr.h" | |
7 #include "base/run_loop.h" | 10 #include "base/run_loop.h" |
8 #include "content/browser/webui/url_data_manager_backend.h" | |
9 #include "content/public/test/mock_resource_context.h" | 11 #include "content/public/test/mock_resource_context.h" |
10 #include "content/public/test/test_browser_thread_bundle.h" | 12 #include "content/public/test/test_browser_thread_bundle.h" |
11 #include "net/http/http_response_headers.h" | 13 #include "net/http/http_response_headers.h" |
12 #include "net/http/http_response_info.h" | 14 #include "net/http/http_response_info.h" |
13 #include "net/url_request/url_request_context.h" | 15 #include "net/url_request/url_request_context.h" |
14 #include "net/url_request/url_request_job.h" | 16 #include "net/url_request/url_request_job.h" |
15 #include "net/url_request/url_request_job_factory_impl.h" | 17 #include "net/url_request/url_request_job_factory_impl.h" |
16 #include "net/url_request/url_request_test_util.h" | 18 #include "net/url_request/url_request_test_util.h" |
17 #include "testing/gtest/include/gtest/gtest.h" | 19 #include "testing/gtest/include/gtest/gtest.h" |
18 | 20 |
(...skipping 23 matching lines...) Expand all Loading... |
42 class UrlDataManagerBackendTest : public testing::Test { | 44 class UrlDataManagerBackendTest : public testing::Test { |
43 public: | 45 public: |
44 UrlDataManagerBackendTest() { | 46 UrlDataManagerBackendTest() { |
45 // URLRequestJobFactory takes ownership of the passed in ProtocolHandler. | 47 // URLRequestJobFactory takes ownership of the passed in ProtocolHandler. |
46 url_request_job_factory_.SetProtocolHandler( | 48 url_request_job_factory_.SetProtocolHandler( |
47 "chrome", URLDataManagerBackend::CreateProtocolHandler( | 49 "chrome", URLDataManagerBackend::CreateProtocolHandler( |
48 &resource_context_, false, nullptr)); | 50 &resource_context_, false, nullptr)); |
49 url_request_context_.set_job_factory(&url_request_job_factory_); | 51 url_request_context_.set_job_factory(&url_request_job_factory_); |
50 } | 52 } |
51 | 53 |
52 scoped_ptr<net::URLRequest> CreateRequest(net::URLRequest::Delegate* delegate, | 54 std::unique_ptr<net::URLRequest> CreateRequest( |
53 const char* origin) { | 55 net::URLRequest::Delegate* delegate, |
54 scoped_ptr<net::URLRequest> request = url_request_context_.CreateRequest( | 56 const char* origin) { |
55 GURL("chrome://resources/polymer/v1_0/polymer/polymer-extracted.js"), | 57 std::unique_ptr<net::URLRequest> request = |
56 net::HIGHEST, delegate); | 58 url_request_context_.CreateRequest( |
| 59 GURL( |
| 60 "chrome://resources/polymer/v1_0/polymer/polymer-extracted.js"), |
| 61 net::HIGHEST, delegate); |
57 request->SetExtraRequestHeaderByName("Origin", origin, true); | 62 request->SetExtraRequestHeaderByName("Origin", origin, true); |
58 return request; | 63 return request; |
59 } | 64 } |
60 | 65 |
61 protected: | 66 protected: |
62 TestBrowserThreadBundle thread_bundle_; | 67 TestBrowserThreadBundle thread_bundle_; |
63 MockResourceContext resource_context_; | 68 MockResourceContext resource_context_; |
64 net::URLRequestJobFactoryImpl url_request_job_factory_; | 69 net::URLRequestJobFactoryImpl url_request_job_factory_; |
65 net::URLRequestContext url_request_context_; | 70 net::URLRequestContext url_request_context_; |
66 net::TestDelegate delegate_; | 71 net::TestDelegate delegate_; |
67 }; | 72 }; |
68 | 73 |
69 TEST_F(UrlDataManagerBackendTest, AccessControlAllowOriginChromeUrl) { | 74 TEST_F(UrlDataManagerBackendTest, AccessControlAllowOriginChromeUrl) { |
70 scoped_ptr<net::URLRequest> request( | 75 std::unique_ptr<net::URLRequest> request( |
71 CreateRequest(&delegate_, "chrome://webui")); | 76 CreateRequest(&delegate_, "chrome://webui")); |
72 request->Start(); | 77 request->Start(); |
73 base::RunLoop().RunUntilIdle(); | 78 base::RunLoop().RunUntilIdle(); |
74 EXPECT_TRUE(request->response_headers()->HasHeaderValue( | 79 EXPECT_TRUE(request->response_headers()->HasHeaderValue( |
75 "Access-Control-Allow-Origin", "chrome://webui")); | 80 "Access-Control-Allow-Origin", "chrome://webui")); |
76 } | 81 } |
77 | 82 |
78 TEST_F(UrlDataManagerBackendTest, AccessControlAllowOriginNonChromeUrl) { | 83 TEST_F(UrlDataManagerBackendTest, AccessControlAllowOriginNonChromeUrl) { |
79 scoped_ptr<net::URLRequest> request( | 84 std::unique_ptr<net::URLRequest> request( |
80 CreateRequest(&delegate_, "http://www.example.com")); | 85 CreateRequest(&delegate_, "http://www.example.com")); |
81 request->Start(); | 86 request->Start(); |
82 base::RunLoop().RunUntilIdle(); | 87 base::RunLoop().RunUntilIdle(); |
83 EXPECT_TRUE(request->response_headers()->HasHeaderValue( | 88 EXPECT_TRUE(request->response_headers()->HasHeaderValue( |
84 "Access-Control-Allow-Origin", "null")); | 89 "Access-Control-Allow-Origin", "null")); |
85 } | 90 } |
86 | 91 |
87 // Check that the URLRequest isn't passed headers after cancellation. | 92 // Check that the URLRequest isn't passed headers after cancellation. |
88 TEST_F(UrlDataManagerBackendTest, CancelBeforeResponseStarts) { | 93 TEST_F(UrlDataManagerBackendTest, CancelBeforeResponseStarts) { |
89 scoped_ptr<net::URLRequest> request( | 94 std::unique_ptr<net::URLRequest> request( |
90 CreateRequest(&delegate_, "chrome://webui")); | 95 CreateRequest(&delegate_, "chrome://webui")); |
91 request->Start(); | 96 request->Start(); |
92 request->Cancel(); | 97 request->Cancel(); |
93 base::RunLoop().RunUntilIdle(); | 98 base::RunLoop().RunUntilIdle(); |
94 EXPECT_EQ(net::URLRequestStatus::CANCELED, request->status().status()); | 99 EXPECT_EQ(net::URLRequestStatus::CANCELED, request->status().status()); |
95 EXPECT_EQ(1, delegate_.response_started_count()); | 100 EXPECT_EQ(1, delegate_.response_started_count()); |
96 } | 101 } |
97 | 102 |
98 // Check that the URLRequest isn't passed data after cancellation. | 103 // Check that the URLRequest isn't passed data after cancellation. |
99 TEST_F(UrlDataManagerBackendTest, CancelAfterFirstReadStarted) { | 104 TEST_F(UrlDataManagerBackendTest, CancelAfterFirstReadStarted) { |
100 CancelAfterFirstReadURLRequestDelegate cancel_delegate; | 105 CancelAfterFirstReadURLRequestDelegate cancel_delegate; |
101 scoped_ptr<net::URLRequest> request( | 106 std::unique_ptr<net::URLRequest> request( |
102 CreateRequest(&cancel_delegate, "chrome://webui")); | 107 CreateRequest(&cancel_delegate, "chrome://webui")); |
103 request->Start(); | 108 request->Start(); |
104 base::RunLoop().RunUntilIdle(); | 109 base::RunLoop().RunUntilIdle(); |
105 | 110 |
106 EXPECT_EQ(net::URLRequestStatus::CANCELED, request->status().status()); | 111 EXPECT_EQ(net::URLRequestStatus::CANCELED, request->status().status()); |
107 EXPECT_EQ(1, cancel_delegate.response_started_count()); | 112 EXPECT_EQ(1, cancel_delegate.response_started_count()); |
108 EXPECT_EQ("", cancel_delegate.data_received()); | 113 EXPECT_EQ("", cancel_delegate.data_received()); |
109 } | 114 } |
110 | 115 |
111 // Check for a network error page request via chrome://network-error/. | 116 // Check for a network error page request via chrome://network-error/. |
112 TEST_F(UrlDataManagerBackendTest, ChromeNetworkErrorPageRequest) { | 117 TEST_F(UrlDataManagerBackendTest, ChromeNetworkErrorPageRequest) { |
113 scoped_ptr<net::URLRequest> error_request = | 118 std::unique_ptr<net::URLRequest> error_request = |
114 url_request_context_.CreateRequest( | 119 url_request_context_.CreateRequest(GURL("chrome://network-error/-105"), |
115 GURL("chrome://network-error/-105"), net::HIGHEST, &delegate_); | 120 net::HIGHEST, &delegate_); |
116 error_request->Start(); | 121 error_request->Start(); |
117 base::RunLoop().Run(); | 122 base::RunLoop().Run(); |
118 EXPECT_EQ(net::URLRequestStatus::FAILED, error_request->status().status()); | 123 EXPECT_EQ(net::URLRequestStatus::FAILED, error_request->status().status()); |
119 EXPECT_EQ(net::ERR_NAME_NOT_RESOLVED, error_request->status().error()); | 124 EXPECT_EQ(net::ERR_NAME_NOT_RESOLVED, error_request->status().error()); |
120 } | 125 } |
121 | 126 |
122 // Check for an invalid network error page request via chrome://network-error/. | 127 // Check for an invalid network error page request via chrome://network-error/. |
123 TEST_F(UrlDataManagerBackendTest, ChromeNetworkErrorPageRequestFailed) { | 128 TEST_F(UrlDataManagerBackendTest, ChromeNetworkErrorPageRequestFailed) { |
124 scoped_ptr<net::URLRequest> error_request = | 129 std::unique_ptr<net::URLRequest> error_request = |
125 url_request_context_.CreateRequest( | 130 url_request_context_.CreateRequest( |
126 GURL("chrome://network-error/-123456789"), net::HIGHEST, &delegate_); | 131 GURL("chrome://network-error/-123456789"), net::HIGHEST, &delegate_); |
127 error_request->Start(); | 132 error_request->Start(); |
128 base::RunLoop().Run(); | 133 base::RunLoop().Run(); |
129 EXPECT_EQ(net::URLRequestStatus::FAILED, error_request->status().status()); | 134 EXPECT_EQ(net::URLRequestStatus::FAILED, error_request->status().status()); |
130 EXPECT_EQ(net::ERR_INVALID_URL, error_request->status().error()); | 135 EXPECT_EQ(net::ERR_INVALID_URL, error_request->status().error()); |
131 } | 136 } |
132 | 137 |
133 } // namespace content | 138 } // namespace content |
OLD | NEW |