| Index: webrtc/test/fake_network_pipe_unittest.cc
|
| diff --git a/webrtc/test/fake_network_pipe_unittest.cc b/webrtc/test/fake_network_pipe_unittest.cc
|
| index 65573438c521dc5fe9a70e3ed69668506b6eb6e0..a753fc5312e3ba89866fc2c63debca66cea60e4f 100644
|
| --- a/webrtc/test/fake_network_pipe_unittest.cc
|
| +++ b/webrtc/test/fake_network_pipe_unittest.cc
|
| @@ -29,12 +29,13 @@ class MockReceiver : public PacketReceiver {
|
| virtual ~MockReceiver() {}
|
|
|
| void IncomingPacket(const uint8_t* data, size_t length) {
|
| - DeliverPacket(MediaType::ANY, data, length);
|
| + DeliverPacket(MediaType::ANY, data, length, PacketTime());
|
| delete [] data;
|
| }
|
|
|
| - MOCK_METHOD3(DeliverPacket,
|
| - DeliveryStatus(MediaType, const uint8_t*, size_t));
|
| + MOCK_METHOD4(
|
| + DeliverPacket,
|
| + DeliveryStatus(MediaType, const uint8_t*, size_t, const PacketTime&));
|
| };
|
|
|
| class FakeNetworkPipeTest : public ::testing::Test {
|
| @@ -42,7 +43,7 @@ class FakeNetworkPipeTest : public ::testing::Test {
|
| virtual void SetUp() {
|
| TickTime::UseFakeClock(12345);
|
| receiver_.reset(new MockReceiver());
|
| - ON_CALL(*receiver_, DeliverPacket(_, _, _))
|
| + ON_CALL(*receiver_, DeliverPacket(_, _, _, _))
|
| .WillByDefault(Return(PacketReceiver::DELIVERY_OK));
|
| }
|
|
|
| @@ -84,26 +85,22 @@ TEST_F(FakeNetworkPipeTest, CapacityTest) {
|
| kPacketSize);
|
|
|
| // Time haven't increased yet, so we souldn't get any packets.
|
| - EXPECT_CALL(*receiver_, DeliverPacket(_, _, _))
|
| - .Times(0);
|
| + EXPECT_CALL(*receiver_, DeliverPacket(_, _, _, _)).Times(0);
|
| pipe->Process();
|
|
|
| // Advance enough time to release one packet.
|
| TickTime::AdvanceFakeClock(kPacketTimeMs);
|
| - EXPECT_CALL(*receiver_, DeliverPacket(_, _, _))
|
| - .Times(1);
|
| + EXPECT_CALL(*receiver_, DeliverPacket(_, _, _, _)).Times(1);
|
| pipe->Process();
|
|
|
| // Release all but one packet
|
| TickTime::AdvanceFakeClock(9 * kPacketTimeMs - 1);
|
| - EXPECT_CALL(*receiver_, DeliverPacket(_, _, _))
|
| - .Times(8);
|
| + EXPECT_CALL(*receiver_, DeliverPacket(_, _, _, _)).Times(8);
|
| pipe->Process();
|
|
|
| // And the last one.
|
| TickTime::AdvanceFakeClock(1);
|
| - EXPECT_CALL(*receiver_, DeliverPacket(_, _, _))
|
| - .Times(1);
|
| + EXPECT_CALL(*receiver_, DeliverPacket(_, _, _, _)).Times(1);
|
| pipe->Process();
|
| }
|
|
|
| @@ -126,20 +123,17 @@ TEST_F(FakeNetworkPipeTest, ExtraDelayTest) {
|
|
|
| // Increase more than kPacketTimeMs, but not more than the extra delay.
|
| TickTime::AdvanceFakeClock(kPacketTimeMs);
|
| - EXPECT_CALL(*receiver_, DeliverPacket(_, _, _))
|
| - .Times(0);
|
| + EXPECT_CALL(*receiver_, DeliverPacket(_, _, _, _)).Times(0);
|
| pipe->Process();
|
|
|
| // Advance the network delay to get the first packet.
|
| TickTime::AdvanceFakeClock(config.queue_delay_ms);
|
| - EXPECT_CALL(*receiver_, DeliverPacket(_, _, _))
|
| - .Times(1);
|
| + EXPECT_CALL(*receiver_, DeliverPacket(_, _, _, _)).Times(1);
|
| pipe->Process();
|
|
|
| // Advance one more kPacketTimeMs to get the last packet.
|
| TickTime::AdvanceFakeClock(kPacketTimeMs);
|
| - EXPECT_CALL(*receiver_, DeliverPacket(_, _, _))
|
| - .Times(1);
|
| + EXPECT_CALL(*receiver_, DeliverPacket(_, _, _, _)).Times(1);
|
| pipe->Process();
|
| }
|
|
|
| @@ -162,8 +156,7 @@ TEST_F(FakeNetworkPipeTest, QueueLengthTest) {
|
| // Increase time enough to deliver all three packets, verify only two are
|
| // delivered.
|
| TickTime::AdvanceFakeClock(3 * kPacketTimeMs);
|
| - EXPECT_CALL(*receiver_, DeliverPacket(_, _, _))
|
| - .Times(2);
|
| + EXPECT_CALL(*receiver_, DeliverPacket(_, _, _, _)).Times(2);
|
| pipe->Process();
|
| }
|
|
|
| @@ -184,8 +177,7 @@ TEST_F(FakeNetworkPipeTest, StatisticsTest) {
|
| SendPackets(pipe.get(), 3, kPacketSize);
|
| TickTime::AdvanceFakeClock(3 * kPacketTimeMs + config.queue_delay_ms);
|
|
|
| - EXPECT_CALL(*receiver_, DeliverPacket(_, _, _))
|
| - .Times(2);
|
| + EXPECT_CALL(*receiver_, DeliverPacket(_, _, _, _)).Times(2);
|
| pipe->Process();
|
|
|
| // Packet 1: kPacketTimeMs + config.queue_delay_ms,
|
| @@ -215,13 +207,13 @@ TEST_F(FakeNetworkPipeTest, ChangingCapacityWithEmptyPipeTest) {
|
| int packet_time_ms = PacketTimeMs(config.link_capacity_kbps, kPacketSize);
|
|
|
| // Time hasn't increased yet, so we souldn't get any packets.
|
| - EXPECT_CALL(*receiver_, DeliverPacket(_, _, _)).Times(0);
|
| + EXPECT_CALL(*receiver_, DeliverPacket(_, _, _, _)).Times(0);
|
| pipe->Process();
|
|
|
| // Advance time in steps to release one packet at a time.
|
| for (int i = 0; i < kNumPackets; ++i) {
|
| TickTime::AdvanceFakeClock(packet_time_ms);
|
| - EXPECT_CALL(*receiver_, DeliverPacket(_, _, _)).Times(1);
|
| + EXPECT_CALL(*receiver_, DeliverPacket(_, _, _, _)).Times(1);
|
| pipe->Process();
|
| }
|
|
|
| @@ -237,20 +229,20 @@ TEST_F(FakeNetworkPipeTest, ChangingCapacityWithEmptyPipeTest) {
|
| packet_time_ms = PacketTimeMs(config.link_capacity_kbps, kPacketSize);
|
|
|
| // Time hasn't increased yet, so we souldn't get any packets.
|
| - EXPECT_CALL(*receiver_, DeliverPacket(_, _, _)).Times(0);
|
| + EXPECT_CALL(*receiver_, DeliverPacket(_, _, _, _)).Times(0);
|
| pipe->Process();
|
|
|
| // Advance time in steps to release one packet at a time.
|
| for (int i = 0; i < kNumPackets; ++i) {
|
| TickTime::AdvanceFakeClock(packet_time_ms);
|
| - EXPECT_CALL(*receiver_, DeliverPacket(_, _, _)).Times(1);
|
| + EXPECT_CALL(*receiver_, DeliverPacket(_, _, _, _)).Times(1);
|
| pipe->Process();
|
| }
|
|
|
| // Check that all the packets were sent.
|
| EXPECT_EQ(static_cast<size_t>(2 * kNumPackets), pipe->sent_packets());
|
| TickTime::AdvanceFakeClock(pipe->TimeUntilNextProcess());
|
| - EXPECT_CALL(*receiver_, DeliverPacket(_, _, _)).Times(0);
|
| + EXPECT_CALL(*receiver_, DeliverPacket(_, _, _, _)).Times(0);
|
| pipe->Process();
|
| }
|
|
|
| @@ -283,27 +275,27 @@ TEST_F(FakeNetworkPipeTest, ChangingCapacityWithPacketsInPipeTest) {
|
| int packet_time_2_ms = PacketTimeMs(config.link_capacity_kbps, kPacketSize);
|
|
|
| // Time hasn't increased yet, so we souldn't get any packets.
|
| - EXPECT_CALL(*receiver_, DeliverPacket(_, _, _)).Times(0);
|
| + EXPECT_CALL(*receiver_, DeliverPacket(_, _, _, _)).Times(0);
|
| pipe->Process();
|
|
|
| // Advance time in steps to release one packet at a time.
|
| for (int i = 0; i < kNumPackets; ++i) {
|
| TickTime::AdvanceFakeClock(packet_time_1_ms);
|
| - EXPECT_CALL(*receiver_, DeliverPacket(_, _, _)).Times(1);
|
| + EXPECT_CALL(*receiver_, DeliverPacket(_, _, _, _)).Times(1);
|
| pipe->Process();
|
| }
|
|
|
| // Advance time in steps to release one packet at a time.
|
| for (int i = 0; i < kNumPackets; ++i) {
|
| TickTime::AdvanceFakeClock(packet_time_2_ms);
|
| - EXPECT_CALL(*receiver_, DeliverPacket(_, _, _)).Times(1);
|
| + EXPECT_CALL(*receiver_, DeliverPacket(_, _, _, _)).Times(1);
|
| pipe->Process();
|
| }
|
|
|
| // Check that all the packets were sent.
|
| EXPECT_EQ(static_cast<size_t>(2 * kNumPackets), pipe->sent_packets());
|
| TickTime::AdvanceFakeClock(pipe->TimeUntilNextProcess());
|
| - EXPECT_CALL(*receiver_, DeliverPacket(_, _, _)).Times(0);
|
| + EXPECT_CALL(*receiver_, DeliverPacket(_, _, _, _)).Times(0);
|
| pipe->Process();
|
| }
|
| } // namespace webrtc
|
|
|