OLD | NEW |
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 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/socks_client_socket_pool.h" | 5 #include "net/socket/socks_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/time.h" | 9 #include "base/time.h" |
10 #include "net/base/mock_host_resolver.h" | 10 #include "net/base/mock_host_resolver.h" |
(...skipping 83 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
94 SOCKSClientSocketPool pool_; | 94 SOCKSClientSocketPool pool_; |
95 ClientSocketPoolTest test_base_; | 95 ClientSocketPoolTest test_base_; |
96 }; | 96 }; |
97 | 97 |
98 TEST_F(SOCKSClientSocketPoolTest, Simple) { | 98 TEST_F(SOCKSClientSocketPoolTest, Simple) { |
99 SOCKS5MockData data(false); | 99 SOCKS5MockData data(false); |
100 data.data_provider()->set_connect_data(MockConnect(false, 0)); | 100 data.data_provider()->set_connect_data(MockConnect(false, 0)); |
101 transport_client_socket_factory_.AddSocketDataProvider(data.data_provider()); | 101 transport_client_socket_factory_.AddSocketDataProvider(data.data_provider()); |
102 | 102 |
103 ClientSocketHandle handle; | 103 ClientSocketHandle handle; |
104 int rv = handle.Init("a", ignored_socket_params_, LOW, NULL, &pool_, | 104 int rv = handle.Init("a", ignored_socket_params_, LOW, CompletionCallback(), |
105 BoundNetLog()); | 105 &pool_, BoundNetLog()); |
106 EXPECT_EQ(OK, rv); | 106 EXPECT_EQ(OK, rv); |
107 EXPECT_TRUE(handle.is_initialized()); | 107 EXPECT_TRUE(handle.is_initialized()); |
108 EXPECT_TRUE(handle.socket()); | 108 EXPECT_TRUE(handle.socket()); |
109 } | 109 } |
110 | 110 |
111 TEST_F(SOCKSClientSocketPoolTest, Async) { | 111 TEST_F(SOCKSClientSocketPoolTest, Async) { |
112 SOCKS5MockData data(true); | 112 SOCKS5MockData data(true); |
113 transport_client_socket_factory_.AddSocketDataProvider(data.data_provider()); | 113 transport_client_socket_factory_.AddSocketDataProvider(data.data_provider()); |
114 | 114 |
115 TestOldCompletionCallback callback; | 115 TestCompletionCallback callback; |
116 ClientSocketHandle handle; | 116 ClientSocketHandle handle; |
117 int rv = handle.Init("a", ignored_socket_params_, LOW, &callback, &pool_, | 117 int rv = handle.Init("a", ignored_socket_params_, LOW, callback.callback(), |
118 BoundNetLog()); | 118 &pool_, BoundNetLog()); |
119 EXPECT_EQ(ERR_IO_PENDING, rv); | 119 EXPECT_EQ(ERR_IO_PENDING, rv); |
120 EXPECT_FALSE(handle.is_initialized()); | 120 EXPECT_FALSE(handle.is_initialized()); |
121 EXPECT_FALSE(handle.socket()); | 121 EXPECT_FALSE(handle.socket()); |
122 | 122 |
123 EXPECT_EQ(OK, callback.WaitForResult()); | 123 EXPECT_EQ(OK, callback.WaitForResult()); |
124 EXPECT_TRUE(handle.is_initialized()); | 124 EXPECT_TRUE(handle.is_initialized()); |
125 EXPECT_TRUE(handle.socket()); | 125 EXPECT_TRUE(handle.socket()); |
126 } | 126 } |
127 | 127 |
128 TEST_F(SOCKSClientSocketPoolTest, TransportConnectError) { | 128 TEST_F(SOCKSClientSocketPoolTest, TransportConnectError) { |
129 scoped_ptr<SocketDataProvider> socket_data(new StaticSocketDataProvider()); | 129 scoped_ptr<SocketDataProvider> socket_data(new StaticSocketDataProvider()); |
130 socket_data->set_connect_data(MockConnect(false, ERR_CONNECTION_REFUSED)); | 130 socket_data->set_connect_data(MockConnect(false, ERR_CONNECTION_REFUSED)); |
131 transport_client_socket_factory_.AddSocketDataProvider(socket_data.get()); | 131 transport_client_socket_factory_.AddSocketDataProvider(socket_data.get()); |
132 | 132 |
133 ClientSocketHandle handle; | 133 ClientSocketHandle handle; |
134 int rv = handle.Init("a", ignored_socket_params_, LOW, NULL, &pool_, | 134 int rv = handle.Init("a", ignored_socket_params_, LOW, CompletionCallback(), |
135 BoundNetLog()); | 135 &pool_, BoundNetLog()); |
136 EXPECT_EQ(ERR_PROXY_CONNECTION_FAILED, rv); | 136 EXPECT_EQ(ERR_PROXY_CONNECTION_FAILED, rv); |
137 EXPECT_FALSE(handle.is_initialized()); | 137 EXPECT_FALSE(handle.is_initialized()); |
138 EXPECT_FALSE(handle.socket()); | 138 EXPECT_FALSE(handle.socket()); |
139 } | 139 } |
140 | 140 |
141 TEST_F(SOCKSClientSocketPoolTest, AsyncTransportConnectError) { | 141 TEST_F(SOCKSClientSocketPoolTest, AsyncTransportConnectError) { |
142 scoped_ptr<SocketDataProvider> socket_data(new StaticSocketDataProvider()); | 142 scoped_ptr<SocketDataProvider> socket_data(new StaticSocketDataProvider()); |
143 socket_data->set_connect_data(MockConnect(true, ERR_CONNECTION_REFUSED)); | 143 socket_data->set_connect_data(MockConnect(true, ERR_CONNECTION_REFUSED)); |
144 transport_client_socket_factory_.AddSocketDataProvider(socket_data.get()); | 144 transport_client_socket_factory_.AddSocketDataProvider(socket_data.get()); |
145 | 145 |
146 TestOldCompletionCallback callback; | 146 TestCompletionCallback callback; |
147 ClientSocketHandle handle; | 147 ClientSocketHandle handle; |
148 int rv = handle.Init("a", ignored_socket_params_, LOW, &callback, &pool_, | 148 int rv = handle.Init("a", ignored_socket_params_, LOW, callback.callback(), |
149 BoundNetLog()); | 149 &pool_, BoundNetLog()); |
150 EXPECT_EQ(ERR_IO_PENDING, rv); | 150 EXPECT_EQ(ERR_IO_PENDING, rv); |
151 EXPECT_FALSE(handle.is_initialized()); | 151 EXPECT_FALSE(handle.is_initialized()); |
152 EXPECT_FALSE(handle.socket()); | 152 EXPECT_FALSE(handle.socket()); |
153 | 153 |
154 EXPECT_EQ(ERR_PROXY_CONNECTION_FAILED, callback.WaitForResult()); | 154 EXPECT_EQ(ERR_PROXY_CONNECTION_FAILED, callback.WaitForResult()); |
155 EXPECT_FALSE(handle.is_initialized()); | 155 EXPECT_FALSE(handle.is_initialized()); |
156 EXPECT_FALSE(handle.socket()); | 156 EXPECT_FALSE(handle.socket()); |
157 } | 157 } |
158 | 158 |
159 TEST_F(SOCKSClientSocketPoolTest, SOCKSConnectError) { | 159 TEST_F(SOCKSClientSocketPoolTest, SOCKSConnectError) { |
160 MockRead failed_read[] = { | 160 MockRead failed_read[] = { |
161 MockRead(false, 0), | 161 MockRead(false, 0), |
162 }; | 162 }; |
163 scoped_ptr<SocketDataProvider> socket_data(new StaticSocketDataProvider( | 163 scoped_ptr<SocketDataProvider> socket_data(new StaticSocketDataProvider( |
164 failed_read, arraysize(failed_read), NULL, 0)); | 164 failed_read, arraysize(failed_read), NULL, 0)); |
165 socket_data->set_connect_data(MockConnect(false, 0)); | 165 socket_data->set_connect_data(MockConnect(false, 0)); |
166 transport_client_socket_factory_.AddSocketDataProvider(socket_data.get()); | 166 transport_client_socket_factory_.AddSocketDataProvider(socket_data.get()); |
167 | 167 |
168 ClientSocketHandle handle; | 168 ClientSocketHandle handle; |
169 EXPECT_EQ(0, transport_socket_pool_.release_count()); | 169 EXPECT_EQ(0, transport_socket_pool_.release_count()); |
170 int rv = handle.Init("a", ignored_socket_params_, LOW, NULL, &pool_, | 170 int rv = handle.Init("a", ignored_socket_params_, LOW, CompletionCallback(), |
171 BoundNetLog()); | 171 &pool_, BoundNetLog()); |
172 EXPECT_EQ(ERR_SOCKS_CONNECTION_FAILED, rv); | 172 EXPECT_EQ(ERR_SOCKS_CONNECTION_FAILED, rv); |
173 EXPECT_FALSE(handle.is_initialized()); | 173 EXPECT_FALSE(handle.is_initialized()); |
174 EXPECT_FALSE(handle.socket()); | 174 EXPECT_FALSE(handle.socket()); |
175 EXPECT_EQ(1, transport_socket_pool_.release_count()); | 175 EXPECT_EQ(1, transport_socket_pool_.release_count()); |
176 } | 176 } |
177 | 177 |
178 TEST_F(SOCKSClientSocketPoolTest, AsyncSOCKSConnectError) { | 178 TEST_F(SOCKSClientSocketPoolTest, AsyncSOCKSConnectError) { |
179 MockRead failed_read[] = { | 179 MockRead failed_read[] = { |
180 MockRead(true, 0), | 180 MockRead(true, 0), |
181 }; | 181 }; |
182 scoped_ptr<SocketDataProvider> socket_data(new StaticSocketDataProvider( | 182 scoped_ptr<SocketDataProvider> socket_data(new StaticSocketDataProvider( |
183 failed_read, arraysize(failed_read), NULL, 0)); | 183 failed_read, arraysize(failed_read), NULL, 0)); |
184 socket_data->set_connect_data(MockConnect(false, 0)); | 184 socket_data->set_connect_data(MockConnect(false, 0)); |
185 transport_client_socket_factory_.AddSocketDataProvider(socket_data.get()); | 185 transport_client_socket_factory_.AddSocketDataProvider(socket_data.get()); |
186 | 186 |
187 TestOldCompletionCallback callback; | 187 TestCompletionCallback callback; |
188 ClientSocketHandle handle; | 188 ClientSocketHandle handle; |
189 EXPECT_EQ(0, transport_socket_pool_.release_count()); | 189 EXPECT_EQ(0, transport_socket_pool_.release_count()); |
190 int rv = handle.Init("a", ignored_socket_params_, LOW, &callback, &pool_, | 190 int rv = handle.Init("a", ignored_socket_params_, LOW, callback.callback(), |
191 BoundNetLog()); | 191 &pool_, BoundNetLog()); |
192 EXPECT_EQ(ERR_IO_PENDING, rv); | 192 EXPECT_EQ(ERR_IO_PENDING, rv); |
193 EXPECT_FALSE(handle.is_initialized()); | 193 EXPECT_FALSE(handle.is_initialized()); |
194 EXPECT_FALSE(handle.socket()); | 194 EXPECT_FALSE(handle.socket()); |
195 | 195 |
196 EXPECT_EQ(ERR_SOCKS_CONNECTION_FAILED, callback.WaitForResult()); | 196 EXPECT_EQ(ERR_SOCKS_CONNECTION_FAILED, callback.WaitForResult()); |
197 EXPECT_FALSE(handle.is_initialized()); | 197 EXPECT_FALSE(handle.is_initialized()); |
198 EXPECT_FALSE(handle.socket()); | 198 EXPECT_FALSE(handle.socket()); |
199 EXPECT_EQ(1, transport_socket_pool_.release_count()); | 199 EXPECT_EQ(1, transport_socket_pool_.release_count()); |
200 } | 200 } |
201 | 201 |
(...skipping 64 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
266 | 266 |
267 (*requests())[0]->handle()->Reset(); | 267 (*requests())[0]->handle()->Reset(); |
268 (*requests())[1]->handle()->Reset(); | 268 (*requests())[1]->handle()->Reset(); |
269 } | 269 } |
270 | 270 |
271 // It would be nice to also test the timeouts in SOCKSClientSocketPool. | 271 // It would be nice to also test the timeouts in SOCKSClientSocketPool. |
272 | 272 |
273 } // namespace | 273 } // namespace |
274 | 274 |
275 } // namespace net | 275 } // namespace net |
OLD | NEW |