Index: net/socket/socks_client_socket_unittest.cc |
diff --git a/net/socket/socks_client_socket_unittest.cc b/net/socket/socks_client_socket_unittest.cc |
index 764c71bcf33d648941a93170867598e61fd5c0a2..e287783955403c26408ae7fc20f8059701455da9 100644 |
--- a/net/socket/socks_client_socket_unittest.cc |
+++ b/net/socket/socks_client_socket_unittest.cc |
@@ -20,9 +20,14 @@ |
#include "net/socket/client_socket_factory.h" |
#include "net/socket/socket_test_util.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::IsError; |
+using net::test::IsOk; |
+ |
//----------------------------------------------------------------------------- |
namespace net { |
@@ -80,9 +85,9 @@ std::unique_ptr<SOCKSClientSocket> SOCKSClientSocketTest::BuildMockSocket( |
tcp_sock_ = new MockTCPClientSocket(address_list_, net_log, data_.get()); |
int rv = tcp_sock_->Connect(callback.callback()); |
- EXPECT_EQ(ERR_IO_PENDING, rv); |
+ EXPECT_THAT(rv, IsError(ERR_IO_PENDING)); |
rv = callback.WaitForResult(); |
- EXPECT_EQ(OK, rv); |
+ EXPECT_THAT(rv, IsOk()); |
EXPECT_TRUE(tcp_sock_->IsConnected()); |
std::unique_ptr<ClientSocketHandle> connection(new ClientSocketHandle); |
@@ -163,7 +168,7 @@ TEST_F(SOCKSClientSocketTest, CompleteHandshake) { |
EXPECT_FALSE(user_sock_->IsConnected()); |
int rv = user_sock_->Connect(callback_.callback()); |
- EXPECT_EQ(ERR_IO_PENDING, rv); |
+ EXPECT_THAT(rv, IsError(ERR_IO_PENDING)); |
TestNetLogEntry::List entries; |
log.GetEntries(&entries); |
@@ -172,7 +177,7 @@ TEST_F(SOCKSClientSocketTest, CompleteHandshake) { |
EXPECT_FALSE(user_sock_->IsConnected()); |
rv = callback_.WaitForResult(); |
- EXPECT_EQ(OK, rv); |
+ EXPECT_THAT(rv, IsOk()); |
EXPECT_TRUE(user_sock_->IsConnected()); |
log.GetEntries(&entries); |
EXPECT_TRUE(LogContainsEndEvent( |
@@ -182,14 +187,14 @@ TEST_F(SOCKSClientSocketTest, CompleteHandshake) { |
memcpy(buffer->data(), payload_write.data(), payload_write.size()); |
rv = user_sock_->Write( |
buffer.get(), payload_write.size(), callback_.callback()); |
- EXPECT_EQ(ERR_IO_PENDING, rv); |
+ EXPECT_THAT(rv, IsError(ERR_IO_PENDING)); |
rv = callback_.WaitForResult(); |
EXPECT_EQ(static_cast<int>(payload_write.size()), rv); |
buffer = new IOBuffer(payload_read.size()); |
rv = |
user_sock_->Read(buffer.get(), payload_read.size(), callback_.callback()); |
- EXPECT_EQ(ERR_IO_PENDING, rv); |
+ EXPECT_THAT(rv, IsError(ERR_IO_PENDING)); |
rv = callback_.WaitForResult(); |
EXPECT_EQ(static_cast<int>(payload_read.size()), rv); |
EXPECT_EQ(payload_read, std::string(buffer->data(), payload_read.size())); |
@@ -235,7 +240,7 @@ TEST_F(SOCKSClientSocketTest, HandshakeFailures) { |
&log); |
int rv = user_sock_->Connect(callback_.callback()); |
- EXPECT_EQ(ERR_IO_PENDING, rv); |
+ EXPECT_THAT(rv, IsError(ERR_IO_PENDING)); |
TestNetLogEntry::List entries; |
log.GetEntries(&entries); |
@@ -272,14 +277,14 @@ TEST_F(SOCKSClientSocketTest, PartialServerReads) { |
&log); |
int rv = user_sock_->Connect(callback_.callback()); |
- EXPECT_EQ(ERR_IO_PENDING, rv); |
+ EXPECT_THAT(rv, IsError(ERR_IO_PENDING)); |
TestNetLogEntry::List entries; |
log.GetEntries(&entries); |
EXPECT_TRUE(LogContainsBeginEvent( |
entries, 0, NetLog::TYPE_SOCKS_CONNECT)); |
rv = callback_.WaitForResult(); |
- EXPECT_EQ(OK, rv); |
+ EXPECT_THAT(rv, IsOk()); |
EXPECT_TRUE(user_sock_->IsConnected()); |
log.GetEntries(&entries); |
EXPECT_TRUE(LogContainsEndEvent( |
@@ -310,14 +315,14 @@ TEST_F(SOCKSClientSocketTest, PartialClientWrites) { |
&log); |
int rv = user_sock_->Connect(callback_.callback()); |
- EXPECT_EQ(ERR_IO_PENDING, rv); |
+ EXPECT_THAT(rv, IsError(ERR_IO_PENDING)); |
TestNetLogEntry::List entries; |
log.GetEntries(&entries); |
EXPECT_TRUE(LogContainsBeginEvent( |
entries, 0, NetLog::TYPE_SOCKS_CONNECT)); |
rv = callback_.WaitForResult(); |
- EXPECT_EQ(OK, rv); |
+ EXPECT_THAT(rv, IsOk()); |
EXPECT_TRUE(user_sock_->IsConnected()); |
log.GetEntries(&entries); |
EXPECT_TRUE(LogContainsEndEvent( |
@@ -342,14 +347,14 @@ TEST_F(SOCKSClientSocketTest, FailedSocketRead) { |
&log); |
int rv = user_sock_->Connect(callback_.callback()); |
- EXPECT_EQ(ERR_IO_PENDING, rv); |
+ EXPECT_THAT(rv, IsError(ERR_IO_PENDING)); |
TestNetLogEntry::List entries; |
log.GetEntries(&entries); |
EXPECT_TRUE(LogContainsBeginEvent( |
entries, 0, NetLog::TYPE_SOCKS_CONNECT)); |
rv = callback_.WaitForResult(); |
- EXPECT_EQ(ERR_CONNECTION_CLOSED, rv); |
+ EXPECT_THAT(rv, IsError(ERR_CONNECTION_CLOSED)); |
EXPECT_FALSE(user_sock_->IsConnected()); |
log.GetEntries(&entries); |
EXPECT_TRUE(LogContainsEndEvent( |
@@ -372,14 +377,14 @@ TEST_F(SOCKSClientSocketTest, FailedDNS) { |
&log); |
int rv = user_sock_->Connect(callback_.callback()); |
- EXPECT_EQ(ERR_IO_PENDING, rv); |
+ EXPECT_THAT(rv, IsError(ERR_IO_PENDING)); |
TestNetLogEntry::List entries; |
log.GetEntries(&entries); |
EXPECT_TRUE(LogContainsBeginEvent( |
entries, 0, NetLog::TYPE_SOCKS_CONNECT)); |
rv = callback_.WaitForResult(); |
- EXPECT_EQ(ERR_NAME_NOT_RESOLVED, rv); |
+ EXPECT_THAT(rv, IsError(ERR_NAME_NOT_RESOLVED)); |
EXPECT_FALSE(user_sock_->IsConnected()); |
log.GetEntries(&entries); |
EXPECT_TRUE(LogContainsEndEvent( |
@@ -404,7 +409,7 @@ TEST_F(SOCKSClientSocketTest, DisconnectWhileHostResolveInProgress) { |
// Start connecting (will get stuck waiting for the host to resolve). |
int rv = user_sock_->Connect(callback_.callback()); |
- EXPECT_EQ(ERR_IO_PENDING, rv); |
+ EXPECT_THAT(rv, IsError(ERR_IO_PENDING)); |
EXPECT_FALSE(user_sock_->IsConnected()); |
EXPECT_FALSE(user_sock_->IsConnectedAndIdle()); |