pwestin@webrtc.org | 1cd1162 | 2012-04-19 12:13:52 +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 | 2e10b8e | 2013-07-16 12:54:53 +0000 | [diff] [blame] | 11 | #include "testing/gtest/include/gtest/gtest.h" |
pwestin@webrtc.org | 1cd1162 | 2012-04-19 12:13:52 +0000 | [diff] [blame] | 12 | |
| 13 | #include <algorithm> |
| 14 | #include <vector> |
| 15 | |
pbos@webrtc.org | 2e10b8e | 2013-07-16 12:54:53 +0000 | [diff] [blame] | 16 | #include "webrtc/modules/bitrate_controller/include/bitrate_controller.h" |
| 17 | #include "webrtc/modules/rtp_rtcp/interface/rtp_rtcp_defines.h" |
pwestin@webrtc.org | 1cd1162 | 2012-04-19 12:13:52 +0000 | [diff] [blame] | 18 | |
| 19 | using webrtc::RtcpBandwidthObserver; |
| 20 | using webrtc::BitrateObserver; |
| 21 | using webrtc::BitrateController; |
| 22 | |
stefan@webrtc.org | 28a331e | 2013-09-17 07:49:56 +0000 | [diff] [blame] | 23 | uint8_t WeightedLoss(int num_packets1, uint8_t fraction_loss1, |
| 24 | int num_packets2, uint8_t fraction_loss2) { |
| 25 | int weighted_sum = num_packets1 * fraction_loss1 + |
| 26 | num_packets2 * fraction_loss2; |
| 27 | int total_num_packets = num_packets1 + num_packets2; |
| 28 | return (weighted_sum + total_num_packets / 2) / total_num_packets; |
| 29 | } |
| 30 | |
| 31 | webrtc::RTCPReportBlock CreateReportBlock( |
| 32 | uint32_t remote_ssrc, uint32_t source_ssrc, |
| 33 | uint8_t fraction_lost, uint32_t extended_high_sequence_number) { |
| 34 | return webrtc::RTCPReportBlock(remote_ssrc, source_ssrc, fraction_lost, 0, |
| 35 | extended_high_sequence_number, 0, 0, 0); |
| 36 | } |
| 37 | |
pwestin@webrtc.org | 1cd1162 | 2012-04-19 12:13:52 +0000 | [diff] [blame] | 38 | class TestBitrateObserver: public BitrateObserver { |
| 39 | public: |
| 40 | TestBitrateObserver() |
kjellander@webrtc.org | 2e84c11 | 2012-05-31 13:55:01 +0000 | [diff] [blame] | 41 | : last_bitrate_(0), |
| 42 | last_fraction_loss_(0), |
| 43 | last_rtt_(0) { |
pwestin@webrtc.org | 1cd1162 | 2012-04-19 12:13:52 +0000 | [diff] [blame] | 44 | } |
| 45 | |
| 46 | virtual void OnNetworkChanged(const uint32_t bitrate, |
| 47 | const uint8_t fraction_loss, |
| 48 | const uint32_t rtt) { |
kjellander@webrtc.org | 2e84c11 | 2012-05-31 13:55:01 +0000 | [diff] [blame] | 49 | last_bitrate_ = bitrate; |
| 50 | last_fraction_loss_ = fraction_loss; |
| 51 | last_rtt_ = rtt; |
pwestin@webrtc.org | 1cd1162 | 2012-04-19 12:13:52 +0000 | [diff] [blame] | 52 | } |
kjellander@webrtc.org | 2e84c11 | 2012-05-31 13:55:01 +0000 | [diff] [blame] | 53 | uint32_t last_bitrate_; |
| 54 | uint8_t last_fraction_loss_; |
| 55 | uint32_t last_rtt_; |
pwestin@webrtc.org | 1cd1162 | 2012-04-19 12:13:52 +0000 | [diff] [blame] | 56 | }; |
| 57 | |
| 58 | class BitrateControllerTest : public ::testing::Test { |
| 59 | protected: |
henrik.lundin@webrtc.org | 29dd0de | 2013-10-21 14:00:01 +0000 | [diff] [blame] | 60 | BitrateControllerTest() : enforce_min_bitrate_(true) {} |
pwestin@webrtc.org | 1cd1162 | 2012-04-19 12:13:52 +0000 | [diff] [blame] | 61 | ~BitrateControllerTest() {} |
| 62 | |
| 63 | virtual void SetUp() { |
henrik.lundin@webrtc.org | 29dd0de | 2013-10-21 14:00:01 +0000 | [diff] [blame] | 64 | controller_ = |
| 65 | BitrateController::CreateBitrateController(enforce_min_bitrate_); |
pwestin@webrtc.org | 1cd1162 | 2012-04-19 12:13:52 +0000 | [diff] [blame] | 66 | bandwidth_observer_ = controller_->CreateRtcpBandwidthObserver(); |
| 67 | } |
| 68 | |
| 69 | virtual void TearDown() { |
| 70 | delete bandwidth_observer_; |
| 71 | delete controller_; |
| 72 | } |
henrik.lundin@webrtc.org | 29dd0de | 2013-10-21 14:00:01 +0000 | [diff] [blame] | 73 | bool enforce_min_bitrate_; |
pwestin@webrtc.org | 1cd1162 | 2012-04-19 12:13:52 +0000 | [diff] [blame] | 74 | BitrateController* controller_; |
| 75 | RtcpBandwidthObserver* bandwidth_observer_; |
| 76 | }; |
| 77 | |
| 78 | TEST_F(BitrateControllerTest, Basic) { |
| 79 | TestBitrateObserver bitrate_observer; |
| 80 | controller_->SetBitrateObserver(&bitrate_observer, 200000, 100000, 300000); |
| 81 | controller_->RemoveBitrateObserver(&bitrate_observer); |
| 82 | } |
| 83 | |
| 84 | TEST_F(BitrateControllerTest, OneBitrateObserverOneRtcpObserver) { |
| 85 | TestBitrateObserver bitrate_observer; |
| 86 | controller_->SetBitrateObserver(&bitrate_observer, 200000, 100000, 300000); |
| 87 | |
| 88 | // Receive a high remb, test bitrate inc. |
| 89 | bandwidth_observer_->OnReceivedEstimatedBitrate(400000); |
| 90 | |
| 91 | // Test start bitrate. |
stefan@webrtc.org | 28a331e | 2013-09-17 07:49:56 +0000 | [diff] [blame] | 92 | webrtc::ReportBlockList report_blocks; |
| 93 | report_blocks.push_back(CreateReportBlock(1, 2, 0, 1)); |
| 94 | bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, 1); |
andresp@webrtc.org | 07bc734 | 2014-03-21 16:51:01 +0000 | [diff] [blame^] | 95 | EXPECT_EQ(200000u, bitrate_observer.last_bitrate_); |
kjellander@webrtc.org | 2e84c11 | 2012-05-31 13:55:01 +0000 | [diff] [blame] | 96 | EXPECT_EQ(0, bitrate_observer.last_fraction_loss_); |
andresp@webrtc.org | 07bc734 | 2014-03-21 16:51:01 +0000 | [diff] [blame^] | 97 | EXPECT_EQ(50u, bitrate_observer.last_rtt_); |
pwestin@webrtc.org | 1cd1162 | 2012-04-19 12:13:52 +0000 | [diff] [blame] | 98 | |
| 99 | // Test bitrate increase 8% per second. |
stefan@webrtc.org | 28a331e | 2013-09-17 07:49:56 +0000 | [diff] [blame] | 100 | report_blocks.clear(); |
| 101 | report_blocks.push_back(CreateReportBlock(1, 2, 0, 21)); |
| 102 | bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, 1001); |
kjellander@webrtc.org | 2e84c11 | 2012-05-31 13:55:01 +0000 | [diff] [blame] | 103 | EXPECT_EQ(217000u, bitrate_observer.last_bitrate_); |
| 104 | EXPECT_EQ(0, bitrate_observer.last_fraction_loss_); |
| 105 | EXPECT_EQ(50u, bitrate_observer.last_rtt_); |
pwestin@webrtc.org | 1cd1162 | 2012-04-19 12:13:52 +0000 | [diff] [blame] | 106 | |
stefan@webrtc.org | 28a331e | 2013-09-17 07:49:56 +0000 | [diff] [blame] | 107 | report_blocks.clear(); |
| 108 | report_blocks.push_back(CreateReportBlock(1, 2, 0, 41)); |
| 109 | bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, 2001); |
kjellander@webrtc.org | 2e84c11 | 2012-05-31 13:55:01 +0000 | [diff] [blame] | 110 | EXPECT_EQ(235360u, bitrate_observer.last_bitrate_); |
pwestin@webrtc.org | 1cd1162 | 2012-04-19 12:13:52 +0000 | [diff] [blame] | 111 | |
stefan@webrtc.org | 28a331e | 2013-09-17 07:49:56 +0000 | [diff] [blame] | 112 | report_blocks.clear(); |
| 113 | report_blocks.push_back(CreateReportBlock(1, 2, 0, 61)); |
| 114 | bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, 3001); |
kjellander@webrtc.org | 2e84c11 | 2012-05-31 13:55:01 +0000 | [diff] [blame] | 115 | EXPECT_EQ(255189u, bitrate_observer.last_bitrate_); |
pwestin@webrtc.org | 1cd1162 | 2012-04-19 12:13:52 +0000 | [diff] [blame] | 116 | |
stefan@webrtc.org | 28a331e | 2013-09-17 07:49:56 +0000 | [diff] [blame] | 117 | report_blocks.clear(); |
| 118 | report_blocks.push_back(CreateReportBlock(1, 2, 0, 801)); |
| 119 | bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, 4001); |
kjellander@webrtc.org | 2e84c11 | 2012-05-31 13:55:01 +0000 | [diff] [blame] | 120 | EXPECT_EQ(276604u, bitrate_observer.last_bitrate_); |
pwestin@webrtc.org | 1cd1162 | 2012-04-19 12:13:52 +0000 | [diff] [blame] | 121 | |
stefan@webrtc.org | 28a331e | 2013-09-17 07:49:56 +0000 | [diff] [blame] | 122 | report_blocks.clear(); |
| 123 | report_blocks.push_back(CreateReportBlock(1, 2, 0, 101)); |
| 124 | bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, 5001); |
kjellander@webrtc.org | 2e84c11 | 2012-05-31 13:55:01 +0000 | [diff] [blame] | 125 | EXPECT_EQ(299732u, bitrate_observer.last_bitrate_); |
pwestin@webrtc.org | 1cd1162 | 2012-04-19 12:13:52 +0000 | [diff] [blame] | 126 | |
stefan@webrtc.org | 28a331e | 2013-09-17 07:49:56 +0000 | [diff] [blame] | 127 | report_blocks.clear(); |
| 128 | report_blocks.push_back(CreateReportBlock(1, 2, 0, 121)); |
| 129 | bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, 6001); |
kjellander@webrtc.org | 2e84c11 | 2012-05-31 13:55:01 +0000 | [diff] [blame] | 130 | EXPECT_EQ(300000u, bitrate_observer.last_bitrate_); // Max cap. |
pwestin@webrtc.org | 1cd1162 | 2012-04-19 12:13:52 +0000 | [diff] [blame] | 131 | |
stefan@webrtc.org | 28a331e | 2013-09-17 07:49:56 +0000 | [diff] [blame] | 132 | report_blocks.clear(); |
| 133 | report_blocks.push_back(CreateReportBlock(1, 2, 0, 141)); |
| 134 | bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, 7001); |
kjellander@webrtc.org | 2e84c11 | 2012-05-31 13:55:01 +0000 | [diff] [blame] | 135 | EXPECT_EQ(300000u, bitrate_observer.last_bitrate_); // Max cap. |
pwestin@webrtc.org | 1cd1162 | 2012-04-19 12:13:52 +0000 | [diff] [blame] | 136 | |
| 137 | // Test that a low REMB trigger immediately. |
| 138 | bandwidth_observer_->OnReceivedEstimatedBitrate(250000); |
kjellander@webrtc.org | 2e84c11 | 2012-05-31 13:55:01 +0000 | [diff] [blame] | 139 | EXPECT_EQ(250000u, bitrate_observer.last_bitrate_); |
| 140 | EXPECT_EQ(0, bitrate_observer.last_fraction_loss_); |
| 141 | EXPECT_EQ(50u, bitrate_observer.last_rtt_); |
pwestin@webrtc.org | 1cd1162 | 2012-04-19 12:13:52 +0000 | [diff] [blame] | 142 | |
| 143 | bandwidth_observer_->OnReceivedEstimatedBitrate(1000); |
kjellander@webrtc.org | 2e84c11 | 2012-05-31 13:55:01 +0000 | [diff] [blame] | 144 | EXPECT_EQ(100000u, bitrate_observer.last_bitrate_); // Min cap. |
pwestin@webrtc.org | 1cd1162 | 2012-04-19 12:13:52 +0000 | [diff] [blame] | 145 | controller_->RemoveBitrateObserver(&bitrate_observer); |
| 146 | } |
| 147 | |
| 148 | TEST_F(BitrateControllerTest, OneBitrateObserverTwoRtcpObservers) { |
| 149 | TestBitrateObserver bitrate_observer; |
| 150 | controller_->SetBitrateObserver(&bitrate_observer, 200000, 100000, 300000); |
| 151 | |
| 152 | RtcpBandwidthObserver* second_bandwidth_observer = |
| 153 | controller_->CreateRtcpBandwidthObserver(); |
| 154 | |
| 155 | // Receive a high remb, test bitrate inc. |
| 156 | bandwidth_observer_->OnReceivedEstimatedBitrate(400000); |
| 157 | |
| 158 | // Test start bitrate. |
stefan@webrtc.org | 28a331e | 2013-09-17 07:49:56 +0000 | [diff] [blame] | 159 | webrtc::ReportBlockList report_blocks; |
| 160 | report_blocks.push_back(CreateReportBlock(1, 2, 0, 1)); |
| 161 | bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, 1); |
| 162 | second_bandwidth_observer->OnReceivedRtcpReceiverReport( |
| 163 | report_blocks, 100, 1); |
andresp@webrtc.org | 07bc734 | 2014-03-21 16:51:01 +0000 | [diff] [blame^] | 164 | EXPECT_EQ(200000u, bitrate_observer.last_bitrate_); |
kjellander@webrtc.org | 2e84c11 | 2012-05-31 13:55:01 +0000 | [diff] [blame] | 165 | EXPECT_EQ(0, bitrate_observer.last_fraction_loss_); |
andresp@webrtc.org | 07bc734 | 2014-03-21 16:51:01 +0000 | [diff] [blame^] | 166 | EXPECT_EQ(100u, bitrate_observer.last_rtt_); |
pwestin@webrtc.org | 1cd1162 | 2012-04-19 12:13:52 +0000 | [diff] [blame] | 167 | |
| 168 | // Test bitrate increase 8% per second. |
stefan@webrtc.org | 28a331e | 2013-09-17 07:49:56 +0000 | [diff] [blame] | 169 | report_blocks.clear(); |
| 170 | report_blocks.push_back(CreateReportBlock(1, 2, 0, 21)); |
| 171 | bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, 501); |
| 172 | second_bandwidth_observer->OnReceivedRtcpReceiverReport(report_blocks, 100, |
| 173 | 1001); |
kjellander@webrtc.org | 2e84c11 | 2012-05-31 13:55:01 +0000 | [diff] [blame] | 174 | EXPECT_EQ(217000u, bitrate_observer.last_bitrate_); |
| 175 | EXPECT_EQ(0, bitrate_observer.last_fraction_loss_); |
| 176 | EXPECT_EQ(100u, bitrate_observer.last_rtt_); |
pwestin@webrtc.org | 1cd1162 | 2012-04-19 12:13:52 +0000 | [diff] [blame] | 177 | |
| 178 | // Extra report should not change estimate. |
stefan@webrtc.org | 28a331e | 2013-09-17 07:49:56 +0000 | [diff] [blame] | 179 | report_blocks.clear(); |
| 180 | report_blocks.push_back(CreateReportBlock(1, 2, 0, 31)); |
| 181 | second_bandwidth_observer->OnReceivedRtcpReceiverReport(report_blocks, 100, |
| 182 | 1501); |
kjellander@webrtc.org | 2e84c11 | 2012-05-31 13:55:01 +0000 | [diff] [blame] | 183 | EXPECT_EQ(217000u, bitrate_observer.last_bitrate_); |
pwestin@webrtc.org | 1cd1162 | 2012-04-19 12:13:52 +0000 | [diff] [blame] | 184 | |
stefan@webrtc.org | 28a331e | 2013-09-17 07:49:56 +0000 | [diff] [blame] | 185 | report_blocks.clear(); |
| 186 | report_blocks.push_back(CreateReportBlock(1, 2, 0, 41)); |
| 187 | bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, 2001); |
kjellander@webrtc.org | 2e84c11 | 2012-05-31 13:55:01 +0000 | [diff] [blame] | 188 | EXPECT_EQ(235360u, bitrate_observer.last_bitrate_); |
pwestin@webrtc.org | a2cd732 | 2012-04-23 08:32:47 +0000 | [diff] [blame] | 189 | |
pwestin@webrtc.org | 1cd1162 | 2012-04-19 12:13:52 +0000 | [diff] [blame] | 190 | // Second report should not change estimate. |
stefan@webrtc.org | 28a331e | 2013-09-17 07:49:56 +0000 | [diff] [blame] | 191 | report_blocks.clear(); |
| 192 | report_blocks.push_back(CreateReportBlock(1, 2, 0, 41)); |
| 193 | second_bandwidth_observer->OnReceivedRtcpReceiverReport(report_blocks, 100, |
| 194 | 2001); |
kjellander@webrtc.org | 2e84c11 | 2012-05-31 13:55:01 +0000 | [diff] [blame] | 195 | EXPECT_EQ(235360u, bitrate_observer.last_bitrate_); |
pwestin@webrtc.org | 1cd1162 | 2012-04-19 12:13:52 +0000 | [diff] [blame] | 196 | |
| 197 | // Reports from only one bandwidth observer is ok. |
stefan@webrtc.org | 28a331e | 2013-09-17 07:49:56 +0000 | [diff] [blame] | 198 | report_blocks.clear(); |
| 199 | report_blocks.push_back(CreateReportBlock(1, 2, 0, 61)); |
| 200 | second_bandwidth_observer->OnReceivedRtcpReceiverReport(report_blocks, 50, |
| 201 | 3001); |
kjellander@webrtc.org | 2e84c11 | 2012-05-31 13:55:01 +0000 | [diff] [blame] | 202 | EXPECT_EQ(255189u, bitrate_observer.last_bitrate_); |
pwestin@webrtc.org | 1cd1162 | 2012-04-19 12:13:52 +0000 | [diff] [blame] | 203 | |
stefan@webrtc.org | 28a331e | 2013-09-17 07:49:56 +0000 | [diff] [blame] | 204 | report_blocks.clear(); |
| 205 | report_blocks.push_back(CreateReportBlock(1, 2, 0, 81)); |
| 206 | second_bandwidth_observer->OnReceivedRtcpReceiverReport(report_blocks, 50, |
| 207 | 4001); |
kjellander@webrtc.org | 2e84c11 | 2012-05-31 13:55:01 +0000 | [diff] [blame] | 208 | EXPECT_EQ(276604u, bitrate_observer.last_bitrate_); |
pwestin@webrtc.org | 1cd1162 | 2012-04-19 12:13:52 +0000 | [diff] [blame] | 209 | |
stefan@webrtc.org | 28a331e | 2013-09-17 07:49:56 +0000 | [diff] [blame] | 210 | report_blocks.clear(); |
| 211 | report_blocks.push_back(CreateReportBlock(1, 2, 0, 101)); |
| 212 | second_bandwidth_observer->OnReceivedRtcpReceiverReport(report_blocks, 50, |
| 213 | 5001); |
kjellander@webrtc.org | 2e84c11 | 2012-05-31 13:55:01 +0000 | [diff] [blame] | 214 | EXPECT_EQ(299732u, bitrate_observer.last_bitrate_); |
pwestin@webrtc.org | 1cd1162 | 2012-04-19 12:13:52 +0000 | [diff] [blame] | 215 | |
stefan@webrtc.org | 28a331e | 2013-09-17 07:49:56 +0000 | [diff] [blame] | 216 | report_blocks.clear(); |
| 217 | report_blocks.push_back(CreateReportBlock(1, 2, 0, 121)); |
| 218 | second_bandwidth_observer->OnReceivedRtcpReceiverReport(report_blocks, 50, |
| 219 | 6001); |
kjellander@webrtc.org | 2e84c11 | 2012-05-31 13:55:01 +0000 | [diff] [blame] | 220 | EXPECT_EQ(300000u, bitrate_observer.last_bitrate_); // Max cap. |
pwestin@webrtc.org | 1cd1162 | 2012-04-19 12:13:52 +0000 | [diff] [blame] | 221 | |
stefan@webrtc.org | 28a331e | 2013-09-17 07:49:56 +0000 | [diff] [blame] | 222 | report_blocks.clear(); |
| 223 | report_blocks.push_back(CreateReportBlock(1, 2, 0, 141)); |
| 224 | second_bandwidth_observer->OnReceivedRtcpReceiverReport(report_blocks, 50, |
| 225 | 7001); |
kjellander@webrtc.org | 2e84c11 | 2012-05-31 13:55:01 +0000 | [diff] [blame] | 226 | EXPECT_EQ(300000u, bitrate_observer.last_bitrate_); // Max cap. |
pwestin@webrtc.org | 1cd1162 | 2012-04-19 12:13:52 +0000 | [diff] [blame] | 227 | |
| 228 | // Test that a low REMB trigger immediately. |
| 229 | // We don't care which bandwidth observer that delivers the REMB. |
| 230 | second_bandwidth_observer->OnReceivedEstimatedBitrate(250000); |
kjellander@webrtc.org | 2e84c11 | 2012-05-31 13:55:01 +0000 | [diff] [blame] | 231 | EXPECT_EQ(250000u, bitrate_observer.last_bitrate_); |
| 232 | EXPECT_EQ(0, bitrate_observer.last_fraction_loss_); |
| 233 | EXPECT_EQ(50u, bitrate_observer.last_rtt_); |
pwestin@webrtc.org | 1cd1162 | 2012-04-19 12:13:52 +0000 | [diff] [blame] | 234 | |
| 235 | bandwidth_observer_->OnReceivedEstimatedBitrate(1000); |
kjellander@webrtc.org | 2e84c11 | 2012-05-31 13:55:01 +0000 | [diff] [blame] | 236 | EXPECT_EQ(100000u, bitrate_observer.last_bitrate_); // Min cap. |
pwestin@webrtc.org | 1cd1162 | 2012-04-19 12:13:52 +0000 | [diff] [blame] | 237 | controller_->RemoveBitrateObserver(&bitrate_observer); |
| 238 | delete second_bandwidth_observer; |
| 239 | } |
| 240 | |
stefan@webrtc.org | 28a331e | 2013-09-17 07:49:56 +0000 | [diff] [blame] | 241 | TEST_F(BitrateControllerTest, OneBitrateObserverMultipleReportBlocks) { |
| 242 | TestBitrateObserver bitrate_observer; |
| 243 | uint32_t sequence_number[2] = {0, 0xFF00}; |
| 244 | const uint32_t kStartBitrate = 200000; |
| 245 | const uint32_t kMinBitrate = 100000; |
| 246 | const uint32_t kMaxBitrate = 300000; |
| 247 | controller_->SetBitrateObserver(&bitrate_observer, kStartBitrate, kMinBitrate, |
| 248 | kMaxBitrate); |
| 249 | |
| 250 | // Receive a high REMB, test bitrate increase. |
| 251 | bandwidth_observer_->OnReceivedEstimatedBitrate(400000); |
| 252 | |
| 253 | webrtc::ReportBlockList report_blocks; |
| 254 | int64_t time_ms = 1001; |
| 255 | uint32_t last_bitrate = 0; |
| 256 | // Ramp up to max bitrate. |
| 257 | for (int i = 0; i < 6; ++i) { |
| 258 | report_blocks.push_back(CreateReportBlock(1, 2, 0, sequence_number[0])); |
| 259 | report_blocks.push_back(CreateReportBlock(1, 3, 0, sequence_number[1])); |
| 260 | bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, |
| 261 | time_ms); |
| 262 | EXPECT_GT(bitrate_observer.last_bitrate_, last_bitrate); |
| 263 | EXPECT_EQ(0, bitrate_observer.last_fraction_loss_); |
| 264 | EXPECT_EQ(50u, bitrate_observer.last_rtt_); |
| 265 | last_bitrate = bitrate_observer.last_bitrate_; |
| 266 | time_ms += 1000; |
| 267 | sequence_number[0] += 20; |
| 268 | sequence_number[1] += 1; |
| 269 | report_blocks.clear(); |
| 270 | } |
| 271 | |
| 272 | EXPECT_EQ(kMaxBitrate, bitrate_observer.last_bitrate_); |
| 273 | |
| 274 | // Packet loss on the first stream. Verify that bitrate decreases. |
| 275 | report_blocks.push_back(CreateReportBlock(1, 2, 50, sequence_number[0])); |
| 276 | report_blocks.push_back(CreateReportBlock(1, 3, 0, sequence_number[1])); |
| 277 | bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, time_ms); |
| 278 | EXPECT_LT(bitrate_observer.last_bitrate_, last_bitrate); |
| 279 | EXPECT_EQ(WeightedLoss(20, 50, 1, 0), bitrate_observer.last_fraction_loss_); |
| 280 | EXPECT_EQ(50u, bitrate_observer.last_rtt_); |
| 281 | last_bitrate = bitrate_observer.last_bitrate_; |
| 282 | sequence_number[0] += 20; |
| 283 | sequence_number[1] += 20; |
| 284 | time_ms += 1000; |
| 285 | report_blocks.clear(); |
| 286 | |
| 287 | // Packet loss on the second stream. Verify that bitrate decreases. |
| 288 | report_blocks.push_back(CreateReportBlock(1, 2, 0, sequence_number[0])); |
| 289 | report_blocks.push_back(CreateReportBlock(1, 3, 75, sequence_number[1])); |
| 290 | bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, time_ms); |
| 291 | EXPECT_LT(bitrate_observer.last_bitrate_, last_bitrate); |
| 292 | EXPECT_EQ(WeightedLoss(20, 0, 20, 75), bitrate_observer.last_fraction_loss_); |
| 293 | EXPECT_EQ(50u, bitrate_observer.last_rtt_); |
| 294 | last_bitrate = bitrate_observer.last_bitrate_; |
| 295 | sequence_number[0] += 20; |
| 296 | sequence_number[1] += 1; |
| 297 | time_ms += 1000; |
| 298 | report_blocks.clear(); |
| 299 | |
| 300 | // All packets lost on stream with few packets, no back-off. |
| 301 | report_blocks.push_back(CreateReportBlock(1, 2, 1, sequence_number[0])); |
| 302 | report_blocks.push_back(CreateReportBlock(1, 3, 255, sequence_number[1])); |
| 303 | bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, time_ms); |
| 304 | EXPECT_EQ(bitrate_observer.last_bitrate_, last_bitrate); |
| 305 | EXPECT_EQ(WeightedLoss(20, 1, 1, 255), bitrate_observer.last_fraction_loss_); |
| 306 | EXPECT_EQ(50u, bitrate_observer.last_rtt_); |
| 307 | last_bitrate = bitrate_observer.last_bitrate_; |
| 308 | sequence_number[0] += 20; |
| 309 | sequence_number[1] += 1; |
| 310 | report_blocks.clear(); |
| 311 | } |
| 312 | |
pwestin@webrtc.org | 1cd1162 | 2012-04-19 12:13:52 +0000 | [diff] [blame] | 313 | TEST_F(BitrateControllerTest, TwoBitrateObserversOneRtcpObserver) { |
| 314 | TestBitrateObserver bitrate_observer_1; |
| 315 | TestBitrateObserver bitrate_observer_2; |
pwestin@webrtc.org | 1cd1162 | 2012-04-19 12:13:52 +0000 | [diff] [blame] | 316 | controller_->SetBitrateObserver(&bitrate_observer_2, 200000, 200000, 300000); |
stefan@webrtc.org | 1281dc0 | 2012-08-13 16:13:09 +0000 | [diff] [blame] | 317 | controller_->SetBitrateObserver(&bitrate_observer_1, 200000, 100000, 300000); |
pwestin@webrtc.org | 1cd1162 | 2012-04-19 12:13:52 +0000 | [diff] [blame] | 318 | |
| 319 | // Receive a high remb, test bitrate inc. |
| 320 | bandwidth_observer_->OnReceivedEstimatedBitrate(400000); |
| 321 | |
| 322 | // Test too low start bitrate, hence lower than sum of min. |
stefan@webrtc.org | 28a331e | 2013-09-17 07:49:56 +0000 | [diff] [blame] | 323 | webrtc::ReportBlockList report_blocks; |
| 324 | report_blocks.push_back(CreateReportBlock(1, 2, 0, 1)); |
| 325 | bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, 1); |
kjellander@webrtc.org | 2e84c11 | 2012-05-31 13:55:01 +0000 | [diff] [blame] | 326 | EXPECT_EQ(100000u, bitrate_observer_1.last_bitrate_); |
| 327 | EXPECT_EQ(0, bitrate_observer_1.last_fraction_loss_); |
| 328 | EXPECT_EQ(50u, bitrate_observer_1.last_rtt_); |
pwestin@webrtc.org | 1cd1162 | 2012-04-19 12:13:52 +0000 | [diff] [blame] | 329 | |
kjellander@webrtc.org | 2e84c11 | 2012-05-31 13:55:01 +0000 | [diff] [blame] | 330 | EXPECT_EQ(200000u, bitrate_observer_2.last_bitrate_); |
| 331 | EXPECT_EQ(0, bitrate_observer_2.last_fraction_loss_); |
| 332 | EXPECT_EQ(50u, bitrate_observer_2.last_rtt_); |
pwestin@webrtc.org | 1cd1162 | 2012-04-19 12:13:52 +0000 | [diff] [blame] | 333 | |
andresp@webrtc.org | 07bc734 | 2014-03-21 16:51:01 +0000 | [diff] [blame^] | 334 | // Test bitrate increase 8% per second, distributed equally. |
stefan@webrtc.org | 28a331e | 2013-09-17 07:49:56 +0000 | [diff] [blame] | 335 | report_blocks.clear(); |
andresp@webrtc.org | 07bc734 | 2014-03-21 16:51:01 +0000 | [diff] [blame^] | 336 | report_blocks.push_back(CreateReportBlock(1, 2, 0, 21)); |
| 337 | bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, 1001); |
kjellander@webrtc.org | 2e84c11 | 2012-05-31 13:55:01 +0000 | [diff] [blame] | 338 | EXPECT_EQ(112500u, bitrate_observer_1.last_bitrate_); |
| 339 | EXPECT_EQ(212500u, bitrate_observer_2.last_bitrate_); |
pwestin@webrtc.org | 1cd1162 | 2012-04-19 12:13:52 +0000 | [diff] [blame] | 340 | |
stefan@webrtc.org | 28a331e | 2013-09-17 07:49:56 +0000 | [diff] [blame] | 341 | report_blocks.clear(); |
andresp@webrtc.org | 07bc734 | 2014-03-21 16:51:01 +0000 | [diff] [blame^] | 342 | report_blocks.push_back(CreateReportBlock(1, 2, 0, 41)); |
| 343 | bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, 2001); |
kjellander@webrtc.org | 2e84c11 | 2012-05-31 13:55:01 +0000 | [diff] [blame] | 344 | EXPECT_EQ(126000u, bitrate_observer_1.last_bitrate_); |
| 345 | EXPECT_EQ(226000u, bitrate_observer_2.last_bitrate_); |
pwestin@webrtc.org | 1cd1162 | 2012-04-19 12:13:52 +0000 | [diff] [blame] | 346 | |
stefan@webrtc.org | 28a331e | 2013-09-17 07:49:56 +0000 | [diff] [blame] | 347 | report_blocks.clear(); |
andresp@webrtc.org | 07bc734 | 2014-03-21 16:51:01 +0000 | [diff] [blame^] | 348 | report_blocks.push_back(CreateReportBlock(1, 2, 0, 61)); |
| 349 | bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, 3001); |
kjellander@webrtc.org | 2e84c11 | 2012-05-31 13:55:01 +0000 | [diff] [blame] | 350 | EXPECT_EQ(140580u, bitrate_observer_1.last_bitrate_); |
| 351 | EXPECT_EQ(240580u, bitrate_observer_2.last_bitrate_); |
pwestin@webrtc.org | 1cd1162 | 2012-04-19 12:13:52 +0000 | [diff] [blame] | 352 | |
| 353 | // Check that the bitrate sum honor our REMB. |
stefan@webrtc.org | 28a331e | 2013-09-17 07:49:56 +0000 | [diff] [blame] | 354 | report_blocks.clear(); |
| 355 | report_blocks.push_back(CreateReportBlock(1, 2, 0, 101)); |
| 356 | bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, 5001); |
kjellander@webrtc.org | 2e84c11 | 2012-05-31 13:55:01 +0000 | [diff] [blame] | 357 | EXPECT_EQ(150000u, bitrate_observer_1.last_bitrate_); |
| 358 | EXPECT_EQ(250000u, bitrate_observer_2.last_bitrate_); |
pwestin@webrtc.org | 1cd1162 | 2012-04-19 12:13:52 +0000 | [diff] [blame] | 359 | |
| 360 | // Remove REMB cap, higher than sum of max. |
| 361 | bandwidth_observer_->OnReceivedEstimatedBitrate(700000); |
| 362 | |
stefan@webrtc.org | 28a331e | 2013-09-17 07:49:56 +0000 | [diff] [blame] | 363 | report_blocks.clear(); |
| 364 | report_blocks.push_back(CreateReportBlock(1, 2, 0, 121)); |
| 365 | bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, 6001); |
kjellander@webrtc.org | 2e84c11 | 2012-05-31 13:55:01 +0000 | [diff] [blame] | 366 | EXPECT_EQ(166500u, bitrate_observer_1.last_bitrate_); |
| 367 | EXPECT_EQ(266500u, bitrate_observer_2.last_bitrate_); |
pwestin@webrtc.org | 1cd1162 | 2012-04-19 12:13:52 +0000 | [diff] [blame] | 368 | |
stefan@webrtc.org | 28a331e | 2013-09-17 07:49:56 +0000 | [diff] [blame] | 369 | report_blocks.clear(); |
| 370 | report_blocks.push_back(CreateReportBlock(1, 2, 0, 141)); |
| 371 | bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, 7001); |
kjellander@webrtc.org | 2e84c11 | 2012-05-31 13:55:01 +0000 | [diff] [blame] | 372 | EXPECT_EQ(184320u, bitrate_observer_1.last_bitrate_); |
| 373 | EXPECT_EQ(284320u, bitrate_observer_2.last_bitrate_); |
pwestin@webrtc.org | 1cd1162 | 2012-04-19 12:13:52 +0000 | [diff] [blame] | 374 | |
stefan@webrtc.org | 28a331e | 2013-09-17 07:49:56 +0000 | [diff] [blame] | 375 | report_blocks.clear(); |
| 376 | report_blocks.push_back(CreateReportBlock(1, 2, 0, 161)); |
| 377 | bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, 8001); |
kjellander@webrtc.org | 2e84c11 | 2012-05-31 13:55:01 +0000 | [diff] [blame] | 378 | EXPECT_EQ(207130u, bitrate_observer_1.last_bitrate_); |
| 379 | EXPECT_EQ(300000u, bitrate_observer_2.last_bitrate_); // Max cap. |
pwestin@webrtc.org | 1cd1162 | 2012-04-19 12:13:52 +0000 | [diff] [blame] | 380 | |
stefan@webrtc.org | 28a331e | 2013-09-17 07:49:56 +0000 | [diff] [blame] | 381 | report_blocks.clear(); |
| 382 | report_blocks.push_back(CreateReportBlock(1, 2, 0, 181)); |
| 383 | bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, 9001); |
kjellander@webrtc.org | 2e84c11 | 2012-05-31 13:55:01 +0000 | [diff] [blame] | 384 | EXPECT_EQ(248700u, bitrate_observer_1.last_bitrate_); |
| 385 | EXPECT_EQ(300000u, bitrate_observer_2.last_bitrate_); |
pwestin@webrtc.org | 1cd1162 | 2012-04-19 12:13:52 +0000 | [diff] [blame] | 386 | |
stefan@webrtc.org | 28a331e | 2013-09-17 07:49:56 +0000 | [diff] [blame] | 387 | report_blocks.clear(); |
| 388 | report_blocks.push_back(CreateReportBlock(1, 2, 0, 201)); |
| 389 | bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, 10001); |
kjellander@webrtc.org | 2e84c11 | 2012-05-31 13:55:01 +0000 | [diff] [blame] | 390 | EXPECT_EQ(293596u, bitrate_observer_1.last_bitrate_); |
| 391 | EXPECT_EQ(300000u, bitrate_observer_2.last_bitrate_); |
pwestin@webrtc.org | 1cd1162 | 2012-04-19 12:13:52 +0000 | [diff] [blame] | 392 | |
stefan@webrtc.org | 28a331e | 2013-09-17 07:49:56 +0000 | [diff] [blame] | 393 | report_blocks.clear(); |
| 394 | report_blocks.push_back(CreateReportBlock(1, 2, 0, 221)); |
| 395 | bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, 11001); |
kjellander@webrtc.org | 2e84c11 | 2012-05-31 13:55:01 +0000 | [diff] [blame] | 396 | EXPECT_EQ(300000u, bitrate_observer_1.last_bitrate_); // Max cap. |
| 397 | EXPECT_EQ(300000u, bitrate_observer_2.last_bitrate_); |
pwestin@webrtc.org | 1cd1162 | 2012-04-19 12:13:52 +0000 | [diff] [blame] | 398 | |
| 399 | // Test that a low REMB trigger immediately. |
| 400 | bandwidth_observer_->OnReceivedEstimatedBitrate(350000); |
kjellander@webrtc.org | 2e84c11 | 2012-05-31 13:55:01 +0000 | [diff] [blame] | 401 | EXPECT_EQ(125000u, bitrate_observer_1.last_bitrate_); |
| 402 | EXPECT_EQ(0, bitrate_observer_1.last_fraction_loss_); |
| 403 | EXPECT_EQ(50u, bitrate_observer_1.last_rtt_); |
| 404 | EXPECT_EQ(225000u, bitrate_observer_2.last_bitrate_); |
| 405 | EXPECT_EQ(0, bitrate_observer_2.last_fraction_loss_); |
| 406 | EXPECT_EQ(50u, bitrate_observer_2.last_rtt_); |
pwestin@webrtc.org | 1cd1162 | 2012-04-19 12:13:52 +0000 | [diff] [blame] | 407 | |
| 408 | bandwidth_observer_->OnReceivedEstimatedBitrate(1000); |
kjellander@webrtc.org | 2e84c11 | 2012-05-31 13:55:01 +0000 | [diff] [blame] | 409 | EXPECT_EQ(100000u, bitrate_observer_1.last_bitrate_); // Min cap. |
| 410 | EXPECT_EQ(200000u, bitrate_observer_2.last_bitrate_); // Min cap. |
pwestin@webrtc.org | 1cd1162 | 2012-04-19 12:13:52 +0000 | [diff] [blame] | 411 | controller_->RemoveBitrateObserver(&bitrate_observer_1); |
| 412 | controller_->RemoveBitrateObserver(&bitrate_observer_2); |
| 413 | } |
henrik.lundin@webrtc.org | 29dd0de | 2013-10-21 14:00:01 +0000 | [diff] [blame] | 414 | |
| 415 | class BitrateControllerTestNoEnforceMin : public BitrateControllerTest { |
| 416 | protected: |
| 417 | BitrateControllerTestNoEnforceMin() : BitrateControllerTest() { |
| 418 | enforce_min_bitrate_ = false; |
| 419 | } |
| 420 | }; |
| 421 | |
| 422 | // The following three tests verify that the EnforceMinBitrate() method works |
| 423 | // as intended. |
| 424 | TEST_F(BitrateControllerTestNoEnforceMin, OneBitrateObserver) { |
| 425 | TestBitrateObserver bitrate_observer_1; |
| 426 | controller_->SetBitrateObserver(&bitrate_observer_1, 200000, 100000, 400000); |
| 427 | |
| 428 | // High REMB. |
| 429 | bandwidth_observer_->OnReceivedEstimatedBitrate(150000); |
| 430 | EXPECT_EQ(150000u, bitrate_observer_1.last_bitrate_); |
| 431 | |
| 432 | // Low REMB. |
andresp@webrtc.org | 07bc734 | 2014-03-21 16:51:01 +0000 | [diff] [blame^] | 433 | bandwidth_observer_->OnReceivedEstimatedBitrate(10000); |
| 434 | EXPECT_EQ(10000u, bitrate_observer_1.last_bitrate_); |
| 435 | |
| 436 | // Keeps at least 10 kbps. |
| 437 | bandwidth_observer_->OnReceivedEstimatedBitrate(9000); |
| 438 | EXPECT_EQ(10000u, bitrate_observer_1.last_bitrate_); |
henrik.lundin@webrtc.org | 29dd0de | 2013-10-21 14:00:01 +0000 | [diff] [blame] | 439 | |
| 440 | controller_->RemoveBitrateObserver(&bitrate_observer_1); |
| 441 | } |
| 442 | |
| 443 | TEST_F(BitrateControllerTestNoEnforceMin, ThreeBitrateObservers) { |
| 444 | TestBitrateObserver bitrate_observer_1; |
| 445 | TestBitrateObserver bitrate_observer_2; |
| 446 | TestBitrateObserver bitrate_observer_3; |
| 447 | // Set up the observers with min bitrates at 100000, 200000, and 300000. |
| 448 | // Note: The start bitrate of bitrate_observer_1 (700000) is used as the |
| 449 | // overall start bitrate. |
| 450 | controller_->SetBitrateObserver(&bitrate_observer_1, 700000, 100000, 400000); |
| 451 | controller_->SetBitrateObserver(&bitrate_observer_2, 200000, 200000, 400000); |
| 452 | controller_->SetBitrateObserver(&bitrate_observer_3, 200000, 300000, 400000); |
| 453 | |
| 454 | // High REMB. Make sure the controllers get a fair share of the surplus |
| 455 | // (i.e., what is left after each controller gets its min rate). |
| 456 | bandwidth_observer_->OnReceivedEstimatedBitrate(690000); |
| 457 | // Verify that each observer gets its min rate (sum of min rates is 600000), |
| 458 | // and that the remaining 90000 is divided equally among the three. |
| 459 | EXPECT_EQ(130000u, bitrate_observer_1.last_bitrate_); |
| 460 | EXPECT_EQ(230000u, bitrate_observer_2.last_bitrate_); |
| 461 | EXPECT_EQ(330000u, bitrate_observer_3.last_bitrate_); |
| 462 | |
| 463 | // High REMB, but below the sum of min bitrates. |
| 464 | bandwidth_observer_->OnReceivedEstimatedBitrate(500000); |
| 465 | // Verify that the first and second observers get their min bitrates, and the |
| 466 | // third gets the remainder. |
| 467 | EXPECT_EQ(100000u, bitrate_observer_1.last_bitrate_); // Min bitrate. |
| 468 | EXPECT_EQ(200000u, bitrate_observer_2.last_bitrate_); // Min bitrate. |
| 469 | EXPECT_EQ(200000u, bitrate_observer_3.last_bitrate_); // Remainder. |
| 470 | |
| 471 | // Low REMB. |
andresp@webrtc.org | 07bc734 | 2014-03-21 16:51:01 +0000 | [diff] [blame^] | 472 | bandwidth_observer_->OnReceivedEstimatedBitrate(10000); |
henrik.lundin@webrtc.org | 29dd0de | 2013-10-21 14:00:01 +0000 | [diff] [blame] | 473 | // Verify that the first observer gets all the rate, and the rest get zero. |
andresp@webrtc.org | 07bc734 | 2014-03-21 16:51:01 +0000 | [diff] [blame^] | 474 | EXPECT_EQ(10000u, bitrate_observer_1.last_bitrate_); |
| 475 | EXPECT_EQ(0u, bitrate_observer_2.last_bitrate_); |
| 476 | EXPECT_EQ(0u, bitrate_observer_3.last_bitrate_); |
| 477 | |
| 478 | // Verify it keeps an estimate of at least 10kbps. |
| 479 | bandwidth_observer_->OnReceivedEstimatedBitrate(9000); |
| 480 | EXPECT_EQ(10000u, bitrate_observer_1.last_bitrate_); |
henrik.lundin@webrtc.org | 29dd0de | 2013-10-21 14:00:01 +0000 | [diff] [blame] | 481 | EXPECT_EQ(0u, bitrate_observer_2.last_bitrate_); |
| 482 | EXPECT_EQ(0u, bitrate_observer_3.last_bitrate_); |
| 483 | |
| 484 | controller_->RemoveBitrateObserver(&bitrate_observer_1); |
| 485 | controller_->RemoveBitrateObserver(&bitrate_observer_2); |
| 486 | controller_->RemoveBitrateObserver(&bitrate_observer_3); |
| 487 | } |
| 488 | |
| 489 | TEST_F(BitrateControllerTest, ThreeBitrateObserversLowRembEnforceMin) { |
| 490 | TestBitrateObserver bitrate_observer_1; |
| 491 | TestBitrateObserver bitrate_observer_2; |
| 492 | TestBitrateObserver bitrate_observer_3; |
| 493 | controller_->SetBitrateObserver(&bitrate_observer_1, 200000, 100000, 300000); |
| 494 | controller_->SetBitrateObserver(&bitrate_observer_2, 200000, 200000, 300000); |
| 495 | controller_->SetBitrateObserver(&bitrate_observer_3, 200000, 300000, 300000); |
| 496 | |
| 497 | // Low REMB. Verify that all observers still get their respective min bitrate. |
| 498 | bandwidth_observer_->OnReceivedEstimatedBitrate(1000); |
| 499 | EXPECT_EQ(100000u, bitrate_observer_1.last_bitrate_); // Min cap. |
| 500 | EXPECT_EQ(200000u, bitrate_observer_2.last_bitrate_); // Min cap. |
| 501 | EXPECT_EQ(300000u, bitrate_observer_3.last_bitrate_); // Min cap. |
| 502 | |
| 503 | controller_->RemoveBitrateObserver(&bitrate_observer_1); |
| 504 | controller_->RemoveBitrateObserver(&bitrate_observer_2); |
| 505 | controller_->RemoveBitrateObserver(&bitrate_observer_3); |
| 506 | } |