OLD | NEW |
| (Empty) |
1 // Copyright (c) 2016 The Chromium Authors. All rights reserved. | |
2 // Use of this source code is governed by a BSD-style license that can be | |
3 // found in the LICENSE file. | |
4 | |
5 #include "net/quic/core/quic_frames.h" | |
6 | |
7 #include "testing/gmock/include/gmock/gmock.h" | |
8 #include "testing/gtest/include/gtest/gtest.h" | |
9 | |
10 namespace net { | |
11 namespace test { | |
12 namespace { | |
13 | |
14 using testing::_; | |
15 | |
16 TEST(QuicFramesTest, AckFrameToString) { | |
17 QuicAckFrame frame; | |
18 frame.largest_observed = 2; | |
19 frame.ack_delay_time = QuicTime::Delta::FromMicroseconds(3); | |
20 frame.packets.Add(4); | |
21 frame.packets.Add(5); | |
22 frame.received_packet_times = { | |
23 {6, QuicTime::Zero() + QuicTime::Delta::FromMicroseconds(7)}}; | |
24 std::ostringstream stream; | |
25 stream << frame; | |
26 EXPECT_EQ( | |
27 "{ largest_observed: 2, ack_delay_time: 3, " | |
28 "packets: [ 4 5 ], received_packets: [ 6 at 7 ] }\n", | |
29 stream.str()); | |
30 } | |
31 | |
32 TEST(QuicFramesTest, PaddingFrameToString) { | |
33 QuicPaddingFrame frame; | |
34 frame.num_padding_bytes = 1; | |
35 std::ostringstream stream; | |
36 stream << frame; | |
37 EXPECT_EQ("{ num_padding_bytes: 1 }\n", stream.str()); | |
38 } | |
39 | |
40 TEST(QuicFramesTest, RstStreamFrameToString) { | |
41 QuicRstStreamFrame frame; | |
42 frame.stream_id = 1; | |
43 frame.error_code = QUIC_STREAM_CANCELLED; | |
44 std::ostringstream stream; | |
45 stream << frame; | |
46 EXPECT_EQ("{ stream_id: 1, error_code: 6 }\n", stream.str()); | |
47 } | |
48 | |
49 TEST(QuicFramesTest, ConnectionCloseFrameToString) { | |
50 QuicConnectionCloseFrame frame; | |
51 frame.error_code = QUIC_NETWORK_IDLE_TIMEOUT; | |
52 frame.error_details = "No recent network activity."; | |
53 std::ostringstream stream; | |
54 stream << frame; | |
55 EXPECT_EQ( | |
56 "{ error_code: 25, error_details: 'No recent network activity.' }\n", | |
57 stream.str()); | |
58 } | |
59 | |
60 TEST(QuicFramesTest, GoAwayFrameToString) { | |
61 QuicGoAwayFrame frame; | |
62 frame.error_code = QUIC_NETWORK_IDLE_TIMEOUT; | |
63 frame.last_good_stream_id = 2; | |
64 frame.reason_phrase = "Reason"; | |
65 std::ostringstream stream; | |
66 stream << frame; | |
67 EXPECT_EQ( | |
68 "{ error_code: 25, last_good_stream_id: 2, reason_phrase: 'Reason' }\n", | |
69 stream.str()); | |
70 } | |
71 | |
72 TEST(QuicFramesTest, WindowUpdateFrameToString) { | |
73 QuicWindowUpdateFrame frame; | |
74 std::ostringstream stream; | |
75 frame.stream_id = 1; | |
76 frame.byte_offset = 2; | |
77 stream << frame; | |
78 EXPECT_EQ("{ stream_id: 1, byte_offset: 2 }\n", stream.str()); | |
79 } | |
80 | |
81 TEST(QuicFramesTest, BlockedFrameToString) { | |
82 QuicBlockedFrame frame; | |
83 frame.stream_id = 1; | |
84 std::ostringstream stream; | |
85 stream << frame; | |
86 EXPECT_EQ("{ stream_id: 1 }\n", stream.str()); | |
87 } | |
88 | |
89 TEST(QuicFramesTest, StreamFrameToString) { | |
90 QuicStreamFrame frame; | |
91 frame.stream_id = 1; | |
92 frame.fin = false; | |
93 frame.offset = 2; | |
94 frame.data_length = 3; | |
95 std::ostringstream stream; | |
96 stream << frame; | |
97 EXPECT_EQ("{ stream_id: 1, fin: 0, offset: 2, length: 3 }\n", stream.str()); | |
98 } | |
99 | |
100 TEST(QuicFramesTest, StopWaitingFrameToString) { | |
101 QuicStopWaitingFrame frame; | |
102 frame.least_unacked = 2; | |
103 std::ostringstream stream; | |
104 stream << frame; | |
105 EXPECT_EQ("{ least_unacked: 2 }\n", stream.str()); | |
106 } | |
107 | |
108 TEST(QuicFramesTest, PathCloseFrameToString) { | |
109 QuicPathCloseFrame frame; | |
110 frame.path_id = 1; | |
111 std::ostringstream stream; | |
112 stream << frame; | |
113 EXPECT_EQ("{ path_id: 1 }\n", stream.str()); | |
114 } | |
115 | |
116 TEST(QuicFramesTest, IsAwaitingPacket) { | |
117 QuicAckFrame ack_frame1; | |
118 ack_frame1.largest_observed = 10u; | |
119 ack_frame1.packets.Add(1, 11); | |
120 EXPECT_TRUE(IsAwaitingPacket(ack_frame1, 11u, 0u)); | |
121 EXPECT_FALSE(IsAwaitingPacket(ack_frame1, 1u, 0u)); | |
122 | |
123 ack_frame1.packets.Remove(10); | |
124 EXPECT_TRUE(IsAwaitingPacket(ack_frame1, 10u, 0u)); | |
125 | |
126 QuicAckFrame ack_frame2; | |
127 ack_frame2.largest_observed = 100u; | |
128 ack_frame2.packets.Add(21, 100); | |
129 EXPECT_FALSE(IsAwaitingPacket(ack_frame2, 11u, 20u)); | |
130 EXPECT_FALSE(IsAwaitingPacket(ack_frame2, 80u, 20u)); | |
131 EXPECT_TRUE(IsAwaitingPacket(ack_frame2, 101u, 20u)); | |
132 | |
133 ack_frame2.packets.Remove(50); | |
134 EXPECT_TRUE(IsAwaitingPacket(ack_frame2, 50u, 20u)); | |
135 } | |
136 | |
137 // Tests that a queue contains the expected data after calls to Add(). | |
138 TEST(PacketNumberQueueTest, AddRange) { | |
139 PacketNumberQueue queue; | |
140 queue.Add(1, 51); | |
141 queue.Add(53); | |
142 | |
143 EXPECT_FALSE(queue.Contains(0)); | |
144 for (int i = 1; i < 51; ++i) { | |
145 EXPECT_TRUE(queue.Contains(i)); | |
146 } | |
147 EXPECT_FALSE(queue.Contains(51)); | |
148 EXPECT_FALSE(queue.Contains(52)); | |
149 EXPECT_TRUE(queue.Contains(53)); | |
150 EXPECT_FALSE(queue.Contains(54)); | |
151 EXPECT_EQ(51u, queue.NumPacketsSlow()); | |
152 EXPECT_EQ(1u, queue.Min()); | |
153 EXPECT_EQ(53u, queue.Max()); | |
154 | |
155 queue.Add(70); | |
156 EXPECT_EQ(70u, queue.Max()); | |
157 } | |
158 | |
159 // Tests that a queue contains the expected data after calls to Remove(). | |
160 TEST(PacketNumberQueueTest, Removal) { | |
161 PacketNumberQueue queue; | |
162 queue.Add(0, 100); | |
163 | |
164 EXPECT_TRUE(queue.RemoveUpTo(51)); | |
165 EXPECT_FALSE(queue.RemoveUpTo(51)); | |
166 queue.Remove(53); | |
167 | |
168 EXPECT_FALSE(queue.Contains(0)); | |
169 for (int i = 1; i < 51; ++i) { | |
170 EXPECT_FALSE(queue.Contains(i)); | |
171 } | |
172 EXPECT_TRUE(queue.Contains(51)); | |
173 EXPECT_TRUE(queue.Contains(52)); | |
174 EXPECT_FALSE(queue.Contains(53)); | |
175 EXPECT_TRUE(queue.Contains(54)); | |
176 EXPECT_EQ(48u, queue.NumPacketsSlow()); | |
177 EXPECT_EQ(51u, queue.Min()); | |
178 EXPECT_EQ(99u, queue.Max()); | |
179 | |
180 queue.Remove(51); | |
181 EXPECT_EQ(52u, queue.Min()); | |
182 queue.Remove(99); | |
183 EXPECT_EQ(98u, queue.Max()); | |
184 } | |
185 | |
186 // Tests that a queue is empty when all of its elements are removed. | |
187 TEST(PacketNumberQueueTest, Empty) { | |
188 PacketNumberQueue queue; | |
189 EXPECT_TRUE(queue.Empty()); | |
190 EXPECT_EQ(0u, queue.NumPacketsSlow()); | |
191 | |
192 queue.Add(1, 100); | |
193 EXPECT_TRUE(queue.RemoveUpTo(100)); | |
194 EXPECT_TRUE(queue.Empty()); | |
195 EXPECT_EQ(0u, queue.NumPacketsSlow()); | |
196 } | |
197 | |
198 // Tests that logging the state of a PacketNumberQueue does not crash. | |
199 TEST(PacketNumberQueueTest, LogDoesNotCrash) { | |
200 std::ostringstream oss; | |
201 PacketNumberQueue queue; | |
202 oss << queue; | |
203 | |
204 queue.Add(1); | |
205 queue.Add(50, 100); | |
206 oss << queue; | |
207 } | |
208 | |
209 // Tests that the iterators returned from a packet queue iterate over the queue. | |
210 TEST(PacketNumberQueueTest, Iterators) { | |
211 PacketNumberQueue queue; | |
212 queue.Add(1, 100); | |
213 | |
214 const std::vector<Interval<QuicPacketNumber>> actual_intervals(queue.begin(), | |
215 queue.end()); | |
216 | |
217 std::vector<Interval<QuicPacketNumber>> expected_intervals; | |
218 expected_intervals.push_back(Interval<QuicPacketNumber>(1, 100)); | |
219 | |
220 EXPECT_EQ(expected_intervals, actual_intervals); | |
221 } | |
222 | |
223 TEST(PacketNumberQueueTest, LowerBoundEquals) { | |
224 PacketNumberQueue queue; | |
225 queue.Add(1, 100); | |
226 | |
227 PacketNumberQueue::const_iterator it = queue.lower_bound(10); | |
228 ASSERT_NE(queue.end(), it); | |
229 EXPECT_TRUE(it->Contains(10u)); | |
230 | |
231 it = queue.lower_bound(101); | |
232 EXPECT_TRUE(queue.end() == it); | |
233 } | |
234 | |
235 TEST(PacketNumberQueueTest, LowerBoundGreater) { | |
236 PacketNumberQueue queue; | |
237 queue.Add(15, 25); | |
238 queue.Add(50, 100); | |
239 | |
240 PacketNumberQueue::const_iterator it = queue.lower_bound(10); | |
241 ASSERT_NE(queue.end(), it); | |
242 EXPECT_EQ(15u, it->min()); | |
243 EXPECT_EQ(25u, it->max()); | |
244 } | |
245 | |
246 TEST(PacketNumberQueueTest, IntervalLengthAndRemoveInterval) { | |
247 PacketNumberQueue queue; | |
248 queue.Add(1, 10); | |
249 queue.Add(20, 30); | |
250 queue.Add(40, 50); | |
251 EXPECT_EQ(3u, queue.NumIntervals()); | |
252 EXPECT_EQ(10u, queue.LastIntervalLength()); | |
253 queue.Remove(9, 21); | |
254 EXPECT_EQ(3u, queue.NumIntervals()); | |
255 EXPECT_FALSE(queue.Contains(9)); | |
256 EXPECT_FALSE(queue.Contains(20)); | |
257 } | |
258 | |
259 TEST(PacketNumberQueueTest, Complement) { | |
260 PacketNumberQueue queue; | |
261 queue.Add(1, 10); | |
262 queue.Add(12, 20); | |
263 queue.Add(22, 30); | |
264 queue.Complement(); | |
265 EXPECT_EQ(2u, queue.NumIntervals()); | |
266 EXPECT_TRUE(queue.Contains(10)); | |
267 EXPECT_TRUE(queue.Contains(11)); | |
268 EXPECT_TRUE(queue.Contains(20)); | |
269 EXPECT_TRUE(queue.Contains(21)); | |
270 } | |
271 | |
272 } // namespace | |
273 } // namespace test | |
274 } // namespace net | |
OLD | NEW |