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

Unified Diff: net/socket/tcp_socket_unittest.cc

Issue 2109503009: Refactor net tests to use GMock matchers for checking net::Error results (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Revert changes to contents.txt files Created 4 years, 6 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 side-by-side diff with in-line comments
Download patch
« no previous file with comments | « net/socket/tcp_server_socket_unittest.cc ('k') | net/socket/transport_client_socket_pool_unittest.cc » ('j') | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
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());
« no previous file with comments | « net/socket/tcp_server_socket_unittest.cc ('k') | net/socket/transport_client_socket_pool_unittest.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698