OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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 "net/http/http_proxy_client_socket_pool.h" | 5 #include "net/http/http_proxy_client_socket_pool.h" |
6 | 6 |
7 #include "base/callback.h" | 7 #include "base/callback.h" |
8 #include "base/compiler_specific.h" | 8 #include "base/compiler_specific.h" |
9 #include "base/string_util.h" | 9 #include "base/string_util.h" |
10 #include "base/utf_string_conversions.h" | 10 #include "base/utf_string_conversions.h" |
11 #include "net/base/mock_cert_verifier.h" | |
12 #include "net/base/mock_host_resolver.h" | |
13 #include "net/base/net_errors.h" | 11 #include "net/base/net_errors.h" |
14 #include "net/base/ssl_config_service_defaults.h" | |
15 #include "net/base/test_completion_callback.h" | 12 #include "net/base/test_completion_callback.h" |
16 #include "net/http/http_auth_handler_factory.h" | |
17 #include "net/http/http_network_session.h" | 13 #include "net/http/http_network_session.h" |
18 #include "net/http/http_proxy_client_socket.h" | 14 #include "net/http/http_proxy_client_socket.h" |
19 #include "net/http/http_response_headers.h" | 15 #include "net/http/http_response_headers.h" |
20 #include "net/http/http_server_properties_impl.h" | |
21 #include "net/proxy/proxy_service.h" | |
22 #include "net/socket/client_socket_handle.h" | 16 #include "net/socket/client_socket_handle.h" |
23 #include "net/socket/client_socket_pool_histograms.h" | 17 #include "net/socket/client_socket_pool_histograms.h" |
24 #include "net/socket/socket_test_util.h" | 18 #include "net/socket/socket_test_util.h" |
25 #include "net/spdy/spdy_protocol.h" | 19 #include "net/spdy/spdy_protocol.h" |
26 #include "net/spdy/spdy_test_util_spdy2.h" | 20 #include "net/spdy/spdy_test_util_spdy2.h" |
27 #include "testing/gtest/include/gtest/gtest.h" | 21 #include "testing/gtest/include/gtest/gtest.h" |
28 | 22 |
29 using namespace net::test_spdy2; | 23 using namespace net::test_spdy2; |
30 | 24 |
31 namespace net { | 25 namespace net { |
(...skipping 25 matching lines...) Expand all Loading... |
57 HostPortPair("proxy", 80), LOWEST, false, false, | 51 HostPortPair("proxy", 80), LOWEST, false, false, |
58 OnHostResolutionCallback())), | 52 OnHostResolutionCallback())), |
59 ignored_ssl_socket_params_(new SSLSocketParams( | 53 ignored_ssl_socket_params_(new SSLSocketParams( |
60 ignored_transport_socket_params_, NULL, NULL, | 54 ignored_transport_socket_params_, NULL, NULL, |
61 ProxyServer::SCHEME_DIRECT, HostPortPair("www.google.com", 443), | 55 ProxyServer::SCHEME_DIRECT, HostPortPair("www.google.com", 443), |
62 ssl_config_, 0, false, false)), | 56 ssl_config_, 0, false, false)), |
63 tcp_histograms_("MockTCP"), | 57 tcp_histograms_("MockTCP"), |
64 transport_socket_pool_( | 58 transport_socket_pool_( |
65 kMaxSockets, kMaxSocketsPerGroup, | 59 kMaxSockets, kMaxSocketsPerGroup, |
66 &tcp_histograms_, | 60 &tcp_histograms_, |
67 &socket_factory_), | 61 session_deps_.deterministic_socket_factory.get()), |
68 ssl_histograms_("MockSSL"), | 62 ssl_histograms_("MockSSL"), |
69 cert_verifier_(new MockCertVerifier), | |
70 proxy_service_(ProxyService::CreateDirect()), | |
71 ssl_config_service_(new SSLConfigServiceDefaults), | |
72 ssl_socket_pool_(kMaxSockets, kMaxSocketsPerGroup, | 63 ssl_socket_pool_(kMaxSockets, kMaxSocketsPerGroup, |
73 &ssl_histograms_, | 64 &ssl_histograms_, |
74 &host_resolver_, | 65 session_deps_.host_resolver.get(), |
75 cert_verifier_.get(), | 66 session_deps_.cert_verifier.get(), |
76 NULL /* server_bound_cert_store */, | 67 NULL /* server_bound_cert_store */, |
77 NULL /* transport_security_state */, | 68 NULL /* transport_security_state */, |
78 "" /* ssl_session_cache_shard */, | 69 "" /* ssl_session_cache_shard */, |
79 &socket_factory_, | 70 session_deps_.deterministic_socket_factory.get(), |
80 &transport_socket_pool_, | 71 &transport_socket_pool_, |
81 NULL, | 72 NULL, |
82 NULL, | 73 NULL, |
83 ssl_config_service_.get(), | 74 session_deps_.ssl_config_service.get(), |
84 BoundNetLog().net_log()), | 75 BoundNetLog().net_log()), |
85 http_auth_handler_factory_( | |
86 HttpAuthHandlerFactory::CreateDefault(&host_resolver_)), | |
87 session_(CreateNetworkSession()), | 76 session_(CreateNetworkSession()), |
88 http_proxy_histograms_("HttpProxyUnitTest"), | 77 http_proxy_histograms_("HttpProxyUnitTest"), |
89 ssl_data_(NULL), | 78 ssl_data_(NULL), |
90 data_(NULL), | 79 data_(NULL), |
91 pool_(kMaxSockets, kMaxSocketsPerGroup, | 80 pool_(kMaxSockets, kMaxSocketsPerGroup, |
92 &http_proxy_histograms_, | 81 &http_proxy_histograms_, |
93 NULL, | 82 NULL, |
94 &transport_socket_pool_, | 83 &transport_socket_pool_, |
95 &ssl_socket_pool_, | 84 &ssl_socket_pool_, |
96 NULL) { | 85 NULL) { |
(...skipping 44 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
141 | 130 |
142 scoped_refptr<HttpProxySocketParams> GetTunnelParams() { | 131 scoped_refptr<HttpProxySocketParams> GetTunnelParams() { |
143 return GetParams(true); | 132 return GetParams(true); |
144 } | 133 } |
145 | 134 |
146 scoped_refptr<HttpProxySocketParams> GetNoTunnelParams() { | 135 scoped_refptr<HttpProxySocketParams> GetNoTunnelParams() { |
147 return GetParams(false); | 136 return GetParams(false); |
148 } | 137 } |
149 | 138 |
150 DeterministicMockClientSocketFactory& socket_factory() { | 139 DeterministicMockClientSocketFactory& socket_factory() { |
151 return socket_factory_; | 140 return *session_deps_.deterministic_socket_factory.get(); |
152 } | 141 } |
153 | 142 |
154 void Initialize(MockRead* reads, size_t reads_count, | 143 void Initialize(MockRead* reads, size_t reads_count, |
155 MockWrite* writes, size_t writes_count, | 144 MockWrite* writes, size_t writes_count, |
156 MockRead* spdy_reads, size_t spdy_reads_count, | 145 MockRead* spdy_reads, size_t spdy_reads_count, |
157 MockWrite* spdy_writes, size_t spdy_writes_count) { | 146 MockWrite* spdy_writes, size_t spdy_writes_count) { |
158 if (GetParam() == SPDY) { | 147 if (GetParam() == SPDY) { |
159 data_.reset(new DeterministicSocketData(spdy_reads, spdy_reads_count, | 148 data_.reset(new DeterministicSocketData(spdy_reads, spdy_reads_count, |
160 spdy_writes, spdy_writes_count)); | 149 spdy_writes, spdy_writes_count)); |
161 } else { | 150 } else { |
162 data_.reset(new DeterministicSocketData(reads, reads_count, writes, | 151 data_.reset(new DeterministicSocketData(reads, reads_count, writes, |
163 writes_count)); | 152 writes_count)); |
164 } | 153 } |
165 | 154 |
166 data_->set_connect_data(MockConnect(SYNCHRONOUS, OK)); | 155 data_->set_connect_data(MockConnect(SYNCHRONOUS, OK)); |
167 data_->StopAfter(2); // Request / Response | 156 data_->StopAfter(2); // Request / Response |
168 | 157 |
169 socket_factory_.AddSocketDataProvider(data_.get()); | 158 socket_factory().AddSocketDataProvider(data_.get()); |
170 | 159 |
171 if (GetParam() != HTTP) { | 160 if (GetParam() != HTTP) { |
172 ssl_data_.reset(new SSLSocketDataProvider(SYNCHRONOUS, OK)); | 161 ssl_data_.reset(new SSLSocketDataProvider(SYNCHRONOUS, OK)); |
173 if (GetParam() == SPDY) { | 162 if (GetParam() == SPDY) { |
174 InitializeSpdySsl(); | 163 InitializeSpdySsl(); |
175 } | 164 } |
176 socket_factory_.AddSSLSocketDataProvider(ssl_data_.get()); | 165 socket_factory().AddSSLSocketDataProvider(ssl_data_.get()); |
177 } | 166 } |
178 } | 167 } |
179 | 168 |
180 void InitializeSpdySsl() { | 169 void InitializeSpdySsl() { |
181 ssl_data_->SetNextProto(kProtoSPDY2); | 170 ssl_data_->SetNextProto(kProtoSPDY2); |
182 } | 171 } |
183 | 172 |
184 HttpNetworkSession* CreateNetworkSession() { | 173 HttpNetworkSession* CreateNetworkSession() { |
185 HttpNetworkSession::Params params; | 174 return SpdySessionDependencies::SpdyCreateSessionDeterministic( |
186 params.host_resolver = &host_resolver_; | 175 &session_deps_); |
187 params.cert_verifier = cert_verifier_.get(); | |
188 params.proxy_service = proxy_service_.get(); | |
189 params.client_socket_factory = &socket_factory_; | |
190 params.ssl_config_service = ssl_config_service_; | |
191 params.http_auth_handler_factory = http_auth_handler_factory_.get(); | |
192 params.http_server_properties = &http_server_properties_; | |
193 HttpNetworkSession* session = new HttpNetworkSession(params); | |
194 SpdySessionPoolPeer pool_peer(session->spdy_session_pool()); | |
195 pool_peer.EnableSendingInitialSettings(false); | |
196 return session; | |
197 } | 176 } |
198 | 177 |
199 private: | 178 private: |
| 179 SpdySessionDependencies session_deps_; |
200 SSLConfig ssl_config_; | 180 SSLConfig ssl_config_; |
201 | 181 |
202 scoped_refptr<TransportSocketParams> ignored_transport_socket_params_; | 182 scoped_refptr<TransportSocketParams> ignored_transport_socket_params_; |
203 scoped_refptr<SSLSocketParams> ignored_ssl_socket_params_; | 183 scoped_refptr<SSLSocketParams> ignored_ssl_socket_params_; |
204 ClientSocketPoolHistograms tcp_histograms_; | 184 ClientSocketPoolHistograms tcp_histograms_; |
205 DeterministicMockClientSocketFactory socket_factory_; | |
206 MockTransportClientSocketPool transport_socket_pool_; | 185 MockTransportClientSocketPool transport_socket_pool_; |
207 ClientSocketPoolHistograms ssl_histograms_; | 186 ClientSocketPoolHistograms ssl_histograms_; |
208 MockHostResolver host_resolver_; | 187 MockHostResolver host_resolver_; |
209 scoped_ptr<CertVerifier> cert_verifier_; | 188 scoped_ptr<CertVerifier> cert_verifier_; |
210 const scoped_ptr<ProxyService> proxy_service_; | |
211 const scoped_refptr<SSLConfigService> ssl_config_service_; | |
212 SSLClientSocketPool ssl_socket_pool_; | 189 SSLClientSocketPool ssl_socket_pool_; |
213 | 190 |
214 const scoped_ptr<HttpAuthHandlerFactory> http_auth_handler_factory_; | |
215 HttpServerPropertiesImpl http_server_properties_; | |
216 const scoped_refptr<HttpNetworkSession> session_; | 191 const scoped_refptr<HttpNetworkSession> session_; |
217 ClientSocketPoolHistograms http_proxy_histograms_; | 192 ClientSocketPoolHistograms http_proxy_histograms_; |
218 SpdyTestStateHelper spdy_state_; | 193 SpdyTestStateHelper spdy_state_; |
219 | 194 |
220 protected: | 195 protected: |
221 scoped_ptr<SSLSocketDataProvider> ssl_data_; | 196 scoped_ptr<SSLSocketDataProvider> ssl_data_; |
222 scoped_ptr<DeterministicSocketData> data_; | 197 scoped_ptr<DeterministicSocketData> data_; |
223 HttpProxyClientSocketPool pool_; | 198 HttpProxyClientSocketPool pool_; |
224 ClientSocketHandle handle_; | 199 ClientSocketHandle handle_; |
225 TestCompletionCallback callback_; | 200 TestCompletionCallback callback_; |
(...skipping 389 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
615 // Make sure Location header was included and correct. | 590 // Make sure Location header was included and correct. |
616 std::string location; | 591 std::string location; |
617 EXPECT_TRUE(headers->IsRedirect(&location)); | 592 EXPECT_TRUE(headers->IsRedirect(&location)); |
618 EXPECT_EQ(location, redirectTarget); | 593 EXPECT_EQ(location, redirectTarget); |
619 } | 594 } |
620 } | 595 } |
621 | 596 |
622 // It would be nice to also test the timeouts in HttpProxyClientSocketPool. | 597 // It would be nice to also test the timeouts in HttpProxyClientSocketPool. |
623 | 598 |
624 } // namespace net | 599 } // namespace net |
OLD | NEW |