| Index: net/websockets/websocket_deflate_stream_test.cc
|
| diff --git a/net/websockets/websocket_deflate_stream_test.cc b/net/websockets/websocket_deflate_stream_test.cc
|
| index 5106df710cdcbbcd1a09f93c19e44b1c590b3478..f4dde0872e591d1ef69da374c0bb378c1ac2a9ff 100644
|
| --- a/net/websockets/websocket_deflate_stream_test.cc
|
| +++ b/net/websockets/websocket_deflate_stream_test.cc
|
| @@ -21,6 +21,7 @@
|
| #include "net/base/completion_callback.h"
|
| #include "net/base/io_buffer.h"
|
| #include "net/base/net_errors.h"
|
| +#include "net/test/gtest_util.h"
|
| #include "net/websockets/websocket_deflate_parameters.h"
|
| #include "net/websockets/websocket_deflate_predictor.h"
|
| #include "net/websockets/websocket_deflater.h"
|
| @@ -31,6 +32,9 @@
|
| #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 {
|
|
|
| @@ -369,7 +373,8 @@ TEST_F(WebSocketDeflateStreamTest, ReadFailedImmediately) {
|
| EXPECT_CALL(*mock_stream_, ReadFrames(&frames, _))
|
| .WillOnce(Return(ERR_FAILED));
|
| }
|
| - EXPECT_EQ(ERR_FAILED, deflate_stream_->ReadFrames(&frames, callback));
|
| + EXPECT_THAT(deflate_stream_->ReadFrames(&frames, callback),
|
| + IsError(ERR_FAILED));
|
| }
|
|
|
| TEST_F(WebSocketDeflateStreamTest, ReadUncompressedFrameImmediately) {
|
| @@ -387,7 +392,7 @@ TEST_F(WebSocketDeflateStreamTest, ReadUncompressedFrameImmediately) {
|
| .WillOnce(Invoke(&stub, &ReadFramesStub::Call));
|
| }
|
| CompletionCallback callback;
|
| - ASSERT_EQ(OK, deflate_stream_->ReadFrames(&frames, callback));
|
| + ASSERT_THAT(deflate_stream_->ReadFrames(&frames, callback), IsOk());
|
| ASSERT_EQ(1u, frames.size());
|
| EXPECT_EQ(WebSocketFrameHeader::kOpCodeText, frames[0]->header.opcode);
|
| EXPECT_TRUE(frames[0]->header.final);
|
| @@ -409,7 +414,8 @@ TEST_F(WebSocketDeflateStreamTest, ReadUncompressedFrameAsync) {
|
| EXPECT_CALL(checkpoint, Call(0));
|
| EXPECT_CALL(mock_callback, Call(OK));
|
| }
|
| - ASSERT_EQ(ERR_IO_PENDING, deflate_stream_->ReadFrames(&frames, callback));
|
| + ASSERT_THAT(deflate_stream_->ReadFrames(&frames, callback),
|
| + IsError(ERR_IO_PENDING));
|
| ASSERT_EQ(0u, frames.size());
|
|
|
| checkpoint.Call(0);
|
| @@ -440,7 +446,8 @@ TEST_F(WebSocketDeflateStreamTest, ReadFailedAsync) {
|
| EXPECT_CALL(checkpoint, Call(0));
|
| EXPECT_CALL(mock_callback, Call(ERR_FAILED));
|
| }
|
| - ASSERT_EQ(ERR_IO_PENDING, deflate_stream_->ReadFrames(&frames, callback));
|
| + ASSERT_THAT(deflate_stream_->ReadFrames(&frames, callback),
|
| + IsError(ERR_IO_PENDING));
|
| ASSERT_EQ(0u, frames.size());
|
|
|
| checkpoint.Call(0);
|
| @@ -467,7 +474,7 @@ TEST_F(WebSocketDeflateStreamTest, ReadCompressedFrameImmediately) {
|
| EXPECT_CALL(*mock_stream_, ReadFrames(&frames, _))
|
| .WillOnce(Invoke(&stub, &ReadFramesStub::Call));
|
| }
|
| - ASSERT_EQ(OK, deflate_stream_->ReadFrames(&frames, callback));
|
| + ASSERT_THAT(deflate_stream_->ReadFrames(&frames, callback), IsOk());
|
| ASSERT_EQ(1u, frames.size());
|
| EXPECT_EQ(WebSocketFrameHeader::kOpCodeText, frames[0]->header.opcode);
|
| EXPECT_TRUE(frames[0]->header.final);
|
| @@ -488,7 +495,8 @@ TEST_F(WebSocketDeflateStreamTest, ReadCompressedFrameAsync) {
|
| EXPECT_CALL(checkpoint, Call(0));
|
| EXPECT_CALL(mock_callback, Call(OK));
|
| }
|
| - ASSERT_EQ(ERR_IO_PENDING, deflate_stream_->ReadFrames(&frames, callback));
|
| + ASSERT_THAT(deflate_stream_->ReadFrames(&frames, callback),
|
| + IsError(ERR_IO_PENDING));
|
|
|
| checkpoint.Call(0);
|
|
|
| @@ -528,7 +536,8 @@ TEST_F(WebSocketDeflateStreamTest,
|
| EXPECT_CALL(checkpoint, Call(0));
|
| EXPECT_CALL(mock_callback, Call(OK));
|
| }
|
| - ASSERT_EQ(ERR_IO_PENDING, deflate_stream_->ReadFrames(&frames, callback));
|
| + ASSERT_THAT(deflate_stream_->ReadFrames(&frames, callback),
|
| + IsError(ERR_IO_PENDING));
|
| ASSERT_EQ(0u, frames.size());
|
|
|
| AppendTo(stub2.frames_passed(),
|
| @@ -588,7 +597,7 @@ TEST_F(WebSocketDeflateStreamTest, MergeMultipleFramesInReadFrames) {
|
| EXPECT_CALL(*mock_stream_, ReadFrames(&frames, _))
|
| .WillOnce(Invoke(&stub, &ReadFramesStub::Call));
|
| }
|
| - ASSERT_EQ(OK, deflate_stream_->ReadFrames(&frames, callback));
|
| + ASSERT_THAT(deflate_stream_->ReadFrames(&frames, callback), IsOk());
|
| ASSERT_EQ(1u, frames.size());
|
| EXPECT_EQ(WebSocketFrameHeader::kOpCodeText, frames[0]->header.opcode);
|
| EXPECT_TRUE(frames[0]->header.final);
|
| @@ -613,7 +622,7 @@ TEST_F(WebSocketDeflateStreamTest, ReadUncompressedEmptyFrames) {
|
| EXPECT_CALL(*mock_stream_, ReadFrames(&frames, _))
|
| .WillOnce(Invoke(&stub, &ReadFramesStub::Call));
|
| }
|
| - ASSERT_EQ(OK, deflate_stream_->ReadFrames(&frames, callback));
|
| + ASSERT_THAT(deflate_stream_->ReadFrames(&frames, callback), IsOk());
|
| ASSERT_EQ(2u, frames.size());
|
| EXPECT_EQ(WebSocketFrameHeader::kOpCodeText, frames[0]->header.opcode);
|
| EXPECT_FALSE(frames[0]->header.final);
|
| @@ -644,7 +653,7 @@ TEST_F(WebSocketDeflateStreamTest, ReadCompressedEmptyFrames) {
|
| EXPECT_CALL(*mock_stream_, ReadFrames(&frames, _))
|
| .WillOnce(Invoke(&stub, &ReadFramesStub::Call));
|
| }
|
| - ASSERT_EQ(OK, deflate_stream_->ReadFrames(&frames, callback));
|
| + ASSERT_THAT(deflate_stream_->ReadFrames(&frames, callback), IsOk());
|
| ASSERT_EQ(1u, frames.size());
|
| EXPECT_EQ(WebSocketFrameHeader::kOpCodeText, frames[0]->header.opcode);
|
| EXPECT_TRUE(frames[0]->header.final);
|
| @@ -672,7 +681,7 @@ TEST_F(WebSocketDeflateStreamTest,
|
| EXPECT_CALL(*mock_stream_, ReadFrames(&frames, _))
|
| .WillOnce(Invoke(&stub, &ReadFramesStub::Call));
|
| }
|
| - ASSERT_EQ(OK, deflate_stream_->ReadFrames(&frames, callback));
|
| + ASSERT_THAT(deflate_stream_->ReadFrames(&frames, callback), IsOk());
|
| ASSERT_EQ(1u, frames.size());
|
| EXPECT_EQ(WebSocketFrameHeader::kOpCodeText, frames[0]->header.opcode);
|
| EXPECT_TRUE(frames[0]->header.final);
|
| @@ -699,7 +708,7 @@ TEST_F(WebSocketDeflateStreamTest, ReadControlFrameBetweenDataFrames) {
|
| EXPECT_CALL(*mock_stream_, ReadFrames(&frames, _))
|
| .WillOnce(Invoke(&stub, &ReadFramesStub::Call));
|
| }
|
| - ASSERT_EQ(OK, deflate_stream_->ReadFrames(&frames, callback));
|
| + ASSERT_THAT(deflate_stream_->ReadFrames(&frames, callback), IsOk());
|
| ASSERT_EQ(2u, frames.size());
|
| EXPECT_EQ(WebSocketFrameHeader::kOpCodePing, frames[0]->header.opcode);
|
| EXPECT_TRUE(frames[0]->header.final);
|
| @@ -733,7 +742,7 @@ TEST_F(WebSocketDeflateStreamTest, SplitToMultipleFramesInReadFrames) {
|
| .WillOnce(Invoke(&stub, &ReadFramesStub::Call));
|
| }
|
|
|
| - ASSERT_EQ(OK, deflate_stream_->ReadFrames(&frames, callback));
|
| + ASSERT_THAT(deflate_stream_->ReadFrames(&frames, callback), IsOk());
|
| ASSERT_EQ(3u, frames.size());
|
| EXPECT_EQ(WebSocketFrameHeader::kOpCodeBinary, frames[0]->header.opcode);
|
| EXPECT_FALSE(frames[0]->header.final);
|
| @@ -779,7 +788,7 @@ TEST_F(WebSocketDeflateStreamTest, InflaterInternalDataCanBeEmpty) {
|
| .WillOnce(Invoke(&stub, &ReadFramesStub::Call));
|
| }
|
|
|
| - ASSERT_EQ(OK, deflate_stream_->ReadFrames(&frames, callback));
|
| + ASSERT_THAT(deflate_stream_->ReadFrames(&frames, callback), IsOk());
|
| ASSERT_EQ(2u, frames.size());
|
| EXPECT_EQ(WebSocketFrameHeader::kOpCodeBinary, frames[0]->header.opcode);
|
| EXPECT_FALSE(frames[0]->header.final);
|
| @@ -864,7 +873,7 @@ TEST_F(WebSocketDeflateStreamTest, ReadCompressedMessages) {
|
| EXPECT_CALL(*mock_stream_, ReadFrames(&frames, _))
|
| .WillOnce(Invoke(&stub, &ReadFramesStub::Call));
|
| }
|
| - ASSERT_EQ(OK, deflate_stream_->ReadFrames(&frames, callback));
|
| + ASSERT_THAT(deflate_stream_->ReadFrames(&frames, callback), IsOk());
|
| ASSERT_EQ(2u, frames.size());
|
| EXPECT_EQ(WebSocketFrameHeader::kOpCodeText, frames[0]->header.opcode);
|
| EXPECT_TRUE(frames[0]->header.final);
|
| @@ -895,7 +904,7 @@ TEST_F(WebSocketDeflateStreamTest, ReadUncompressedMessages) {
|
| EXPECT_CALL(*mock_stream_, ReadFrames(&frames, _))
|
| .WillOnce(Invoke(&stub, &ReadFramesStub::Call));
|
| }
|
| - ASSERT_EQ(OK, deflate_stream_->ReadFrames(&frames, callback));
|
| + ASSERT_THAT(deflate_stream_->ReadFrames(&frames, callback), IsOk());
|
| ASSERT_EQ(2u, frames.size());
|
| EXPECT_EQ(WebSocketFrameHeader::kOpCodeText, frames[0]->header.opcode);
|
| EXPECT_TRUE(frames[0]->header.final);
|
| @@ -928,7 +937,7 @@ TEST_F(WebSocketDeflateStreamTest,
|
| EXPECT_CALL(*mock_stream_, ReadFrames(&frames, _))
|
| .WillOnce(Invoke(&stub, &ReadFramesStub::Call));
|
| }
|
| - ASSERT_EQ(OK, deflate_stream_->ReadFrames(&frames, callback));
|
| + ASSERT_THAT(deflate_stream_->ReadFrames(&frames, callback), IsOk());
|
| ASSERT_EQ(2u, frames.size());
|
| EXPECT_EQ(WebSocketFrameHeader::kOpCodeText, frames[0]->header.opcode);
|
| EXPECT_TRUE(frames[0]->header.final);
|
| @@ -961,7 +970,7 @@ TEST_F(WebSocketDeflateStreamTest,
|
| EXPECT_CALL(*mock_stream_, ReadFrames(&frames, _))
|
| .WillOnce(Invoke(&stub, &ReadFramesStub::Call));
|
| }
|
| - ASSERT_EQ(OK, deflate_stream_->ReadFrames(&frames, callback));
|
| + ASSERT_THAT(deflate_stream_->ReadFrames(&frames, callback), IsOk());
|
| ASSERT_EQ(2u, frames.size());
|
| EXPECT_EQ(WebSocketFrameHeader::kOpCodeText, frames[0]->header.opcode);
|
| EXPECT_TRUE(frames[0]->header.final);
|
| @@ -994,7 +1003,8 @@ TEST_F(WebSocketDeflateStreamTest, ReadEmptyAsyncFrame) {
|
| EXPECT_CALL(mock_callback, Call(OK));
|
| }
|
|
|
| - ASSERT_EQ(ERR_IO_PENDING, deflate_stream_->ReadFrames(&frames, callback));
|
| + ASSERT_THAT(deflate_stream_->ReadFrames(&frames, callback),
|
| + IsError(ERR_IO_PENDING));
|
| AppendTo(stub_vector[0]->frames_passed(),
|
| WebSocketFrameHeader::kOpCodeText,
|
| kReserved1,
|
| @@ -1017,7 +1027,7 @@ TEST_F(WebSocketDeflateStreamTest, WriteEmpty) {
|
| InSequence s;
|
| EXPECT_CALL(*mock_stream_, WriteFrames(&frames, _)).Times(0);
|
| }
|
| - EXPECT_EQ(OK, deflate_stream_->WriteFrames(&frames, callback));
|
| + EXPECT_THAT(deflate_stream_->WriteFrames(&frames, callback), IsOk());
|
| }
|
|
|
| TEST_F(WebSocketDeflateStreamTest, WriteFailedImmediately) {
|
| @@ -1031,7 +1041,8 @@ TEST_F(WebSocketDeflateStreamTest, WriteFailedImmediately) {
|
|
|
| AppendTo(&frames, WebSocketFrameHeader::kOpCodeText, kFinal, "hello");
|
| predictor_->AddFramesToBeInput(frames);
|
| - EXPECT_EQ(ERR_FAILED, deflate_stream_->WriteFrames(&frames, callback));
|
| + EXPECT_THAT(deflate_stream_->WriteFrames(&frames, callback),
|
| + IsError(ERR_FAILED));
|
| predictor_->Clear();
|
| }
|
|
|
| @@ -1046,7 +1057,7 @@ TEST_F(WebSocketDeflateStreamTest, WriteFrameImmediately) {
|
| EXPECT_CALL(*mock_stream_, WriteFrames(_, _))
|
| .WillOnce(Invoke(&stub, &WriteFramesStub::Call));
|
| }
|
| - ASSERT_EQ(OK, deflate_stream_->WriteFrames(&frames, callback));
|
| + ASSERT_THAT(deflate_stream_->WriteFrames(&frames, callback), IsOk());
|
| const std::vector<std::unique_ptr<WebSocketFrame>>& frames_passed =
|
| *stub.frames();
|
| ASSERT_EQ(1u, frames_passed.size());
|
| @@ -1072,7 +1083,8 @@ TEST_F(WebSocketDeflateStreamTest, WriteFrameAsync) {
|
| }
|
| AppendTo(&frames, WebSocketFrameHeader::kOpCodeText, kFinal, "Hello");
|
| predictor_->AddFramesToBeInput(frames);
|
| - ASSERT_EQ(ERR_IO_PENDING, deflate_stream_->WriteFrames(&frames, callback));
|
| + ASSERT_THAT(deflate_stream_->WriteFrames(&frames, callback),
|
| + IsError(ERR_IO_PENDING));
|
|
|
| checkpoint.Call(0);
|
| stub.callback().Run(OK);
|
| @@ -1101,7 +1113,7 @@ TEST_F(WebSocketDeflateStreamTest, WriteControlFrameBetweenDataFrames) {
|
| EXPECT_CALL(*mock_stream_, WriteFrames(&frames, _))
|
| .WillOnce(Invoke(&stub, &WriteFramesStub::Call));
|
| }
|
| - ASSERT_EQ(OK, deflate_stream_->WriteFrames(&frames, callback));
|
| + ASSERT_THAT(deflate_stream_->WriteFrames(&frames, callback), IsOk());
|
| const std::vector<std::unique_ptr<WebSocketFrame>>& frames_passed =
|
| *stub.frames();
|
| ASSERT_EQ(2u, frames_passed.size());
|
| @@ -1127,7 +1139,7 @@ TEST_F(WebSocketDeflateStreamTest, WriteEmptyMessage) {
|
| EXPECT_CALL(*mock_stream_, WriteFrames(&frames, _))
|
| .WillOnce(Invoke(&stub, &WriteFramesStub::Call));
|
| }
|
| - ASSERT_EQ(OK, deflate_stream_->WriteFrames(&frames, callback));
|
| + ASSERT_THAT(deflate_stream_->WriteFrames(&frames, callback), IsOk());
|
| const std::vector<std::unique_ptr<WebSocketFrame>>& frames_passed =
|
| *stub.frames();
|
| ASSERT_EQ(1u, frames_passed.size());
|
| @@ -1152,7 +1164,7 @@ TEST_F(WebSocketDeflateStreamTest, WriteUncompressedMessage) {
|
| EXPECT_CALL(*mock_stream_, WriteFrames(&frames, _))
|
| .WillOnce(Invoke(&stub, &WriteFramesStub::Call));
|
| }
|
| - ASSERT_EQ(OK, deflate_stream_->WriteFrames(&frames, callback));
|
| + ASSERT_THAT(deflate_stream_->WriteFrames(&frames, callback), IsOk());
|
| const std::vector<std::unique_ptr<WebSocketFrame>>& frames_passed =
|
| *stub.frames();
|
| ASSERT_EQ(2u, frames_passed.size());
|
| @@ -1192,7 +1204,7 @@ TEST_F(WebSocketDeflateStreamTest, LargeDeflatedFramesShouldBeSplit) {
|
| FrameFlag flag = is_final ? kFinal : kNoFlag;
|
| AppendTo(&frames, WebSocketFrameHeader::kOpCodeBinary, flag, data);
|
| predictor_->AddFramesToBeInput(frames);
|
| - ASSERT_EQ(OK, deflate_stream_->WriteFrames(&frames, callback));
|
| + ASSERT_THAT(deflate_stream_->WriteFrames(&frames, callback), IsOk());
|
| total_compressed_frames.insert(
|
| total_compressed_frames.end(),
|
| std::make_move_iterator(stub.frames()->begin()),
|
| @@ -1236,7 +1248,7 @@ TEST_F(WebSocketDeflateStreamTest, WriteMultipleMessages) {
|
| EXPECT_CALL(*mock_stream_, WriteFrames(&frames, _))
|
| .WillOnce(Invoke(&stub, &WriteFramesStub::Call));
|
| }
|
| - ASSERT_EQ(OK, deflate_stream_->WriteFrames(&frames, callback));
|
| + ASSERT_THAT(deflate_stream_->WriteFrames(&frames, callback), IsOk());
|
| const std::vector<std::unique_ptr<WebSocketFrame>>& frames_passed =
|
| *stub.frames();
|
| ASSERT_EQ(2u, frames_passed.size());
|
| @@ -1265,7 +1277,7 @@ TEST_F(WebSocketDeflateStreamWithDoNotTakeOverContextTest,
|
| EXPECT_CALL(*mock_stream_, WriteFrames(&frames, _))
|
| .WillOnce(Invoke(&stub, &WriteFramesStub::Call));
|
| }
|
| - ASSERT_EQ(OK, deflate_stream_->WriteFrames(&frames, callback));
|
| + ASSERT_THAT(deflate_stream_->WriteFrames(&frames, callback), IsOk());
|
| const std::vector<std::unique_ptr<WebSocketFrame>>& frames_passed =
|
| *stub.frames();
|
| ASSERT_EQ(2u, frames_passed.size());
|
| @@ -1302,7 +1314,7 @@ TEST_F(WebSocketDeflateStreamWithDoNotTakeOverContextTest,
|
| EXPECT_CALL(*mock_stream_, WriteFrames(&frames, _))
|
| .WillOnce(Invoke(&stub, &WriteFramesStub::Call));
|
| }
|
| - ASSERT_EQ(OK, deflate_stream_->WriteFrames(&frames, callback));
|
| + ASSERT_THAT(deflate_stream_->WriteFrames(&frames, callback), IsOk());
|
| const std::vector<std::unique_ptr<WebSocketFrame>>& frames_passed =
|
| *stub.frames();
|
| ASSERT_EQ(5u, frames_passed.size());
|
| @@ -1345,7 +1357,7 @@ TEST_F(WebSocketDeflateStreamWithClientWindowBitsTest, WindowBits8) {
|
| EXPECT_CALL(*mock_stream_, WriteFrames(_, _))
|
| .WillOnce(Invoke(&stub, &WriteFramesStub::Call));
|
| }
|
| - ASSERT_EQ(OK, deflate_stream_->WriteFrames(&frames_, callback));
|
| + ASSERT_THAT(deflate_stream_->WriteFrames(&frames_, callback), IsOk());
|
| const std::vector<std::unique_ptr<WebSocketFrame>>& frames_passed =
|
| *stub.frames();
|
| ASSERT_EQ(1u, frames_passed.size());
|
| @@ -1365,7 +1377,7 @@ TEST_F(WebSocketDeflateStreamWithClientWindowBitsTest, WindowBits10) {
|
| EXPECT_CALL(*mock_stream_, WriteFrames(_, _))
|
| .WillOnce(Invoke(&stub, &WriteFramesStub::Call));
|
| }
|
| - ASSERT_EQ(OK, deflate_stream_->WriteFrames(&frames_, callback));
|
| + ASSERT_THAT(deflate_stream_->WriteFrames(&frames_, callback), IsOk());
|
| const std::vector<std::unique_ptr<WebSocketFrame>>& frames_passed =
|
| *stub.frames();
|
| ASSERT_EQ(1u, frames_passed.size());
|
|
|