Index: net/quic/quic_network_transaction_unittest.cc |
diff --git a/net/quic/quic_network_transaction_unittest.cc b/net/quic/quic_network_transaction_unittest.cc |
index 3d4bce6ed6badc54ed1c15e00b0f585de119037e..12a94cc13a4afb1159c71e5eb3d6b5074c61b3f9 100644 |
--- a/net/quic/quic_network_transaction_unittest.cc |
+++ b/net/quic/quic_network_transaction_unittest.cc |
@@ -54,11 +54,16 @@ |
#include "net/spdy/spdy_framer.h" |
#include "net/ssl/ssl_config_service_defaults.h" |
#include "net/test/cert_test_util.h" |
+#include "net/test/gtest_util.h" |
#include "net/test/test_data_directory.h" |
+#include "testing/gmock/include/gmock/gmock.h" |
#include "testing/gtest/include/gtest/gtest.h" |
#include "testing/platform_test.h" |
#include "url/gurl.h" |
+using net::test::IsError; |
+using net::test::IsOk; |
+ |
namespace net { |
namespace test { |
@@ -589,15 +594,15 @@ class QuicNetworkTransactionTest |
void CheckResponseData(const std::unique_ptr<HttpNetworkTransaction>& trans, |
const std::string& expected) { |
std::string response_data; |
- ASSERT_EQ(OK, ReadTransaction(trans.get(), &response_data)); |
+ ASSERT_THAT(ReadTransaction(trans.get(), &response_data), IsOk()); |
EXPECT_EQ(expected, response_data); |
} |
void RunTransaction(const std::unique_ptr<HttpNetworkTransaction>& trans) { |
TestCompletionCallback callback; |
int rv = trans->Start(&request_, callback.callback(), net_log_.bound()); |
- EXPECT_EQ(ERR_IO_PENDING, rv); |
- EXPECT_EQ(OK, callback.WaitForResult()); |
+ EXPECT_THAT(rv, IsError(ERR_IO_PENDING)); |
+ EXPECT_THAT(callback.WaitForResult(), IsOk()); |
} |
void SendRequestAndExpectHttpResponse(const std::string& expected) { |
@@ -924,8 +929,8 @@ TEST_P(QuicNetworkTransactionTest, ForceQuicWithErrorConnecting) { |
new HttpNetworkTransaction(DEFAULT_PRIORITY, session_.get())); |
TestCompletionCallback callback; |
int rv = trans->Start(&request_, callback.callback(), net_log_.bound()); |
- EXPECT_EQ(ERR_IO_PENDING, rv); |
- EXPECT_EQ(ERR_CONNECTION_CLOSED, callback.WaitForResult()); |
+ EXPECT_THAT(rv, IsError(ERR_IO_PENDING)); |
+ EXPECT_THAT(callback.WaitForResult(), IsError(ERR_CONNECTION_CLOSED)); |
EXPECT_EQ(1 + i, test_socket_performance_watcher_factory_.watcher_count()); |
} |
} |
@@ -1167,11 +1172,11 @@ TEST_P(QuicNetworkTransactionTest, GoAwayWithConnectionMigrationOnPortsOnly) { |
new HttpNetworkTransaction(DEFAULT_PRIORITY, session_.get())); |
TestCompletionCallback callback; |
int rv = trans->Start(&request_, callback.callback(), net_log_.bound()); |
- EXPECT_EQ(ERR_IO_PENDING, rv); |
+ EXPECT_THAT(rv, IsError(ERR_IO_PENDING)); |
crypto_client_stream_factory_.last_stream()->SendOnCryptoHandshakeEvent( |
QuicSession::HANDSHAKE_CONFIRMED); |
- EXPECT_EQ(OK, callback.WaitForResult()); |
+ EXPECT_THAT(callback.WaitForResult(), IsOk()); |
// Check whether this transaction is correctly marked as received a go-away |
// because of migrating port. |
@@ -1784,11 +1789,11 @@ TEST_P(QuicNetworkTransactionTest, ZeroRTTWithConfirmationRequired) { |
new HttpNetworkTransaction(DEFAULT_PRIORITY, session_.get())); |
TestCompletionCallback callback; |
int rv = trans->Start(&request_, callback.callback(), net_log_.bound()); |
- EXPECT_EQ(ERR_IO_PENDING, rv); |
+ EXPECT_THAT(rv, IsError(ERR_IO_PENDING)); |
crypto_client_stream_factory_.last_stream()->SendOnCryptoHandshakeEvent( |
QuicSession::HANDSHAKE_CONFIRMED); |
- EXPECT_EQ(OK, callback.WaitForResult()); |
+ EXPECT_THAT(callback.WaitForResult(), IsOk()); |
CheckWasQuicResponse(trans); |
CheckResponseData(trans, "hello!"); |
@@ -1830,11 +1835,11 @@ TEST_P(QuicNetworkTransactionTest, |
new HttpNetworkTransaction(DEFAULT_PRIORITY, session_.get())); |
TestCompletionCallback callback; |
int rv = trans->Start(&request_, callback.callback(), net_log_.bound()); |
- EXPECT_EQ(ERR_IO_PENDING, rv); |
+ EXPECT_THAT(rv, IsError(ERR_IO_PENDING)); |
crypto_client_stream_factory_.last_stream()->SendOnCryptoHandshakeEvent( |
QuicSession::HANDSHAKE_CONFIRMED); |
- EXPECT_EQ(ERR_QUIC_PROTOCOL_ERROR, callback.WaitForResult()); |
+ EXPECT_THAT(callback.WaitForResult(), IsError(ERR_QUIC_PROTOCOL_ERROR)); |
NetErrorDetails details; |
EXPECT_EQ(QUIC_NO_ERROR, details.quic_connection_error); |
@@ -1885,11 +1890,11 @@ TEST_P(QuicNetworkTransactionTest, |
new HttpNetworkTransaction(DEFAULT_PRIORITY, session_.get())); |
TestCompletionCallback callback; |
int rv = trans->Start(&request_, callback.callback(), net_log_.bound()); |
- EXPECT_EQ(ERR_IO_PENDING, rv); |
+ EXPECT_THAT(rv, IsError(ERR_IO_PENDING)); |
crypto_client_stream_factory_.last_stream()->SendOnCryptoHandshakeEvent( |
QuicSession::HANDSHAKE_CONFIRMED); |
- EXPECT_EQ(ERR_QUIC_PROTOCOL_ERROR, callback.WaitForResult()); |
+ EXPECT_THAT(callback.WaitForResult(), IsError(ERR_QUIC_PROTOCOL_ERROR)); |
NetErrorDetails details; |
EXPECT_EQ(QUIC_NO_ERROR, details.quic_connection_error); |
@@ -1936,12 +1941,12 @@ TEST_P(QuicNetworkTransactionTest, RstSteamErrorHandling) { |
new HttpNetworkTransaction(DEFAULT_PRIORITY, session_.get())); |
TestCompletionCallback callback; |
int rv = trans->Start(&request_, callback.callback(), net_log_.bound()); |
- EXPECT_EQ(ERR_IO_PENDING, rv); |
+ EXPECT_THAT(rv, IsError(ERR_IO_PENDING)); |
crypto_client_stream_factory_.last_stream()->SendOnCryptoHandshakeEvent( |
QuicSession::HANDSHAKE_CONFIRMED); |
// Read the headers. |
- EXPECT_EQ(OK, callback.WaitForResult()); |
+ EXPECT_THAT(callback.WaitForResult(), IsOk()); |
const HttpResponseInfo* response = trans->GetResponseInfo(); |
ASSERT_TRUE(response != nullptr); |
@@ -1992,12 +1997,12 @@ TEST_P(QuicNetworkTransactionTest, RstSteamBeforeHeaders) { |
new HttpNetworkTransaction(DEFAULT_PRIORITY, session_.get())); |
TestCompletionCallback callback; |
int rv = trans->Start(&request_, callback.callback(), net_log_.bound()); |
- EXPECT_EQ(ERR_IO_PENDING, rv); |
+ EXPECT_THAT(rv, IsError(ERR_IO_PENDING)); |
crypto_client_stream_factory_.last_stream()->SendOnCryptoHandshakeEvent( |
QuicSession::HANDSHAKE_CONFIRMED); |
// Read the headers. |
- EXPECT_EQ(ERR_QUIC_PROTOCOL_ERROR, callback.WaitForResult()); |
+ EXPECT_THAT(callback.WaitForResult(), IsError(ERR_QUIC_PROTOCOL_ERROR)); |
} |
TEST_P(QuicNetworkTransactionTest, BrokenAlternateProtocol) { |
@@ -2085,8 +2090,8 @@ TEST_P(QuicNetworkTransactionTest, NoBrokenAlternateProtocolIfTcpFails) { |
new HttpNetworkTransaction(DEFAULT_PRIORITY, session_.get())); |
TestCompletionCallback callback; |
int rv = trans->Start(&request_, callback.callback(), net_log_.bound()); |
- EXPECT_EQ(ERR_IO_PENDING, rv); |
- EXPECT_EQ(ERR_SOCKET_NOT_CONNECTED, callback.WaitForResult()); |
+ EXPECT_THAT(rv, IsError(ERR_IO_PENDING)); |
+ EXPECT_THAT(callback.WaitForResult(), IsError(ERR_SOCKET_NOT_CONNECTED)); |
ExpectQuicAlternateProtocolMapping(); |
} |
@@ -2267,7 +2272,7 @@ TEST_P(QuicNetworkTransactionTest, QuicUpload) { |
new HttpNetworkTransaction(DEFAULT_PRIORITY, session_.get())); |
TestCompletionCallback callback; |
int rv = trans->Start(&request_, callback.callback(), net_log_.bound()); |
- EXPECT_EQ(ERR_IO_PENDING, rv); |
+ EXPECT_THAT(rv, IsError(ERR_IO_PENDING)); |
EXPECT_NE(OK, callback.WaitForResult()); |
} |
@@ -2509,10 +2514,10 @@ class QuicNetworkTransactionWithDestinationTest |
request.method = "GET"; |
TestCompletionCallback callback; |
int rv = trans.Start(&request, callback.callback(), net_log_.bound()); |
- EXPECT_EQ(OK, callback.GetResult(rv)); |
+ EXPECT_THAT(callback.GetResult(rv), IsOk()); |
std::string response_data; |
- ASSERT_EQ(OK, ReadTransaction(&trans, &response_data)); |
+ ASSERT_THAT(ReadTransaction(&trans, &response_data), IsOk()); |
EXPECT_EQ("hello", response_data); |
const HttpResponseInfo* response = trans.GetResponseInfo(); |
@@ -2595,7 +2600,7 @@ TEST_P(QuicNetworkTransactionWithDestinationTest, InvalidCertificate) { |
HttpNetworkTransaction trans(DEFAULT_PRIORITY, session_.get()); |
TestCompletionCallback callback; |
int rv = trans.Start(&request, callback.callback(), net_log_.bound()); |
- EXPECT_EQ(ERR_CONNECTION_REFUSED, callback.GetResult(rv)); |
+ EXPECT_THAT(callback.GetResult(rv), IsError(ERR_CONNECTION_REFUSED)); |
EXPECT_TRUE(AllDataConsumed()); |
} |