| 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 6141435253269330cf995c19384647b1d5ea6e2f..df8e6301a73945c99e0ed9614f4b322e0afba7a9 100644
|
| --- a/net/quic/reliable_quic_stream_test.cc
|
| +++ b/net/quic/reliable_quic_stream_test.cc
|
| @@ -157,7 +157,7 @@ TEST_F(ReliableQuicStreamTest, WriteAllData) {
|
| QuicConnectionPeer::GetPacketCreator(connection_)->set_max_packet_length(
|
| length);
|
|
|
| - EXPECT_CALL(*session_, WritevData(kHeadersStreamId, _, _, _, _)).WillOnce(
|
| + EXPECT_CALL(*session_, WritevData(kHeadersStreamId, _, _, _, _, _)).WillOnce(
|
| Return(QuicConsumedData(kDataLen, true)));
|
| stream_->WriteOrBufferData(kData1, false, NULL);
|
| EXPECT_FALSE(HasWriteBlockedStreams());
|
| @@ -177,7 +177,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_, WritevData(kHeadersStreamId, _, _, _, _))
|
| + EXPECT_CALL(*session_, WritevData(kHeadersStreamId, _, _, _, _, _))
|
| .WillOnce(Return(QuicConsumedData(1, false)));
|
| stream_->WriteOrBufferData(StringPiece(kData1, 2), false, NULL);
|
| ASSERT_EQ(1u, write_blocked_list_->NumBlockedStreams());
|
| @@ -190,7 +190,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_, WritevData(kHeadersStreamId, _, _, _, _))
|
| + EXPECT_CALL(*session_, WritevData(kHeadersStreamId, _, _, _, _, _))
|
| .WillOnce(Return(QuicConsumedData(2, false)));
|
| stream_->WriteOrBufferData(StringPiece(kData1, 2), true, NULL);
|
| ASSERT_EQ(1u, write_blocked_list_->NumBlockedStreams());
|
| @@ -201,7 +201,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_, WritevData(kHeadersStreamId, _, _, _, _))
|
| + EXPECT_CALL(*session_, WritevData(kHeadersStreamId, _, _, _, _, _))
|
| .WillOnce(Return(QuicConsumedData(0, false)));
|
| stream_->WriteOrBufferData(StringPiece(), true, NULL);
|
| ASSERT_EQ(1u, write_blocked_list_->NumBlockedStreams());
|
| @@ -217,7 +217,7 @@ TEST_F(ReliableQuicStreamTest, WriteOrBufferData) {
|
| QuicConnectionPeer::GetPacketCreator(connection_)->set_max_packet_length(
|
| length);
|
|
|
| - EXPECT_CALL(*session_, WritevData(_, _, _, _, _)).WillOnce(
|
| + EXPECT_CALL(*session_, WritevData(_, _, _, _, _, _)).WillOnce(
|
| Return(QuicConsumedData(kDataLen - 1, false)));
|
| stream_->WriteOrBufferData(kData1, false, NULL);
|
| EXPECT_TRUE(HasWriteBlockedStreams());
|
| @@ -227,14 +227,14 @@ TEST_F(ReliableQuicStreamTest, WriteOrBufferData) {
|
|
|
| // Make sure we get the tail of the first write followed by the bytes_consumed
|
| InSequence s;
|
| - EXPECT_CALL(*session_, WritevData(_, _, _, _, _)).
|
| + EXPECT_CALL(*session_, WritevData(_, _, _, _, _, _)).
|
| WillOnce(Return(QuicConsumedData(1, false)));
|
| - EXPECT_CALL(*session_, WritevData(_, _, _, _, _)).
|
| + EXPECT_CALL(*session_, WritevData(_, _, _, _, _, _)).
|
| WillOnce(Return(QuicConsumedData(kDataLen - 2, false)));
|
| stream_->OnCanWrite();
|
|
|
| // And finally the end of the bytes_consumed.
|
| - EXPECT_CALL(*session_, WritevData(_, _, _, _, _)).
|
| + EXPECT_CALL(*session_, WritevData(_, _, _, _, _, _)).
|
| WillOnce(Return(QuicConsumedData(2, true)));
|
| stream_->OnCanWrite();
|
| }
|
| @@ -261,7 +261,7 @@ TEST_F(ReliableQuicStreamTest, RstAlwaysSentIfNoFinSent) {
|
| EXPECT_FALSE(rst_sent());
|
|
|
| // Write some data, with no FIN.
|
| - EXPECT_CALL(*session_, WritevData(kHeadersStreamId, _, _, _, _))
|
| + EXPECT_CALL(*session_, WritevData(kHeadersStreamId, _, _, _, _, _))
|
| .WillOnce(Return(QuicConsumedData(1, false)));
|
| stream_->WriteOrBufferData(StringPiece(kData1, 1), false, NULL);
|
| EXPECT_FALSE(fin_sent());
|
| @@ -284,7 +284,7 @@ TEST_F(ReliableQuicStreamTest, RstNotSentIfFinSent) {
|
| EXPECT_FALSE(rst_sent());
|
|
|
| // Write some data, with FIN.
|
| - EXPECT_CALL(*session_, WritevData(kHeadersStreamId, _, _, _, _))
|
| + EXPECT_CALL(*session_, WritevData(kHeadersStreamId, _, _, _, _, _))
|
| .WillOnce(Return(QuicConsumedData(1, true)));
|
| stream_->WriteOrBufferData(StringPiece(kData1, 1), true, NULL);
|
| EXPECT_TRUE(fin_sent());
|
| @@ -405,26 +405,26 @@ TEST_F(ReliableQuicStreamTest, WriteOrBufferDataWithQuicAckNotifier) {
|
|
|
| scoped_refptr<QuicAckNotifier::DelegateInterface> proxy_delegate;
|
|
|
| - EXPECT_CALL(*session_, WritevData(kHeadersStreamId, _, _, _, _))
|
| - .WillOnce(DoAll(WithArgs<4>(Invoke(CreateFunctor(
|
| + EXPECT_CALL(*session_, WritevData(kHeadersStreamId, _, _, _, _, _))
|
| + .WillOnce(DoAll(WithArgs<5>(Invoke(CreateFunctor(
|
| &SaveProxyAckNotifierDelegate, &proxy_delegate))),
|
| Return(QuicConsumedData(kFirstWriteSize, false))));
|
| stream_->WriteOrBufferData(kData, false, delegate.get());
|
| EXPECT_TRUE(HasWriteBlockedStreams());
|
|
|
| EXPECT_CALL(*session_,
|
| - WritevData(kHeadersStreamId, _, _, _, proxy_delegate.get()))
|
| + WritevData(kHeadersStreamId, _, _, _, _, proxy_delegate.get()))
|
| .WillOnce(Return(QuicConsumedData(kSecondWriteSize, false)));
|
| stream_->OnCanWrite();
|
|
|
| // No ack expected for an empty write.
|
| EXPECT_CALL(*session_,
|
| - WritevData(kHeadersStreamId, _, _, _, proxy_delegate.get()))
|
| + WritevData(kHeadersStreamId, _, _, _, _, proxy_delegate.get()))
|
| .WillOnce(Return(QuicConsumedData(0, false)));
|
| stream_->OnCanWrite();
|
|
|
| EXPECT_CALL(*session_,
|
| - WritevData(kHeadersStreamId, _, _, _, proxy_delegate.get()))
|
| + WritevData(kHeadersStreamId, _, _, _, _, proxy_delegate.get()))
|
| .WillOnce(Return(QuicConsumedData(kLastWriteSize, false)));
|
| stream_->OnCanWrite();
|
|
|
| @@ -460,8 +460,8 @@ TEST_F(ReliableQuicStreamTest, WriteOrBufferDataAckNotificationBeforeFlush) {
|
|
|
| scoped_refptr<QuicAckNotifier::DelegateInterface> proxy_delegate;
|
|
|
| - EXPECT_CALL(*session_, WritevData(kHeadersStreamId, _, _, _, _))
|
| - .WillOnce(DoAll(WithArgs<4>(Invoke(CreateFunctor(
|
| + EXPECT_CALL(*session_, WritevData(kHeadersStreamId, _, _, _, _, _))
|
| + .WillOnce(DoAll(WithArgs<5>(Invoke(CreateFunctor(
|
| &SaveProxyAckNotifierDelegate, &proxy_delegate))),
|
| Return(QuicConsumedData(kInitialWriteSize, false))));
|
| stream_->WriteOrBufferData(kData, false, delegate.get());
|
| @@ -471,8 +471,8 @@ TEST_F(ReliableQuicStreamTest, WriteOrBufferDataAckNotificationBeforeFlush) {
|
| proxy_delegate->OnAckNotification(1, 2, 3, 4, zero_);
|
| proxy_delegate = NULL;
|
|
|
| - EXPECT_CALL(*session_, WritevData(kHeadersStreamId, _, _, _, _)).WillOnce(
|
| - DoAll(WithArgs<4>(Invoke(CreateFunctor(
|
| + EXPECT_CALL(*session_, WritevData(kHeadersStreamId, _, _, _, _, _)).WillOnce(
|
| + DoAll(WithArgs<5>(Invoke(CreateFunctor(
|
| &SaveProxyAckNotifierDelegate, &proxy_delegate))),
|
| Return(QuicConsumedData(kDataSize - kInitialWriteSize, false))));
|
| stream_->OnCanWrite();
|
| @@ -491,8 +491,8 @@ TEST_F(ReliableQuicStreamTest, WriteAndBufferDataWithAckNotiferNoBuffer) {
|
|
|
| scoped_refptr<QuicAckNotifier::DelegateInterface> proxy_delegate;
|
|
|
| - EXPECT_CALL(*session_, WritevData(kHeadersStreamId, _, _, _, _))
|
| - .WillOnce(DoAll(WithArgs<4>(Invoke(CreateFunctor(
|
| + EXPECT_CALL(*session_, WritevData(kHeadersStreamId, _, _, _, _, _))
|
| + .WillOnce(DoAll(WithArgs<5>(Invoke(CreateFunctor(
|
| &SaveProxyAckNotifierDelegate, &proxy_delegate))),
|
| Return(QuicConsumedData(kDataLen, true))));
|
| stream_->WriteOrBufferData(kData1, true, delegate.get());
|
| @@ -512,13 +512,13 @@ TEST_F(ReliableQuicStreamTest, BufferOnWriteAndBufferDataWithAckNotifer) {
|
|
|
| scoped_refptr<QuicAckNotifier::DelegateInterface> proxy_delegate;
|
|
|
| - EXPECT_CALL(*session_, WritevData(kHeadersStreamId, _, _, _, _))
|
| + EXPECT_CALL(*session_, WritevData(kHeadersStreamId, _, _, _, _, _))
|
| .WillOnce(Return(QuicConsumedData(0, false)));
|
| stream_->WriteOrBufferData(kData1, true, delegate.get());
|
| EXPECT_TRUE(HasWriteBlockedStreams());
|
|
|
| - EXPECT_CALL(*session_, WritevData(kHeadersStreamId, _, _, _, _))
|
| - .WillOnce(DoAll(WithArgs<4>(Invoke(CreateFunctor(
|
| + EXPECT_CALL(*session_, WritevData(kHeadersStreamId, _, _, _, _, _))
|
| + .WillOnce(DoAll(WithArgs<5>(Invoke(CreateFunctor(
|
| &SaveProxyAckNotifierDelegate, &proxy_delegate))),
|
| Return(QuicConsumedData(kDataLen, true))));
|
| stream_->OnCanWrite();
|
| @@ -538,15 +538,15 @@ TEST_F(ReliableQuicStreamTest, WriteAndBufferDataWithAckNotiferOnlyFinRemains) {
|
|
|
| scoped_refptr<QuicAckNotifier::DelegateInterface> proxy_delegate;
|
|
|
| - EXPECT_CALL(*session_, WritevData(kHeadersStreamId, _, _, _, _))
|
| - .WillOnce(DoAll(WithArgs<4>(Invoke(CreateFunctor(
|
| + EXPECT_CALL(*session_, WritevData(kHeadersStreamId, _, _, _, _, _))
|
| + .WillOnce(DoAll(WithArgs<5>(Invoke(CreateFunctor(
|
| &SaveProxyAckNotifierDelegate, &proxy_delegate))),
|
| Return(QuicConsumedData(kDataLen, false))));
|
| stream_->WriteOrBufferData(kData1, true, delegate.get());
|
| EXPECT_TRUE(HasWriteBlockedStreams());
|
|
|
| - EXPECT_CALL(*session_, WritevData(kHeadersStreamId, _, _, _, _))
|
| - .WillOnce(DoAll(WithArgs<4>(Invoke(CreateFunctor(
|
| + EXPECT_CALL(*session_, WritevData(kHeadersStreamId, _, _, _, _, _))
|
| + .WillOnce(DoAll(WithArgs<5>(Invoke(CreateFunctor(
|
| &SaveProxyAckNotifierDelegate, &proxy_delegate))),
|
| Return(QuicConsumedData(0, true))));
|
| stream_->OnCanWrite();
|
|
|