| Index: net/socket/socks5_client_socket_unittest.cc
|
| diff --git a/net/socket/socks5_client_socket_unittest.cc b/net/socket/socks5_client_socket_unittest.cc
|
| index a32f449fac36d41868fb0292ec9d1cf0f87d00bc..97bfeca1f9f0edd17499df1b90dddc9ba13776ce 100644
|
| --- a/net/socket/socks5_client_socket_unittest.cc
|
| +++ b/net/socket/socks5_client_socket_unittest.cc
|
| @@ -22,9 +22,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 {
|
| @@ -81,9 +86,9 @@ void SOCKS5ClientSocketTest::SetUp() {
|
| callback.callback(),
|
| NULL,
|
| BoundNetLog());
|
| - ASSERT_EQ(ERR_IO_PENDING, rv);
|
| + ASSERT_THAT(rv, IsError(ERR_IO_PENDING));
|
| rv = callback.WaitForResult();
|
| - ASSERT_EQ(OK, rv);
|
| + ASSERT_THAT(rv, IsOk());
|
| }
|
|
|
| std::unique_ptr<SOCKS5ClientSocket> SOCKS5ClientSocketTest::BuildMockSocket(
|
| @@ -100,9 +105,9 @@ std::unique_ptr<SOCKS5ClientSocket> SOCKS5ClientSocketTest::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);
|
| @@ -148,7 +153,7 @@ TEST_F(SOCKS5ClientSocketTest, 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));
|
| EXPECT_FALSE(user_sock_->IsConnected());
|
|
|
| TestNetLogEntry::List net_log_entries;
|
| @@ -158,7 +163,7 @@ TEST_F(SOCKS5ClientSocketTest, CompleteHandshake) {
|
|
|
| rv = callback_.WaitForResult();
|
|
|
| - EXPECT_EQ(OK, rv);
|
| + EXPECT_THAT(rv, IsOk());
|
| EXPECT_TRUE(user_sock_->IsConnected());
|
|
|
| net_log_.GetEntries(&net_log_entries);
|
| @@ -169,14 +174,14 @@ TEST_F(SOCKS5ClientSocketTest, 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()));
|
| @@ -216,7 +221,7 @@ TEST_F(SOCKS5ClientSocketTest, ConnectAndDisconnectTwice) {
|
| hostname, 80, NULL);
|
|
|
| int rv = user_sock_->Connect(callback_.callback());
|
| - EXPECT_EQ(OK, rv);
|
| + EXPECT_THAT(rv, IsOk());
|
| EXPECT_TRUE(user_sock_->IsConnected());
|
|
|
| user_sock_->Disconnect();
|
| @@ -241,7 +246,7 @@ TEST_F(SOCKS5ClientSocketTest, LargeHostNameFails) {
|
| // the transport socket first) because the hostname is too long.
|
| TestCompletionCallback callback;
|
| int rv = user_sock_->Connect(callback.callback());
|
| - EXPECT_EQ(ERR_SOCKS_CONNECTION_FAILED, rv);
|
| + EXPECT_THAT(rv, IsError(ERR_SOCKS_CONNECTION_FAILED));
|
| }
|
|
|
| TEST_F(SOCKS5ClientSocketTest, PartialReadWrites) {
|
| @@ -273,7 +278,7 @@ TEST_F(SOCKS5ClientSocketTest, PartialReadWrites) {
|
| data_writes, arraysize(data_writes),
|
| hostname, 80, &net_log_);
|
| int rv = user_sock_->Connect(callback_.callback());
|
| - EXPECT_EQ(ERR_IO_PENDING, rv);
|
| + EXPECT_THAT(rv, IsError(ERR_IO_PENDING));
|
|
|
| TestNetLogEntry::List net_log_entries;
|
| net_log_.GetEntries(&net_log_entries);
|
| @@ -281,7 +286,7 @@ TEST_F(SOCKS5ClientSocketTest, PartialReadWrites) {
|
| NetLog::TYPE_SOCKS5_CONNECT));
|
|
|
| rv = callback_.WaitForResult();
|
| - EXPECT_EQ(OK, rv);
|
| + EXPECT_THAT(rv, IsOk());
|
| EXPECT_TRUE(user_sock_->IsConnected());
|
|
|
| net_log_.GetEntries(&net_log_entries);
|
| @@ -304,14 +309,14 @@ TEST_F(SOCKS5ClientSocketTest, PartialReadWrites) {
|
| data_writes, arraysize(data_writes),
|
| hostname, 80, &net_log_);
|
| int rv = user_sock_->Connect(callback_.callback());
|
| - EXPECT_EQ(ERR_IO_PENDING, rv);
|
| + EXPECT_THAT(rv, IsError(ERR_IO_PENDING));
|
|
|
| TestNetLogEntry::List net_log_entries;
|
| net_log_.GetEntries(&net_log_entries);
|
| EXPECT_TRUE(LogContainsBeginEvent(net_log_entries, 0,
|
| NetLog::TYPE_SOCKS5_CONNECT));
|
| rv = callback_.WaitForResult();
|
| - EXPECT_EQ(OK, rv);
|
| + EXPECT_THAT(rv, IsOk());
|
| EXPECT_TRUE(user_sock_->IsConnected());
|
| net_log_.GetEntries(&net_log_entries);
|
| EXPECT_TRUE(LogContainsEndEvent(net_log_entries, -1,
|
| @@ -334,13 +339,13 @@ TEST_F(SOCKS5ClientSocketTest, PartialReadWrites) {
|
| data_writes, arraysize(data_writes),
|
| hostname, 80, &net_log_);
|
| int rv = user_sock_->Connect(callback_.callback());
|
| - EXPECT_EQ(ERR_IO_PENDING, rv);
|
| + EXPECT_THAT(rv, IsError(ERR_IO_PENDING));
|
| TestNetLogEntry::List net_log_entries;
|
| net_log_.GetEntries(&net_log_entries);
|
| EXPECT_TRUE(LogContainsBeginEvent(net_log_entries, 0,
|
| NetLog::TYPE_SOCKS5_CONNECT));
|
| rv = callback_.WaitForResult();
|
| - EXPECT_EQ(OK, rv);
|
| + EXPECT_THAT(rv, IsOk());
|
| EXPECT_TRUE(user_sock_->IsConnected());
|
| net_log_.GetEntries(&net_log_entries);
|
| EXPECT_TRUE(LogContainsEndEvent(net_log_entries, -1,
|
| @@ -365,13 +370,13 @@ TEST_F(SOCKS5ClientSocketTest, PartialReadWrites) {
|
| data_writes, arraysize(data_writes),
|
| hostname, 80, &net_log_);
|
| int rv = user_sock_->Connect(callback_.callback());
|
| - EXPECT_EQ(ERR_IO_PENDING, rv);
|
| + EXPECT_THAT(rv, IsError(ERR_IO_PENDING));
|
| TestNetLogEntry::List net_log_entries;
|
| net_log_.GetEntries(&net_log_entries);
|
| EXPECT_TRUE(LogContainsBeginEvent(net_log_entries, 0,
|
| NetLog::TYPE_SOCKS5_CONNECT));
|
| rv = callback_.WaitForResult();
|
| - EXPECT_EQ(OK, rv);
|
| + EXPECT_THAT(rv, IsOk());
|
| EXPECT_TRUE(user_sock_->IsConnected());
|
| net_log_.GetEntries(&net_log_entries);
|
| EXPECT_TRUE(LogContainsEndEvent(net_log_entries, -1,
|
|
|