| 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/socket/transport_client_socket_pool.h" | 5 #include "net/socket/transport_client_socket_pool.h" |
| 6 | 6 |
| 7 #include "base/bind.h" | 7 #include "base/bind.h" |
| 8 #include "base/bind_helpers.h" | 8 #include "base/bind_helpers.h" |
| 9 #include "base/callback.h" | 9 #include "base/callback.h" |
| 10 #include "base/message_loop/message_loop.h" | 10 #include "base/message_loop/message_loop.h" |
| (...skipping 23 matching lines...) Expand all Loading... |
| 34 const int kMaxSocketsPerGroup = 6; | 34 const int kMaxSocketsPerGroup = 6; |
| 35 const RequestPriority kDefaultPriority = LOW; | 35 const RequestPriority kDefaultPriority = LOW; |
| 36 | 36 |
| 37 class TransportClientSocketPoolTest : public testing::Test { | 37 class TransportClientSocketPoolTest : public testing::Test { |
| 38 protected: | 38 protected: |
| 39 TransportClientSocketPoolTest() | 39 TransportClientSocketPoolTest() |
| 40 : connect_backup_jobs_enabled_( | 40 : connect_backup_jobs_enabled_( |
| 41 ClientSocketPoolBaseHelper::set_connect_backup_jobs_enabled(true)), | 41 ClientSocketPoolBaseHelper::set_connect_backup_jobs_enabled(true)), |
| 42 params_( | 42 params_( |
| 43 new TransportSocketParams(HostPortPair("www.google.com", 80), | 43 new TransportSocketParams(HostPortPair("www.google.com", 80), |
| 44 false, false, | 44 false, |
| 45 OnHostResolutionCallback())), | 45 false, |
| 46 OnHostResolutionCallback(), |
| 47 AUTO_CONNECT_USE_DEFAULT)), |
| 46 histograms_(new ClientSocketPoolHistograms("TCPUnitTest")), | 48 histograms_(new ClientSocketPoolHistograms("TCPUnitTest")), |
| 47 host_resolver_(new MockHostResolver), | 49 host_resolver_(new MockHostResolver), |
| 48 client_socket_factory_(&net_log_), | 50 client_socket_factory_(&net_log_), |
| 49 pool_(kMaxSockets, | 51 pool_(kMaxSockets, |
| 50 kMaxSocketsPerGroup, | 52 kMaxSocketsPerGroup, |
| 51 histograms_.get(), | 53 histograms_.get(), |
| 52 host_resolver_.get(), | 54 host_resolver_.get(), |
| 53 &client_socket_factory_, | 55 &client_socket_factory_, |
| 54 NULL) { | 56 NULL) { |
| 55 } | 57 } |
| 56 | 58 |
| 57 virtual ~TransportClientSocketPoolTest() { | 59 virtual ~TransportClientSocketPoolTest() { |
| 58 internal::ClientSocketPoolBaseHelper::set_connect_backup_jobs_enabled( | 60 internal::ClientSocketPoolBaseHelper::set_connect_backup_jobs_enabled( |
| 59 connect_backup_jobs_enabled_); | 61 connect_backup_jobs_enabled_); |
| 60 } | 62 } |
| 61 | 63 |
| 62 int StartRequest(const std::string& group_name, RequestPriority priority) { | 64 int StartRequest(const std::string& group_name, RequestPriority priority) { |
| 63 scoped_refptr<TransportSocketParams> params(new TransportSocketParams( | 65 scoped_refptr<TransportSocketParams> params(new TransportSocketParams( |
| 64 HostPortPair("www.google.com", 80), false, false, | 66 HostPortPair("www.google.com", 80), false, false, |
| 65 OnHostResolutionCallback())); | 67 OnHostResolutionCallback(), AUTO_CONNECT_USE_DEFAULT)); |
| 66 return test_base_.StartRequestUsingPool( | 68 return test_base_.StartRequestUsingPool( |
| 67 &pool_, group_name, priority, params); | 69 &pool_, group_name, priority, params); |
| 68 } | 70 } |
| 69 | 71 |
| 70 int GetOrderOfRequest(size_t index) { | 72 int GetOrderOfRequest(size_t index) { |
| 71 return test_base_.GetOrderOfRequest(index); | 73 return test_base_.GetOrderOfRequest(index); |
| 72 } | 74 } |
| 73 | 75 |
| 74 bool ReleaseOneConnection(ClientSocketPoolTest::KeepAlive keep_alive) { | 76 bool ReleaseOneConnection(ClientSocketPoolTest::KeepAlive keep_alive) { |
| 75 return test_base_.ReleaseOneConnection(keep_alive); | 77 return test_base_.ReleaseOneConnection(keep_alive); |
| (...skipping 118 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 194 EXPECT_EQ(priority, host_resolver_->last_request_priority()); | 196 EXPECT_EQ(priority, host_resolver_->last_request_priority()); |
| 195 } | 197 } |
| 196 } | 198 } |
| 197 | 199 |
| 198 TEST_F(TransportClientSocketPoolTest, InitHostResolutionFailure) { | 200 TEST_F(TransportClientSocketPoolTest, InitHostResolutionFailure) { |
| 199 host_resolver_->rules()->AddSimulatedFailure("unresolvable.host.name"); | 201 host_resolver_->rules()->AddSimulatedFailure("unresolvable.host.name"); |
| 200 TestCompletionCallback callback; | 202 TestCompletionCallback callback; |
| 201 ClientSocketHandle handle; | 203 ClientSocketHandle handle; |
| 202 HostPortPair host_port_pair("unresolvable.host.name", 80); | 204 HostPortPair host_port_pair("unresolvable.host.name", 80); |
| 203 scoped_refptr<TransportSocketParams> dest(new TransportSocketParams( | 205 scoped_refptr<TransportSocketParams> dest(new TransportSocketParams( |
| 204 host_port_pair, false, false, | 206 host_port_pair, false, false, OnHostResolutionCallback(), |
| 205 OnHostResolutionCallback())); | 207 AUTO_CONNECT_USE_DEFAULT)); |
| 206 EXPECT_EQ(ERR_IO_PENDING, | 208 EXPECT_EQ(ERR_IO_PENDING, |
| 207 handle.Init("a", dest, kDefaultPriority, callback.callback(), | 209 handle.Init("a", dest, kDefaultPriority, callback.callback(), |
| 208 &pool_, BoundNetLog())); | 210 &pool_, BoundNetLog())); |
| 209 EXPECT_EQ(ERR_NAME_NOT_RESOLVED, callback.WaitForResult()); | 211 EXPECT_EQ(ERR_NAME_NOT_RESOLVED, callback.WaitForResult()); |
| 210 } | 212 } |
| 211 | 213 |
| 212 TEST_F(TransportClientSocketPoolTest, InitConnectionFailure) { | 214 TEST_F(TransportClientSocketPoolTest, InitConnectionFailure) { |
| 213 client_socket_factory_.set_default_client_socket_type( | 215 client_socket_factory_.set_default_client_socket_type( |
| 214 MockTransportClientSocketFactory::MOCK_FAILING_CLIENT_SOCKET); | 216 MockTransportClientSocketFactory::MOCK_FAILING_CLIENT_SOCKET); |
| 215 TestCompletionCallback callback; | 217 TestCompletionCallback callback; |
| (...skipping 254 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 470 handle_->socket()->Disconnect(); | 472 handle_->socket()->Disconnect(); |
| 471 handle_->Reset(); | 473 handle_->Reset(); |
| 472 { | 474 { |
| 473 base::MessageLoop::ScopedNestableTaskAllower allow( | 475 base::MessageLoop::ScopedNestableTaskAllower allow( |
| 474 base::MessageLoop::current()); | 476 base::MessageLoop::current()); |
| 475 base::MessageLoop::current()->RunUntilIdle(); | 477 base::MessageLoop::current()->RunUntilIdle(); |
| 476 } | 478 } |
| 477 within_callback_ = true; | 479 within_callback_ = true; |
| 478 scoped_refptr<TransportSocketParams> dest(new TransportSocketParams( | 480 scoped_refptr<TransportSocketParams> dest(new TransportSocketParams( |
| 479 HostPortPair("www.google.com", 80), false, false, | 481 HostPortPair("www.google.com", 80), false, false, |
| 480 OnHostResolutionCallback())); | 482 OnHostResolutionCallback(), AUTO_CONNECT_USE_DEFAULT)); |
| 481 int rv = handle_->Init("a", dest, LOWEST, callback(), pool_, | 483 int rv = handle_->Init("a", dest, LOWEST, callback(), pool_, |
| 482 BoundNetLog()); | 484 BoundNetLog()); |
| 483 EXPECT_EQ(OK, rv); | 485 EXPECT_EQ(OK, rv); |
| 484 } | 486 } |
| 485 } | 487 } |
| 486 | 488 |
| 487 ClientSocketHandle* const handle_; | 489 ClientSocketHandle* const handle_; |
| 488 TransportClientSocketPool* const pool_; | 490 TransportClientSocketPool* const pool_; |
| 489 bool within_callback_; | 491 bool within_callback_; |
| 490 CompletionCallback callback_; | 492 CompletionCallback callback_; |
| 491 | 493 |
| 492 DISALLOW_COPY_AND_ASSIGN(RequestSocketCallback); | 494 DISALLOW_COPY_AND_ASSIGN(RequestSocketCallback); |
| 493 }; | 495 }; |
| 494 | 496 |
| 495 TEST_F(TransportClientSocketPoolTest, RequestTwice) { | 497 TEST_F(TransportClientSocketPoolTest, RequestTwice) { |
| 496 ClientSocketHandle handle; | 498 ClientSocketHandle handle; |
| 497 RequestSocketCallback callback(&handle, &pool_); | 499 RequestSocketCallback callback(&handle, &pool_); |
| 498 scoped_refptr<TransportSocketParams> dest(new TransportSocketParams( | 500 scoped_refptr<TransportSocketParams> dest(new TransportSocketParams( |
| 499 HostPortPair("www.google.com", 80), false, false, | 501 HostPortPair("www.google.com", 80), false, false, |
| 500 OnHostResolutionCallback())); | 502 OnHostResolutionCallback(), AUTO_CONNECT_USE_DEFAULT)); |
| 501 int rv = handle.Init("a", dest, LOWEST, callback.callback(), &pool_, | 503 int rv = handle.Init("a", dest, LOWEST, callback.callback(), &pool_, |
| 502 BoundNetLog()); | 504 BoundNetLog()); |
| 503 ASSERT_EQ(ERR_IO_PENDING, rv); | 505 ASSERT_EQ(ERR_IO_PENDING, rv); |
| 504 | 506 |
| 505 // The callback is going to request "www.google.com". We want it to complete | 507 // The callback is going to request "www.google.com". We want it to complete |
| 506 // synchronously this time. | 508 // synchronously this time. |
| 507 host_resolver_->set_synchronous_mode(true); | 509 host_resolver_->set_synchronous_mode(true); |
| 508 | 510 |
| 509 EXPECT_EQ(OK, callback.WaitForResult()); | 511 EXPECT_EQ(OK, callback.WaitForResult()); |
| 510 | 512 |
| (...skipping 449 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 960 EXPECT_TRUE(handle.socket()); | 962 EXPECT_TRUE(handle.socket()); |
| 961 IPEndPoint endpoint; | 963 IPEndPoint endpoint; |
| 962 handle.socket()->GetLocalAddress(&endpoint); | 964 handle.socket()->GetLocalAddress(&endpoint); |
| 963 EXPECT_EQ(kIPv4AddressSize, endpoint.address().size()); | 965 EXPECT_EQ(kIPv4AddressSize, endpoint.address().size()); |
| 964 EXPECT_EQ(1, client_socket_factory_.allocation_count()); | 966 EXPECT_EQ(1, client_socket_factory_.allocation_count()); |
| 965 } | 967 } |
| 966 | 968 |
| 967 } // namespace | 969 } // namespace |
| 968 | 970 |
| 969 } // namespace net | 971 } // namespace net |
| OLD | NEW |