| Index: net/quic/reliable_quic_stream_test.cc
|
| diff --git a/net/quic/reliable_quic_stream_test.cc b/net/quic/reliable_quic_stream_test.cc
|
| index 1df13cfc148c388ce88224b00272aa094e131c58..063554d82c9178aeed071619ac82048ee055d6f8 100644
|
| --- a/net/quic/reliable_quic_stream_test.cc
|
| +++ b/net/quic/reliable_quic_stream_test.cc
|
| @@ -126,9 +126,7 @@ TEST_F(ReliableQuicStreamTest, WriteAllData) {
|
| 1 + QuicPacketCreator::StreamFramePacketOverhead(
|
| connection_->version(), PACKET_8BYTE_GUID, !kIncludeVersion,
|
| PACKET_6BYTE_SEQUENCE_NUMBER, NOT_IN_FEC_GROUP);
|
| - // TODO(rch): figure out how to get StrEq working here.
|
| - //EXPECT_CALL(*session_, WriteData(kStreamId, StrEq(kData1), _, _)).WillOnce(
|
| - EXPECT_CALL(*session_, WriteData(kStreamId, _, _, _)).WillOnce(
|
| + EXPECT_CALL(*session_, WritevData(kStreamId, _, 1, _, _)).WillOnce(
|
| Return(QuicConsumedData(kDataLen, true)));
|
| EXPECT_EQ(kDataLen, stream_->WriteData(kData1, false).bytes_consumed);
|
| EXPECT_FALSE(write_blocked_list_->HasWriteBlockedStreams());
|
| @@ -142,7 +140,7 @@ TEST_F(ReliableQuicStreamTest, NoBlockingIfNoDataOrFin) {
|
| // Write no data and no fin. If we consume nothing we should not be write
|
| // blocked.
|
| EXPECT_DEBUG_DEATH({
|
| - EXPECT_CALL(*session_, WriteData(kStreamId, _, _, _)).WillOnce(
|
| + EXPECT_CALL(*session_, WritevData(kStreamId, _, 1, _, _)).WillOnce(
|
| Return(QuicConsumedData(0, false)));
|
| stream_->WriteData(StringPiece(), false);
|
| EXPECT_FALSE(write_blocked_list_->HasWriteBlockedStreams());
|
| @@ -155,7 +153,7 @@ TEST_F(ReliableQuicStreamTest, BlockIfOnlySomeDataConsumed) {
|
|
|
| // Write some data and no fin. If we consume some but not all of the data,
|
| // we should be write blocked a not all the data was consumed.
|
| - EXPECT_CALL(*session_, WriteData(kStreamId, _, _, _)).WillOnce(
|
| + EXPECT_CALL(*session_, WritevData(kStreamId, _, 1, _, _)).WillOnce(
|
| Return(QuicConsumedData(1, false)));
|
| stream_->WriteData(StringPiece(kData1, 2), false);
|
| ASSERT_EQ(1, write_blocked_list_->NumBlockedStreams());
|
| @@ -169,7 +167,7 @@ TEST_F(ReliableQuicStreamTest, BlockIfFinNotConsumedWithData) {
|
| // we should be write blocked because the fin was not consumed.
|
| // (This should never actually happen as the fin should be sent out with the
|
| // last data)
|
| - EXPECT_CALL(*session_, WriteData(kStreamId, _, _, _)).WillOnce(
|
| + EXPECT_CALL(*session_, WritevData(kStreamId, _, 1, _, _)).WillOnce(
|
| Return(QuicConsumedData(2, false)));
|
| stream_->WriteData(StringPiece(kData1, 2), true);
|
| ASSERT_EQ(1, write_blocked_list_->NumBlockedStreams());
|
| @@ -180,7 +178,7 @@ TEST_F(ReliableQuicStreamTest, BlockIfSoloFinNotConsumed) {
|
|
|
| // Write no data and a fin. If we consume nothing we should be write blocked,
|
| // as the fin was not consumed.
|
| - EXPECT_CALL(*session_, WriteData(kStreamId, _, _, _)).WillOnce(
|
| + EXPECT_CALL(*session_, WritevData(kStreamId, _, 1, _, _)).WillOnce(
|
| Return(QuicConsumedData(0, false)));
|
| stream_->WriteData(StringPiece(), true);
|
| ASSERT_EQ(1, write_blocked_list_->NumBlockedStreams());
|
| @@ -194,9 +192,7 @@ TEST_F(ReliableQuicStreamTest, WriteData) {
|
| 1 + QuicPacketCreator::StreamFramePacketOverhead(
|
| connection_->version(), PACKET_8BYTE_GUID, !kIncludeVersion,
|
| PACKET_6BYTE_SEQUENCE_NUMBER, NOT_IN_FEC_GROUP);
|
| - // TODO(rch): figure out how to get StrEq working here.
|
| - //EXPECT_CALL(*session_, WriteData(_, StrEq(kData1), _, _)).WillOnce(
|
| - EXPECT_CALL(*session_, WriteData(_, _, _, _)).WillOnce(
|
| + EXPECT_CALL(*session_, WritevData(_, _, 1, _, _)).WillOnce(
|
| Return(QuicConsumedData(kDataLen - 1, false)));
|
| // The return will be kDataLen, because the last byte gets buffered.
|
| EXPECT_EQ(kDataLen, stream_->WriteData(kData1, false).bytes_consumed);
|
| @@ -207,17 +203,14 @@ TEST_F(ReliableQuicStreamTest, WriteData) {
|
|
|
| // Make sure we get the tail of the first write followed by the bytes_consumed
|
| InSequence s;
|
| - //EXPECT_CALL(*session_, WriteData(_, StrEq(&kData1[kDataLen - 1]), _, _)).
|
| - EXPECT_CALL(*session_, WriteData(_, _, _, _)).
|
| + EXPECT_CALL(*session_, WritevData(_, _, 1, _, _)).
|
| WillOnce(Return(QuicConsumedData(1, false)));
|
| - //EXPECT_CALL(*session_, WriteData(_, StrEq(kData2), _, _)).
|
| - EXPECT_CALL(*session_, WriteData(_, _, _, _)).
|
| + EXPECT_CALL(*session_, WritevData(_, _, 1, _, _)).
|
| WillOnce(Return(QuicConsumedData(kDataLen - 2, false)));
|
| stream_->OnCanWrite();
|
|
|
| - // And finally the end of the bytes_consumed
|
| - //EXPECT_CALL(*session_, WriteData(_, StrEq(&kData2[kDataLen - 2]), _, _)).
|
| - EXPECT_CALL(*session_, WriteData(_, _, _, _)).
|
| + // And finally the end of the bytes_consumed.
|
| + EXPECT_CALL(*session_, WritevData(_, _, 1, _, _)).
|
| WillOnce(Return(QuicConsumedData(2, true)));
|
| stream_->OnCanWrite();
|
| }
|
| @@ -238,19 +231,20 @@ TEST_F(ReliableQuicStreamTest, ProcessHeaders) {
|
| Initialize(kShouldProcessData);
|
|
|
| string compressed_headers =
|
| - compressor_->CompressHeadersWithPriority(0, headers_);
|
| + compressor_->CompressHeadersWithPriority(kHighestPriority, headers_);
|
| QuicStreamFrame frame(kStreamId, false, 0, compressed_headers);
|
|
|
| stream_->OnStreamFrame(frame);
|
| EXPECT_EQ(SpdyUtils::SerializeUncompressedHeaders(headers_), stream_->data());
|
| - EXPECT_EQ(0u, stream_->priority());
|
| + EXPECT_EQ(static_cast<QuicPriority>(kHighestPriority),
|
| + stream_->EffectivePriority());
|
| }
|
|
|
| TEST_F(ReliableQuicStreamTest, ProcessHeadersWithInvalidHeaderId) {
|
| Initialize(kShouldProcessData);
|
|
|
| string compressed_headers =
|
| - compressor_->CompressHeadersWithPriority(0, headers_);
|
| + compressor_->CompressHeadersWithPriority(kHighestPriority, headers_);
|
| compressed_headers.replace(4, 1, 1, '\xFF'); // Illegal header id.
|
| QuicStreamFrame frame(kStreamId, false, 0, compressed_headers);
|
|
|
| @@ -262,7 +256,7 @@ TEST_F(ReliableQuicStreamTest, ProcessHeadersAndBody) {
|
| Initialize(kShouldProcessData);
|
|
|
| string compressed_headers =
|
| - compressor_->CompressHeadersWithPriority(0, headers_);
|
| + compressor_->CompressHeadersWithPriority(kHighestPriority, headers_);
|
| string body = "this is the body";
|
| string data = compressed_headers + body;
|
| QuicStreamFrame frame(kStreamId, false, 0, data);
|
| @@ -276,7 +270,7 @@ TEST_F(ReliableQuicStreamTest, ProcessHeadersAndBodyFragments) {
|
| Initialize(kShouldProcessData);
|
|
|
| string compressed_headers =
|
| - compressor_->CompressHeadersWithPriority(7, headers_);
|
| + compressor_->CompressHeadersWithPriority(kLowestPriority, headers_);
|
| string body = "this is the body";
|
| string data = compressed_headers + body;
|
|
|
| @@ -308,14 +302,15 @@ TEST_F(ReliableQuicStreamTest, ProcessHeadersAndBodyFragments) {
|
| ASSERT_EQ(SpdyUtils::SerializeUncompressedHeaders(headers_) + body,
|
| stream_->data()) << "split_point: " << split_point;
|
| }
|
| - EXPECT_EQ(7u, stream_->priority());
|
| + EXPECT_EQ(static_cast<QuicPriority>(kLowestPriority),
|
| + stream_->EffectivePriority());
|
| }
|
|
|
| TEST_F(ReliableQuicStreamTest, ProcessHeadersAndBodyReadv) {
|
| Initialize(!kShouldProcessData);
|
|
|
| string compressed_headers =
|
| - compressor_->CompressHeadersWithPriority(0, headers_);
|
| + compressor_->CompressHeadersWithPriority(kHighestPriority, headers_);
|
| string body = "this is the body";
|
| string data = compressed_headers + body;
|
| QuicStreamFrame frame(kStreamId, false, 0, data);
|
| @@ -345,7 +340,7 @@ TEST_F(ReliableQuicStreamTest, ProcessHeadersAndBodyIncrementalReadv) {
|
| Initialize(!kShouldProcessData);
|
|
|
| string compressed_headers =
|
| - compressor_->CompressHeadersWithPriority(0, headers_);
|
| + compressor_->CompressHeadersWithPriority(kHighestPriority, headers_);
|
| string body = "this is the body";
|
| string data = compressed_headers + body;
|
| QuicStreamFrame frame(kStreamId, false, 0, data);
|
| @@ -371,7 +366,7 @@ TEST_F(ReliableQuicStreamTest, ProcessHeadersUsingReadvWithMultipleIovecs) {
|
| Initialize(!kShouldProcessData);
|
|
|
| string compressed_headers =
|
| - compressor_->CompressHeadersWithPriority(0, headers_);
|
| + compressor_->CompressHeadersWithPriority(kHighestPriority, headers_);
|
| string body = "this is the body";
|
| string data = compressed_headers + body;
|
| QuicStreamFrame frame(kStreamId, false, 0, data);
|
| @@ -401,14 +396,14 @@ TEST_F(ReliableQuicStreamTest, ProcessCorruptHeadersEarly) {
|
| Initialize(kShouldProcessData);
|
|
|
| string compressed_headers1 =
|
| - compressor_->CompressHeadersWithPriority(0, headers_);
|
| + compressor_->CompressHeadersWithPriority(kHighestPriority, headers_);
|
| QuicStreamFrame frame1(stream_->id(), false, 0, compressed_headers1);
|
| string decompressed_headers1 =
|
| SpdyUtils::SerializeUncompressedHeaders(headers_);
|
|
|
| headers_["content-type"] = "text/plain";
|
| string compressed_headers2 =
|
| - compressor_->CompressHeadersWithPriority(0, headers_);
|
| + compressor_->CompressHeadersWithPriority(kHighestPriority, headers_);
|
| // Corrupt the compressed data.
|
| compressed_headers2[compressed_headers2.length() - 1] ^= 0xA1;
|
| QuicStreamFrame frame2(stream2_->id(), false, 0, compressed_headers2);
|
| @@ -441,14 +436,14 @@ TEST_F(ReliableQuicStreamTest, ProcessPartialHeadersEarly) {
|
| Initialize(kShouldProcessData);
|
|
|
| string compressed_headers1 =
|
| - compressor_->CompressHeadersWithPriority(0, headers_);
|
| + compressor_->CompressHeadersWithPriority(kHighestPriority, headers_);
|
| QuicStreamFrame frame1(stream_->id(), false, 0, compressed_headers1);
|
| string decompressed_headers1 =
|
| SpdyUtils::SerializeUncompressedHeaders(headers_);
|
|
|
| headers_["content-type"] = "text/plain";
|
| string compressed_headers2 =
|
| - compressor_->CompressHeadersWithPriority(0, headers_);
|
| + compressor_->CompressHeadersWithPriority(kHighestPriority, headers_);
|
| string partial_compressed_headers =
|
| compressed_headers2.substr(0, compressed_headers2.length() / 2);
|
| QuicStreamFrame frame2(stream2_->id(), false, 0, partial_compressed_headers);
|
| @@ -492,14 +487,14 @@ TEST_F(ReliableQuicStreamTest, ProcessHeadersEarly) {
|
| Initialize(kShouldProcessData);
|
|
|
| string compressed_headers1 =
|
| - compressor_->CompressHeadersWithPriority(0, headers_);
|
| + compressor_->CompressHeadersWithPriority(kHighestPriority, headers_);
|
| QuicStreamFrame frame1(stream_->id(), false, 0, compressed_headers1);
|
| string decompressed_headers1 =
|
| SpdyUtils::SerializeUncompressedHeaders(headers_);
|
|
|
| headers_["content-type"] = "text/plain";
|
| string compressed_headers2 =
|
| - compressor_->CompressHeadersWithPriority(0, headers_);
|
| + compressor_->CompressHeadersWithPriority(kHighestPriority, headers_);
|
| QuicStreamFrame frame2(stream2_->id(), false, 0, compressed_headers2);
|
| string decompressed_headers2 =
|
| SpdyUtils::SerializeUncompressedHeaders(headers_);
|
| @@ -528,7 +523,7 @@ TEST_F(ReliableQuicStreamTest, ProcessHeadersDelay) {
|
| Initialize(!kShouldProcessData);
|
|
|
| string compressed_headers =
|
| - compressor_->CompressHeadersWithPriority(0, headers_);
|
| + compressor_->CompressHeadersWithPriority(kHighestPriority, headers_);
|
| QuicStreamFrame frame1(stream_->id(), false, 0, compressed_headers);
|
| string decompressed_headers =
|
| SpdyUtils::SerializeUncompressedHeaders(headers_);
|
|
|