| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "net/quic/quic_unacked_packet_map.h" | 5 #include "net/quic/quic_unacked_packet_map.h" |
| 6 | 6 |
| 7 #include "net/quic/test_tools/quic_test_utils.h" | 7 #include "net/quic/test_tools/quic_test_utils.h" |
| 8 #include "testing/gtest/include/gtest/gtest.h" | 8 #include "testing/gtest/include/gtest/gtest.h" |
| 9 | 9 |
| 10 namespace net { | 10 namespace net { |
| (...skipping 17 matching lines...) Expand all Loading... |
| 28 } | 28 } |
| 29 | 29 |
| 30 SerializedPacket CreateNonRetransmittablePacket( | 30 SerializedPacket CreateNonRetransmittablePacket( |
| 31 QuicPacketSequenceNumber sequence_number) { | 31 QuicPacketSequenceNumber sequence_number) { |
| 32 return SerializedPacket( | 32 return SerializedPacket( |
| 33 sequence_number, PACKET_1BYTE_SEQUENCE_NUMBER, NULL, 0, NULL); | 33 sequence_number, PACKET_1BYTE_SEQUENCE_NUMBER, NULL, 0, NULL); |
| 34 } | 34 } |
| 35 | 35 |
| 36 void VerifyInFlightPackets(QuicPacketSequenceNumber* packets, | 36 void VerifyInFlightPackets(QuicPacketSequenceNumber* packets, |
| 37 size_t num_packets) { | 37 size_t num_packets) { |
| 38 unacked_packets_.RemoveObsoletePackets(); | |
| 39 if (num_packets == 0) { | 38 if (num_packets == 0) { |
| 40 EXPECT_FALSE(unacked_packets_.HasInFlightPackets()); | 39 EXPECT_FALSE(unacked_packets_.HasInFlightPackets()); |
| 41 EXPECT_FALSE(unacked_packets_.HasMultipleInFlightPackets()); | 40 EXPECT_FALSE(unacked_packets_.HasMultipleInFlightPackets()); |
| 42 return; | 41 return; |
| 43 } | 42 } |
| 44 if (num_packets == 1) { | 43 if (num_packets == 1) { |
| 45 EXPECT_TRUE(unacked_packets_.HasInFlightPackets()); | 44 EXPECT_TRUE(unacked_packets_.HasInFlightPackets()); |
| 46 EXPECT_FALSE(unacked_packets_.HasMultipleInFlightPackets()); | 45 EXPECT_FALSE(unacked_packets_.HasMultipleInFlightPackets()); |
| 47 ASSERT_TRUE(unacked_packets_.IsUnacked(packets[0])); | 46 ASSERT_TRUE(unacked_packets_.IsUnacked(packets[0])); |
| 48 EXPECT_TRUE(unacked_packets_.GetTransmissionInfo(packets[0]).in_flight); | 47 EXPECT_TRUE(unacked_packets_.GetTransmissionInfo(packets[0]).in_flight); |
| 49 } | 48 } |
| 50 for (size_t i = 0; i < num_packets; ++i) { | 49 for (size_t i = 0; i < num_packets; ++i) { |
| 51 ASSERT_TRUE(unacked_packets_.IsUnacked(packets[i])); | 50 ASSERT_TRUE(unacked_packets_.IsUnacked(packets[i])); |
| 52 EXPECT_TRUE(unacked_packets_.GetTransmissionInfo(packets[i]).in_flight); | 51 EXPECT_TRUE(unacked_packets_.GetTransmissionInfo(packets[i]).in_flight); |
| 53 } | 52 } |
| 54 size_t in_flight_count = 0; | 53 size_t in_flight_count = 0; |
| 55 for (QuicUnackedPacketMap::const_iterator it = unacked_packets_.begin(); | 54 for (QuicUnackedPacketMap::const_iterator it = unacked_packets_.begin(); |
| 56 it != unacked_packets_.end(); ++it) { | 55 it != unacked_packets_.end(); ++it) { |
| 57 if (it->in_flight) { | 56 if (it->second.in_flight) { |
| 58 ++in_flight_count; | 57 ++in_flight_count; |
| 59 } | 58 } |
| 60 } | 59 } |
| 61 EXPECT_EQ(num_packets, in_flight_count); | 60 EXPECT_EQ(num_packets, in_flight_count); |
| 62 } | 61 } |
| 63 | 62 |
| 64 void VerifyUnackedPackets(QuicPacketSequenceNumber* packets, | 63 void VerifyUnackedPackets(QuicPacketSequenceNumber* packets, |
| 65 size_t num_packets) { | 64 size_t num_packets) { |
| 66 unacked_packets_.RemoveObsoletePackets(); | |
| 67 if (num_packets == 0) { | 65 if (num_packets == 0) { |
| 68 EXPECT_FALSE(unacked_packets_.HasUnackedPackets()); | 66 EXPECT_FALSE(unacked_packets_.HasUnackedPackets()); |
| 69 EXPECT_FALSE(unacked_packets_.HasUnackedRetransmittableFrames()); | 67 EXPECT_FALSE(unacked_packets_.HasUnackedRetransmittableFrames()); |
| 70 return; | 68 return; |
| 71 } | 69 } |
| 72 EXPECT_TRUE(unacked_packets_.HasUnackedPackets()); | 70 EXPECT_TRUE(unacked_packets_.HasUnackedPackets()); |
| 73 for (size_t i = 0; i < num_packets; ++i) { | 71 for (size_t i = 0; i < num_packets; ++i) { |
| 74 EXPECT_TRUE(unacked_packets_.IsUnacked(packets[i])) << packets[i]; | 72 EXPECT_TRUE(unacked_packets_.IsUnacked(packets[i])) << packets[i]; |
| 75 } | 73 } |
| 76 EXPECT_EQ(num_packets, unacked_packets_.GetNumUnackedPacketsDebugOnly()); | 74 size_t unacked_count = 0; |
| 75 for (QuicUnackedPacketMap::const_iterator it = unacked_packets_.begin(); |
| 76 it != unacked_packets_.end(); ++it) { |
| 77 ++unacked_count; |
| 78 } |
| 79 EXPECT_EQ(num_packets, unacked_count); |
| 77 } | 80 } |
| 78 | 81 |
| 79 void VerifyRetransmittablePackets(QuicPacketSequenceNumber* packets, | 82 void VerifyRetransmittablePackets(QuicPacketSequenceNumber* packets, |
| 80 size_t num_packets) { | 83 size_t num_packets) { |
| 81 unacked_packets_.RemoveObsoletePackets(); | |
| 82 size_t num_retransmittable_packets = 0; | 84 size_t num_retransmittable_packets = 0; |
| 83 for (QuicUnackedPacketMap::const_iterator it = unacked_packets_.begin(); | 85 for (QuicUnackedPacketMap::const_iterator it = unacked_packets_.begin(); |
| 84 it != unacked_packets_.end(); ++it) { | 86 it != unacked_packets_.end(); ++it) { |
| 85 if (it->retransmittable_frames != NULL) { | 87 if (it->second.retransmittable_frames != NULL) { |
| 86 ++num_retransmittable_packets; | 88 ++num_retransmittable_packets; |
| 87 } | 89 } |
| 88 } | 90 } |
| 89 EXPECT_EQ(num_packets, num_retransmittable_packets); | 91 EXPECT_EQ(num_packets, num_retransmittable_packets); |
| 90 for (size_t i = 0; i < num_packets; ++i) { | 92 for (size_t i = 0; i < num_packets; ++i) { |
| 91 EXPECT_TRUE(unacked_packets_.HasRetransmittableFrames(packets[i])) | 93 EXPECT_TRUE(unacked_packets_.HasRetransmittableFrames(packets[i])) |
| 92 << " packets[" << i << "]:" << packets[i]; | 94 << " packets[" << i << "]:" << packets[i]; |
| 93 } | 95 } |
| 94 } | 96 } |
| 95 | 97 |
| (...skipping 136 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 232 VerifyUnackedPackets(unacked4, arraysize(unacked4)); | 234 VerifyUnackedPackets(unacked4, arraysize(unacked4)); |
| 233 QuicPacketSequenceNumber pending4[] = { 3, 5, 7 }; | 235 QuicPacketSequenceNumber pending4[] = { 3, 5, 7 }; |
| 234 VerifyInFlightPackets(pending4, arraysize(pending4)); | 236 VerifyInFlightPackets(pending4, arraysize(pending4)); |
| 235 QuicPacketSequenceNumber retransmittable4[] = { 7 }; | 237 QuicPacketSequenceNumber retransmittable4[] = { 7 }; |
| 236 VerifyRetransmittablePackets(retransmittable4, arraysize(retransmittable4)); | 238 VerifyRetransmittablePackets(retransmittable4, arraysize(retransmittable4)); |
| 237 } | 239 } |
| 238 | 240 |
| 239 } // namespace | 241 } // namespace |
| 240 } // namespace test | 242 } // namespace test |
| 241 } // namespace net | 243 } // namespace net |
| OLD | NEW |