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 |