Index: net/socket/tcp_socket_unittest.cc |
diff --git a/net/socket/tcp_socket_unittest.cc b/net/socket/tcp_socket_unittest.cc |
deleted file mode 100644 |
index 4bfc1384fbd93101293d84fa43fcd92fb3ac8f67..0000000000000000000000000000000000000000 |
--- a/net/socket/tcp_socket_unittest.cc |
+++ /dev/null |
@@ -1,289 +0,0 @@ |
-// Copyright 2013 The Chromium Authors. All rights reserved. |
-// Use of this source code is governed by a BSD-style license that can be |
-// found in the LICENSE file. |
- |
-#include "net/socket/tcp_socket.h" |
- |
-#include <string.h> |
- |
-#include <string> |
-#include <vector> |
- |
-#include "base/basictypes.h" |
-#include "base/memory/ref_counted.h" |
-#include "base/memory/scoped_ptr.h" |
-#include "net/base/address_list.h" |
-#include "net/base/io_buffer.h" |
-#include "net/base/ip_endpoint.h" |
-#include "net/base/net_errors.h" |
-#include "net/base/test_completion_callback.h" |
-#include "net/socket/tcp_client_socket.h" |
-#include "testing/gtest/include/gtest/gtest.h" |
-#include "testing/platform_test.h" |
- |
-namespace net { |
- |
-namespace { |
-const int kListenBacklog = 5; |
- |
-class TCPSocketTest : public PlatformTest { |
- protected: |
- TCPSocketTest() : socket_(NULL, NetLog::Source()) { |
- } |
- |
- void SetUpListenIPv4() { |
- IPEndPoint address; |
- ParseAddress("127.0.0.1", 0, &address); |
- |
- ASSERT_EQ(OK, socket_.Open(ADDRESS_FAMILY_IPV4)); |
- ASSERT_EQ(OK, socket_.Bind(address)); |
- ASSERT_EQ(OK, socket_.Listen(kListenBacklog)); |
- ASSERT_EQ(OK, socket_.GetLocalAddress(&local_address_)); |
- } |
- |
- void SetUpListenIPv6(bool* success) { |
- *success = false; |
- IPEndPoint address; |
- ParseAddress("::1", 0, &address); |
- |
- if (socket_.Open(ADDRESS_FAMILY_IPV6) != OK || |
- socket_.Bind(address) != OK || |
- socket_.Listen(kListenBacklog) != OK) { |
- LOG(ERROR) << "Failed to listen on ::1 - probably because IPv6 is " |
- "disabled. Skipping the test"; |
- return; |
- } |
- ASSERT_EQ(OK, socket_.GetLocalAddress(&local_address_)); |
- *success = true; |
- } |
- |
- void ParseAddress(const std::string& ip_str, |
- uint16 port, |
- IPEndPoint* address) { |
- IPAddressNumber ip_number; |
- bool rv = ParseIPLiteralToNumber(ip_str, &ip_number); |
- if (!rv) |
- return; |
- *address = IPEndPoint(ip_number, port); |
- } |
- |
- void TestAcceptAsync() { |
- TestCompletionCallback accept_callback; |
- scoped_ptr<TCPSocket> accepted_socket; |
- IPEndPoint accepted_address; |
- ASSERT_EQ(ERR_IO_PENDING, |
- socket_.Accept(&accepted_socket, &accepted_address, |
- accept_callback.callback())); |
- |
- TestCompletionCallback connect_callback; |
- TCPClientSocket connecting_socket(local_address_list(), |
- NULL, NetLog::Source()); |
- connecting_socket.Connect(connect_callback.callback()); |
- |
- EXPECT_EQ(OK, connect_callback.WaitForResult()); |
- EXPECT_EQ(OK, accept_callback.WaitForResult()); |
- |
- EXPECT_TRUE(accepted_socket.get()); |
- |
- // Both sockets should be on the loopback network interface. |
- EXPECT_EQ(accepted_address.address(), local_address_.address()); |
- } |
- |
- AddressList local_address_list() const { |
- return AddressList(local_address_); |
- } |
- |
- TCPSocket socket_; |
- IPEndPoint local_address_; |
-}; |
- |
-// Test listening and accepting with a socket bound to an IPv4 address. |
-TEST_F(TCPSocketTest, Accept) { |
- ASSERT_NO_FATAL_FAILURE(SetUpListenIPv4()); |
- |
- TestCompletionCallback connect_callback; |
- // TODO(yzshen): Switch to use TCPSocket when it supports client socket |
- // operations. |
- TCPClientSocket connecting_socket(local_address_list(), |
- NULL, NetLog::Source()); |
- connecting_socket.Connect(connect_callback.callback()); |
- |
- TestCompletionCallback accept_callback; |
- scoped_ptr<TCPSocket> accepted_socket; |
- IPEndPoint accepted_address; |
- int result = socket_.Accept(&accepted_socket, &accepted_address, |
- accept_callback.callback()); |
- if (result == ERR_IO_PENDING) |
- result = accept_callback.WaitForResult(); |
- ASSERT_EQ(OK, result); |
- |
- EXPECT_TRUE(accepted_socket.get()); |
- |
- // Both sockets should be on the loopback network interface. |
- EXPECT_EQ(accepted_address.address(), local_address_.address()); |
- |
- EXPECT_EQ(OK, connect_callback.WaitForResult()); |
-} |
- |
-// Test Accept() callback. |
-TEST_F(TCPSocketTest, AcceptAsync) { |
- ASSERT_NO_FATAL_FAILURE(SetUpListenIPv4()); |
- TestAcceptAsync(); |
-} |
- |
-#if defined(OS_WIN) |
-// Test Accept() for AdoptListenSocket. |
-TEST_F(TCPSocketTest, AcceptForAdoptedListenSocket) { |
- // Create a socket to be used with AdoptListenSocket. |
- SOCKET existing_socket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP); |
- ASSERT_EQ(OK, socket_.AdoptListenSocket(existing_socket)); |
- |
- IPEndPoint address; |
- ParseAddress("127.0.0.1", 0, &address); |
- SockaddrStorage storage; |
- ASSERT_TRUE(address.ToSockAddr(storage.addr, &storage.addr_len)); |
- ASSERT_EQ(0, bind(existing_socket, storage.addr, storage.addr_len)); |
- |
- ASSERT_EQ(OK, socket_.Listen(kListenBacklog)); |
- ASSERT_EQ(OK, socket_.GetLocalAddress(&local_address_)); |
- |
- TestAcceptAsync(); |
-} |
-#endif |
- |
-// Accept two connections simultaneously. |
-TEST_F(TCPSocketTest, Accept2Connections) { |
- ASSERT_NO_FATAL_FAILURE(SetUpListenIPv4()); |
- |
- TestCompletionCallback accept_callback; |
- scoped_ptr<TCPSocket> accepted_socket; |
- IPEndPoint accepted_address; |
- |
- ASSERT_EQ(ERR_IO_PENDING, |
- socket_.Accept(&accepted_socket, &accepted_address, |
- accept_callback.callback())); |
- |
- TestCompletionCallback connect_callback; |
- TCPClientSocket connecting_socket(local_address_list(), |
- NULL, NetLog::Source()); |
- connecting_socket.Connect(connect_callback.callback()); |
- |
- TestCompletionCallback connect_callback2; |
- TCPClientSocket connecting_socket2(local_address_list(), |
- NULL, NetLog::Source()); |
- connecting_socket2.Connect(connect_callback2.callback()); |
- |
- EXPECT_EQ(OK, accept_callback.WaitForResult()); |
- |
- TestCompletionCallback accept_callback2; |
- scoped_ptr<TCPSocket> accepted_socket2; |
- IPEndPoint accepted_address2; |
- |
- int result = socket_.Accept(&accepted_socket2, &accepted_address2, |
- accept_callback2.callback()); |
- if (result == ERR_IO_PENDING) |
- result = accept_callback2.WaitForResult(); |
- ASSERT_EQ(OK, result); |
- |
- EXPECT_EQ(OK, connect_callback.WaitForResult()); |
- EXPECT_EQ(OK, connect_callback2.WaitForResult()); |
- |
- EXPECT_TRUE(accepted_socket.get()); |
- EXPECT_TRUE(accepted_socket2.get()); |
- EXPECT_NE(accepted_socket.get(), accepted_socket2.get()); |
- |
- EXPECT_EQ(accepted_address.address(), local_address_.address()); |
- EXPECT_EQ(accepted_address2.address(), local_address_.address()); |
-} |
- |
-// Test listening and accepting with a socket bound to an IPv6 address. |
-TEST_F(TCPSocketTest, AcceptIPv6) { |
- bool initialized = false; |
- ASSERT_NO_FATAL_FAILURE(SetUpListenIPv6(&initialized)); |
- if (!initialized) |
- return; |
- |
- TestCompletionCallback connect_callback; |
- TCPClientSocket connecting_socket(local_address_list(), |
- NULL, NetLog::Source()); |
- connecting_socket.Connect(connect_callback.callback()); |
- |
- TestCompletionCallback accept_callback; |
- scoped_ptr<TCPSocket> accepted_socket; |
- IPEndPoint accepted_address; |
- int result = socket_.Accept(&accepted_socket, &accepted_address, |
- accept_callback.callback()); |
- if (result == ERR_IO_PENDING) |
- result = accept_callback.WaitForResult(); |
- ASSERT_EQ(OK, result); |
- |
- EXPECT_TRUE(accepted_socket.get()); |
- |
- // Both sockets should be on the loopback network interface. |
- EXPECT_EQ(accepted_address.address(), local_address_.address()); |
- |
- EXPECT_EQ(OK, connect_callback.WaitForResult()); |
-} |
- |
-TEST_F(TCPSocketTest, ReadWrite) { |
- ASSERT_NO_FATAL_FAILURE(SetUpListenIPv4()); |
- |
- TestCompletionCallback connect_callback; |
- TCPSocket connecting_socket(NULL, NetLog::Source()); |
- int result = connecting_socket.Open(ADDRESS_FAMILY_IPV4); |
- ASSERT_EQ(OK, result); |
- connecting_socket.Connect(local_address_, connect_callback.callback()); |
- |
- TestCompletionCallback accept_callback; |
- scoped_ptr<TCPSocket> accepted_socket; |
- IPEndPoint accepted_address; |
- result = socket_.Accept(&accepted_socket, &accepted_address, |
- accept_callback.callback()); |
- ASSERT_EQ(OK, accept_callback.GetResult(result)); |
- |
- ASSERT_TRUE(accepted_socket.get()); |
- |
- // Both sockets should be on the loopback network interface. |
- EXPECT_EQ(accepted_address.address(), local_address_.address()); |
- |
- EXPECT_EQ(OK, connect_callback.WaitForResult()); |
- |
- const std::string message("test message"); |
- std::vector<char> buffer(message.size()); |
- |
- size_t bytes_written = 0; |
- while (bytes_written < message.size()) { |
- scoped_refptr<IOBufferWithSize> write_buffer( |
- new IOBufferWithSize(message.size() - bytes_written)); |
- memmove(write_buffer->data(), message.data() + bytes_written, |
- message.size() - bytes_written); |
- |
- TestCompletionCallback write_callback; |
- int write_result = accepted_socket->Write( |
- write_buffer.get(), write_buffer->size(), write_callback.callback()); |
- write_result = write_callback.GetResult(write_result); |
- ASSERT_TRUE(write_result >= 0); |
- bytes_written += write_result; |
- ASSERT_TRUE(bytes_written <= message.size()); |
- } |
- |
- size_t bytes_read = 0; |
- while (bytes_read < message.size()) { |
- scoped_refptr<IOBufferWithSize> read_buffer( |
- new IOBufferWithSize(message.size() - bytes_read)); |
- TestCompletionCallback read_callback; |
- int read_result = connecting_socket.Read( |
- read_buffer.get(), read_buffer->size(), read_callback.callback()); |
- read_result = read_callback.GetResult(read_result); |
- ASSERT_TRUE(read_result >= 0); |
- ASSERT_TRUE(bytes_read + read_result <= message.size()); |
- memmove(&buffer[bytes_read], read_buffer->data(), read_result); |
- bytes_read += read_result; |
- } |
- |
- std::string received_message(buffer.begin(), buffer.end()); |
- ASSERT_EQ(message, received_message); |
-} |
- |
-} // namespace |
-} // namespace net |