Index: net/socket/tcp_socket_unittest.cc |
diff --git a/net/socket/tcp_socket_unittest.cc b/net/socket/tcp_socket_unittest.cc |
index b24d104fd69ab0e791938309ba3c0220e431288a..0ac3266742124e17bc9646c365e18875818b7ec1 100644 |
--- a/net/socket/tcp_socket_unittest.cc |
+++ b/net/socket/tcp_socket_unittest.cc |
@@ -17,15 +17,18 @@ |
#include "net/base/address_list.h" |
#include "net/base/io_buffer.h" |
#include "net/base/ip_endpoint.h" |
-#include "net/base/ip_endpoint.h" |
#include "net/base/net_errors.h" |
#include "net/base/sockaddr_storage.h" |
#include "net/base/test_completion_callback.h" |
#include "net/socket/socket_performance_watcher.h" |
#include "net/socket/tcp_client_socket.h" |
+#include "net/test/gtest_util.h" |
+#include "testing/gmock/include/gmock/gmock.h" |
#include "testing/gtest/include/gtest/gtest.h" |
#include "testing/platform_test.h" |
+using net::test::IsOk; |
+ |
namespace net { |
namespace { |
@@ -67,10 +70,11 @@ class TCPSocketTest : public PlatformTest { |
TCPSocketTest() : socket_(NULL, NULL, NetLog::Source()) {} |
void SetUpListenIPv4() { |
- ASSERT_EQ(OK, socket_.Open(ADDRESS_FAMILY_IPV4)); |
- ASSERT_EQ(OK, socket_.Bind(IPEndPoint(IPAddress::IPv4Localhost(), 0))); |
- ASSERT_EQ(OK, socket_.Listen(kListenBacklog)); |
- ASSERT_EQ(OK, socket_.GetLocalAddress(&local_address_)); |
+ ASSERT_THAT(socket_.Open(ADDRESS_FAMILY_IPV4), IsOk()); |
+ ASSERT_THAT(socket_.Bind(IPEndPoint(IPAddress::IPv4Localhost(), 0)), |
+ IsOk()); |
+ ASSERT_THAT(socket_.Listen(kListenBacklog), IsOk()); |
+ ASSERT_THAT(socket_.GetLocalAddress(&local_address_), IsOk()); |
} |
void SetUpListenIPv6(bool* success) { |
@@ -83,7 +87,7 @@ class TCPSocketTest : public PlatformTest { |
"disabled. Skipping the test"; |
return; |
} |
- ASSERT_EQ(OK, socket_.GetLocalAddress(&local_address_)); |
+ ASSERT_THAT(socket_.GetLocalAddress(&local_address_), IsOk()); |
*success = true; |
} |
@@ -100,8 +104,8 @@ class TCPSocketTest : public PlatformTest { |
NetLog::Source()); |
connecting_socket.Connect(connect_callback.callback()); |
- EXPECT_EQ(OK, connect_callback.WaitForResult()); |
- EXPECT_EQ(OK, accept_callback.WaitForResult()); |
+ EXPECT_THAT(connect_callback.WaitForResult(), IsOk()); |
+ EXPECT_THAT(accept_callback.WaitForResult(), IsOk()); |
EXPECT_TRUE(accepted_socket.get()); |
@@ -142,7 +146,7 @@ class TCPSocketTest : public PlatformTest { |
connecting_socket.SetTickClockForTesting(std::move(tick_clock)); |
int result = connecting_socket.Open(ADDRESS_FAMILY_IPV4); |
- ASSERT_EQ(OK, result); |
+ ASSERT_THAT(result, IsOk()); |
connecting_socket.Connect(local_address_, connect_callback.callback()); |
TestCompletionCallback accept_callback; |
@@ -150,14 +154,14 @@ class TCPSocketTest : public PlatformTest { |
IPEndPoint accepted_address; |
result = socket_.Accept(&accepted_socket, &accepted_address, |
accept_callback.callback()); |
- ASSERT_EQ(OK, accept_callback.GetResult(result)); |
+ ASSERT_THAT(accept_callback.GetResult(result), IsOk()); |
ASSERT_TRUE(accepted_socket.get()); |
// Both sockets should be on the loopback network interface. |
EXPECT_EQ(accepted_address.address(), local_address_.address()); |
- ASSERT_EQ(OK, connect_callback.WaitForResult()); |
+ ASSERT_THAT(connect_callback.WaitForResult(), IsOk()); |
for (size_t i = 0; i < num_messages; ++i) { |
tick_clock_ptr->Advance(advance_ticks); |
@@ -216,14 +220,14 @@ TEST_F(TCPSocketTest, Accept) { |
accept_callback.callback()); |
if (result == ERR_IO_PENDING) |
result = accept_callback.WaitForResult(); |
- ASSERT_EQ(OK, result); |
+ ASSERT_THAT(result, IsOk()); |
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()); |
+ EXPECT_THAT(connect_callback.WaitForResult(), IsOk()); |
} |
// Test Accept() callback. |
@@ -237,15 +241,15 @@ TEST_F(TCPSocketTest, AcceptAsync) { |
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)); |
+ ASSERT_THAT(socket_.AdoptListenSocket(existing_socket), IsOk()); |
IPEndPoint address(IPAddress::IPv4Localhost(), 0); |
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_)); |
+ ASSERT_THAT(socket_.Listen(kListenBacklog), IsOk()); |
+ ASSERT_THAT(socket_.GetLocalAddress(&local_address_), IsOk()); |
TestAcceptAsync(); |
} |
@@ -273,7 +277,7 @@ TEST_F(TCPSocketTest, Accept2Connections) { |
NetLog::Source()); |
connecting_socket2.Connect(connect_callback2.callback()); |
- EXPECT_EQ(OK, accept_callback.WaitForResult()); |
+ EXPECT_THAT(accept_callback.WaitForResult(), IsOk()); |
TestCompletionCallback accept_callback2; |
std::unique_ptr<TCPSocket> accepted_socket2; |
@@ -283,10 +287,10 @@ TEST_F(TCPSocketTest, Accept2Connections) { |
accept_callback2.callback()); |
if (result == ERR_IO_PENDING) |
result = accept_callback2.WaitForResult(); |
- ASSERT_EQ(OK, result); |
+ ASSERT_THAT(result, IsOk()); |
- EXPECT_EQ(OK, connect_callback.WaitForResult()); |
- EXPECT_EQ(OK, connect_callback2.WaitForResult()); |
+ EXPECT_THAT(connect_callback.WaitForResult(), IsOk()); |
+ EXPECT_THAT(connect_callback2.WaitForResult(), IsOk()); |
EXPECT_TRUE(accepted_socket.get()); |
EXPECT_TRUE(accepted_socket2.get()); |
@@ -315,14 +319,14 @@ TEST_F(TCPSocketTest, AcceptIPv6) { |
accept_callback.callback()); |
if (result == ERR_IO_PENDING) |
result = accept_callback.WaitForResult(); |
- ASSERT_EQ(OK, result); |
+ ASSERT_THAT(result, IsOk()); |
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()); |
+ EXPECT_THAT(connect_callback.WaitForResult(), IsOk()); |
} |
TEST_F(TCPSocketTest, ReadWrite) { |
@@ -331,7 +335,7 @@ TEST_F(TCPSocketTest, ReadWrite) { |
TestCompletionCallback connect_callback; |
TCPSocket connecting_socket(NULL, NULL, NetLog::Source()); |
int result = connecting_socket.Open(ADDRESS_FAMILY_IPV4); |
- ASSERT_EQ(OK, result); |
+ ASSERT_THAT(result, IsOk()); |
connecting_socket.Connect(local_address_, connect_callback.callback()); |
TestCompletionCallback accept_callback; |
@@ -339,14 +343,14 @@ TEST_F(TCPSocketTest, ReadWrite) { |
IPEndPoint accepted_address; |
result = socket_.Accept(&accepted_socket, &accepted_address, |
accept_callback.callback()); |
- ASSERT_EQ(OK, accept_callback.GetResult(result)); |
+ ASSERT_THAT(accept_callback.GetResult(result), IsOk()); |
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()); |
+ EXPECT_THAT(connect_callback.WaitForResult(), IsOk()); |
const std::string message("test message"); |
std::vector<char> buffer(message.size()); |