OLD | NEW |
1 // Copyright (c) 2009 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2009 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.h" | 5 #include "net/socket/socks_client_socket.h" |
6 | 6 |
7 #include "net/base/address_list.h" | 7 #include "net/base/address_list.h" |
8 #include "net/base/load_log.h" | 8 #include "net/base/load_log.h" |
9 #include "net/base/load_log_unittest.h" | 9 #include "net/base/load_log_unittest.h" |
10 #include "net/base/mock_host_resolver.h" | 10 #include "net/base/mock_host_resolver.h" |
(...skipping 21 matching lines...) Expand all Loading... |
32 SOCKSClientSocket* BuildMockSocket(MockRead reads[], MockWrite writes[], | 32 SOCKSClientSocket* BuildMockSocket(MockRead reads[], MockWrite writes[], |
33 const std::string& hostname, int port); | 33 const std::string& hostname, int port); |
34 virtual void SetUp(); | 34 virtual void SetUp(); |
35 | 35 |
36 protected: | 36 protected: |
37 scoped_ptr<SOCKSClientSocket> user_sock_; | 37 scoped_ptr<SOCKSClientSocket> user_sock_; |
38 AddressList address_list_; | 38 AddressList address_list_; |
39 ClientSocket* tcp_sock_; | 39 ClientSocket* tcp_sock_; |
40 TestCompletionCallback callback_; | 40 TestCompletionCallback callback_; |
41 scoped_refptr<MockHostResolver> host_resolver_; | 41 scoped_refptr<MockHostResolver> host_resolver_; |
42 scoped_ptr<MockSocket> mock_socket_; | 42 scoped_ptr<SocketDataProvider> data_; |
43 | 43 |
44 private: | 44 private: |
45 DISALLOW_COPY_AND_ASSIGN(SOCKSClientSocketTest); | 45 DISALLOW_COPY_AND_ASSIGN(SOCKSClientSocketTest); |
46 }; | 46 }; |
47 | 47 |
48 SOCKSClientSocketTest::SOCKSClientSocketTest() | 48 SOCKSClientSocketTest::SOCKSClientSocketTest() |
49 : host_resolver_(new MockHostResolver) { | 49 : host_resolver_(new MockHostResolver) { |
50 } | 50 } |
51 | 51 |
52 // Set up platform before every test case | 52 // Set up platform before every test case |
53 void SOCKSClientSocketTest::SetUp() { | 53 void SOCKSClientSocketTest::SetUp() { |
54 PlatformTest::SetUp(); | 54 PlatformTest::SetUp(); |
55 } | 55 } |
56 | 56 |
57 SOCKSClientSocket* SOCKSClientSocketTest::BuildMockSocket( | 57 SOCKSClientSocket* SOCKSClientSocketTest::BuildMockSocket( |
58 MockRead reads[], | 58 MockRead reads[], |
59 MockWrite writes[], | 59 MockWrite writes[], |
60 const std::string& hostname, | 60 const std::string& hostname, |
61 int port) { | 61 int port) { |
62 | 62 |
63 TestCompletionCallback callback; | 63 TestCompletionCallback callback; |
64 mock_socket_.reset(new StaticMockSocket(reads, writes)); | 64 data_.reset(new StaticSocketDataProvider(reads, writes)); |
65 tcp_sock_ = new MockTCPClientSocket(address_list_, mock_socket_.get()); | 65 tcp_sock_ = new MockTCPClientSocket(address_list_, data_.get()); |
66 | 66 |
67 int rv = tcp_sock_->Connect(&callback, NULL); | 67 int rv = tcp_sock_->Connect(&callback, NULL); |
68 EXPECT_EQ(ERR_IO_PENDING, rv); | 68 EXPECT_EQ(ERR_IO_PENDING, rv); |
69 rv = callback.WaitForResult(); | 69 rv = callback.WaitForResult(); |
70 EXPECT_EQ(OK, rv); | 70 EXPECT_EQ(OK, rv); |
71 EXPECT_TRUE(tcp_sock_->IsConnected()); | 71 EXPECT_TRUE(tcp_sock_->IsConnected()); |
72 | 72 |
73 return new SOCKSClientSocket(tcp_sock_, | 73 return new SOCKSClientSocket(tcp_sock_, |
74 HostResolver::RequestInfo(hostname, port), | 74 HostResolver::RequestInfo(hostname, port), |
75 host_resolver_); | 75 host_resolver_); |
(...skipping 228 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
304 *log, 0, LoadLog::TYPE_SOCKS_CONNECT, LoadLog::PHASE_BEGIN)); | 304 *log, 0, LoadLog::TYPE_SOCKS_CONNECT, LoadLog::PHASE_BEGIN)); |
305 rv = callback_.WaitForResult(); | 305 rv = callback_.WaitForResult(); |
306 EXPECT_EQ(OK, rv); | 306 EXPECT_EQ(OK, rv); |
307 EXPECT_TRUE(user_sock_->IsConnected()); | 307 EXPECT_TRUE(user_sock_->IsConnected()); |
308 EXPECT_EQ(SOCKSClientSocket::kSOCKS4a, user_sock_->socks_version_); | 308 EXPECT_EQ(SOCKSClientSocket::kSOCKS4a, user_sock_->socks_version_); |
309 EXPECT_TRUE(LogContains( | 309 EXPECT_TRUE(LogContains( |
310 *log, -1, LoadLog::TYPE_SOCKS_CONNECT, LoadLog::PHASE_END)); | 310 *log, -1, LoadLog::TYPE_SOCKS_CONNECT, LoadLog::PHASE_END)); |
311 } | 311 } |
312 | 312 |
313 } // namespace net | 313 } // namespace net |
OLD | NEW |