| 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
|
|
|