Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(443)

Side by Side Diff: net/socket/tcp_server_socket_unittest.cc

Issue 1565303002: Change IPEndpoint::address() to return a net::IPAddress (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Fix Android Created 4 years, 11 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
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/tcp_server_socket.h" 5 #include "net/socket/tcp_server_socket.h"
6 6
7 #include <string> 7 #include <string>
8 #include <vector> 8 #include <vector>
9 9
10 #include "base/compiler_specific.h" 10 #include "base/compiler_specific.h"
(...skipping 74 matching lines...) Expand 10 before | Expand all | Expand 10 after
85 TestCompletionCallback accept_callback; 85 TestCompletionCallback accept_callback;
86 scoped_ptr<StreamSocket> accepted_socket; 86 scoped_ptr<StreamSocket> accepted_socket;
87 int result = socket_.Accept(&accepted_socket, accept_callback.callback()); 87 int result = socket_.Accept(&accepted_socket, accept_callback.callback());
88 if (result == ERR_IO_PENDING) 88 if (result == ERR_IO_PENDING)
89 result = accept_callback.WaitForResult(); 89 result = accept_callback.WaitForResult();
90 ASSERT_EQ(OK, result); 90 ASSERT_EQ(OK, result);
91 91
92 ASSERT_TRUE(accepted_socket.get() != NULL); 92 ASSERT_TRUE(accepted_socket.get() != NULL);
93 93
94 // Both sockets should be on the loopback network interface. 94 // Both sockets should be on the loopback network interface.
95 EXPECT_EQ(GetPeerAddress(accepted_socket.get()).address(), 95 EXPECT_EQ(GetPeerAddress(accepted_socket.get()).address_number(),
96 local_address_.address()); 96 local_address_.address_number());
97 97
98 EXPECT_EQ(OK, connect_callback.WaitForResult()); 98 EXPECT_EQ(OK, connect_callback.WaitForResult());
99 } 99 }
100 100
101 // Test Accept() callback. 101 // Test Accept() callback.
102 TEST_F(TCPServerSocketTest, AcceptAsync) { 102 TEST_F(TCPServerSocketTest, AcceptAsync) {
103 ASSERT_NO_FATAL_FAILURE(SetUpIPv4()); 103 ASSERT_NO_FATAL_FAILURE(SetUpIPv4());
104 104
105 TestCompletionCallback accept_callback; 105 TestCompletionCallback accept_callback;
106 scoped_ptr<StreamSocket> accepted_socket; 106 scoped_ptr<StreamSocket> accepted_socket;
107 107
108 ASSERT_EQ(ERR_IO_PENDING, 108 ASSERT_EQ(ERR_IO_PENDING,
109 socket_.Accept(&accepted_socket, accept_callback.callback())); 109 socket_.Accept(&accepted_socket, accept_callback.callback()));
110 110
111 TestCompletionCallback connect_callback; 111 TestCompletionCallback connect_callback;
112 TCPClientSocket connecting_socket(local_address_list(), 112 TCPClientSocket connecting_socket(local_address_list(),
113 NULL, NetLog::Source()); 113 NULL, NetLog::Source());
114 connecting_socket.Connect(connect_callback.callback()); 114 connecting_socket.Connect(connect_callback.callback());
115 115
116 EXPECT_EQ(OK, connect_callback.WaitForResult()); 116 EXPECT_EQ(OK, connect_callback.WaitForResult());
117 EXPECT_EQ(OK, accept_callback.WaitForResult()); 117 EXPECT_EQ(OK, accept_callback.WaitForResult());
118 118
119 EXPECT_TRUE(accepted_socket != NULL); 119 EXPECT_TRUE(accepted_socket != NULL);
120 120
121 // Both sockets should be on the loopback network interface. 121 // Both sockets should be on the loopback network interface.
122 EXPECT_EQ(GetPeerAddress(accepted_socket.get()).address(), 122 EXPECT_EQ(GetPeerAddress(accepted_socket.get()).address_number(),
123 local_address_.address()); 123 local_address_.address_number());
124 } 124 }
125 125
126 // Accept two connections simultaneously. 126 // Accept two connections simultaneously.
127 TEST_F(TCPServerSocketTest, Accept2Connections) { 127 TEST_F(TCPServerSocketTest, Accept2Connections) {
128 ASSERT_NO_FATAL_FAILURE(SetUpIPv4()); 128 ASSERT_NO_FATAL_FAILURE(SetUpIPv4());
129 129
130 TestCompletionCallback accept_callback; 130 TestCompletionCallback accept_callback;
131 scoped_ptr<StreamSocket> accepted_socket; 131 scoped_ptr<StreamSocket> accepted_socket;
132 132
133 ASSERT_EQ(ERR_IO_PENDING, 133 ASSERT_EQ(ERR_IO_PENDING,
(...skipping 17 matching lines...) Expand all
151 if (result == ERR_IO_PENDING) 151 if (result == ERR_IO_PENDING)
152 result = accept_callback2.WaitForResult(); 152 result = accept_callback2.WaitForResult();
153 ASSERT_EQ(OK, result); 153 ASSERT_EQ(OK, result);
154 154
155 EXPECT_EQ(OK, connect_callback.WaitForResult()); 155 EXPECT_EQ(OK, connect_callback.WaitForResult());
156 156
157 EXPECT_TRUE(accepted_socket != NULL); 157 EXPECT_TRUE(accepted_socket != NULL);
158 EXPECT_TRUE(accepted_socket2 != NULL); 158 EXPECT_TRUE(accepted_socket2 != NULL);
159 EXPECT_NE(accepted_socket.get(), accepted_socket2.get()); 159 EXPECT_NE(accepted_socket.get(), accepted_socket2.get());
160 160
161 EXPECT_EQ(GetPeerAddress(accepted_socket.get()).address(), 161 EXPECT_EQ(GetPeerAddress(accepted_socket.get()).address_number(),
162 local_address_.address()); 162 local_address_.address_number());
163 EXPECT_EQ(GetPeerAddress(accepted_socket2.get()).address(), 163 EXPECT_EQ(GetPeerAddress(accepted_socket2.get()).address_number(),
164 local_address_.address()); 164 local_address_.address_number());
165 } 165 }
166 166
167 TEST_F(TCPServerSocketTest, AcceptIPv6) { 167 TEST_F(TCPServerSocketTest, AcceptIPv6) {
168 bool initialized = false; 168 bool initialized = false;
169 ASSERT_NO_FATAL_FAILURE(SetUpIPv6(&initialized)); 169 ASSERT_NO_FATAL_FAILURE(SetUpIPv6(&initialized));
170 if (!initialized) 170 if (!initialized)
171 return; 171 return;
172 172
173 TestCompletionCallback connect_callback; 173 TestCompletionCallback connect_callback;
174 TCPClientSocket connecting_socket(local_address_list(), 174 TCPClientSocket connecting_socket(local_address_list(),
175 NULL, NetLog::Source()); 175 NULL, NetLog::Source());
176 connecting_socket.Connect(connect_callback.callback()); 176 connecting_socket.Connect(connect_callback.callback());
177 177
178 TestCompletionCallback accept_callback; 178 TestCompletionCallback accept_callback;
179 scoped_ptr<StreamSocket> accepted_socket; 179 scoped_ptr<StreamSocket> accepted_socket;
180 int result = socket_.Accept(&accepted_socket, accept_callback.callback()); 180 int result = socket_.Accept(&accepted_socket, accept_callback.callback());
181 if (result == ERR_IO_PENDING) 181 if (result == ERR_IO_PENDING)
182 result = accept_callback.WaitForResult(); 182 result = accept_callback.WaitForResult();
183 ASSERT_EQ(OK, result); 183 ASSERT_EQ(OK, result);
184 184
185 ASSERT_TRUE(accepted_socket.get() != NULL); 185 ASSERT_TRUE(accepted_socket.get() != NULL);
186 186
187 // Both sockets should be on the loopback network interface. 187 // Both sockets should be on the loopback network interface.
188 EXPECT_EQ(GetPeerAddress(accepted_socket.get()).address(), 188 EXPECT_EQ(GetPeerAddress(accepted_socket.get()).address_number(),
189 local_address_.address()); 189 local_address_.address_number());
190 190
191 EXPECT_EQ(OK, connect_callback.WaitForResult()); 191 EXPECT_EQ(OK, connect_callback.WaitForResult());
192 } 192 }
193 193
194 TEST_F(TCPServerSocketTest, AcceptIO) { 194 TEST_F(TCPServerSocketTest, AcceptIO) {
195 ASSERT_NO_FATAL_FAILURE(SetUpIPv4()); 195 ASSERT_NO_FATAL_FAILURE(SetUpIPv4());
196 196
197 TestCompletionCallback connect_callback; 197 TestCompletionCallback connect_callback;
198 TCPClientSocket connecting_socket(local_address_list(), 198 TCPClientSocket connecting_socket(local_address_list(),
199 NULL, NetLog::Source()); 199 NULL, NetLog::Source());
200 connecting_socket.Connect(connect_callback.callback()); 200 connecting_socket.Connect(connect_callback.callback());
201 201
202 TestCompletionCallback accept_callback; 202 TestCompletionCallback accept_callback;
203 scoped_ptr<StreamSocket> accepted_socket; 203 scoped_ptr<StreamSocket> accepted_socket;
204 int result = socket_.Accept(&accepted_socket, accept_callback.callback()); 204 int result = socket_.Accept(&accepted_socket, accept_callback.callback());
205 ASSERT_EQ(OK, accept_callback.GetResult(result)); 205 ASSERT_EQ(OK, accept_callback.GetResult(result));
206 206
207 ASSERT_TRUE(accepted_socket.get() != NULL); 207 ASSERT_TRUE(accepted_socket.get() != NULL);
208 208
209 // Both sockets should be on the loopback network interface. 209 // Both sockets should be on the loopback network interface.
210 EXPECT_EQ(GetPeerAddress(accepted_socket.get()).address(), 210 EXPECT_EQ(GetPeerAddress(accepted_socket.get()).address_number(),
211 local_address_.address()); 211 local_address_.address_number());
212 212
213 EXPECT_EQ(OK, connect_callback.WaitForResult()); 213 EXPECT_EQ(OK, connect_callback.WaitForResult());
214 214
215 const std::string message("test message"); 215 const std::string message("test message");
216 std::vector<char> buffer(message.size()); 216 std::vector<char> buffer(message.size());
217 217
218 size_t bytes_written = 0; 218 size_t bytes_written = 0;
219 while (bytes_written < message.size()) { 219 while (bytes_written < message.size()) {
220 scoped_refptr<IOBufferWithSize> write_buffer( 220 scoped_refptr<IOBufferWithSize> write_buffer(
221 new IOBufferWithSize(message.size() - bytes_written)); 221 new IOBufferWithSize(message.size() - bytes_written));
(...skipping 22 matching lines...) Expand all
244 bytes_read += read_result; 244 bytes_read += read_result;
245 } 245 }
246 246
247 std::string received_message(buffer.begin(), buffer.end()); 247 std::string received_message(buffer.begin(), buffer.end());
248 ASSERT_EQ(message, received_message); 248 ASSERT_EQ(message, received_message);
249 } 249 }
250 250
251 } // namespace 251 } // namespace
252 252
253 } // namespace net 253 } // namespace net
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698