| Index: net/quic/quic_packet_generator_test.cc
|
| diff --git a/net/quic/quic_packet_generator_test.cc b/net/quic/quic_packet_generator_test.cc
|
| index 081161e1c3adbe203577ff59ee7032dd7e7edd67..0e44d3918f9fd57e7ba5d40f60b1307b9807dd30 100644
|
| --- a/net/quic/quic_packet_generator_test.cc
|
| +++ b/net/quic/quic_packet_generator_test.cc
|
| @@ -42,7 +42,7 @@ class MockDelegate : public QuicPacketGenerator::DelegateInterface {
|
| MOCK_METHOD0(CreateAckFrame, QuicAckFrame*());
|
| MOCK_METHOD0(CreateFeedbackFrame, QuicCongestionFeedbackFrame*());
|
| MOCK_METHOD0(CreateStopWaitingFrame, QuicStopWaitingFrame*());
|
| - MOCK_METHOD1(OnSerializedPacket, bool(const SerializedPacket& packet));
|
| + MOCK_METHOD1(OnSerializedPacket, void(const SerializedPacket& packet));
|
| MOCK_METHOD2(CloseConnection, void(QuicErrorCode, bool));
|
|
|
| void SetCanWriteAnything() {
|
| @@ -271,8 +271,7 @@ TEST_F(QuicPacketGeneratorTest, ShouldSendAck_WritableAndShouldFlush) {
|
| delegate_.SetCanWriteOnlyNonRetransmittable();
|
|
|
| EXPECT_CALL(delegate_, CreateAckFrame()).WillOnce(Return(CreateAckFrame()));
|
| - EXPECT_CALL(delegate_, OnSerializedPacket(_)).WillOnce(
|
| - DoAll(SaveArg<0>(&packet_), Return(true)));
|
| + EXPECT_CALL(delegate_, OnSerializedPacket(_)).WillOnce(SaveArg<0>(&packet_));
|
|
|
| generator_.SetShouldSendAck(false, false);
|
| EXPECT_FALSE(generator_.HasQueuedFrames());
|
| @@ -305,8 +304,7 @@ TEST_F(QuicPacketGeneratorTest,
|
| EXPECT_CALL(delegate_, CreateStopWaitingFrame()).WillOnce(
|
| Return(CreateStopWaitingFrame()));
|
|
|
| - EXPECT_CALL(delegate_, OnSerializedPacket(_)).WillOnce(
|
| - DoAll(SaveArg<0>(&packet_), Return(true)));
|
| + EXPECT_CALL(delegate_, OnSerializedPacket(_)).WillOnce(SaveArg<0>(&packet_));
|
|
|
| generator_.SetShouldSendAck(true, true);
|
| EXPECT_FALSE(generator_.HasQueuedFrames());
|
| @@ -349,8 +347,7 @@ TEST_F(QuicPacketGeneratorTest, AddControlFrame_NotWritableBatchThenFlush) {
|
| generator_.FinishBatchOperations();
|
| EXPECT_TRUE(generator_.HasQueuedFrames());
|
|
|
| - EXPECT_CALL(delegate_, OnSerializedPacket(_)).WillOnce(
|
| - DoAll(SaveArg<0>(&packet_), Return(true)));
|
| + EXPECT_CALL(delegate_, OnSerializedPacket(_)).WillOnce(SaveArg<0>(&packet_));
|
| generator_.FlushAllQueuedFrames();
|
| EXPECT_FALSE(generator_.HasQueuedFrames());
|
|
|
| @@ -362,8 +359,7 @@ TEST_F(QuicPacketGeneratorTest, AddControlFrame_NotWritableBatchThenFlush) {
|
| TEST_F(QuicPacketGeneratorTest, AddControlFrame_WritableAndShouldFlush) {
|
| delegate_.SetCanWriteAnything();
|
|
|
| - EXPECT_CALL(delegate_, OnSerializedPacket(_)).WillOnce(
|
| - DoAll(SaveArg<0>(&packet_), Return(true)));
|
| + EXPECT_CALL(delegate_, OnSerializedPacket(_)).WillOnce(SaveArg<0>(&packet_));
|
|
|
| generator_.AddControlFrame(QuicFrame(CreateRstStreamFrame()));
|
| EXPECT_FALSE(generator_.HasQueuedFrames());
|
| @@ -397,8 +393,7 @@ TEST_F(QuicPacketGeneratorTest, ConsumeData_WritableAndShouldNotFlush) {
|
| TEST_F(QuicPacketGeneratorTest, ConsumeData_WritableAndShouldFlush) {
|
| delegate_.SetCanWriteAnything();
|
|
|
| - EXPECT_CALL(delegate_, OnSerializedPacket(_)).WillOnce(
|
| - DoAll(SaveArg<0>(&packet_), Return(true)));
|
| + EXPECT_CALL(delegate_, OnSerializedPacket(_)).WillOnce(SaveArg<0>(&packet_));
|
| QuicConsumedData consumed = generator_.ConsumeData(
|
| kHeadersStreamId, MakeIOVector("foo"), 2, true, MAY_FEC_PROTECT, NULL);
|
| EXPECT_EQ(3u, consumed.bytes_consumed);
|
| @@ -437,8 +432,7 @@ TEST_F(QuicPacketGeneratorTest, ConsumeData_BatchOperations) {
|
| EXPECT_TRUE(generator_.HasQueuedFrames());
|
|
|
| // Now both frames will be flushed out.
|
| - EXPECT_CALL(delegate_, OnSerializedPacket(_)).WillOnce(
|
| - DoAll(SaveArg<0>(&packet_), Return(true)));
|
| + EXPECT_CALL(delegate_, OnSerializedPacket(_)).WillOnce(SaveArg<0>(&packet_));
|
| generator_.FinishBatchOperations();
|
| EXPECT_FALSE(generator_.HasQueuedFrames());
|
|
|
| @@ -456,15 +450,15 @@ TEST_F(QuicPacketGeneratorTest, ConsumeDataFEC) {
|
| {
|
| InSequence dummy;
|
| EXPECT_CALL(delegate_, OnSerializedPacket(_)).WillOnce(
|
| - DoAll(SaveArg<0>(&packet_), Return(true)));
|
| + SaveArg<0>(&packet_));
|
| EXPECT_CALL(delegate_, OnSerializedPacket(_)).WillOnce(
|
| - DoAll(SaveArg<0>(&packet2_), Return(true)));
|
| + SaveArg<0>(&packet2_));
|
| EXPECT_CALL(delegate_, OnSerializedPacket(_)).WillOnce(
|
| - DoAll(SaveArg<0>(&packet3_), Return(true)));
|
| + SaveArg<0>(&packet3_));
|
| EXPECT_CALL(delegate_, OnSerializedPacket(_)).WillOnce(
|
| - DoAll(SaveArg<0>(&packet4_), Return(true)));
|
| + SaveArg<0>(&packet4_));
|
| EXPECT_CALL(delegate_, OnSerializedPacket(_)).WillOnce(
|
| - DoAll(SaveArg<0>(&packet5_), Return(true)));
|
| + SaveArg<0>(&packet5_));
|
| }
|
|
|
| // Send enough data to create 3 packets: two full and one partial. Send
|
| @@ -493,11 +487,11 @@ TEST_F(QuicPacketGeneratorTest, ConsumeDataSendsFecAtEnd) {
|
| {
|
| InSequence dummy;
|
| EXPECT_CALL(delegate_, OnSerializedPacket(_)).WillOnce(
|
| - DoAll(SaveArg<0>(&packet_), Return(true)));
|
| + SaveArg<0>(&packet_));
|
| EXPECT_CALL(delegate_, OnSerializedPacket(_)).WillOnce(
|
| - DoAll(SaveArg<0>(&packet2_), Return(true)));
|
| + SaveArg<0>(&packet2_));
|
| EXPECT_CALL(delegate_, OnSerializedPacket(_)).WillOnce(
|
| - DoAll(SaveArg<0>(&packet3_), Return(true)));
|
| + SaveArg<0>(&packet3_));
|
| }
|
|
|
| // Send enough data to create 2 packets: one full and one partial. Send
|
| @@ -533,9 +527,9 @@ TEST_F(QuicPacketGeneratorTest, ConsumeData_FramesPreviouslyQueued) {
|
| {
|
| InSequence dummy;
|
| EXPECT_CALL(delegate_, OnSerializedPacket(_)).WillOnce(
|
| - DoAll(SaveArg<0>(&packet_), Return(true)));
|
| + SaveArg<0>(&packet_));
|
| EXPECT_CALL(delegate_, OnSerializedPacket(_)).WillOnce(
|
| - DoAll(SaveArg<0>(&packet2_), Return(true)));
|
| + SaveArg<0>(&packet2_));
|
| }
|
| generator_.StartBatchOperations();
|
| // Queue enough data to prevent a stream frame with a non-zero offset from
|
| @@ -594,11 +588,11 @@ TEST_F(QuicPacketGeneratorTest, FecGroupSizeChangeWithOpenGroup) {
|
| {
|
| InSequence dummy;
|
| EXPECT_CALL(delegate_, OnSerializedPacket(_)).WillOnce(
|
| - DoAll(SaveArg<0>(&packet_), Return(true)));
|
| + SaveArg<0>(&packet_));
|
| EXPECT_CALL(delegate_, OnSerializedPacket(_)).WillOnce(
|
| - DoAll(SaveArg<0>(&packet2_), Return(true)));
|
| + SaveArg<0>(&packet2_));
|
| EXPECT_CALL(delegate_, OnSerializedPacket(_)).WillOnce(
|
| - DoAll(SaveArg<0>(&packet3_), Return(true)));
|
| + SaveArg<0>(&packet3_));
|
| }
|
| size_t data_len = 3 * kDefaultMaxPacketSize + 1;
|
| QuicConsumedData consumed = generator_.ConsumeData(
|
| @@ -615,9 +609,9 @@ TEST_F(QuicPacketGeneratorTest, FecGroupSizeChangeWithOpenGroup) {
|
| {
|
| InSequence dummy;
|
| EXPECT_CALL(delegate_, OnSerializedPacket(_)).WillOnce(
|
| - DoAll(SaveArg<0>(&packet4_), Return(true)));
|
| + SaveArg<0>(&packet4_));
|
| EXPECT_CALL(delegate_, OnSerializedPacket(_)).WillOnce(
|
| - DoAll(SaveArg<0>(&packet5_), Return(true)));
|
| + SaveArg<0>(&packet5_));
|
| }
|
| consumed = generator_.ConsumeData(7, CreateData(kDefaultMaxPacketSize), 0,
|
| true, MAY_FEC_PROTECT, NULL);
|
| @@ -635,7 +629,7 @@ TEST_F(QuicPacketGeneratorTest, SwitchFecOnOff) {
|
|
|
| // Send one unprotected data packet.
|
| EXPECT_CALL(delegate_, OnSerializedPacket(_)).WillOnce(
|
| - DoAll(SaveArg<0>(&packet_), Return(true)));
|
| + SaveArg<0>(&packet_));
|
| QuicConsumedData consumed =
|
| generator_.ConsumeData(5, CreateData(1u), 0, true, MAY_FEC_PROTECT,
|
| NULL);
|
| @@ -650,15 +644,15 @@ TEST_F(QuicPacketGeneratorTest, SwitchFecOnOff) {
|
| {
|
| InSequence dummy;
|
| EXPECT_CALL(delegate_, OnSerializedPacket(_)).WillOnce(
|
| - DoAll(SaveArg<0>(&packet2_), Return(true)));
|
| + SaveArg<0>(&packet2_));
|
| EXPECT_CALL(delegate_, OnSerializedPacket(_)).WillOnce(
|
| - DoAll(SaveArg<0>(&packet3_), Return(true)));
|
| + SaveArg<0>(&packet3_));
|
| EXPECT_CALL(delegate_, OnSerializedPacket(_)).WillOnce(
|
| - DoAll(SaveArg<0>(&packet4_), Return(true)));
|
| + SaveArg<0>(&packet4_));
|
| EXPECT_CALL(delegate_, OnSerializedPacket(_)).WillOnce(
|
| - DoAll(SaveArg<0>(&packet5_), Return(true)));
|
| + SaveArg<0>(&packet5_));
|
| EXPECT_CALL(delegate_, OnSerializedPacket(_)).WillOnce(
|
| - DoAll(SaveArg<0>(&packet6_), Return(true)));
|
| + SaveArg<0>(&packet6_));
|
| }
|
| // Send enough data to create 3 packets with MUST_FEC_PROTECT flag.
|
| size_t data_len = 2 * kDefaultMaxPacketSize + 100;
|
| @@ -676,7 +670,7 @@ TEST_F(QuicPacketGeneratorTest, SwitchFecOnOff) {
|
|
|
| // Send one unprotected data packet.
|
| EXPECT_CALL(delegate_, OnSerializedPacket(_)).WillOnce(
|
| - DoAll(SaveArg<0>(&packet7_), Return(true)));
|
| + SaveArg<0>(&packet7_));
|
| consumed = generator_.ConsumeData(7, CreateData(1u), 0, true,
|
| MAY_FEC_PROTECT, NULL);
|
| EXPECT_EQ(1u, consumed.bytes_consumed);
|
| @@ -701,7 +695,7 @@ TEST_F(QuicPacketGeneratorTest, SwitchFecOnWithPendingFrameInCreator) {
|
|
|
| // Queue protected data for sending. Should cause queued frames to be flushed.
|
| EXPECT_CALL(delegate_, OnSerializedPacket(_)).WillOnce(
|
| - DoAll(SaveArg<0>(&packet_), Return(true)));
|
| + SaveArg<0>(&packet_));
|
| EXPECT_FALSE(creator_->IsFecProtected());
|
| consumed = generator_.ConsumeData(7, CreateData(1u), 0, true,
|
| MUST_FEC_PROTECT, NULL);
|
| @@ -738,7 +732,7 @@ TEST_F(QuicPacketGeneratorTest, SwitchFecOnWithPendingFramesInGenerator) {
|
|
|
| // Queue protected data for sending. Should cause queued frames to be flushed.
|
| EXPECT_CALL(delegate_, OnSerializedPacket(_)).WillOnce(
|
| - DoAll(SaveArg<0>(&packet_), Return(true)));
|
| + SaveArg<0>(&packet_));
|
| QuicConsumedData consumed = generator_.ConsumeData(7, CreateData(1u), 0, true,
|
| MUST_FEC_PROTECT, NULL);
|
| EXPECT_EQ(1u, consumed.bytes_consumed);
|
| @@ -770,8 +764,7 @@ TEST_F(QuicPacketGeneratorTest, SwitchFecOnOffWithSubsequentFramesProtected) {
|
|
|
| // Add enough unprotected data to exceed size of current packet, so that
|
| // current packet is sent. Both frames will be sent out in a single packet.
|
| - EXPECT_CALL(delegate_, OnSerializedPacket(_)).WillOnce(
|
| - DoAll(SaveArg<0>(&packet_), Return(true)));
|
| + EXPECT_CALL(delegate_, OnSerializedPacket(_)).WillOnce(SaveArg<0>(&packet_));
|
| size_t data_len = kDefaultMaxPacketSize;
|
| consumed = generator_.ConsumeData(5, CreateData(data_len), 0, true,
|
| MAY_FEC_PROTECT, NULL);
|
| @@ -793,8 +786,7 @@ TEST_F(QuicPacketGeneratorTest, SwitchFecOnOffWithSubsequentPacketsProtected) {
|
|
|
| generator_.StartBatchOperations();
|
| // Send first packet, FEC protected.
|
| - EXPECT_CALL(delegate_, OnSerializedPacket(_)).WillOnce(
|
| - DoAll(SaveArg<0>(&packet_), Return(true)));
|
| + EXPECT_CALL(delegate_, OnSerializedPacket(_)).WillOnce(SaveArg<0>(&packet_));
|
| // Write enough data to cause a packet to be emitted.
|
| size_t data_len = kDefaultMaxPacketSize;
|
| QuicConsumedData consumed = generator_.ConsumeData(
|
| @@ -814,9 +806,9 @@ TEST_F(QuicPacketGeneratorTest, SwitchFecOnOffWithSubsequentPacketsProtected) {
|
| {
|
| InSequence dummy;
|
| EXPECT_CALL(delegate_, OnSerializedPacket(_)).WillOnce(
|
| - DoAll(SaveArg<0>(&packet2_), Return(true)));
|
| + SaveArg<0>(&packet2_));
|
| EXPECT_CALL(delegate_, OnSerializedPacket(_)).WillOnce(
|
| - DoAll(SaveArg<0>(&packet3_), Return(true)));
|
| + SaveArg<0>(&packet3_));
|
| }
|
| consumed = generator_.ConsumeData(5, CreateData(data_len), 0, true,
|
| MAY_FEC_PROTECT, NULL);
|
| @@ -844,7 +836,7 @@ TEST_F(QuicPacketGeneratorTest, SwitchFecOnOffThenOnWithCreatorProtectionOn) {
|
|
|
| // Add more unprotected data causing first packet to be sent, FEC protected.
|
| EXPECT_CALL(delegate_, OnSerializedPacket(_)).WillOnce(
|
| - DoAll(SaveArg<0>(&packet_), Return(true)));
|
| + SaveArg<0>(&packet_));
|
| size_t data_len = kDefaultMaxPacketSize;
|
| consumed = generator_.ConsumeData(5, CreateData(data_len), 0, true,
|
| MAY_FEC_PROTECT, NULL);
|
| @@ -862,9 +854,9 @@ TEST_F(QuicPacketGeneratorTest, SwitchFecOnOffThenOnWithCreatorProtectionOn) {
|
| {
|
| InSequence dummy;
|
| EXPECT_CALL(delegate_, OnSerializedPacket(_)).WillOnce(
|
| - DoAll(SaveArg<0>(&packet2_), Return(true)));
|
| + SaveArg<0>(&packet2_));
|
| EXPECT_CALL(delegate_, OnSerializedPacket(_)).WillOnce(
|
| - DoAll(SaveArg<0>(&packet3_), Return(true)));
|
| + SaveArg<0>(&packet3_));
|
| }
|
| consumed = generator_.ConsumeData(5, CreateData(data_len), 0, true,
|
| MUST_FEC_PROTECT, NULL);
|
| @@ -899,8 +891,7 @@ TEST_F(QuicPacketGeneratorTest, NotWritableThenBatchOperations) {
|
| generator_.AddControlFrame(QuicFrame(CreateGoAwayFrame()));
|
|
|
| // All five frames will be flushed out in a single packet.
|
| - EXPECT_CALL(delegate_, OnSerializedPacket(_)).WillOnce(
|
| - DoAll(SaveArg<0>(&packet_), Return(true)));
|
| + EXPECT_CALL(delegate_, OnSerializedPacket(_)).WillOnce(SaveArg<0>(&packet_));
|
| generator_.FinishBatchOperations();
|
| EXPECT_FALSE(generator_.HasQueuedFrames());
|
|
|
| @@ -932,11 +923,11 @@ TEST_F(QuicPacketGeneratorTest, NotWritableThenBatchOperations2) {
|
|
|
| {
|
| InSequence dummy;
|
| - // All five frames will be flushed out in a single packet
|
| - EXPECT_CALL(delegate_, OnSerializedPacket(_)).WillOnce(
|
| - DoAll(SaveArg<0>(&packet_), Return(true)));
|
| - EXPECT_CALL(delegate_, OnSerializedPacket(_)).WillOnce(
|
| - DoAll(SaveArg<0>(&packet2_), Return(true)));
|
| + // All five frames will be flushed out in a single packet
|
| + EXPECT_CALL(delegate_, OnSerializedPacket(_)).WillOnce(
|
| + SaveArg<0>(&packet_));
|
| + EXPECT_CALL(delegate_, OnSerializedPacket(_)).WillOnce(
|
| + SaveArg<0>(&packet2_));
|
| }
|
|
|
| // Send enough data to exceed one packet
|
|
|