pwestin@webrtc.org | b518017 | 2012-11-09 20:56:23 +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 | |
pbos@webrtc.org | db6e3f8 | 2013-07-11 09:50:05 +0000 | [diff] [blame] | 11 | #include "testing/gmock/include/gmock/gmock.h" |
| 12 | #include "testing/gtest/include/gtest/gtest.h" |
pwestin@webrtc.org | b518017 | 2012-11-09 20:56:23 +0000 | [diff] [blame] | 13 | |
| 14 | #include "webrtc/modules/pacing/include/paced_sender.h" |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 15 | #include "webrtc/system_wrappers/interface/clock.h" |
pwestin@webrtc.org | b518017 | 2012-11-09 20:56:23 +0000 | [diff] [blame] | 16 | |
pwestin@webrtc.org | db41856 | 2013-03-22 23:39:29 +0000 | [diff] [blame] | 17 | using testing::_; |
stefan@webrtc.org | c3cc375 | 2013-06-04 09:36:56 +0000 | [diff] [blame] | 18 | using testing::Return; |
pwestin@webrtc.org | b518017 | 2012-11-09 20:56:23 +0000 | [diff] [blame] | 19 | |
| 20 | namespace webrtc { |
pwestin@webrtc.org | db41856 | 2013-03-22 23:39:29 +0000 | [diff] [blame] | 21 | namespace test { |
| 22 | |
| 23 | static const int kTargetBitrate = 800; |
pwestin@webrtc.org | 52b4e88 | 2013-05-02 19:02:17 +0000 | [diff] [blame] | 24 | static const float kPaceMultiplier = 1.5f; |
pwestin@webrtc.org | b518017 | 2012-11-09 20:56:23 +0000 | [diff] [blame] | 25 | |
| 26 | class MockPacedSenderCallback : public PacedSender::Callback { |
| 27 | public: |
stefan@webrtc.org | 9b82f5a | 2013-11-13 15:29:21 +0000 | [diff] [blame] | 28 | MOCK_METHOD4(TimeToSendPacket, |
| 29 | bool(uint32_t ssrc, uint16_t sequence_number, int64_t capture_time_ms, |
| 30 | bool retransmission)); |
pwestin@webrtc.org | b518017 | 2012-11-09 20:56:23 +0000 | [diff] [blame] | 31 | MOCK_METHOD1(TimeToSendPadding, |
stefan@webrtc.org | c3cc375 | 2013-06-04 09:36:56 +0000 | [diff] [blame] | 32 | int(int bytes)); |
| 33 | }; |
| 34 | |
| 35 | class PacedSenderPadding : public PacedSender::Callback { |
| 36 | public: |
| 37 | PacedSenderPadding() : padding_sent_(0) {} |
| 38 | |
hclam@chromium.org | 2e402ce | 2013-06-20 20:18:31 +0000 | [diff] [blame] | 39 | bool TimeToSendPacket(uint32_t ssrc, uint16_t sequence_number, |
stefan@webrtc.org | 9b82f5a | 2013-11-13 15:29:21 +0000 | [diff] [blame] | 40 | int64_t capture_time_ms, bool retransmission) { |
hclam@chromium.org | 2e402ce | 2013-06-20 20:18:31 +0000 | [diff] [blame] | 41 | return true; |
stefan@webrtc.org | c3cc375 | 2013-06-04 09:36:56 +0000 | [diff] [blame] | 42 | } |
| 43 | |
| 44 | int TimeToSendPadding(int bytes) { |
| 45 | const int kPaddingPacketSize = 224; |
| 46 | int num_packets = (bytes + kPaddingPacketSize - 1) / kPaddingPacketSize; |
| 47 | padding_sent_ += kPaddingPacketSize * num_packets; |
| 48 | return kPaddingPacketSize * num_packets; |
| 49 | } |
| 50 | |
| 51 | int padding_sent() { return padding_sent_; } |
| 52 | |
| 53 | private: |
| 54 | int padding_sent_; |
pwestin@webrtc.org | b518017 | 2012-11-09 20:56:23 +0000 | [diff] [blame] | 55 | }; |
| 56 | |
| 57 | class PacedSenderTest : public ::testing::Test { |
| 58 | protected: |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 59 | PacedSenderTest() : clock_(123456) { |
stefan@webrtc.org | c3cc375 | 2013-06-04 09:36:56 +0000 | [diff] [blame] | 60 | srand(0); |
pwestin@webrtc.org | b518017 | 2012-11-09 20:56:23 +0000 | [diff] [blame] | 61 | // Need to initialize PacedSender after we initialize clock. |
pbos@webrtc.org | 709e297 | 2014-03-19 10:59:52 +0000 | [diff] [blame] | 62 | send_bucket_.reset( |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 63 | new PacedSender( |
| 64 | &clock_, &callback_, kPaceMultiplier * kTargetBitrate, 0)); |
pwestin@webrtc.org | b518017 | 2012-11-09 20:56:23 +0000 | [diff] [blame] | 65 | } |
stefan@webrtc.org | c3cc375 | 2013-06-04 09:36:56 +0000 | [diff] [blame] | 66 | |
stefan@webrtc.org | 8ccb9f9 | 2013-06-19 14:13:42 +0000 | [diff] [blame] | 67 | void SendAndExpectPacket(PacedSender::Priority priority, |
| 68 | uint32_t ssrc, uint16_t sequence_number, |
stefan@webrtc.org | 9b82f5a | 2013-11-13 15:29:21 +0000 | [diff] [blame] | 69 | int64_t capture_time_ms, int size, |
| 70 | bool retransmission) { |
stefan@webrtc.org | 8ccb9f9 | 2013-06-19 14:13:42 +0000 | [diff] [blame] | 71 | EXPECT_FALSE(send_bucket_->SendPacket(priority, ssrc, |
stefan@webrtc.org | 9b82f5a | 2013-11-13 15:29:21 +0000 | [diff] [blame] | 72 | sequence_number, capture_time_ms, size, retransmission)); |
stefan@webrtc.org | 8ccb9f9 | 2013-06-19 14:13:42 +0000 | [diff] [blame] | 73 | EXPECT_CALL(callback_, TimeToSendPacket( |
stefan@webrtc.org | 9b82f5a | 2013-11-13 15:29:21 +0000 | [diff] [blame] | 74 | ssrc, sequence_number, capture_time_ms, false)) |
hclam@chromium.org | 2e402ce | 2013-06-20 20:18:31 +0000 | [diff] [blame] | 75 | .Times(1) |
| 76 | .WillRepeatedly(Return(true)); |
stefan@webrtc.org | 8ccb9f9 | 2013-06-19 14:13:42 +0000 | [diff] [blame] | 77 | } |
| 78 | |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 79 | SimulatedClock clock_; |
pwestin@webrtc.org | b518017 | 2012-11-09 20:56:23 +0000 | [diff] [blame] | 80 | MockPacedSenderCallback callback_; |
| 81 | scoped_ptr<PacedSender> send_bucket_; |
| 82 | }; |
| 83 | |
| 84 | TEST_F(PacedSenderTest, QueuePacket) { |
| 85 | uint32_t ssrc = 12345; |
| 86 | uint16_t sequence_number = 1234; |
pwestin@webrtc.org | b518017 | 2012-11-09 20:56:23 +0000 | [diff] [blame] | 87 | // Due to the multiplicative factor we can send 3 packets not 2 packets. |
stefan@webrtc.org | 8ccb9f9 | 2013-06-19 14:13:42 +0000 | [diff] [blame] | 88 | SendAndExpectPacket(PacedSender::kNormalPriority, ssrc, sequence_number++, |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 89 | clock_.TimeInMilliseconds(), 250, false); |
stefan@webrtc.org | 8ccb9f9 | 2013-06-19 14:13:42 +0000 | [diff] [blame] | 90 | SendAndExpectPacket(PacedSender::kNormalPriority, ssrc, sequence_number++, |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 91 | clock_.TimeInMilliseconds(), 250, false); |
stefan@webrtc.org | 8ccb9f9 | 2013-06-19 14:13:42 +0000 | [diff] [blame] | 92 | SendAndExpectPacket(PacedSender::kNormalPriority, ssrc, sequence_number++, |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 93 | clock_.TimeInMilliseconds(), 250, false); |
| 94 | int64_t queued_packet_timestamp = clock_.TimeInMilliseconds(); |
pwestin@webrtc.org | b518017 | 2012-11-09 20:56:23 +0000 | [diff] [blame] | 95 | EXPECT_FALSE(send_bucket_->SendPacket(PacedSender::kNormalPriority, ssrc, |
stefan@webrtc.org | 19a40ff | 2013-11-27 14:16:20 +0000 | [diff] [blame] | 96 | sequence_number, queued_packet_timestamp, 250, false)); |
stefan@webrtc.org | 8ccb9f9 | 2013-06-19 14:13:42 +0000 | [diff] [blame] | 97 | send_bucket_->Process(); |
pwestin@webrtc.org | b518017 | 2012-11-09 20:56:23 +0000 | [diff] [blame] | 98 | EXPECT_EQ(5, send_bucket_->TimeUntilNextProcess()); |
pwestin@webrtc.org | db41856 | 2013-03-22 23:39:29 +0000 | [diff] [blame] | 99 | EXPECT_CALL(callback_, TimeToSendPadding(_)).Times(0); |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 100 | clock_.AdvanceTimeMilliseconds(4); |
pwestin@webrtc.org | b518017 | 2012-11-09 20:56:23 +0000 | [diff] [blame] | 101 | EXPECT_EQ(1, send_bucket_->TimeUntilNextProcess()); |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 102 | clock_.AdvanceTimeMilliseconds(1); |
pwestin@webrtc.org | b518017 | 2012-11-09 20:56:23 +0000 | [diff] [blame] | 103 | EXPECT_EQ(0, send_bucket_->TimeUntilNextProcess()); |
stefan@webrtc.org | 8ccb9f9 | 2013-06-19 14:13:42 +0000 | [diff] [blame] | 104 | EXPECT_CALL(callback_, TimeToSendPacket( |
stefan@webrtc.org | 19a40ff | 2013-11-27 14:16:20 +0000 | [diff] [blame] | 105 | ssrc, sequence_number++, queued_packet_timestamp, false)) |
hclam@chromium.org | 2e402ce | 2013-06-20 20:18:31 +0000 | [diff] [blame] | 106 | .Times(1) |
| 107 | .WillRepeatedly(Return(true)); |
stefan@webrtc.org | 8ccb9f9 | 2013-06-19 14:13:42 +0000 | [diff] [blame] | 108 | send_bucket_->Process(); |
pwestin@webrtc.org | b518017 | 2012-11-09 20:56:23 +0000 | [diff] [blame] | 109 | sequence_number++; |
stefan@webrtc.org | 8ccb9f9 | 2013-06-19 14:13:42 +0000 | [diff] [blame] | 110 | SendAndExpectPacket(PacedSender::kNormalPriority, ssrc, sequence_number++, |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 111 | clock_.TimeInMilliseconds(), 250, false); |
stefan@webrtc.org | 8ccb9f9 | 2013-06-19 14:13:42 +0000 | [diff] [blame] | 112 | SendAndExpectPacket(PacedSender::kNormalPriority, ssrc, sequence_number++, |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 113 | clock_.TimeInMilliseconds(), 250, false); |
pwestin@webrtc.org | b518017 | 2012-11-09 20:56:23 +0000 | [diff] [blame] | 114 | EXPECT_FALSE(send_bucket_->SendPacket(PacedSender::kNormalPriority, ssrc, |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 115 | sequence_number++, clock_.TimeInMilliseconds(), 250, false)); |
stefan@webrtc.org | 8ccb9f9 | 2013-06-19 14:13:42 +0000 | [diff] [blame] | 116 | send_bucket_->Process(); |
pwestin@webrtc.org | b518017 | 2012-11-09 20:56:23 +0000 | [diff] [blame] | 117 | } |
| 118 | |
| 119 | TEST_F(PacedSenderTest, PaceQueuedPackets) { |
| 120 | uint32_t ssrc = 12345; |
| 121 | uint16_t sequence_number = 1234; |
pwestin@webrtc.org | b518017 | 2012-11-09 20:56:23 +0000 | [diff] [blame] | 122 | |
| 123 | // Due to the multiplicative factor we can send 3 packets not 2 packets. |
| 124 | for (int i = 0; i < 3; ++i) { |
stefan@webrtc.org | 8ccb9f9 | 2013-06-19 14:13:42 +0000 | [diff] [blame] | 125 | SendAndExpectPacket(PacedSender::kNormalPriority, ssrc, sequence_number++, |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 126 | clock_.TimeInMilliseconds(), 250, false); |
pwestin@webrtc.org | b518017 | 2012-11-09 20:56:23 +0000 | [diff] [blame] | 127 | } |
| 128 | for (int j = 0; j < 30; ++j) { |
| 129 | EXPECT_FALSE(send_bucket_->SendPacket(PacedSender::kNormalPriority, ssrc, |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 130 | sequence_number++, clock_.TimeInMilliseconds(), 250, false)); |
pwestin@webrtc.org | b518017 | 2012-11-09 20:56:23 +0000 | [diff] [blame] | 131 | } |
stefan@webrtc.org | 8ccb9f9 | 2013-06-19 14:13:42 +0000 | [diff] [blame] | 132 | send_bucket_->Process(); |
pwestin@webrtc.org | db41856 | 2013-03-22 23:39:29 +0000 | [diff] [blame] | 133 | EXPECT_CALL(callback_, TimeToSendPadding(_)).Times(0); |
pwestin@webrtc.org | b518017 | 2012-11-09 20:56:23 +0000 | [diff] [blame] | 134 | for (int k = 0; k < 10; ++k) { |
| 135 | EXPECT_EQ(5, send_bucket_->TimeUntilNextProcess()); |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 136 | clock_.AdvanceTimeMilliseconds(5); |
pwestin@webrtc.org | b518017 | 2012-11-09 20:56:23 +0000 | [diff] [blame] | 137 | EXPECT_CALL(callback_, |
stefan@webrtc.org | 19a40ff | 2013-11-27 14:16:20 +0000 | [diff] [blame] | 138 | TimeToSendPacket(ssrc, _, _, false)) |
hclam@chromium.org | 2e402ce | 2013-06-20 20:18:31 +0000 | [diff] [blame] | 139 | .Times(3) |
| 140 | .WillRepeatedly(Return(true)); |
pwestin@webrtc.org | b518017 | 2012-11-09 20:56:23 +0000 | [diff] [blame] | 141 | EXPECT_EQ(0, send_bucket_->TimeUntilNextProcess()); |
| 142 | EXPECT_EQ(0, send_bucket_->Process()); |
| 143 | } |
| 144 | EXPECT_EQ(5, send_bucket_->TimeUntilNextProcess()); |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 145 | clock_.AdvanceTimeMilliseconds(5); |
pwestin@webrtc.org | b518017 | 2012-11-09 20:56:23 +0000 | [diff] [blame] | 146 | EXPECT_EQ(0, send_bucket_->TimeUntilNextProcess()); |
| 147 | EXPECT_EQ(0, send_bucket_->Process()); |
stefan@webrtc.org | 8ccb9f9 | 2013-06-19 14:13:42 +0000 | [diff] [blame] | 148 | SendAndExpectPacket(PacedSender::kNormalPriority, ssrc, sequence_number++, |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 149 | clock_.TimeInMilliseconds(), 250, false); |
stefan@webrtc.org | 8ccb9f9 | 2013-06-19 14:13:42 +0000 | [diff] [blame] | 150 | SendAndExpectPacket(PacedSender::kNormalPriority, ssrc, sequence_number++, |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 151 | clock_.TimeInMilliseconds(), 250, false); |
stefan@webrtc.org | 8ccb9f9 | 2013-06-19 14:13:42 +0000 | [diff] [blame] | 152 | SendAndExpectPacket(PacedSender::kNormalPriority, ssrc, sequence_number++, |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 153 | clock_.TimeInMilliseconds(), 250, false); |
pwestin@webrtc.org | b518017 | 2012-11-09 20:56:23 +0000 | [diff] [blame] | 154 | EXPECT_FALSE(send_bucket_->SendPacket(PacedSender::kNormalPriority, ssrc, |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 155 | sequence_number, clock_.TimeInMilliseconds(), 250, false)); |
stefan@webrtc.org | 8ccb9f9 | 2013-06-19 14:13:42 +0000 | [diff] [blame] | 156 | send_bucket_->Process(); |
pwestin@webrtc.org | b518017 | 2012-11-09 20:56:23 +0000 | [diff] [blame] | 157 | } |
| 158 | |
pwestin@webrtc.org | 52aa019 | 2013-04-25 17:35:56 +0000 | [diff] [blame] | 159 | TEST_F(PacedSenderTest, PaceQueuedPacketsWithDuplicates) { |
| 160 | uint32_t ssrc = 12345; |
| 161 | uint16_t sequence_number = 1234; |
pwestin@webrtc.org | 52aa019 | 2013-04-25 17:35:56 +0000 | [diff] [blame] | 162 | uint16_t queued_sequence_number; |
| 163 | |
| 164 | // Due to the multiplicative factor we can send 3 packets not 2 packets. |
| 165 | for (int i = 0; i < 3; ++i) { |
stefan@webrtc.org | 8ccb9f9 | 2013-06-19 14:13:42 +0000 | [diff] [blame] | 166 | SendAndExpectPacket(PacedSender::kNormalPriority, ssrc, sequence_number++, |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 167 | clock_.TimeInMilliseconds(), 250, false); |
pwestin@webrtc.org | 52aa019 | 2013-04-25 17:35:56 +0000 | [diff] [blame] | 168 | } |
| 169 | queued_sequence_number = sequence_number; |
| 170 | |
| 171 | for (int j = 0; j < 30; ++j) { |
| 172 | // Send in duplicate packets. |
| 173 | EXPECT_FALSE(send_bucket_->SendPacket(PacedSender::kNormalPriority, ssrc, |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 174 | sequence_number, clock_.TimeInMilliseconds(), 250, false)); |
pwestin@webrtc.org | 52aa019 | 2013-04-25 17:35:56 +0000 | [diff] [blame] | 175 | EXPECT_FALSE(send_bucket_->SendPacket(PacedSender::kNormalPriority, ssrc, |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 176 | sequence_number++, clock_.TimeInMilliseconds(), 250, false)); |
pwestin@webrtc.org | 52aa019 | 2013-04-25 17:35:56 +0000 | [diff] [blame] | 177 | } |
| 178 | EXPECT_CALL(callback_, TimeToSendPadding(_)).Times(0); |
stefan@webrtc.org | 8ccb9f9 | 2013-06-19 14:13:42 +0000 | [diff] [blame] | 179 | send_bucket_->Process(); |
pwestin@webrtc.org | 52aa019 | 2013-04-25 17:35:56 +0000 | [diff] [blame] | 180 | for (int k = 0; k < 10; ++k) { |
| 181 | EXPECT_EQ(5, send_bucket_->TimeUntilNextProcess()); |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 182 | clock_.AdvanceTimeMilliseconds(5); |
pwestin@webrtc.org | 52aa019 | 2013-04-25 17:35:56 +0000 | [diff] [blame] | 183 | |
| 184 | for (int i = 0; i < 3; ++i) { |
| 185 | EXPECT_CALL(callback_, TimeToSendPacket(ssrc, queued_sequence_number++, |
stefan@webrtc.org | 19a40ff | 2013-11-27 14:16:20 +0000 | [diff] [blame] | 186 | _, |
| 187 | false)) |
hclam@chromium.org | 2e402ce | 2013-06-20 20:18:31 +0000 | [diff] [blame] | 188 | .Times(1) |
| 189 | .WillRepeatedly(Return(true)); |
pwestin@webrtc.org | 52aa019 | 2013-04-25 17:35:56 +0000 | [diff] [blame] | 190 | } |
| 191 | EXPECT_EQ(0, send_bucket_->TimeUntilNextProcess()); |
| 192 | EXPECT_EQ(0, send_bucket_->Process()); |
| 193 | } |
| 194 | EXPECT_EQ(5, send_bucket_->TimeUntilNextProcess()); |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 195 | clock_.AdvanceTimeMilliseconds(5); |
pwestin@webrtc.org | 52aa019 | 2013-04-25 17:35:56 +0000 | [diff] [blame] | 196 | EXPECT_EQ(0, send_bucket_->TimeUntilNextProcess()); |
| 197 | EXPECT_EQ(0, send_bucket_->Process()); |
stefan@webrtc.org | 8ccb9f9 | 2013-06-19 14:13:42 +0000 | [diff] [blame] | 198 | SendAndExpectPacket(PacedSender::kNormalPriority, ssrc, sequence_number++, |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 199 | clock_.TimeInMilliseconds(), 250, false); |
stefan@webrtc.org | 8ccb9f9 | 2013-06-19 14:13:42 +0000 | [diff] [blame] | 200 | SendAndExpectPacket(PacedSender::kNormalPriority, ssrc, sequence_number++, |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 201 | clock_.TimeInMilliseconds(), 250, false); |
stefan@webrtc.org | 8ccb9f9 | 2013-06-19 14:13:42 +0000 | [diff] [blame] | 202 | SendAndExpectPacket(PacedSender::kNormalPriority, ssrc, sequence_number++, |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 203 | clock_.TimeInMilliseconds(), 250, false); |
pwestin@webrtc.org | 52aa019 | 2013-04-25 17:35:56 +0000 | [diff] [blame] | 204 | EXPECT_FALSE(send_bucket_->SendPacket(PacedSender::kNormalPriority, ssrc, |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 205 | sequence_number++, clock_.TimeInMilliseconds(), 250, false)); |
stefan@webrtc.org | 8ccb9f9 | 2013-06-19 14:13:42 +0000 | [diff] [blame] | 206 | send_bucket_->Process(); |
pwestin@webrtc.org | 52aa019 | 2013-04-25 17:35:56 +0000 | [diff] [blame] | 207 | } |
| 208 | |
pbos@webrtc.org | 03c817e | 2014-07-07 10:20:35 +0000 | [diff] [blame] | 209 | TEST_F(PacedSenderTest, CanQueuePacketsWithSameSequenceNumberOnDifferentSsrcs) { |
| 210 | uint32_t ssrc = 12345; |
| 211 | uint16_t sequence_number = 1234; |
| 212 | |
| 213 | SendAndExpectPacket(PacedSender::kNormalPriority, |
| 214 | ssrc, |
| 215 | sequence_number, |
| 216 | clock_.TimeInMilliseconds(), |
| 217 | 250, |
| 218 | false); |
| 219 | |
| 220 | // Expect packet on second ssrc to be queued and sent as well. |
| 221 | SendAndExpectPacket(PacedSender::kNormalPriority, |
| 222 | ssrc + 1, |
| 223 | sequence_number, |
| 224 | clock_.TimeInMilliseconds(), |
| 225 | 250, |
| 226 | false); |
| 227 | |
| 228 | clock_.AdvanceTimeMilliseconds(1000); |
pbos@webrtc.org | 03c817e | 2014-07-07 10:20:35 +0000 | [diff] [blame] | 229 | send_bucket_->Process(); |
| 230 | } |
| 231 | |
pwestin@webrtc.org | b518017 | 2012-11-09 20:56:23 +0000 | [diff] [blame] | 232 | TEST_F(PacedSenderTest, Padding) { |
| 233 | uint32_t ssrc = 12345; |
| 234 | uint16_t sequence_number = 1234; |
pwestin@webrtc.org | b518017 | 2012-11-09 20:56:23 +0000 | [diff] [blame] | 235 | |
pbos@webrtc.org | 709e297 | 2014-03-19 10:59:52 +0000 | [diff] [blame] | 236 | send_bucket_->UpdateBitrate(kPaceMultiplier * kTargetBitrate, kTargetBitrate); |
pwestin@webrtc.org | b518017 | 2012-11-09 20:56:23 +0000 | [diff] [blame] | 237 | // Due to the multiplicative factor we can send 3 packets not 2 packets. |
stefan@webrtc.org | 8ccb9f9 | 2013-06-19 14:13:42 +0000 | [diff] [blame] | 238 | SendAndExpectPacket(PacedSender::kNormalPriority, ssrc, sequence_number++, |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 239 | clock_.TimeInMilliseconds(), 250, false); |
stefan@webrtc.org | 8ccb9f9 | 2013-06-19 14:13:42 +0000 | [diff] [blame] | 240 | SendAndExpectPacket(PacedSender::kNormalPriority, ssrc, sequence_number++, |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 241 | clock_.TimeInMilliseconds(), 250, false); |
stefan@webrtc.org | 8ccb9f9 | 2013-06-19 14:13:42 +0000 | [diff] [blame] | 242 | SendAndExpectPacket(PacedSender::kNormalPriority, ssrc, sequence_number++, |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 243 | clock_.TimeInMilliseconds(), 250, false); |
stefan@webrtc.org | c3cc375 | 2013-06-04 09:36:56 +0000 | [diff] [blame] | 244 | // No padding is expected since we have sent too much already. |
| 245 | EXPECT_CALL(callback_, TimeToSendPadding(_)).Times(0); |
pwestin@webrtc.org | b518017 | 2012-11-09 20:56:23 +0000 | [diff] [blame] | 246 | EXPECT_EQ(5, send_bucket_->TimeUntilNextProcess()); |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 247 | clock_.AdvanceTimeMilliseconds(5); |
pwestin@webrtc.org | b518017 | 2012-11-09 20:56:23 +0000 | [diff] [blame] | 248 | EXPECT_EQ(0, send_bucket_->TimeUntilNextProcess()); |
| 249 | EXPECT_EQ(0, send_bucket_->Process()); |
| 250 | |
stefan@webrtc.org | c3cc375 | 2013-06-04 09:36:56 +0000 | [diff] [blame] | 251 | // 5 milliseconds later we have enough budget to send some padding. |
| 252 | EXPECT_CALL(callback_, TimeToSendPadding(250)).Times(1). |
| 253 | WillOnce(Return(250)); |
pwestin@webrtc.org | b518017 | 2012-11-09 20:56:23 +0000 | [diff] [blame] | 254 | EXPECT_EQ(5, send_bucket_->TimeUntilNextProcess()); |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 255 | clock_.AdvanceTimeMilliseconds(5); |
pwestin@webrtc.org | b518017 | 2012-11-09 20:56:23 +0000 | [diff] [blame] | 256 | EXPECT_EQ(0, send_bucket_->TimeUntilNextProcess()); |
| 257 | EXPECT_EQ(0, send_bucket_->Process()); |
| 258 | } |
| 259 | |
stefan@webrtc.org | 80865fd | 2013-08-09 11:31:23 +0000 | [diff] [blame] | 260 | TEST_F(PacedSenderTest, NoPaddingWhenDisabled) { |
| 261 | send_bucket_->SetStatus(false); |
pbos@webrtc.org | 709e297 | 2014-03-19 10:59:52 +0000 | [diff] [blame] | 262 | send_bucket_->UpdateBitrate(kPaceMultiplier * kTargetBitrate, kTargetBitrate); |
stefan@webrtc.org | 80865fd | 2013-08-09 11:31:23 +0000 | [diff] [blame] | 263 | // No padding is expected since the pacer is disabled. |
| 264 | EXPECT_CALL(callback_, TimeToSendPadding(_)).Times(0); |
| 265 | EXPECT_EQ(5, send_bucket_->TimeUntilNextProcess()); |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 266 | clock_.AdvanceTimeMilliseconds(5); |
stefan@webrtc.org | 80865fd | 2013-08-09 11:31:23 +0000 | [diff] [blame] | 267 | EXPECT_EQ(0, send_bucket_->TimeUntilNextProcess()); |
| 268 | EXPECT_EQ(0, send_bucket_->Process()); |
| 269 | EXPECT_CALL(callback_, TimeToSendPadding(_)).Times(0); |
| 270 | EXPECT_EQ(5, send_bucket_->TimeUntilNextProcess()); |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 271 | clock_.AdvanceTimeMilliseconds(5); |
stefan@webrtc.org | 80865fd | 2013-08-09 11:31:23 +0000 | [diff] [blame] | 272 | EXPECT_EQ(0, send_bucket_->TimeUntilNextProcess()); |
| 273 | EXPECT_EQ(0, send_bucket_->Process()); |
| 274 | } |
| 275 | |
stefan@webrtc.org | c3cc375 | 2013-06-04 09:36:56 +0000 | [diff] [blame] | 276 | TEST_F(PacedSenderTest, VerifyPaddingUpToBitrate) { |
| 277 | uint32_t ssrc = 12345; |
| 278 | uint16_t sequence_number = 1234; |
| 279 | int64_t capture_time_ms = 56789; |
| 280 | const int kTimeStep = 5; |
| 281 | const int64_t kBitrateWindow = 100; |
pbos@webrtc.org | 709e297 | 2014-03-19 10:59:52 +0000 | [diff] [blame] | 282 | send_bucket_->UpdateBitrate(kPaceMultiplier * kTargetBitrate, kTargetBitrate); |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 283 | int64_t start_time = clock_.TimeInMilliseconds(); |
| 284 | while (clock_.TimeInMilliseconds() - start_time < kBitrateWindow) { |
stefan@webrtc.org | 8ccb9f9 | 2013-06-19 14:13:42 +0000 | [diff] [blame] | 285 | SendAndExpectPacket(PacedSender::kNormalPriority, ssrc, sequence_number++, |
stefan@webrtc.org | 9b82f5a | 2013-11-13 15:29:21 +0000 | [diff] [blame] | 286 | capture_time_ms, 250, false); |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 287 | clock_.AdvanceTimeMilliseconds(kTimeStep); |
stefan@webrtc.org | c3cc375 | 2013-06-04 09:36:56 +0000 | [diff] [blame] | 288 | EXPECT_CALL(callback_, TimeToSendPadding(250)).Times(1). |
| 289 | WillOnce(Return(250)); |
| 290 | send_bucket_->Process(); |
| 291 | } |
| 292 | } |
| 293 | |
stefan@webrtc.org | c3cc375 | 2013-06-04 09:36:56 +0000 | [diff] [blame] | 294 | TEST_F(PacedSenderTest, VerifyAverageBitrateVaryingMediaPayload) { |
| 295 | uint32_t ssrc = 12345; |
| 296 | uint16_t sequence_number = 1234; |
| 297 | int64_t capture_time_ms = 56789; |
| 298 | const int kTimeStep = 5; |
| 299 | const int64_t kBitrateWindow = 10000; |
| 300 | PacedSenderPadding callback; |
pbos@webrtc.org | 709e297 | 2014-03-19 10:59:52 +0000 | [diff] [blame] | 301 | send_bucket_.reset( |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 302 | new PacedSender(&clock_, &callback, kPaceMultiplier * kTargetBitrate, 0)); |
pbos@webrtc.org | 709e297 | 2014-03-19 10:59:52 +0000 | [diff] [blame] | 303 | send_bucket_->UpdateBitrate(kPaceMultiplier * kTargetBitrate, kTargetBitrate); |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 304 | int64_t start_time = clock_.TimeInMilliseconds(); |
stefan@webrtc.org | c3cc375 | 2013-06-04 09:36:56 +0000 | [diff] [blame] | 305 | int media_bytes = 0; |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 306 | while (clock_.TimeInMilliseconds() - start_time < kBitrateWindow) { |
stefan@webrtc.org | c3cc375 | 2013-06-04 09:36:56 +0000 | [diff] [blame] | 307 | int media_payload = rand() % 100 + 200; // [200, 300] bytes. |
stefan@webrtc.org | 8ccb9f9 | 2013-06-19 14:13:42 +0000 | [diff] [blame] | 308 | EXPECT_FALSE(send_bucket_->SendPacket(PacedSender::kNormalPriority, ssrc, |
| 309 | sequence_number++, capture_time_ms, |
stefan@webrtc.org | 9b82f5a | 2013-11-13 15:29:21 +0000 | [diff] [blame] | 310 | media_payload, false)); |
stefan@webrtc.org | c3cc375 | 2013-06-04 09:36:56 +0000 | [diff] [blame] | 311 | media_bytes += media_payload; |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 312 | clock_.AdvanceTimeMilliseconds(kTimeStep); |
stefan@webrtc.org | c3cc375 | 2013-06-04 09:36:56 +0000 | [diff] [blame] | 313 | send_bucket_->Process(); |
| 314 | } |
| 315 | EXPECT_NEAR(kTargetBitrate, 8 * (media_bytes + callback.padding_sent()) / |
| 316 | kBitrateWindow, 1); |
| 317 | } |
| 318 | |
pwestin@webrtc.org | b518017 | 2012-11-09 20:56:23 +0000 | [diff] [blame] | 319 | TEST_F(PacedSenderTest, Priority) { |
| 320 | uint32_t ssrc_low_priority = 12345; |
| 321 | uint32_t ssrc = 12346; |
| 322 | uint16_t sequence_number = 1234; |
| 323 | int64_t capture_time_ms = 56789; |
| 324 | int64_t capture_time_ms_low_priority = 1234567; |
| 325 | |
| 326 | // Due to the multiplicative factor we can send 3 packets not 2 packets. |
stefan@webrtc.org | 8ccb9f9 | 2013-06-19 14:13:42 +0000 | [diff] [blame] | 327 | SendAndExpectPacket(PacedSender::kLowPriority, ssrc, sequence_number++, |
stefan@webrtc.org | 9b82f5a | 2013-11-13 15:29:21 +0000 | [diff] [blame] | 328 | capture_time_ms, 250, false); |
stefan@webrtc.org | 8ccb9f9 | 2013-06-19 14:13:42 +0000 | [diff] [blame] | 329 | SendAndExpectPacket(PacedSender::kNormalPriority, ssrc, sequence_number++, |
stefan@webrtc.org | 9b82f5a | 2013-11-13 15:29:21 +0000 | [diff] [blame] | 330 | capture_time_ms, 250, false); |
stefan@webrtc.org | 8ccb9f9 | 2013-06-19 14:13:42 +0000 | [diff] [blame] | 331 | SendAndExpectPacket(PacedSender::kNormalPriority, ssrc, sequence_number++, |
stefan@webrtc.org | 9b82f5a | 2013-11-13 15:29:21 +0000 | [diff] [blame] | 332 | capture_time_ms, 250, false); |
stefan@webrtc.org | 8ccb9f9 | 2013-06-19 14:13:42 +0000 | [diff] [blame] | 333 | send_bucket_->Process(); |
pwestin@webrtc.org | b518017 | 2012-11-09 20:56:23 +0000 | [diff] [blame] | 334 | |
| 335 | // Expect normal and low priority to be queued and high to pass through. |
| 336 | EXPECT_FALSE(send_bucket_->SendPacket(PacedSender::kLowPriority, |
stefan@webrtc.org | 9b82f5a | 2013-11-13 15:29:21 +0000 | [diff] [blame] | 337 | ssrc_low_priority, sequence_number++, capture_time_ms_low_priority, 250, |
| 338 | false)); |
pwestin@webrtc.org | b518017 | 2012-11-09 20:56:23 +0000 | [diff] [blame] | 339 | EXPECT_FALSE(send_bucket_->SendPacket(PacedSender::kNormalPriority, |
stefan@webrtc.org | 9b82f5a | 2013-11-13 15:29:21 +0000 | [diff] [blame] | 340 | ssrc, sequence_number++, capture_time_ms, 250, false)); |
pwestin@webrtc.org | b518017 | 2012-11-09 20:56:23 +0000 | [diff] [blame] | 341 | EXPECT_FALSE(send_bucket_->SendPacket(PacedSender::kNormalPriority, |
stefan@webrtc.org | 9b82f5a | 2013-11-13 15:29:21 +0000 | [diff] [blame] | 342 | ssrc, sequence_number++, capture_time_ms, 250, false)); |
pwestin@webrtc.org | db41856 | 2013-03-22 23:39:29 +0000 | [diff] [blame] | 343 | EXPECT_FALSE(send_bucket_->SendPacket(PacedSender::kHighPriority, |
stefan@webrtc.org | 9b82f5a | 2013-11-13 15:29:21 +0000 | [diff] [blame] | 344 | ssrc, sequence_number++, capture_time_ms, 250, false)); |
pwestin@webrtc.org | b518017 | 2012-11-09 20:56:23 +0000 | [diff] [blame] | 345 | |
pwestin@webrtc.org | db41856 | 2013-03-22 23:39:29 +0000 | [diff] [blame] | 346 | // Expect all high and normal priority to be sent out first. |
| 347 | EXPECT_CALL(callback_, TimeToSendPadding(_)).Times(0); |
stefan@webrtc.org | 9b82f5a | 2013-11-13 15:29:21 +0000 | [diff] [blame] | 348 | EXPECT_CALL(callback_, TimeToSendPacket(ssrc, _, capture_time_ms, false)) |
hclam@chromium.org | 2e402ce | 2013-06-20 20:18:31 +0000 | [diff] [blame] | 349 | .Times(3) |
| 350 | .WillRepeatedly(Return(true)); |
pwestin@webrtc.org | b518017 | 2012-11-09 20:56:23 +0000 | [diff] [blame] | 351 | |
| 352 | EXPECT_EQ(5, send_bucket_->TimeUntilNextProcess()); |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 353 | clock_.AdvanceTimeMilliseconds(5); |
pwestin@webrtc.org | b518017 | 2012-11-09 20:56:23 +0000 | [diff] [blame] | 354 | EXPECT_EQ(0, send_bucket_->TimeUntilNextProcess()); |
| 355 | EXPECT_EQ(0, send_bucket_->Process()); |
| 356 | |
pwestin@webrtc.org | db41856 | 2013-03-22 23:39:29 +0000 | [diff] [blame] | 357 | EXPECT_CALL(callback_, TimeToSendPacket( |
stefan@webrtc.org | 9b82f5a | 2013-11-13 15:29:21 +0000 | [diff] [blame] | 358 | ssrc_low_priority, _, capture_time_ms_low_priority, false)) |
hclam@chromium.org | 2e402ce | 2013-06-20 20:18:31 +0000 | [diff] [blame] | 359 | .Times(1) |
| 360 | .WillRepeatedly(Return(true)); |
pwestin@webrtc.org | b518017 | 2012-11-09 20:56:23 +0000 | [diff] [blame] | 361 | |
| 362 | EXPECT_EQ(5, send_bucket_->TimeUntilNextProcess()); |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 363 | clock_.AdvanceTimeMilliseconds(5); |
pwestin@webrtc.org | b518017 | 2012-11-09 20:56:23 +0000 | [diff] [blame] | 364 | EXPECT_EQ(0, send_bucket_->TimeUntilNextProcess()); |
| 365 | EXPECT_EQ(0, send_bucket_->Process()); |
| 366 | } |
| 367 | |
pwestin@webrtc.org | db41856 | 2013-03-22 23:39:29 +0000 | [diff] [blame] | 368 | TEST_F(PacedSenderTest, Pause) { |
| 369 | uint32_t ssrc_low_priority = 12345; |
| 370 | uint32_t ssrc = 12346; |
| 371 | uint16_t sequence_number = 1234; |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 372 | int64_t capture_time_ms = clock_.TimeInMilliseconds(); |
stefan@webrtc.org | bfacda6 | 2013-03-27 16:36:01 +0000 | [diff] [blame] | 373 | |
| 374 | EXPECT_EQ(0, send_bucket_->QueueInMs()); |
pwestin@webrtc.org | db41856 | 2013-03-22 23:39:29 +0000 | [diff] [blame] | 375 | |
| 376 | // Due to the multiplicative factor we can send 3 packets not 2 packets. |
stefan@webrtc.org | 8ccb9f9 | 2013-06-19 14:13:42 +0000 | [diff] [blame] | 377 | SendAndExpectPacket(PacedSender::kLowPriority, ssrc, sequence_number++, |
stefan@webrtc.org | 9b82f5a | 2013-11-13 15:29:21 +0000 | [diff] [blame] | 378 | capture_time_ms, 250, false); |
stefan@webrtc.org | 8ccb9f9 | 2013-06-19 14:13:42 +0000 | [diff] [blame] | 379 | SendAndExpectPacket(PacedSender::kNormalPriority, ssrc, sequence_number++, |
stefan@webrtc.org | 9b82f5a | 2013-11-13 15:29:21 +0000 | [diff] [blame] | 380 | capture_time_ms, 250, false); |
stefan@webrtc.org | 8ccb9f9 | 2013-06-19 14:13:42 +0000 | [diff] [blame] | 381 | SendAndExpectPacket(PacedSender::kNormalPriority, ssrc, sequence_number++, |
stefan@webrtc.org | 9b82f5a | 2013-11-13 15:29:21 +0000 | [diff] [blame] | 382 | capture_time_ms, 250, false); |
stefan@webrtc.org | 8ccb9f9 | 2013-06-19 14:13:42 +0000 | [diff] [blame] | 383 | send_bucket_->Process(); |
pwestin@webrtc.org | db41856 | 2013-03-22 23:39:29 +0000 | [diff] [blame] | 384 | |
| 385 | send_bucket_->Pause(); |
| 386 | |
pwestin@webrtc.org | db41856 | 2013-03-22 23:39:29 +0000 | [diff] [blame] | 387 | EXPECT_FALSE(send_bucket_->SendPacket(PacedSender::kNormalPriority, |
stefan@webrtc.org | 9b82f5a | 2013-11-13 15:29:21 +0000 | [diff] [blame] | 388 | ssrc, sequence_number++, capture_time_ms, 250, false)); |
pwestin@webrtc.org | db41856 | 2013-03-22 23:39:29 +0000 | [diff] [blame] | 389 | EXPECT_FALSE(send_bucket_->SendPacket(PacedSender::kNormalPriority, |
stefan@webrtc.org | 9b82f5a | 2013-11-13 15:29:21 +0000 | [diff] [blame] | 390 | ssrc, sequence_number++, capture_time_ms, 250, false)); |
pwestin@webrtc.org | db41856 | 2013-03-22 23:39:29 +0000 | [diff] [blame] | 391 | EXPECT_FALSE(send_bucket_->SendPacket(PacedSender::kHighPriority, |
stefan@webrtc.org | 9b82f5a | 2013-11-13 15:29:21 +0000 | [diff] [blame] | 392 | ssrc, sequence_number++, capture_time_ms, 250, false)); |
pwestin@webrtc.org | db41856 | 2013-03-22 23:39:29 +0000 | [diff] [blame] | 393 | |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 394 | clock_.AdvanceTimeMilliseconds(10000); |
| 395 | int64_t second_capture_time_ms = clock_.TimeInMilliseconds(); |
stefan@webrtc.org | dd393e7 | 2013-12-13 22:03:27 +0000 | [diff] [blame] | 396 | |
| 397 | // Expect everything to be queued. |
| 398 | EXPECT_FALSE(send_bucket_->SendPacket(PacedSender::kLowPriority, |
| 399 | ssrc_low_priority, sequence_number++, second_capture_time_ms, 250, |
| 400 | false)); |
| 401 | |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 402 | EXPECT_EQ(clock_.TimeInMilliseconds() - capture_time_ms, |
stefan@webrtc.org | bfacda6 | 2013-03-27 16:36:01 +0000 | [diff] [blame] | 403 | send_bucket_->QueueInMs()); |
| 404 | |
pwestin@webrtc.org | db41856 | 2013-03-22 23:39:29 +0000 | [diff] [blame] | 405 | // Expect no packet to come out while paused. |
| 406 | EXPECT_CALL(callback_, TimeToSendPadding(_)).Times(0); |
stefan@webrtc.org | 9b82f5a | 2013-11-13 15:29:21 +0000 | [diff] [blame] | 407 | EXPECT_CALL(callback_, TimeToSendPacket(_, _, _, _)).Times(0); |
pwestin@webrtc.org | db41856 | 2013-03-22 23:39:29 +0000 | [diff] [blame] | 408 | |
| 409 | for (int i = 0; i < 10; ++i) { |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 410 | clock_.AdvanceTimeMilliseconds(5); |
pwestin@webrtc.org | db41856 | 2013-03-22 23:39:29 +0000 | [diff] [blame] | 411 | EXPECT_EQ(0, send_bucket_->TimeUntilNextProcess()); |
| 412 | EXPECT_EQ(0, send_bucket_->Process()); |
| 413 | } |
| 414 | // Expect high prio packets to come out first followed by all packets in the |
| 415 | // way they were added. |
stefan@webrtc.org | 9b82f5a | 2013-11-13 15:29:21 +0000 | [diff] [blame] | 416 | EXPECT_CALL(callback_, TimeToSendPacket(_, _, capture_time_ms, false)) |
hclam@chromium.org | 2e402ce | 2013-06-20 20:18:31 +0000 | [diff] [blame] | 417 | .Times(3) |
| 418 | .WillRepeatedly(Return(true)); |
pwestin@webrtc.org | db41856 | 2013-03-22 23:39:29 +0000 | [diff] [blame] | 419 | send_bucket_->Resume(); |
| 420 | |
| 421 | EXPECT_EQ(5, send_bucket_->TimeUntilNextProcess()); |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 422 | clock_.AdvanceTimeMilliseconds(5); |
pwestin@webrtc.org | db41856 | 2013-03-22 23:39:29 +0000 | [diff] [blame] | 423 | EXPECT_EQ(0, send_bucket_->TimeUntilNextProcess()); |
| 424 | EXPECT_EQ(0, send_bucket_->Process()); |
| 425 | |
jiayl@webrtc.org | 9fd8d87 | 2014-02-27 22:32:40 +0000 | [diff] [blame] | 426 | EXPECT_CALL( |
| 427 | callback_, TimeToSendPacket(_, _, second_capture_time_ms, false)) |
hclam@chromium.org | 2e402ce | 2013-06-20 20:18:31 +0000 | [diff] [blame] | 428 | .Times(1) |
| 429 | .WillRepeatedly(Return(true)); |
pwestin@webrtc.org | db41856 | 2013-03-22 23:39:29 +0000 | [diff] [blame] | 430 | EXPECT_EQ(5, send_bucket_->TimeUntilNextProcess()); |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 431 | clock_.AdvanceTimeMilliseconds(5); |
pwestin@webrtc.org | db41856 | 2013-03-22 23:39:29 +0000 | [diff] [blame] | 432 | EXPECT_EQ(0, send_bucket_->TimeUntilNextProcess()); |
| 433 | EXPECT_EQ(0, send_bucket_->Process()); |
stefan@webrtc.org | bfacda6 | 2013-03-27 16:36:01 +0000 | [diff] [blame] | 434 | EXPECT_EQ(0, send_bucket_->QueueInMs()); |
pwestin@webrtc.org | db41856 | 2013-03-22 23:39:29 +0000 | [diff] [blame] | 435 | } |
| 436 | |
hclam@chromium.org | 2e402ce | 2013-06-20 20:18:31 +0000 | [diff] [blame] | 437 | TEST_F(PacedSenderTest, ResendPacket) { |
| 438 | uint32_t ssrc = 12346; |
| 439 | uint16_t sequence_number = 1234; |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 440 | int64_t capture_time_ms = clock_.TimeInMilliseconds(); |
hclam@chromium.org | 2e402ce | 2013-06-20 20:18:31 +0000 | [diff] [blame] | 441 | EXPECT_EQ(0, send_bucket_->QueueInMs()); |
| 442 | |
| 443 | EXPECT_FALSE(send_bucket_->SendPacket(PacedSender::kNormalPriority, |
| 444 | ssrc, |
| 445 | sequence_number, |
| 446 | capture_time_ms, |
stefan@webrtc.org | 9b82f5a | 2013-11-13 15:29:21 +0000 | [diff] [blame] | 447 | 250, |
| 448 | false)); |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 449 | clock_.AdvanceTimeMilliseconds(1); |
hclam@chromium.org | 2e402ce | 2013-06-20 20:18:31 +0000 | [diff] [blame] | 450 | EXPECT_FALSE(send_bucket_->SendPacket(PacedSender::kNormalPriority, |
| 451 | ssrc, |
| 452 | sequence_number + 1, |
| 453 | capture_time_ms + 1, |
stefan@webrtc.org | 9b82f5a | 2013-11-13 15:29:21 +0000 | [diff] [blame] | 454 | 250, |
| 455 | false)); |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 456 | clock_.AdvanceTimeMilliseconds(9999); |
| 457 | EXPECT_EQ(clock_.TimeInMilliseconds() - capture_time_ms, |
hclam@chromium.org | 2e402ce | 2013-06-20 20:18:31 +0000 | [diff] [blame] | 458 | send_bucket_->QueueInMs()); |
| 459 | // Fails to send first packet so only one call. |
| 460 | EXPECT_CALL(callback_, TimeToSendPacket( |
stefan@webrtc.org | 9b82f5a | 2013-11-13 15:29:21 +0000 | [diff] [blame] | 461 | ssrc, sequence_number, capture_time_ms, false)) |
hclam@chromium.org | 2e402ce | 2013-06-20 20:18:31 +0000 | [diff] [blame] | 462 | .Times(1) |
| 463 | .WillOnce(Return(false)); |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 464 | clock_.AdvanceTimeMilliseconds(10000); |
hclam@chromium.org | 2e402ce | 2013-06-20 20:18:31 +0000 | [diff] [blame] | 465 | send_bucket_->Process(); |
| 466 | |
| 467 | // Queue remains unchanged. |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 468 | EXPECT_EQ(clock_.TimeInMilliseconds() - capture_time_ms, |
hclam@chromium.org | 2e402ce | 2013-06-20 20:18:31 +0000 | [diff] [blame] | 469 | send_bucket_->QueueInMs()); |
| 470 | |
| 471 | // Fails to send second packet. |
| 472 | EXPECT_CALL(callback_, TimeToSendPacket( |
stefan@webrtc.org | 9b82f5a | 2013-11-13 15:29:21 +0000 | [diff] [blame] | 473 | ssrc, sequence_number, capture_time_ms, false)) |
hclam@chromium.org | 2e402ce | 2013-06-20 20:18:31 +0000 | [diff] [blame] | 474 | .Times(1) |
| 475 | .WillOnce(Return(true)); |
| 476 | EXPECT_CALL(callback_, TimeToSendPacket( |
stefan@webrtc.org | 9b82f5a | 2013-11-13 15:29:21 +0000 | [diff] [blame] | 477 | ssrc, sequence_number + 1, capture_time_ms + 1, false)) |
hclam@chromium.org | 2e402ce | 2013-06-20 20:18:31 +0000 | [diff] [blame] | 478 | .Times(1) |
| 479 | .WillOnce(Return(false)); |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 480 | clock_.AdvanceTimeMilliseconds(10000); |
hclam@chromium.org | 2e402ce | 2013-06-20 20:18:31 +0000 | [diff] [blame] | 481 | send_bucket_->Process(); |
| 482 | |
| 483 | // Queue is reduced by 1 packet. |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 484 | EXPECT_EQ(clock_.TimeInMilliseconds() - capture_time_ms - 1, |
hclam@chromium.org | 2e402ce | 2013-06-20 20:18:31 +0000 | [diff] [blame] | 485 | send_bucket_->QueueInMs()); |
| 486 | |
| 487 | // Send second packet and queue becomes empty. |
| 488 | EXPECT_CALL(callback_, TimeToSendPacket( |
stefan@webrtc.org | 9b82f5a | 2013-11-13 15:29:21 +0000 | [diff] [blame] | 489 | ssrc, sequence_number + 1, capture_time_ms + 1, false)) |
hclam@chromium.org | 2e402ce | 2013-06-20 20:18:31 +0000 | [diff] [blame] | 490 | .Times(1) |
| 491 | .WillOnce(Return(true)); |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 492 | clock_.AdvanceTimeMilliseconds(10000); |
hclam@chromium.org | 2e402ce | 2013-06-20 20:18:31 +0000 | [diff] [blame] | 493 | send_bucket_->Process(); |
| 494 | EXPECT_EQ(0, send_bucket_->QueueInMs()); |
| 495 | } |
| 496 | |
stefan@webrtc.org | 19a40ff | 2013-11-27 14:16:20 +0000 | [diff] [blame] | 497 | TEST_F(PacedSenderTest, MaxQueueLength) { |
| 498 | uint32_t ssrc = 12346; |
| 499 | uint16_t sequence_number = 1234; |
| 500 | EXPECT_EQ(0, send_bucket_->QueueInMs()); |
| 501 | |
pbos@webrtc.org | 709e297 | 2014-03-19 10:59:52 +0000 | [diff] [blame] | 502 | send_bucket_->UpdateBitrate(kPaceMultiplier * 30, 0); |
stefan@webrtc.org | 19a40ff | 2013-11-27 14:16:20 +0000 | [diff] [blame] | 503 | for (int i = 0; i < 30; ++i) { |
| 504 | SendAndExpectPacket(PacedSender::kNormalPriority, |
| 505 | ssrc, |
| 506 | sequence_number++, |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 507 | clock_.TimeInMilliseconds(), |
stefan@webrtc.org | 19a40ff | 2013-11-27 14:16:20 +0000 | [diff] [blame] | 508 | 1200, |
| 509 | false); |
| 510 | } |
| 511 | |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 512 | clock_.AdvanceTimeMilliseconds(2001); |
stefan@webrtc.org | 19a40ff | 2013-11-27 14:16:20 +0000 | [diff] [blame] | 513 | SendAndExpectPacket(PacedSender::kNormalPriority, |
| 514 | ssrc, |
| 515 | sequence_number++, |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 516 | clock_.TimeInMilliseconds(), |
stefan@webrtc.org | 19a40ff | 2013-11-27 14:16:20 +0000 | [diff] [blame] | 517 | 1200, |
| 518 | false); |
| 519 | EXPECT_EQ(2001, send_bucket_->QueueInMs()); |
| 520 | send_bucket_->Process(); |
| 521 | EXPECT_EQ(0, send_bucket_->QueueInMs()); |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 522 | clock_.AdvanceTimeMilliseconds(31); |
stefan@webrtc.org | 168f23f | 2014-07-11 13:44:02 +0000 | [diff] [blame] | 523 | |
stefan@webrtc.org | 19a40ff | 2013-11-27 14:16:20 +0000 | [diff] [blame] | 524 | send_bucket_->Process(); |
| 525 | } |
stefan@webrtc.org | dd393e7 | 2013-12-13 22:03:27 +0000 | [diff] [blame] | 526 | |
| 527 | TEST_F(PacedSenderTest, QueueTimeGrowsOverTime) { |
| 528 | uint32_t ssrc = 12346; |
| 529 | uint16_t sequence_number = 1234; |
| 530 | EXPECT_EQ(0, send_bucket_->QueueInMs()); |
| 531 | |
pbos@webrtc.org | 709e297 | 2014-03-19 10:59:52 +0000 | [diff] [blame] | 532 | send_bucket_->UpdateBitrate(kPaceMultiplier * 30, 0); |
stefan@webrtc.org | dd393e7 | 2013-12-13 22:03:27 +0000 | [diff] [blame] | 533 | SendAndExpectPacket(PacedSender::kNormalPriority, |
| 534 | ssrc, |
| 535 | sequence_number, |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 536 | clock_.TimeInMilliseconds(), |
stefan@webrtc.org | dd393e7 | 2013-12-13 22:03:27 +0000 | [diff] [blame] | 537 | 1200, |
| 538 | false); |
| 539 | |
stefan@webrtc.org | 88e0dda | 2014-07-04 09:20:42 +0000 | [diff] [blame] | 540 | clock_.AdvanceTimeMilliseconds(500); |
stefan@webrtc.org | dd393e7 | 2013-12-13 22:03:27 +0000 | [diff] [blame] | 541 | EXPECT_EQ(500, send_bucket_->QueueInMs()); |
| 542 | send_bucket_->Process(); |
| 543 | EXPECT_EQ(0, send_bucket_->QueueInMs()); |
| 544 | } |
pwestin@webrtc.org | db41856 | 2013-03-22 23:39:29 +0000 | [diff] [blame] | 545 | } // namespace test |
pwestin@webrtc.org | b518017 | 2012-11-09 20:56:23 +0000 | [diff] [blame] | 546 | } // namespace webrtc |