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 |