| Index: net/http/http_network_transaction_unittest.cc
|
| diff --git a/net/http/http_network_transaction_unittest.cc b/net/http/http_network_transaction_unittest.cc
|
| index f5afd74a88d82e7cf8d2ea4d13b364e5f1a97392..43adf0653381998da9a246830029baa7f434dd12 100644
|
| --- a/net/http/http_network_transaction_unittest.cc
|
| +++ b/net/http/http_network_transaction_unittest.cc
|
| @@ -45,6 +45,7 @@
|
| #include "net/proxy/proxy_resolver.h"
|
| #include "net/proxy/proxy_service.h"
|
| #include "net/socket/client_socket_factory.h"
|
| +#include "net/socket/mock_client_socket_pool_manager.h"
|
| #include "net/socket/socket_test_util.h"
|
| #include "net/socket/ssl_client_socket.h"
|
| #include "net/spdy/spdy_framer.h"
|
| @@ -3306,11 +3307,11 @@ TEST_F(HttpNetworkTransactionTest, DontRecycleTransportSocketForSSLTunnel) {
|
|
|
| // We now check to make sure the TCPClientSocket was not added back to
|
| // the pool.
|
| - EXPECT_EQ(0, session->transport_socket_pool()->IdleSocketCount());
|
| + EXPECT_EQ(0, session->GetTransportSocketPool()->IdleSocketCount());
|
| trans.reset();
|
| MessageLoop::current()->RunAllPending();
|
| // Make sure that the socket didn't get recycled after calling the destructor.
|
| - EXPECT_EQ(0, session->transport_socket_pool()->IdleSocketCount());
|
| + EXPECT_EQ(0, session->GetTransportSocketPool()->IdleSocketCount());
|
| }
|
|
|
| // Make sure that we recycle a socket after reading all of the response body.
|
| @@ -3353,7 +3354,7 @@ TEST_F(HttpNetworkTransactionTest, RecycleSocket) {
|
| std::string status_line = response->headers->GetStatusLine();
|
| EXPECT_EQ("HTTP/1.1 200 OK", status_line);
|
|
|
| - EXPECT_EQ(0, session->transport_socket_pool()->IdleSocketCount());
|
| + EXPECT_EQ(0, session->GetTransportSocketPool()->IdleSocketCount());
|
|
|
| std::string response_data;
|
| rv = ReadTransaction(trans.get(), &response_data);
|
| @@ -3365,7 +3366,7 @@ TEST_F(HttpNetworkTransactionTest, RecycleSocket) {
|
| MessageLoop::current()->RunAllPending();
|
|
|
| // We now check to make sure the socket was added back to the pool.
|
| - EXPECT_EQ(1, session->transport_socket_pool()->IdleSocketCount());
|
| + EXPECT_EQ(1, session->GetTransportSocketPool()->IdleSocketCount());
|
| }
|
|
|
| // Make sure that we recycle a SSL socket after reading all of the response
|
| @@ -3412,7 +3413,7 @@ TEST_F(HttpNetworkTransactionTest, RecycleSSLSocket) {
|
| ASSERT_TRUE(response->headers != NULL);
|
| EXPECT_EQ("HTTP/1.1 200 OK", response->headers->GetStatusLine());
|
|
|
| - EXPECT_EQ(0, session->transport_socket_pool()->IdleSocketCount());
|
| + EXPECT_EQ(0, session->GetTransportSocketPool()->IdleSocketCount());
|
|
|
| std::string response_data;
|
| rv = ReadTransaction(trans.get(), &response_data);
|
| @@ -3424,7 +3425,7 @@ TEST_F(HttpNetworkTransactionTest, RecycleSSLSocket) {
|
| MessageLoop::current()->RunAllPending();
|
|
|
| // We now check to make sure the socket was added back to the pool.
|
| - EXPECT_EQ(1, session->ssl_socket_pool()->IdleSocketCount());
|
| + EXPECT_EQ(1, session->GetSSLSocketPool()->IdleSocketCount());
|
| }
|
|
|
| // Grab a SSL socket, use it, and put it back into the pool. Then, reuse it
|
| @@ -3480,7 +3481,7 @@ TEST_F(HttpNetworkTransactionTest, RecycleDeadSSLSocket) {
|
| ASSERT_TRUE(response->headers != NULL);
|
| EXPECT_EQ("HTTP/1.1 200 OK", response->headers->GetStatusLine());
|
|
|
| - EXPECT_EQ(0, session->transport_socket_pool()->IdleSocketCount());
|
| + EXPECT_EQ(0, session->GetTransportSocketPool()->IdleSocketCount());
|
|
|
| std::string response_data;
|
| rv = ReadTransaction(trans.get(), &response_data);
|
| @@ -3492,7 +3493,7 @@ TEST_F(HttpNetworkTransactionTest, RecycleDeadSSLSocket) {
|
| MessageLoop::current()->RunAllPending();
|
|
|
| // We now check to make sure the socket was added back to the pool.
|
| - EXPECT_EQ(1, session->ssl_socket_pool()->IdleSocketCount());
|
| + EXPECT_EQ(1, session->GetSSLSocketPool()->IdleSocketCount());
|
|
|
| // Now start the second transaction, which should reuse the previous socket.
|
|
|
| @@ -3508,7 +3509,7 @@ TEST_F(HttpNetworkTransactionTest, RecycleDeadSSLSocket) {
|
| ASSERT_TRUE(response->headers != NULL);
|
| EXPECT_EQ("HTTP/1.1 200 OK", response->headers->GetStatusLine());
|
|
|
| - EXPECT_EQ(0, session->transport_socket_pool()->IdleSocketCount());
|
| + EXPECT_EQ(0, session->GetTransportSocketPool()->IdleSocketCount());
|
|
|
| rv = ReadTransaction(trans.get(), &response_data);
|
| EXPECT_EQ(OK, rv);
|
| @@ -3519,7 +3520,7 @@ TEST_F(HttpNetworkTransactionTest, RecycleDeadSSLSocket) {
|
| MessageLoop::current()->RunAllPending();
|
|
|
| // We now check to make sure the socket was added back to the pool.
|
| - EXPECT_EQ(1, session->ssl_socket_pool()->IdleSocketCount());
|
| + EXPECT_EQ(1, session->GetSSLSocketPool()->IdleSocketCount());
|
| }
|
|
|
| // Make sure that we recycle a socket after a zero-length response.
|
| @@ -3564,7 +3565,7 @@ TEST_F(HttpNetworkTransactionTest, RecycleSocketAfterZeroContentLength) {
|
| std::string status_line = response->headers->GetStatusLine();
|
| EXPECT_EQ("HTTP/1.1 204 No Content", status_line);
|
|
|
| - EXPECT_EQ(0, session->transport_socket_pool()->IdleSocketCount());
|
| + EXPECT_EQ(0, session->GetTransportSocketPool()->IdleSocketCount());
|
|
|
| std::string response_data;
|
| rv = ReadTransaction(trans.get(), &response_data);
|
| @@ -3576,7 +3577,7 @@ TEST_F(HttpNetworkTransactionTest, RecycleSocketAfterZeroContentLength) {
|
| MessageLoop::current()->RunAllPending();
|
|
|
| // We now check to make sure the socket was added back to the pool.
|
| - EXPECT_EQ(1, session->transport_socket_pool()->IdleSocketCount());
|
| + EXPECT_EQ(1, session->GetTransportSocketPool()->IdleSocketCount());
|
| }
|
|
|
| TEST_F(HttpNetworkTransactionTest, ResendRequestOnWriteBodyError) {
|
| @@ -5626,10 +5627,13 @@ TEST_F(HttpNetworkTransactionTest, GroupNameForDirectConnections) {
|
| HttpNetworkSessionPeer peer(session);
|
| CaptureGroupNameTransportSocketPool* transport_conn_pool =
|
| new CaptureGroupNameTransportSocketPool(NULL, NULL);
|
| - peer.SetTransportSocketPool(transport_conn_pool);
|
| CaptureGroupNameSSLSocketPool* ssl_conn_pool =
|
| new CaptureGroupNameSSLSocketPool(NULL, NULL);
|
| - peer.SetSSLSocketPool(ssl_conn_pool);
|
| + MockClientSocketPoolManager* mock_pool_manager =
|
| + new MockClientSocketPoolManager;
|
| + mock_pool_manager->SetTransportSocketPool(transport_conn_pool);
|
| + mock_pool_manager->SetSSLSocketPool(ssl_conn_pool);
|
| + peer.SetClientSocketPoolManager(mock_pool_manager);
|
|
|
| EXPECT_EQ(ERR_IO_PENDING,
|
| GroupNameTransactionHelper(tests[i].url, session));
|
| @@ -5682,10 +5686,14 @@ TEST_F(HttpNetworkTransactionTest, GroupNameForHTTPProxyConnections) {
|
| HostPortPair proxy_host("http_proxy", 80);
|
| CaptureGroupNameHttpProxySocketPool* http_proxy_pool =
|
| new CaptureGroupNameHttpProxySocketPool(NULL, NULL);
|
| - peer.SetSocketPoolForHTTPProxy(proxy_host, http_proxy_pool);
|
| CaptureGroupNameSSLSocketPool* ssl_conn_pool =
|
| new CaptureGroupNameSSLSocketPool(NULL, NULL);
|
| - peer.SetSocketPoolForSSLWithProxy(proxy_host, ssl_conn_pool);
|
| +
|
| + MockClientSocketPoolManager* mock_pool_manager =
|
| + new MockClientSocketPoolManager;
|
| + mock_pool_manager->SetSocketPoolForHTTPProxy(proxy_host, http_proxy_pool);
|
| + mock_pool_manager->SetSocketPoolForSSLWithProxy(proxy_host, ssl_conn_pool);
|
| + peer.SetClientSocketPoolManager(mock_pool_manager);
|
|
|
| EXPECT_EQ(ERR_IO_PENDING,
|
| GroupNameTransactionHelper(tests[i].url, session));
|
| @@ -5750,10 +5758,14 @@ TEST_F(HttpNetworkTransactionTest, GroupNameForSOCKSConnections) {
|
| HostPortPair proxy_host("socks_proxy", 1080);
|
| CaptureGroupNameSOCKSSocketPool* socks_conn_pool =
|
| new CaptureGroupNameSOCKSSocketPool(NULL, NULL);
|
| - peer.SetSocketPoolForSOCKSProxy(proxy_host, socks_conn_pool);
|
| CaptureGroupNameSSLSocketPool* ssl_conn_pool =
|
| new CaptureGroupNameSSLSocketPool(NULL, NULL);
|
| - peer.SetSocketPoolForSSLWithProxy(proxy_host, ssl_conn_pool);
|
| +
|
| + MockClientSocketPoolManager* mock_pool_manager =
|
| + new MockClientSocketPoolManager;
|
| + mock_pool_manager->SetSocketPoolForSOCKSProxy(proxy_host, socks_conn_pool);
|
| + mock_pool_manager->SetSocketPoolForSSLWithProxy(proxy_host, ssl_conn_pool);
|
| + peer.SetClientSocketPoolManager(mock_pool_manager);
|
|
|
| scoped_ptr<HttpTransaction> trans(new HttpNetworkTransaction(session));
|
|
|
| @@ -7269,7 +7281,7 @@ TEST_F(HttpNetworkTransactionTest,
|
| transport_params,
|
| LOWEST,
|
| &callback,
|
| - session->transport_socket_pool(),
|
| + session->GetTransportSocketPool(),
|
| BoundNetLog()));
|
| EXPECT_EQ(OK, callback.WaitForResult());
|
|
|
| @@ -7761,7 +7773,10 @@ TEST_F(HttpNetworkTransactionTest, MultiRoundAuth) {
|
| session_deps.host_resolver.get(),
|
| &session_deps.socket_factory,
|
| session_deps.net_log);
|
| - session_peer.SetTransportSocketPool(transport_pool);
|
| + MockClientSocketPoolManager* mock_pool_manager =
|
| + new MockClientSocketPoolManager;
|
| + mock_pool_manager->SetTransportSocketPool(transport_pool);
|
| + session_peer.SetClientSocketPoolManager(mock_pool_manager);
|
|
|
| scoped_ptr<HttpTransaction> trans(new HttpNetworkTransaction(session));
|
| TestOldCompletionCallback callback;
|
| @@ -8524,7 +8539,7 @@ TEST_F(HttpNetworkTransactionTest, PreconnectWithExistingSpdySession) {
|
| EXPECT_EQ(ERR_IO_PENDING,
|
| connection->Init(host_port_pair.ToString(), transport_params,
|
| LOWEST, &callback,
|
| - session->transport_socket_pool(), BoundNetLog()));
|
| + session->GetTransportSocketPool(), BoundNetLog()));
|
| EXPECT_EQ(OK, callback.WaitForResult());
|
| spdy_session->InitializeWithSocket(connection.release(), false, OK);
|
|
|
|
|