mflodman@webrtc.org | eaf7cf2 | 2012-12-11 11:47:22 +0000 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (c) 2012 The WebRTC project authors. All Rights Reserved. |
| 3 | * |
| 4 | * Use of this source code is governed by a BSD-style license |
| 5 | * that can be found in the LICENSE file in the root of the source |
| 6 | * tree. An additional intellectual property rights grant can be found |
| 7 | * in the file PATENTS. All contributing project authors may |
| 8 | * be found in the AUTHORS file in the root of the source tree. |
| 9 | */ |
| 10 | |
kwiberg | bfefb03 | 2016-05-01 14:53:46 -0700 | [diff] [blame] | 11 | #include <memory> |
| 12 | |
Mirko Bonadei | 92ea95e | 2017-09-15 06:47:31 +0200 | [diff] [blame] | 13 | #include "call/call.h" |
| 14 | #include "modules/rtp_rtcp/include/rtp_header_parser.h" |
| 15 | #include "system_wrappers/include/clock.h" |
| 16 | #include "test/fake_network_pipe.h" |
| 17 | #include "test/gmock.h" |
| 18 | #include "test/gtest.h" |
mflodman@webrtc.org | eaf7cf2 | 2012-12-11 11:47:22 +0000 | [diff] [blame] | 19 | |
| 20 | using ::testing::_; |
| 21 | using ::testing::AnyNumber; |
| 22 | using ::testing::Return; |
| 23 | using ::testing::Invoke; |
| 24 | |
| 25 | namespace webrtc { |
| 26 | |
minyue | 20c84cc | 2017-04-10 16:57:57 -0700 | [diff] [blame] | 27 | class TestDemuxer : public Demuxer { |
mflodman@webrtc.org | eaf7cf2 | 2012-12-11 11:47:22 +0000 | [diff] [blame] | 28 | public: |
minyue | 20c84cc | 2017-04-10 16:57:57 -0700 | [diff] [blame] | 29 | void IncomingPacket(NetworkPacket* packet) { |
| 30 | DeliverPacket(packet, PacketTime()); |
mflodman@webrtc.org | eaf7cf2 | 2012-12-11 11:47:22 +0000 | [diff] [blame] | 31 | } |
| 32 | |
minyue | 20c84cc | 2017-04-10 16:57:57 -0700 | [diff] [blame] | 33 | MOCK_METHOD1(SetReceiver, void(PacketReceiver* receiver)); |
| 34 | MOCK_METHOD2(DeliverPacket, |
| 35 | void(const NetworkPacket* packet, |
| 36 | const PacketTime& packet_time)); |
mflodman@webrtc.org | eaf7cf2 | 2012-12-11 11:47:22 +0000 | [diff] [blame] | 37 | }; |
| 38 | |
minyue | 20c84cc | 2017-04-10 16:57:57 -0700 | [diff] [blame] | 39 | class ReorderTestDemuxer : public TestDemuxer { |
philipel | a2c5523 | 2016-01-26 08:41:53 -0800 | [diff] [blame] | 40 | public: |
minyue | 20c84cc | 2017-04-10 16:57:57 -0700 | [diff] [blame] | 41 | void DeliverPacket(const NetworkPacket* packet, |
| 42 | const PacketTime& packet_time) override { |
| 43 | RTC_DCHECK_GE(packet->data_length(), sizeof(int)); |
philipel | a2c5523 | 2016-01-26 08:41:53 -0800 | [diff] [blame] | 44 | int seq_num; |
minyue | 20c84cc | 2017-04-10 16:57:57 -0700 | [diff] [blame] | 45 | memcpy(&seq_num, packet->data(), sizeof(int)); |
philipel | a2c5523 | 2016-01-26 08:41:53 -0800 | [diff] [blame] | 46 | delivered_sequence_numbers_.push_back(seq_num); |
philipel | a2c5523 | 2016-01-26 08:41:53 -0800 | [diff] [blame] | 47 | } |
| 48 | std::vector<int> delivered_sequence_numbers_; |
| 49 | }; |
| 50 | |
minyue | 20c84cc | 2017-04-10 16:57:57 -0700 | [diff] [blame] | 51 | class MockReceiver : public PacketReceiver { |
| 52 | public: |
| 53 | MOCK_METHOD4( |
| 54 | DeliverPacket, |
| 55 | DeliveryStatus(MediaType, const uint8_t*, size_t, const PacketTime&)); |
| 56 | }; |
| 57 | |
mflodman@webrtc.org | eaf7cf2 | 2012-12-11 11:47:22 +0000 | [diff] [blame] | 58 | class FakeNetworkPipeTest : public ::testing::Test { |
Peter Boström | d3c9447 | 2015-12-09 11:20:58 +0100 | [diff] [blame] | 59 | public: |
| 60 | FakeNetworkPipeTest() : fake_clock_(12345) {} |
| 61 | |
mflodman@webrtc.org | eaf7cf2 | 2012-12-11 11:47:22 +0000 | [diff] [blame] | 62 | protected: |
philipel | a2c5523 | 2016-01-26 08:41:53 -0800 | [diff] [blame] | 63 | void SendPackets(FakeNetworkPipe* pipe, int number_packets, int packet_size) { |
kwiberg | 352444f | 2016-11-28 15:58:53 -0800 | [diff] [blame] | 64 | RTC_DCHECK_GE(packet_size, sizeof(int)); |
kwiberg | bfefb03 | 2016-05-01 14:53:46 -0700 | [diff] [blame] | 65 | std::unique_ptr<uint8_t[]> packet(new uint8_t[packet_size]); |
mflodman@webrtc.org | eaf7cf2 | 2012-12-11 11:47:22 +0000 | [diff] [blame] | 66 | for (int i = 0; i < number_packets; ++i) { |
philipel | a2c5523 | 2016-01-26 08:41:53 -0800 | [diff] [blame] | 67 | // Set a sequence number for the packets by |
| 68 | // using the first bytes in the packet. |
| 69 | memcpy(packet.get(), &i, sizeof(int)); |
| 70 | pipe->SendPacket(packet.get(), packet_size); |
mflodman@webrtc.org | eaf7cf2 | 2012-12-11 11:47:22 +0000 | [diff] [blame] | 71 | } |
| 72 | } |
| 73 | |
philipel | a2c5523 | 2016-01-26 08:41:53 -0800 | [diff] [blame] | 74 | int PacketTimeMs(int capacity_kbps, int packet_size) const { |
| 75 | return 8 * packet_size / capacity_kbps; |
mflodman@webrtc.org | eaf7cf2 | 2012-12-11 11:47:22 +0000 | [diff] [blame] | 76 | } |
| 77 | |
Peter Boström | d3c9447 | 2015-12-09 11:20:58 +0100 | [diff] [blame] | 78 | SimulatedClock fake_clock_; |
mflodman@webrtc.org | eaf7cf2 | 2012-12-11 11:47:22 +0000 | [diff] [blame] | 79 | }; |
| 80 | |
mflodman@webrtc.org | eaf7cf2 | 2012-12-11 11:47:22 +0000 | [diff] [blame] | 81 | // Test the capacity link and verify we get as many packets as we expect. |
| 82 | TEST_F(FakeNetworkPipeTest, CapacityTest) { |
stefan@webrtc.org | faada6e | 2013-12-18 20:28:25 +0000 | [diff] [blame] | 83 | FakeNetworkPipe::Config config; |
stefan@webrtc.org | b8e9e44 | 2014-07-09 11:29:06 +0000 | [diff] [blame] | 84 | config.queue_length_packets = 20; |
mflodman@webrtc.org | 7acb65a | 2012-12-13 15:53:11 +0000 | [diff] [blame] | 85 | config.link_capacity_kbps = 80; |
minyue | 20c84cc | 2017-04-10 16:57:57 -0700 | [diff] [blame] | 86 | TestDemuxer* demuxer = new TestDemuxer(); |
| 87 | std::unique_ptr<FakeNetworkPipe> pipe(new FakeNetworkPipe( |
| 88 | &fake_clock_, config, std::unique_ptr<Demuxer>(demuxer))); |
mflodman@webrtc.org | eaf7cf2 | 2012-12-11 11:47:22 +0000 | [diff] [blame] | 89 | |
| 90 | // Add 10 packets of 1000 bytes, = 80 kb, and verify it takes one second to |
| 91 | // get through the pipe. |
| 92 | const int kNumPackets = 10; |
| 93 | const int kPacketSize = 1000; |
minyue | 20c84cc | 2017-04-10 16:57:57 -0700 | [diff] [blame] | 94 | SendPackets(pipe.get(), kNumPackets, kPacketSize); |
mflodman@webrtc.org | eaf7cf2 | 2012-12-11 11:47:22 +0000 | [diff] [blame] | 95 | |
| 96 | // Time to get one packet through the link. |
mflodman@webrtc.org | 7acb65a | 2012-12-13 15:53:11 +0000 | [diff] [blame] | 97 | const int kPacketTimeMs = PacketTimeMs(config.link_capacity_kbps, |
| 98 | kPacketSize); |
mflodman@webrtc.org | eaf7cf2 | 2012-12-11 11:47:22 +0000 | [diff] [blame] | 99 | |
| 100 | // Time haven't increased yet, so we souldn't get any packets. |
minyue | 20c84cc | 2017-04-10 16:57:57 -0700 | [diff] [blame] | 101 | EXPECT_CALL(*demuxer, DeliverPacket(_, _)).Times(0); |
stefan@webrtc.org | faada6e | 2013-12-18 20:28:25 +0000 | [diff] [blame] | 102 | pipe->Process(); |
mflodman@webrtc.org | eaf7cf2 | 2012-12-11 11:47:22 +0000 | [diff] [blame] | 103 | |
| 104 | // Advance enough time to release one packet. |
Peter Boström | d3c9447 | 2015-12-09 11:20:58 +0100 | [diff] [blame] | 105 | fake_clock_.AdvanceTimeMilliseconds(kPacketTimeMs); |
minyue | 20c84cc | 2017-04-10 16:57:57 -0700 | [diff] [blame] | 106 | EXPECT_CALL(*demuxer, DeliverPacket(_, _)).Times(1); |
stefan@webrtc.org | faada6e | 2013-12-18 20:28:25 +0000 | [diff] [blame] | 107 | pipe->Process(); |
mflodman@webrtc.org | eaf7cf2 | 2012-12-11 11:47:22 +0000 | [diff] [blame] | 108 | |
| 109 | // Release all but one packet |
Peter Boström | d3c9447 | 2015-12-09 11:20:58 +0100 | [diff] [blame] | 110 | fake_clock_.AdvanceTimeMilliseconds(9 * kPacketTimeMs - 1); |
minyue | 20c84cc | 2017-04-10 16:57:57 -0700 | [diff] [blame] | 111 | EXPECT_CALL(*demuxer, DeliverPacket(_, _)).Times(8); |
stefan@webrtc.org | faada6e | 2013-12-18 20:28:25 +0000 | [diff] [blame] | 112 | pipe->Process(); |
mflodman@webrtc.org | eaf7cf2 | 2012-12-11 11:47:22 +0000 | [diff] [blame] | 113 | |
| 114 | // And the last one. |
Peter Boström | d3c9447 | 2015-12-09 11:20:58 +0100 | [diff] [blame] | 115 | fake_clock_.AdvanceTimeMilliseconds(1); |
minyue | 20c84cc | 2017-04-10 16:57:57 -0700 | [diff] [blame] | 116 | EXPECT_CALL(*demuxer, DeliverPacket(_, _)).Times(1); |
stefan@webrtc.org | faada6e | 2013-12-18 20:28:25 +0000 | [diff] [blame] | 117 | pipe->Process(); |
mflodman@webrtc.org | eaf7cf2 | 2012-12-11 11:47:22 +0000 | [diff] [blame] | 118 | } |
| 119 | |
| 120 | // Test the extra network delay. |
| 121 | TEST_F(FakeNetworkPipeTest, ExtraDelayTest) { |
stefan@webrtc.org | faada6e | 2013-12-18 20:28:25 +0000 | [diff] [blame] | 122 | FakeNetworkPipe::Config config; |
stefan@webrtc.org | b8e9e44 | 2014-07-09 11:29:06 +0000 | [diff] [blame] | 123 | config.queue_length_packets = 20; |
mflodman@webrtc.org | 7acb65a | 2012-12-13 15:53:11 +0000 | [diff] [blame] | 124 | config.queue_delay_ms = 100; |
| 125 | config.link_capacity_kbps = 80; |
minyue | 20c84cc | 2017-04-10 16:57:57 -0700 | [diff] [blame] | 126 | TestDemuxer* demuxer = new TestDemuxer(); |
| 127 | std::unique_ptr<FakeNetworkPipe> pipe(new FakeNetworkPipe( |
| 128 | &fake_clock_, config, std::unique_ptr<Demuxer>(demuxer))); |
mflodman@webrtc.org | eaf7cf2 | 2012-12-11 11:47:22 +0000 | [diff] [blame] | 129 | |
| 130 | const int kNumPackets = 2; |
| 131 | const int kPacketSize = 1000; |
minyue | 20c84cc | 2017-04-10 16:57:57 -0700 | [diff] [blame] | 132 | SendPackets(pipe.get(), kNumPackets, kPacketSize); |
mflodman@webrtc.org | eaf7cf2 | 2012-12-11 11:47:22 +0000 | [diff] [blame] | 133 | |
| 134 | // Time to get one packet through the link. |
mflodman@webrtc.org | 7acb65a | 2012-12-13 15:53:11 +0000 | [diff] [blame] | 135 | const int kPacketTimeMs = PacketTimeMs(config.link_capacity_kbps, |
| 136 | kPacketSize); |
mflodman@webrtc.org | eaf7cf2 | 2012-12-11 11:47:22 +0000 | [diff] [blame] | 137 | |
| 138 | // Increase more than kPacketTimeMs, but not more than the extra delay. |
Peter Boström | d3c9447 | 2015-12-09 11:20:58 +0100 | [diff] [blame] | 139 | fake_clock_.AdvanceTimeMilliseconds(kPacketTimeMs); |
minyue | 20c84cc | 2017-04-10 16:57:57 -0700 | [diff] [blame] | 140 | EXPECT_CALL(*demuxer, DeliverPacket(_, _)).Times(0); |
stefan@webrtc.org | faada6e | 2013-12-18 20:28:25 +0000 | [diff] [blame] | 141 | pipe->Process(); |
mflodman@webrtc.org | eaf7cf2 | 2012-12-11 11:47:22 +0000 | [diff] [blame] | 142 | |
| 143 | // Advance the network delay to get the first packet. |
Peter Boström | d3c9447 | 2015-12-09 11:20:58 +0100 | [diff] [blame] | 144 | fake_clock_.AdvanceTimeMilliseconds(config.queue_delay_ms); |
minyue | 20c84cc | 2017-04-10 16:57:57 -0700 | [diff] [blame] | 145 | EXPECT_CALL(*demuxer, DeliverPacket(_, _)).Times(1); |
stefan@webrtc.org | faada6e | 2013-12-18 20:28:25 +0000 | [diff] [blame] | 146 | pipe->Process(); |
mflodman@webrtc.org | eaf7cf2 | 2012-12-11 11:47:22 +0000 | [diff] [blame] | 147 | |
| 148 | // Advance one more kPacketTimeMs to get the last packet. |
Peter Boström | d3c9447 | 2015-12-09 11:20:58 +0100 | [diff] [blame] | 149 | fake_clock_.AdvanceTimeMilliseconds(kPacketTimeMs); |
minyue | 20c84cc | 2017-04-10 16:57:57 -0700 | [diff] [blame] | 150 | EXPECT_CALL(*demuxer, DeliverPacket(_, _)).Times(1); |
stefan@webrtc.org | faada6e | 2013-12-18 20:28:25 +0000 | [diff] [blame] | 151 | pipe->Process(); |
mflodman@webrtc.org | eaf7cf2 | 2012-12-11 11:47:22 +0000 | [diff] [blame] | 152 | } |
| 153 | |
| 154 | // Test the number of buffers and packets are dropped when sending too many |
| 155 | // packets too quickly. |
| 156 | TEST_F(FakeNetworkPipeTest, QueueLengthTest) { |
stefan@webrtc.org | faada6e | 2013-12-18 20:28:25 +0000 | [diff] [blame] | 157 | FakeNetworkPipe::Config config; |
stefan@webrtc.org | b8e9e44 | 2014-07-09 11:29:06 +0000 | [diff] [blame] | 158 | config.queue_length_packets = 2; |
mflodman@webrtc.org | 7acb65a | 2012-12-13 15:53:11 +0000 | [diff] [blame] | 159 | config.link_capacity_kbps = 80; |
minyue | 20c84cc | 2017-04-10 16:57:57 -0700 | [diff] [blame] | 160 | TestDemuxer* demuxer = new TestDemuxer(); |
| 161 | std::unique_ptr<FakeNetworkPipe> pipe(new FakeNetworkPipe( |
| 162 | &fake_clock_, config, std::unique_ptr<Demuxer>(demuxer))); |
mflodman@webrtc.org | eaf7cf2 | 2012-12-11 11:47:22 +0000 | [diff] [blame] | 163 | |
| 164 | const int kPacketSize = 1000; |
mflodman@webrtc.org | 7acb65a | 2012-12-13 15:53:11 +0000 | [diff] [blame] | 165 | const int kPacketTimeMs = PacketTimeMs(config.link_capacity_kbps, |
| 166 | kPacketSize); |
mflodman@webrtc.org | eaf7cf2 | 2012-12-11 11:47:22 +0000 | [diff] [blame] | 167 | |
| 168 | // Send three packets and verify only 2 are delivered. |
| 169 | SendPackets(pipe.get(), 3, kPacketSize); |
| 170 | |
| 171 | // Increase time enough to deliver all three packets, verify only two are |
| 172 | // delivered. |
Peter Boström | d3c9447 | 2015-12-09 11:20:58 +0100 | [diff] [blame] | 173 | fake_clock_.AdvanceTimeMilliseconds(3 * kPacketTimeMs); |
minyue | 20c84cc | 2017-04-10 16:57:57 -0700 | [diff] [blame] | 174 | EXPECT_CALL(*demuxer, DeliverPacket(_, _)).Times(2); |
stefan@webrtc.org | faada6e | 2013-12-18 20:28:25 +0000 | [diff] [blame] | 175 | pipe->Process(); |
mflodman@webrtc.org | eaf7cf2 | 2012-12-11 11:47:22 +0000 | [diff] [blame] | 176 | } |
| 177 | |
| 178 | // Test we get statistics as expected. |
| 179 | TEST_F(FakeNetworkPipeTest, StatisticsTest) { |
stefan@webrtc.org | faada6e | 2013-12-18 20:28:25 +0000 | [diff] [blame] | 180 | FakeNetworkPipe::Config config; |
stefan@webrtc.org | b8e9e44 | 2014-07-09 11:29:06 +0000 | [diff] [blame] | 181 | config.queue_length_packets = 2; |
mflodman@webrtc.org | 7acb65a | 2012-12-13 15:53:11 +0000 | [diff] [blame] | 182 | config.queue_delay_ms = 20; |
| 183 | config.link_capacity_kbps = 80; |
minyue | 20c84cc | 2017-04-10 16:57:57 -0700 | [diff] [blame] | 184 | TestDemuxer* demuxer = new TestDemuxer(); |
| 185 | std::unique_ptr<FakeNetworkPipe> pipe(new FakeNetworkPipe( |
| 186 | &fake_clock_, config, std::unique_ptr<Demuxer>(demuxer))); |
mflodman@webrtc.org | eaf7cf2 | 2012-12-11 11:47:22 +0000 | [diff] [blame] | 187 | |
| 188 | const int kPacketSize = 1000; |
mflodman@webrtc.org | 7acb65a | 2012-12-13 15:53:11 +0000 | [diff] [blame] | 189 | const int kPacketTimeMs = PacketTimeMs(config.link_capacity_kbps, |
| 190 | kPacketSize); |
mflodman@webrtc.org | eaf7cf2 | 2012-12-11 11:47:22 +0000 | [diff] [blame] | 191 | |
| 192 | // Send three packets and verify only 2 are delivered. |
| 193 | SendPackets(pipe.get(), 3, kPacketSize); |
Peter Boström | d3c9447 | 2015-12-09 11:20:58 +0100 | [diff] [blame] | 194 | fake_clock_.AdvanceTimeMilliseconds(3 * kPacketTimeMs + |
| 195 | config.queue_delay_ms); |
mflodman@webrtc.org | eaf7cf2 | 2012-12-11 11:47:22 +0000 | [diff] [blame] | 196 | |
minyue | 20c84cc | 2017-04-10 16:57:57 -0700 | [diff] [blame] | 197 | EXPECT_CALL(*demuxer, DeliverPacket(_, _)).Times(2); |
stefan@webrtc.org | faada6e | 2013-12-18 20:28:25 +0000 | [diff] [blame] | 198 | pipe->Process(); |
mflodman@webrtc.org | eaf7cf2 | 2012-12-11 11:47:22 +0000 | [diff] [blame] | 199 | |
mflodman@webrtc.org | 7acb65a | 2012-12-13 15:53:11 +0000 | [diff] [blame] | 200 | // Packet 1: kPacketTimeMs + config.queue_delay_ms, |
| 201 | // packet 2: 2 * kPacketTimeMs + config.queue_delay_ms => 170 ms average. |
mflodman@webrtc.org | eaf7cf2 | 2012-12-11 11:47:22 +0000 | [diff] [blame] | 202 | EXPECT_EQ(pipe->AverageDelay(), 170); |
stefan@webrtc.org | faada6e | 2013-12-18 20:28:25 +0000 | [diff] [blame] | 203 | EXPECT_EQ(pipe->sent_packets(), 2u); |
| 204 | EXPECT_EQ(pipe->dropped_packets(), 1u); |
mflodman@webrtc.org | eaf7cf2 | 2012-12-11 11:47:22 +0000 | [diff] [blame] | 205 | EXPECT_EQ(pipe->PercentageLoss(), 1/3.f); |
| 206 | } |
| 207 | |
henrik.lundin@webrtc.org | c0e9aeb | 2014-02-26 13:34:52 +0000 | [diff] [blame] | 208 | // Change the link capacity half-way through the test and verify that the |
| 209 | // delivery times change accordingly. |
| 210 | TEST_F(FakeNetworkPipeTest, ChangingCapacityWithEmptyPipeTest) { |
| 211 | FakeNetworkPipe::Config config; |
stefan@webrtc.org | b8e9e44 | 2014-07-09 11:29:06 +0000 | [diff] [blame] | 212 | config.queue_length_packets = 20; |
henrik.lundin@webrtc.org | c0e9aeb | 2014-02-26 13:34:52 +0000 | [diff] [blame] | 213 | config.link_capacity_kbps = 80; |
minyue | 20c84cc | 2017-04-10 16:57:57 -0700 | [diff] [blame] | 214 | TestDemuxer* demuxer = new TestDemuxer(); |
| 215 | std::unique_ptr<FakeNetworkPipe> pipe(new FakeNetworkPipe( |
| 216 | &fake_clock_, config, std::unique_ptr<Demuxer>(demuxer))); |
henrik.lundin@webrtc.org | c0e9aeb | 2014-02-26 13:34:52 +0000 | [diff] [blame] | 217 | |
| 218 | // Add 10 packets of 1000 bytes, = 80 kb, and verify it takes one second to |
| 219 | // get through the pipe. |
| 220 | const int kNumPackets = 10; |
| 221 | const int kPacketSize = 1000; |
| 222 | SendPackets(pipe.get(), kNumPackets, kPacketSize); |
| 223 | |
| 224 | // Time to get one packet through the link. |
| 225 | int packet_time_ms = PacketTimeMs(config.link_capacity_kbps, kPacketSize); |
| 226 | |
| 227 | // Time hasn't increased yet, so we souldn't get any packets. |
minyue | 20c84cc | 2017-04-10 16:57:57 -0700 | [diff] [blame] | 228 | EXPECT_CALL(*demuxer, DeliverPacket(_, _)).Times(0); |
henrik.lundin@webrtc.org | c0e9aeb | 2014-02-26 13:34:52 +0000 | [diff] [blame] | 229 | pipe->Process(); |
| 230 | |
| 231 | // Advance time in steps to release one packet at a time. |
| 232 | for (int i = 0; i < kNumPackets; ++i) { |
Peter Boström | d3c9447 | 2015-12-09 11:20:58 +0100 | [diff] [blame] | 233 | fake_clock_.AdvanceTimeMilliseconds(packet_time_ms); |
minyue | 20c84cc | 2017-04-10 16:57:57 -0700 | [diff] [blame] | 234 | EXPECT_CALL(*demuxer, DeliverPacket(_, _)).Times(1); |
henrik.lundin@webrtc.org | c0e9aeb | 2014-02-26 13:34:52 +0000 | [diff] [blame] | 235 | pipe->Process(); |
| 236 | } |
| 237 | |
| 238 | // Change the capacity. |
| 239 | config.link_capacity_kbps /= 2; // Reduce to 50%. |
| 240 | pipe->SetConfig(config); |
| 241 | |
| 242 | // Add another 10 packets of 1000 bytes, = 80 kb, and verify it takes two |
| 243 | // seconds to get them through the pipe. |
| 244 | SendPackets(pipe.get(), kNumPackets, kPacketSize); |
| 245 | |
| 246 | // Time to get one packet through the link. |
| 247 | packet_time_ms = PacketTimeMs(config.link_capacity_kbps, kPacketSize); |
| 248 | |
| 249 | // Time hasn't increased yet, so we souldn't get any packets. |
minyue | 20c84cc | 2017-04-10 16:57:57 -0700 | [diff] [blame] | 250 | EXPECT_CALL(*demuxer, DeliverPacket(_, _)).Times(0); |
henrik.lundin@webrtc.org | c0e9aeb | 2014-02-26 13:34:52 +0000 | [diff] [blame] | 251 | pipe->Process(); |
| 252 | |
| 253 | // Advance time in steps to release one packet at a time. |
| 254 | for (int i = 0; i < kNumPackets; ++i) { |
Peter Boström | d3c9447 | 2015-12-09 11:20:58 +0100 | [diff] [blame] | 255 | fake_clock_.AdvanceTimeMilliseconds(packet_time_ms); |
minyue | 20c84cc | 2017-04-10 16:57:57 -0700 | [diff] [blame] | 256 | EXPECT_CALL(*demuxer, DeliverPacket(_, _)).Times(1); |
henrik.lundin@webrtc.org | c0e9aeb | 2014-02-26 13:34:52 +0000 | [diff] [blame] | 257 | pipe->Process(); |
| 258 | } |
| 259 | |
| 260 | // Check that all the packets were sent. |
| 261 | EXPECT_EQ(static_cast<size_t>(2 * kNumPackets), pipe->sent_packets()); |
Peter Boström | d3c9447 | 2015-12-09 11:20:58 +0100 | [diff] [blame] | 262 | fake_clock_.AdvanceTimeMilliseconds(pipe->TimeUntilNextProcess()); |
minyue | 20c84cc | 2017-04-10 16:57:57 -0700 | [diff] [blame] | 263 | EXPECT_CALL(*demuxer, DeliverPacket(_, _)).Times(0); |
henrik.lundin@webrtc.org | c0e9aeb | 2014-02-26 13:34:52 +0000 | [diff] [blame] | 264 | pipe->Process(); |
| 265 | } |
| 266 | |
| 267 | // Change the link capacity half-way through the test and verify that the |
| 268 | // delivery times change accordingly. |
| 269 | TEST_F(FakeNetworkPipeTest, ChangingCapacityWithPacketsInPipeTest) { |
| 270 | FakeNetworkPipe::Config config; |
stefan@webrtc.org | b8e9e44 | 2014-07-09 11:29:06 +0000 | [diff] [blame] | 271 | config.queue_length_packets = 20; |
henrik.lundin@webrtc.org | c0e9aeb | 2014-02-26 13:34:52 +0000 | [diff] [blame] | 272 | config.link_capacity_kbps = 80; |
minyue | 20c84cc | 2017-04-10 16:57:57 -0700 | [diff] [blame] | 273 | TestDemuxer* demuxer = new TestDemuxer(); |
| 274 | std::unique_ptr<FakeNetworkPipe> pipe(new FakeNetworkPipe( |
| 275 | &fake_clock_, config, std::unique_ptr<Demuxer>(demuxer))); |
henrik.lundin@webrtc.org | c0e9aeb | 2014-02-26 13:34:52 +0000 | [diff] [blame] | 276 | |
| 277 | // Add 10 packets of 1000 bytes, = 80 kb. |
| 278 | const int kNumPackets = 10; |
| 279 | const int kPacketSize = 1000; |
| 280 | SendPackets(pipe.get(), kNumPackets, kPacketSize); |
| 281 | |
| 282 | // Time to get one packet through the link at the initial speed. |
| 283 | int packet_time_1_ms = PacketTimeMs(config.link_capacity_kbps, kPacketSize); |
| 284 | |
| 285 | // Change the capacity. |
| 286 | config.link_capacity_kbps *= 2; // Double the capacity. |
| 287 | pipe->SetConfig(config); |
| 288 | |
| 289 | // Add another 10 packets of 1000 bytes, = 80 kb, and verify it takes two |
| 290 | // seconds to get them through the pipe. |
| 291 | SendPackets(pipe.get(), kNumPackets, kPacketSize); |
| 292 | |
| 293 | // Time to get one packet through the link at the new capacity. |
| 294 | int packet_time_2_ms = PacketTimeMs(config.link_capacity_kbps, kPacketSize); |
| 295 | |
| 296 | // Time hasn't increased yet, so we souldn't get any packets. |
minyue | 20c84cc | 2017-04-10 16:57:57 -0700 | [diff] [blame] | 297 | EXPECT_CALL(*demuxer, DeliverPacket(_, _)).Times(0); |
henrik.lundin@webrtc.org | c0e9aeb | 2014-02-26 13:34:52 +0000 | [diff] [blame] | 298 | pipe->Process(); |
| 299 | |
| 300 | // Advance time in steps to release one packet at a time. |
| 301 | for (int i = 0; i < kNumPackets; ++i) { |
Peter Boström | d3c9447 | 2015-12-09 11:20:58 +0100 | [diff] [blame] | 302 | fake_clock_.AdvanceTimeMilliseconds(packet_time_1_ms); |
minyue | 20c84cc | 2017-04-10 16:57:57 -0700 | [diff] [blame] | 303 | EXPECT_CALL(*demuxer, DeliverPacket(_, _)).Times(1); |
henrik.lundin@webrtc.org | c0e9aeb | 2014-02-26 13:34:52 +0000 | [diff] [blame] | 304 | pipe->Process(); |
| 305 | } |
| 306 | |
| 307 | // Advance time in steps to release one packet at a time. |
| 308 | for (int i = 0; i < kNumPackets; ++i) { |
Peter Boström | d3c9447 | 2015-12-09 11:20:58 +0100 | [diff] [blame] | 309 | fake_clock_.AdvanceTimeMilliseconds(packet_time_2_ms); |
minyue | 20c84cc | 2017-04-10 16:57:57 -0700 | [diff] [blame] | 310 | EXPECT_CALL(*demuxer, DeliverPacket(_, _)).Times(1); |
henrik.lundin@webrtc.org | c0e9aeb | 2014-02-26 13:34:52 +0000 | [diff] [blame] | 311 | pipe->Process(); |
| 312 | } |
| 313 | |
| 314 | // Check that all the packets were sent. |
| 315 | EXPECT_EQ(static_cast<size_t>(2 * kNumPackets), pipe->sent_packets()); |
Peter Boström | d3c9447 | 2015-12-09 11:20:58 +0100 | [diff] [blame] | 316 | fake_clock_.AdvanceTimeMilliseconds(pipe->TimeUntilNextProcess()); |
minyue | 20c84cc | 2017-04-10 16:57:57 -0700 | [diff] [blame] | 317 | EXPECT_CALL(*demuxer, DeliverPacket(_, _)).Times(0); |
henrik.lundin@webrtc.org | c0e9aeb | 2014-02-26 13:34:52 +0000 | [diff] [blame] | 318 | pipe->Process(); |
| 319 | } |
philipel | a2c5523 | 2016-01-26 08:41:53 -0800 | [diff] [blame] | 320 | |
| 321 | // At first disallow reordering and then allow reordering. |
| 322 | TEST_F(FakeNetworkPipeTest, DisallowReorderingThenAllowReordering) { |
| 323 | FakeNetworkPipe::Config config; |
| 324 | config.queue_length_packets = 1000; |
| 325 | config.link_capacity_kbps = 800; |
| 326 | config.queue_delay_ms = 100; |
| 327 | config.delay_standard_deviation_ms = 10; |
minyue | 20c84cc | 2017-04-10 16:57:57 -0700 | [diff] [blame] | 328 | ReorderTestDemuxer* demuxer = new ReorderTestDemuxer(); |
| 329 | std::unique_ptr<FakeNetworkPipe> pipe(new FakeNetworkPipe( |
| 330 | &fake_clock_, config, std::unique_ptr<Demuxer>(demuxer))); |
philipel | a2c5523 | 2016-01-26 08:41:53 -0800 | [diff] [blame] | 331 | |
| 332 | const uint32_t kNumPackets = 100; |
| 333 | const int kPacketSize = 10; |
| 334 | SendPackets(pipe.get(), kNumPackets, kPacketSize); |
| 335 | fake_clock_.AdvanceTimeMilliseconds(1000); |
| 336 | pipe->Process(); |
| 337 | |
| 338 | // Confirm that all packets have been delivered in order. |
minyue | 20c84cc | 2017-04-10 16:57:57 -0700 | [diff] [blame] | 339 | EXPECT_EQ(kNumPackets, demuxer->delivered_sequence_numbers_.size()); |
philipel | a2c5523 | 2016-01-26 08:41:53 -0800 | [diff] [blame] | 340 | int last_seq_num = -1; |
minyue | 20c84cc | 2017-04-10 16:57:57 -0700 | [diff] [blame] | 341 | for (int seq_num : demuxer->delivered_sequence_numbers_) { |
philipel | a2c5523 | 2016-01-26 08:41:53 -0800 | [diff] [blame] | 342 | EXPECT_GT(seq_num, last_seq_num); |
| 343 | last_seq_num = seq_num; |
| 344 | } |
| 345 | |
| 346 | config.allow_reordering = true; |
| 347 | pipe->SetConfig(config); |
| 348 | SendPackets(pipe.get(), kNumPackets, kPacketSize); |
| 349 | fake_clock_.AdvanceTimeMilliseconds(1000); |
minyue | 20c84cc | 2017-04-10 16:57:57 -0700 | [diff] [blame] | 350 | demuxer->delivered_sequence_numbers_.clear(); |
philipel | a2c5523 | 2016-01-26 08:41:53 -0800 | [diff] [blame] | 351 | pipe->Process(); |
| 352 | |
| 353 | // Confirm that all packets have been delivered |
| 354 | // and that reordering has occured. |
minyue | 20c84cc | 2017-04-10 16:57:57 -0700 | [diff] [blame] | 355 | EXPECT_EQ(kNumPackets, demuxer->delivered_sequence_numbers_.size()); |
philipel | a2c5523 | 2016-01-26 08:41:53 -0800 | [diff] [blame] | 356 | bool reordering_has_occured = false; |
| 357 | last_seq_num = -1; |
minyue | 20c84cc | 2017-04-10 16:57:57 -0700 | [diff] [blame] | 358 | for (int seq_num : demuxer->delivered_sequence_numbers_) { |
philipel | a2c5523 | 2016-01-26 08:41:53 -0800 | [diff] [blame] | 359 | if (last_seq_num > seq_num) { |
| 360 | reordering_has_occured = true; |
| 361 | break; |
| 362 | } |
| 363 | last_seq_num = seq_num; |
| 364 | } |
| 365 | EXPECT_TRUE(reordering_has_occured); |
| 366 | } |
philipel | 536378b | 2016-05-31 03:20:23 -0700 | [diff] [blame] | 367 | |
| 368 | TEST_F(FakeNetworkPipeTest, BurstLoss) { |
| 369 | const int kLossPercent = 5; |
| 370 | const int kAvgBurstLength = 3; |
| 371 | const int kNumPackets = 10000; |
| 372 | const int kPacketSize = 10; |
| 373 | |
| 374 | FakeNetworkPipe::Config config; |
| 375 | config.queue_length_packets = kNumPackets; |
| 376 | config.loss_percent = kLossPercent; |
| 377 | config.avg_burst_loss_length = kAvgBurstLength; |
minyue | 20c84cc | 2017-04-10 16:57:57 -0700 | [diff] [blame] | 378 | ReorderTestDemuxer* demuxer = new ReorderTestDemuxer(); |
| 379 | std::unique_ptr<FakeNetworkPipe> pipe(new FakeNetworkPipe( |
| 380 | &fake_clock_, config, std::unique_ptr<Demuxer>(demuxer))); |
philipel | 536378b | 2016-05-31 03:20:23 -0700 | [diff] [blame] | 381 | |
| 382 | SendPackets(pipe.get(), kNumPackets, kPacketSize); |
| 383 | fake_clock_.AdvanceTimeMilliseconds(1000); |
| 384 | pipe->Process(); |
| 385 | |
| 386 | // Check that the average loss is |kLossPercent| percent. |
minyue | 20c84cc | 2017-04-10 16:57:57 -0700 | [diff] [blame] | 387 | int lost_packets = kNumPackets - demuxer->delivered_sequence_numbers_.size(); |
philipel | 536378b | 2016-05-31 03:20:23 -0700 | [diff] [blame] | 388 | double loss_fraction = lost_packets / static_cast<double>(kNumPackets); |
| 389 | |
| 390 | EXPECT_NEAR(kLossPercent / 100.0, loss_fraction, 0.05); |
| 391 | |
| 392 | // Find the number of bursts that has occurred. |
minyue | 20c84cc | 2017-04-10 16:57:57 -0700 | [diff] [blame] | 393 | size_t received_packets = demuxer->delivered_sequence_numbers_.size(); |
philipel | 536378b | 2016-05-31 03:20:23 -0700 | [diff] [blame] | 394 | int num_bursts = 0; |
| 395 | for (size_t i = 0; i < received_packets - 1; ++i) { |
minyue | 20c84cc | 2017-04-10 16:57:57 -0700 | [diff] [blame] | 396 | int diff = demuxer->delivered_sequence_numbers_[i + 1] - |
| 397 | demuxer->delivered_sequence_numbers_[i]; |
philipel | 536378b | 2016-05-31 03:20:23 -0700 | [diff] [blame] | 398 | if (diff > 1) |
| 399 | ++num_bursts; |
| 400 | } |
| 401 | |
| 402 | double average_burst_length = static_cast<double>(lost_packets) / num_bursts; |
| 403 | |
| 404 | EXPECT_NEAR(kAvgBurstLength, average_burst_length, 0.3); |
| 405 | } |
minyue | 20c84cc | 2017-04-10 16:57:57 -0700 | [diff] [blame] | 406 | |
| 407 | TEST_F(FakeNetworkPipeTest, SetReceiver) { |
| 408 | FakeNetworkPipe::Config config; |
| 409 | TestDemuxer* demuxer = new TestDemuxer(); |
| 410 | std::unique_ptr<FakeNetworkPipe> pipe(new FakeNetworkPipe( |
| 411 | &fake_clock_, config, std::unique_ptr<Demuxer>(demuxer))); |
| 412 | MockReceiver packet_receiver; |
| 413 | EXPECT_CALL(*demuxer, SetReceiver(&packet_receiver)).Times(1); |
| 414 | pipe->SetReceiver(&packet_receiver); |
| 415 | } |
| 416 | |
| 417 | TEST(DemuxerImplTest, Demuxing) { |
| 418 | constexpr uint8_t kVideoPayloadType = 100; |
| 419 | constexpr uint8_t kAudioPayloadType = 101; |
| 420 | constexpr int64_t kTimeNow = 12345; |
| 421 | constexpr int64_t kArrivalTime = kTimeNow - 1; |
| 422 | constexpr size_t kPacketSize = 10; |
| 423 | DemuxerImpl demuxer({{kVideoPayloadType, MediaType::VIDEO}, |
| 424 | {kAudioPayloadType, MediaType::AUDIO}}); |
| 425 | |
| 426 | MockReceiver mock_receiver; |
| 427 | demuxer.SetReceiver(&mock_receiver); |
| 428 | |
| 429 | std::vector<uint8_t> data(kPacketSize); |
| 430 | data[1] = kVideoPayloadType; |
| 431 | std::unique_ptr<NetworkPacket> packet( |
| 432 | new NetworkPacket(&data[0], kPacketSize, kTimeNow, kArrivalTime)); |
| 433 | EXPECT_CALL(mock_receiver, DeliverPacket(MediaType::VIDEO, _, _, _)) |
| 434 | .WillOnce(Return(PacketReceiver::DELIVERY_OK)); |
| 435 | demuxer.DeliverPacket(packet.get(), PacketTime()); |
| 436 | |
| 437 | data[1] = kAudioPayloadType; |
| 438 | packet.reset( |
| 439 | new NetworkPacket(&data[0], kPacketSize, kTimeNow, kArrivalTime)); |
| 440 | EXPECT_CALL(mock_receiver, DeliverPacket(MediaType::AUDIO, _, _, _)) |
| 441 | .WillOnce(Return(PacketReceiver::DELIVERY_OK)); |
| 442 | demuxer.DeliverPacket(packet.get(), PacketTime()); |
| 443 | } |
| 444 | |
mflodman@webrtc.org | eaf7cf2 | 2012-12-11 11:47:22 +0000 | [diff] [blame] | 445 | } // namespace webrtc |