@@ -29,20 +29,21 @@ class MockReceiver : public PacketReceiver {
2929 virtual ~MockReceiver () {}
3030
3131 void IncomingPacket (const uint8_t * data, size_t length) {
32- DeliverPacket (MediaType::ANY, data, length);
32+ DeliverPacket (MediaType::ANY, data, length, PacketTime () );
3333 delete [] data;
3434 }
3535
36- MOCK_METHOD3 (DeliverPacket,
37- DeliveryStatus (MediaType, const uint8_t *, size_t ));
36+ MOCK_METHOD4 (
37+ DeliverPacket,
38+ DeliveryStatus (MediaType, const uint8_t *, size_t , const PacketTime&));
3839};
3940
4041class FakeNetworkPipeTest : public ::testing::Test {
4142 protected:
4243 virtual void SetUp () {
4344 TickTime::UseFakeClock (12345 );
4445 receiver_.reset (new MockReceiver ());
45- ON_CALL (*receiver_, DeliverPacket (_, _, _))
46+ ON_CALL (*receiver_, DeliverPacket (_, _, _, _ ))
4647 .WillByDefault (Return (PacketReceiver::DELIVERY_OK));
4748 }
4849
@@ -84,26 +85,22 @@ TEST_F(FakeNetworkPipeTest, CapacityTest) {
8485 kPacketSize );
8586
8687 // Time haven't increased yet, so we souldn't get any packets.
87- EXPECT_CALL (*receiver_, DeliverPacket (_, _, _))
88- .Times (0 );
88+ EXPECT_CALL (*receiver_, DeliverPacket (_, _, _, _)).Times (0 );
8989 pipe->Process ();
9090
9191 // Advance enough time to release one packet.
9292 TickTime::AdvanceFakeClock (kPacketTimeMs );
93- EXPECT_CALL (*receiver_, DeliverPacket (_, _, _))
94- .Times (1 );
93+ EXPECT_CALL (*receiver_, DeliverPacket (_, _, _, _)).Times (1 );
9594 pipe->Process ();
9695
9796 // Release all but one packet
9897 TickTime::AdvanceFakeClock (9 * kPacketTimeMs - 1 );
99- EXPECT_CALL (*receiver_, DeliverPacket (_, _, _))
100- .Times (8 );
98+ EXPECT_CALL (*receiver_, DeliverPacket (_, _, _, _)).Times (8 );
10199 pipe->Process ();
102100
103101 // And the last one.
104102 TickTime::AdvanceFakeClock (1 );
105- EXPECT_CALL (*receiver_, DeliverPacket (_, _, _))
106- .Times (1 );
103+ EXPECT_CALL (*receiver_, DeliverPacket (_, _, _, _)).Times (1 );
107104 pipe->Process ();
108105}
109106
@@ -126,20 +123,17 @@ TEST_F(FakeNetworkPipeTest, ExtraDelayTest) {
126123
127124 // Increase more than kPacketTimeMs, but not more than the extra delay.
128125 TickTime::AdvanceFakeClock (kPacketTimeMs );
129- EXPECT_CALL (*receiver_, DeliverPacket (_, _, _))
130- .Times (0 );
126+ EXPECT_CALL (*receiver_, DeliverPacket (_, _, _, _)).Times (0 );
131127 pipe->Process ();
132128
133129 // Advance the network delay to get the first packet.
134130 TickTime::AdvanceFakeClock (config.queue_delay_ms );
135- EXPECT_CALL (*receiver_, DeliverPacket (_, _, _))
136- .Times (1 );
131+ EXPECT_CALL (*receiver_, DeliverPacket (_, _, _, _)).Times (1 );
137132 pipe->Process ();
138133
139134 // Advance one more kPacketTimeMs to get the last packet.
140135 TickTime::AdvanceFakeClock (kPacketTimeMs );
141- EXPECT_CALL (*receiver_, DeliverPacket (_, _, _))
142- .Times (1 );
136+ EXPECT_CALL (*receiver_, DeliverPacket (_, _, _, _)).Times (1 );
143137 pipe->Process ();
144138}
145139
@@ -162,8 +156,7 @@ TEST_F(FakeNetworkPipeTest, QueueLengthTest) {
162156 // Increase time enough to deliver all three packets, verify only two are
163157 // delivered.
164158 TickTime::AdvanceFakeClock (3 * kPacketTimeMs );
165- EXPECT_CALL (*receiver_, DeliverPacket (_, _, _))
166- .Times (2 );
159+ EXPECT_CALL (*receiver_, DeliverPacket (_, _, _, _)).Times (2 );
167160 pipe->Process ();
168161}
169162
@@ -184,8 +177,7 @@ TEST_F(FakeNetworkPipeTest, StatisticsTest) {
184177 SendPackets (pipe.get (), 3 , kPacketSize );
185178 TickTime::AdvanceFakeClock (3 * kPacketTimeMs + config.queue_delay_ms );
186179
187- EXPECT_CALL (*receiver_, DeliverPacket (_, _, _))
188- .Times (2 );
180+ EXPECT_CALL (*receiver_, DeliverPacket (_, _, _, _)).Times (2 );
189181 pipe->Process ();
190182
191183 // Packet 1: kPacketTimeMs + config.queue_delay_ms,
@@ -215,13 +207,13 @@ TEST_F(FakeNetworkPipeTest, ChangingCapacityWithEmptyPipeTest) {
215207 int packet_time_ms = PacketTimeMs (config.link_capacity_kbps , kPacketSize );
216208
217209 // Time hasn't increased yet, so we souldn't get any packets.
218- EXPECT_CALL (*receiver_, DeliverPacket (_, _, _)).Times (0 );
210+ EXPECT_CALL (*receiver_, DeliverPacket (_, _, _, _ )).Times (0 );
219211 pipe->Process ();
220212
221213 // Advance time in steps to release one packet at a time.
222214 for (int i = 0 ; i < kNumPackets ; ++i) {
223215 TickTime::AdvanceFakeClock (packet_time_ms);
224- EXPECT_CALL (*receiver_, DeliverPacket (_, _, _)).Times (1 );
216+ EXPECT_CALL (*receiver_, DeliverPacket (_, _, _, _ )).Times (1 );
225217 pipe->Process ();
226218 }
227219
@@ -237,20 +229,20 @@ TEST_F(FakeNetworkPipeTest, ChangingCapacityWithEmptyPipeTest) {
237229 packet_time_ms = PacketTimeMs (config.link_capacity_kbps , kPacketSize );
238230
239231 // Time hasn't increased yet, so we souldn't get any packets.
240- EXPECT_CALL (*receiver_, DeliverPacket (_, _, _)).Times (0 );
232+ EXPECT_CALL (*receiver_, DeliverPacket (_, _, _, _ )).Times (0 );
241233 pipe->Process ();
242234
243235 // Advance time in steps to release one packet at a time.
244236 for (int i = 0 ; i < kNumPackets ; ++i) {
245237 TickTime::AdvanceFakeClock (packet_time_ms);
246- EXPECT_CALL (*receiver_, DeliverPacket (_, _, _)).Times (1 );
238+ EXPECT_CALL (*receiver_, DeliverPacket (_, _, _, _ )).Times (1 );
247239 pipe->Process ();
248240 }
249241
250242 // Check that all the packets were sent.
251243 EXPECT_EQ (static_cast <size_t >(2 * kNumPackets ), pipe->sent_packets ());
252244 TickTime::AdvanceFakeClock (pipe->TimeUntilNextProcess ());
253- EXPECT_CALL (*receiver_, DeliverPacket (_, _, _)).Times (0 );
245+ EXPECT_CALL (*receiver_, DeliverPacket (_, _, _, _ )).Times (0 );
254246 pipe->Process ();
255247}
256248
@@ -283,27 +275,27 @@ TEST_F(FakeNetworkPipeTest, ChangingCapacityWithPacketsInPipeTest) {
283275 int packet_time_2_ms = PacketTimeMs (config.link_capacity_kbps , kPacketSize );
284276
285277 // Time hasn't increased yet, so we souldn't get any packets.
286- EXPECT_CALL (*receiver_, DeliverPacket (_, _, _)).Times (0 );
278+ EXPECT_CALL (*receiver_, DeliverPacket (_, _, _, _ )).Times (0 );
287279 pipe->Process ();
288280
289281 // Advance time in steps to release one packet at a time.
290282 for (int i = 0 ; i < kNumPackets ; ++i) {
291283 TickTime::AdvanceFakeClock (packet_time_1_ms);
292- EXPECT_CALL (*receiver_, DeliverPacket (_, _, _)).Times (1 );
284+ EXPECT_CALL (*receiver_, DeliverPacket (_, _, _, _ )).Times (1 );
293285 pipe->Process ();
294286 }
295287
296288 // Advance time in steps to release one packet at a time.
297289 for (int i = 0 ; i < kNumPackets ; ++i) {
298290 TickTime::AdvanceFakeClock (packet_time_2_ms);
299- EXPECT_CALL (*receiver_, DeliverPacket (_, _, _)).Times (1 );
291+ EXPECT_CALL (*receiver_, DeliverPacket (_, _, _, _ )).Times (1 );
300292 pipe->Process ();
301293 }
302294
303295 // Check that all the packets were sent.
304296 EXPECT_EQ (static_cast <size_t >(2 * kNumPackets ), pipe->sent_packets ());
305297 TickTime::AdvanceFakeClock (pipe->TimeUntilNextProcess ());
306- EXPECT_CALL (*receiver_, DeliverPacket (_, _, _)).Times (0 );
298+ EXPECT_CALL (*receiver_, DeliverPacket (_, _, _, _ )).Times (0 );
307299 pipe->Process ();
308300}
309301} // namespace webrtc
0 commit comments