Index: net/quic/bidirectional_stream_quic_impl_unittest.cc |
diff --git a/net/quic/bidirectional_stream_quic_impl_unittest.cc b/net/quic/bidirectional_stream_quic_impl_unittest.cc |
index abad19b225b98af0b5e520bd61ac08fd1c61bf9f..a27cbb267da2a1bc29943983ee5b3f2b696b76d7 100644 |
--- a/net/quic/bidirectional_stream_quic_impl_unittest.cc |
+++ b/net/quic/bidirectional_stream_quic_impl_unittest.cc |
@@ -36,9 +36,13 @@ |
#include "net/quic/test_tools/quic_test_utils.h" |
#include "net/quic/test_tools/test_task_runner.h" |
#include "net/socket/socket_test_util.h" |
+#include "net/test/gtest_util.h" |
#include "testing/gmock/include/gmock/gmock.h" |
#include "testing/gtest/include/gtest/gtest.h" |
+using net::test::IsError; |
+using net::test::IsOk; |
+ |
namespace net { |
namespace test { |
@@ -641,7 +645,7 @@ TEST_P(BidirectionalStreamQuicImplTest, GetRequest) { |
delegate->WaitUntilNextCallback(); // OnHeadersReceived |
TestCompletionCallback cb; |
int rv = delegate->ReadData(cb.callback()); |
- EXPECT_EQ(ERR_IO_PENDING, rv); |
+ EXPECT_THAT(rv, IsError(ERR_IO_PENDING)); |
EXPECT_EQ("200", delegate->response_headers().find(":status")->second); |
const char kResponseBody[] = "Hello world!"; |
// Server sends data. |
@@ -651,7 +655,7 @@ TEST_P(BidirectionalStreamQuicImplTest, GetRequest) { |
EXPECT_EQ(std::string(kResponseBody), delegate->data_received()); |
TestCompletionCallback cb2; |
- EXPECT_EQ(ERR_IO_PENDING, delegate->ReadData(cb2.callback())); |
+ EXPECT_THAT(delegate->ReadData(cb2.callback()), IsError(ERR_IO_PENDING)); |
SpdyHeaderBlock trailers; |
size_t spdy_trailers_frame_length; |
@@ -662,11 +666,11 @@ TEST_P(BidirectionalStreamQuicImplTest, GetRequest) { |
4, kFin, trailers.Clone(), &spdy_trailers_frame_length, &offset)); |
delegate->WaitUntilNextCallback(); // OnTrailersReceived |
- EXPECT_EQ(OK, cb2.WaitForResult()); |
+ EXPECT_THAT(cb2.WaitForResult(), IsOk()); |
trailers.erase(kFinalOffsetHeaderKey); |
EXPECT_EQ(trailers, delegate->trailers()); |
- EXPECT_EQ(OK, delegate->ReadData(cb2.callback())); |
+ EXPECT_THAT(delegate->ReadData(cb2.callback()), IsOk()); |
base::RunLoop().RunUntilIdle(); |
EXPECT_EQ(2, delegate->on_data_read_count()); |
@@ -759,7 +763,7 @@ TEST_P(BidirectionalStreamQuicImplTest, CoalesceDataBuffersNotHeadersFrame) { |
delegate->WaitUntilNextCallback(); // OnHeadersReceived |
TestCompletionCallback cb; |
int rv = delegate->ReadData(cb.callback()); |
- EXPECT_EQ(ERR_IO_PENDING, rv); |
+ EXPECT_THAT(rv, IsError(ERR_IO_PENDING)); |
EXPECT_EQ("200", delegate->response_headers().find(":status")->second); |
const char kResponseBody[] = "Hello world!"; |
// Server sends data. |
@@ -788,7 +792,7 @@ TEST_P(BidirectionalStreamQuicImplTest, CoalesceDataBuffersNotHeadersFrame) { |
delegate->WaitUntilNextCallback(); // OnTrailersReceived |
trailers.erase(kFinalOffsetHeaderKey); |
EXPECT_EQ(trailers, delegate->trailers()); |
- EXPECT_EQ(OK, delegate->ReadData(cb.callback())); |
+ EXPECT_THAT(delegate->ReadData(cb.callback()), IsOk()); |
EXPECT_EQ(1, delegate->on_data_read_count()); |
EXPECT_EQ(2, delegate->on_data_sent_count()); |
@@ -858,7 +862,7 @@ TEST_P(BidirectionalStreamQuicImplTest, |
delegate->WaitUntilNextCallback(); // OnHeadersReceived |
TestCompletionCallback cb; |
int rv = delegate->ReadData(cb.callback()); |
- EXPECT_EQ(ERR_IO_PENDING, rv); |
+ EXPECT_THAT(rv, IsError(ERR_IO_PENDING)); |
EXPECT_EQ("200", delegate->response_headers().find(":status")->second); |
const char kResponseBody[] = "Hello world!"; |
// Server sends data. |
@@ -884,7 +888,7 @@ TEST_P(BidirectionalStreamQuicImplTest, |
delegate->WaitUntilNextCallback(); // OnTrailersReceived |
trailers.erase(kFinalOffsetHeaderKey); |
EXPECT_EQ(trailers, delegate->trailers()); |
- EXPECT_EQ(OK, delegate->ReadData(cb.callback())); |
+ EXPECT_THAT(delegate->ReadData(cb.callback()), IsOk()); |
EXPECT_EQ(1, delegate->on_data_read_count()); |
EXPECT_EQ(2, delegate->on_data_sent_count()); |
@@ -958,7 +962,7 @@ TEST_P(BidirectionalStreamQuicImplTest, |
delegate->WaitUntilNextCallback(); // OnHeadersReceived |
TestCompletionCallback cb; |
int rv = delegate->ReadData(cb.callback()); |
- EXPECT_EQ(ERR_IO_PENDING, rv); |
+ EXPECT_THAT(rv, IsError(ERR_IO_PENDING)); |
EXPECT_EQ("200", delegate->response_headers().find(":status")->second); |
const char kResponseBody[] = "Hello world!"; |
// Server sends data. |
@@ -987,7 +991,7 @@ TEST_P(BidirectionalStreamQuicImplTest, |
delegate->WaitUntilNextCallback(); // OnTrailersReceived |
trailers.erase(kFinalOffsetHeaderKey); |
EXPECT_EQ(trailers, delegate->trailers()); |
- EXPECT_EQ(OK, delegate->ReadData(cb.callback())); |
+ EXPECT_THAT(delegate->ReadData(cb.callback()), IsOk()); |
EXPECT_EQ(1, delegate->on_data_read_count()); |
EXPECT_EQ(2, delegate->on_data_sent_count()); |
@@ -1047,7 +1051,7 @@ TEST_P(BidirectionalStreamQuicImplTest, PostRequest) { |
delegate->WaitUntilNextCallback(); // OnHeadersReceived |
TestCompletionCallback cb; |
int rv = delegate->ReadData(cb.callback()); |
- EXPECT_EQ(ERR_IO_PENDING, rv); |
+ EXPECT_THAT(rv, IsError(ERR_IO_PENDING)); |
EXPECT_EQ("200", delegate->response_headers().find(":status")->second); |
const char kResponseBody[] = "Hello world!"; |
// Server sends data. |
@@ -1067,7 +1071,7 @@ TEST_P(BidirectionalStreamQuicImplTest, PostRequest) { |
delegate->WaitUntilNextCallback(); // OnTrailersReceived |
trailers.erase(kFinalOffsetHeaderKey); |
EXPECT_EQ(trailers, delegate->trailers()); |
- EXPECT_EQ(OK, delegate->ReadData(cb.callback())); |
+ EXPECT_THAT(delegate->ReadData(cb.callback()), IsOk()); |
EXPECT_EQ(1, delegate->on_data_read_count()); |
EXPECT_EQ(1, delegate->on_data_sent_count()); |
@@ -1124,7 +1128,7 @@ TEST_P(BidirectionalStreamQuicImplTest, PutRequest) { |
delegate->WaitUntilNextCallback(); // OnHeadersReceived |
TestCompletionCallback cb; |
int rv = delegate->ReadData(cb.callback()); |
- EXPECT_EQ(ERR_IO_PENDING, rv); |
+ EXPECT_THAT(rv, IsError(ERR_IO_PENDING)); |
EXPECT_EQ("200", delegate->response_headers().find(":status")->second); |
const char kResponseBody[] = "Hello world!"; |
// Server sends data. |
@@ -1144,7 +1148,7 @@ TEST_P(BidirectionalStreamQuicImplTest, PutRequest) { |
delegate->WaitUntilNextCallback(); // OnTrailersReceived |
trailers.erase(kFinalOffsetHeaderKey); |
EXPECT_EQ(trailers, delegate->trailers()); |
- EXPECT_EQ(OK, delegate->ReadData(cb.callback())); |
+ EXPECT_THAT(delegate->ReadData(cb.callback()), IsOk()); |
EXPECT_EQ(1, delegate->on_data_read_count()); |
EXPECT_EQ(1, delegate->on_data_sent_count()); |
@@ -1204,7 +1208,7 @@ TEST_P(BidirectionalStreamQuicImplTest, InterleaveReadDataAndSendData) { |
TestCompletionCallback cb; |
int rv = delegate->ReadData(cb.callback()); |
- EXPECT_EQ(ERR_IO_PENDING, rv); |
+ EXPECT_THAT(rv, IsError(ERR_IO_PENDING)); |
const char kResponseBody[] = "Hello world!"; |
// Server sends a data packet. |
@@ -1220,7 +1224,7 @@ TEST_P(BidirectionalStreamQuicImplTest, InterleaveReadDataAndSendData) { |
TestCompletionCallback cb2; |
rv = delegate->ReadData(cb2.callback()); |
- EXPECT_EQ(ERR_IO_PENDING, rv); |
+ EXPECT_THAT(rv, IsError(ERR_IO_PENDING)); |
ProcessPacket(ConstructAckAndDataPacket(4, !kIncludeVersion, 3, 1, kFin, |
strlen(kResponseBody), kResponseBody, |
&server_maker_)); |
@@ -1231,7 +1235,7 @@ TEST_P(BidirectionalStreamQuicImplTest, InterleaveReadDataAndSendData) { |
expected_body.append(kResponseBody); |
EXPECT_EQ(expected_body, delegate->data_received()); |
- EXPECT_EQ(OK, delegate->ReadData(cb.callback())); |
+ EXPECT_THAT(delegate->ReadData(cb.callback()), IsOk()); |
EXPECT_EQ(2, delegate->on_data_read_count()); |
EXPECT_EQ(2, delegate->on_data_sent_count()); |
EXPECT_EQ(kProtoQUIC1SPDY3, delegate->GetProtocol()); |
@@ -1268,11 +1272,12 @@ TEST_P(BidirectionalStreamQuicImplTest, ServerSendsRstAfterHeaders) { |
delegate->WaitUntilNextCallback(); // OnFailed |
TestCompletionCallback cb; |
- EXPECT_EQ(ERR_QUIC_PROTOCOL_ERROR, delegate->ReadData(cb.callback())); |
+ EXPECT_THAT(delegate->ReadData(cb.callback()), |
+ IsError(ERR_QUIC_PROTOCOL_ERROR)); |
base::RunLoop().RunUntilIdle(); |
- EXPECT_EQ(ERR_QUIC_PROTOCOL_ERROR, delegate->error()); |
+ EXPECT_THAT(delegate->error(), IsError(ERR_QUIC_PROTOCOL_ERROR)); |
EXPECT_EQ(0, delegate->on_data_read_count()); |
EXPECT_EQ(0, delegate->on_data_sent_count()); |
EXPECT_EQ(static_cast<int64_t>(spdy_request_headers_frame_length), |
@@ -1320,15 +1325,16 @@ TEST_P(BidirectionalStreamQuicImplTest, ServerSendsRstAfterReadData) { |
TestCompletionCallback cb; |
int rv = delegate->ReadData(cb.callback()); |
- EXPECT_EQ(ERR_IO_PENDING, rv); |
+ EXPECT_THAT(rv, IsError(ERR_IO_PENDING)); |
// Server sends a Rst. |
ProcessPacket(ConstructServerRstStreamPacket(3)); |
delegate->WaitUntilNextCallback(); // OnFailed |
- EXPECT_EQ(ERR_QUIC_PROTOCOL_ERROR, delegate->ReadData(cb.callback())); |
- EXPECT_EQ(ERR_QUIC_PROTOCOL_ERROR, delegate->error()); |
+ EXPECT_THAT(delegate->ReadData(cb.callback()), |
+ IsError(ERR_QUIC_PROTOCOL_ERROR)); |
+ EXPECT_THAT(delegate->error(), IsError(ERR_QUIC_PROTOCOL_ERROR)); |
EXPECT_EQ(0, delegate->on_data_read_count()); |
EXPECT_EQ(0, delegate->on_data_sent_count()); |
EXPECT_EQ(static_cast<int64_t>(spdy_request_headers_frame_length), |
@@ -1434,7 +1440,7 @@ TEST_P(BidirectionalStreamQuicImplTest, SessionClosedBeforeReadData) { |
delegate->WaitUntilNextCallback(); // OnHeadersReceived |
TestCompletionCallback cb; |
int rv = delegate->ReadData(cb.callback()); |
- EXPECT_EQ(ERR_IO_PENDING, rv); |
+ EXPECT_THAT(rv, IsError(ERR_IO_PENDING)); |
session()->connection()->CloseConnection( |
QUIC_NO_ERROR, "test", ConnectionCloseBehavior::SILENT_CLOSE); |
delegate->WaitUntilNextCallback(); // OnFailed |
@@ -1445,8 +1451,8 @@ TEST_P(BidirectionalStreamQuicImplTest, SessionClosedBeforeReadData) { |
delegate->SendData(buf, buf->size(), false); |
base::RunLoop().RunUntilIdle(); |
- EXPECT_EQ(ERR_UNEXPECTED, delegate->ReadData(cb.callback())); |
- EXPECT_EQ(ERR_UNEXPECTED, delegate->error()); |
+ EXPECT_THAT(delegate->ReadData(cb.callback()), IsError(ERR_UNEXPECTED)); |
+ EXPECT_THAT(delegate->error(), IsError(ERR_UNEXPECTED)); |
EXPECT_EQ(0, delegate->on_data_read_count()); |
EXPECT_EQ(0, delegate->on_data_sent_count()); |
EXPECT_EQ(kProtoQUIC1SPDY3, delegate->GetProtocol()); |
@@ -1493,7 +1499,7 @@ TEST_P(BidirectionalStreamQuicImplTest, CancelStreamAfterReadData) { |
// Cancel the stream after ReadData returns ERR_IO_PENDING. |
TestCompletionCallback cb; |
- EXPECT_EQ(ERR_IO_PENDING, delegate->ReadData(cb.callback())); |
+ EXPECT_THAT(delegate->ReadData(cb.callback()), IsError(ERR_IO_PENDING)); |
delegate->CancelStream(); |
base::RunLoop().RunUntilIdle(); |
@@ -1590,7 +1596,7 @@ TEST_P(BidirectionalStreamQuicImplTest, DeleteStreamDuringOnDataRead) { |
TestCompletionCallback cb; |
int rv = delegate->ReadData(cb.callback()); |
- EXPECT_EQ(ERR_IO_PENDING, rv); |
+ EXPECT_THAT(rv, IsError(ERR_IO_PENDING)); |
const char kResponseBody[] = "Hello world!"; |
// Server sends data. |
ProcessPacket( |
@@ -1643,7 +1649,7 @@ TEST_P(BidirectionalStreamQuicImplTest, DeleteStreamDuringOnTrailersReceived) { |
TestCompletionCallback cb; |
int rv = delegate->ReadData(cb.callback()); |
- EXPECT_EQ(ERR_IO_PENDING, rv); |
+ EXPECT_THAT(rv, IsError(ERR_IO_PENDING)); |
const char kResponseBody[] = "Hello world!"; |
// Server sends data. |
ProcessPacket( |