blob: edfe34377594a67c7e6651d1f521dea0f657c46d [file] [log] [blame]
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +00001/*
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.org2e10b8e2013-07-16 12:54:53 +000011#include "testing/gtest/include/gtest/gtest.h"
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +000012
13#include <algorithm>
14#include <vector>
15
pbos@webrtc.org2e10b8e2013-07-16 12:54:53 +000016#include "webrtc/modules/bitrate_controller/include/bitrate_controller.h"
17#include "webrtc/modules/rtp_rtcp/interface/rtp_rtcp_defines.h"
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +000018
19using webrtc::RtcpBandwidthObserver;
20using webrtc::BitrateObserver;
21using webrtc::BitrateController;
22
stefan@webrtc.org28a331e2013-09-17 07:49:56 +000023uint8_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
31webrtc::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.org1cd11622012-04-19 12:13:52 +000038class TestBitrateObserver: public BitrateObserver {
39 public:
40 TestBitrateObserver()
kjellander@webrtc.org2e84c112012-05-31 13:55:01 +000041 : last_bitrate_(0),
42 last_fraction_loss_(0),
43 last_rtt_(0) {
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +000044 }
45
46 virtual void OnNetworkChanged(const uint32_t bitrate,
47 const uint8_t fraction_loss,
48 const uint32_t rtt) {
kjellander@webrtc.org2e84c112012-05-31 13:55:01 +000049 last_bitrate_ = bitrate;
50 last_fraction_loss_ = fraction_loss;
51 last_rtt_ = rtt;
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +000052 }
kjellander@webrtc.org2e84c112012-05-31 13:55:01 +000053 uint32_t last_bitrate_;
54 uint8_t last_fraction_loss_;
55 uint32_t last_rtt_;
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +000056};
57
58class BitrateControllerTest : public ::testing::Test {
59 protected:
andresp@webrtc.orgda077372014-03-25 12:48:42 +000060 BitrateControllerTest() : clock_(0), enforce_min_bitrate_(true) {}
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +000061 ~BitrateControllerTest() {}
62
63 virtual void SetUp() {
andresp@webrtc.orgda077372014-03-25 12:48:42 +000064 controller_ = BitrateController::CreateBitrateController(
65 &clock_, enforce_min_bitrate_);
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +000066 bandwidth_observer_ = controller_->CreateRtcpBandwidthObserver();
67 }
68
69 virtual void TearDown() {
70 delete bandwidth_observer_;
71 delete controller_;
72 }
andresp@webrtc.orgda077372014-03-25 12:48:42 +000073
74 webrtc::SimulatedClock clock_;
henrik.lundin@webrtc.org29dd0de2013-10-21 14:00:01 +000075 bool enforce_min_bitrate_;
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +000076 BitrateController* controller_;
77 RtcpBandwidthObserver* bandwidth_observer_;
78};
79
80TEST_F(BitrateControllerTest, Basic) {
81 TestBitrateObserver bitrate_observer;
82 controller_->SetBitrateObserver(&bitrate_observer, 200000, 100000, 300000);
83 controller_->RemoveBitrateObserver(&bitrate_observer);
84}
85
86TEST_F(BitrateControllerTest, OneBitrateObserverOneRtcpObserver) {
87 TestBitrateObserver bitrate_observer;
88 controller_->SetBitrateObserver(&bitrate_observer, 200000, 100000, 300000);
89
90 // Receive a high remb, test bitrate inc.
91 bandwidth_observer_->OnReceivedEstimatedBitrate(400000);
andresp@webrtc.orgda077372014-03-25 12:48:42 +000092 EXPECT_EQ(200000u, bitrate_observer.last_bitrate_);
93 EXPECT_EQ(0, bitrate_observer.last_fraction_loss_);
94 EXPECT_EQ(0u, bitrate_observer.last_rtt_);
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +000095
andresp@webrtc.orgda077372014-03-25 12:48:42 +000096 // Test bitrate increase 8% per second.
stefan@webrtc.org28a331e2013-09-17 07:49:56 +000097 webrtc::ReportBlockList report_blocks;
98 report_blocks.push_back(CreateReportBlock(1, 2, 0, 1));
99 bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, 1);
andresp@webrtc.orgda077372014-03-25 12:48:42 +0000100 EXPECT_EQ(217000u, bitrate_observer.last_bitrate_);
kjellander@webrtc.org2e84c112012-05-31 13:55:01 +0000101 EXPECT_EQ(0, bitrate_observer.last_fraction_loss_);
andresp@webrtc.org07bc7342014-03-21 16:51:01 +0000102 EXPECT_EQ(50u, bitrate_observer.last_rtt_);
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000103
stefan@webrtc.org28a331e2013-09-17 07:49:56 +0000104 report_blocks.clear();
105 report_blocks.push_back(CreateReportBlock(1, 2, 0, 21));
106 bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, 1001);
andresp@webrtc.orgda077372014-03-25 12:48:42 +0000107 EXPECT_EQ(235360u, bitrate_observer.last_bitrate_);
kjellander@webrtc.org2e84c112012-05-31 13:55:01 +0000108 EXPECT_EQ(0, bitrate_observer.last_fraction_loss_);
109 EXPECT_EQ(50u, bitrate_observer.last_rtt_);
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000110
stefan@webrtc.org28a331e2013-09-17 07:49:56 +0000111 report_blocks.clear();
112 report_blocks.push_back(CreateReportBlock(1, 2, 0, 41));
113 bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, 2001);
andresp@webrtc.orgda077372014-03-25 12:48:42 +0000114 EXPECT_EQ(255189u, bitrate_observer.last_bitrate_);
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000115
stefan@webrtc.org28a331e2013-09-17 07:49:56 +0000116 report_blocks.clear();
117 report_blocks.push_back(CreateReportBlock(1, 2, 0, 61));
118 bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, 3001);
andresp@webrtc.orgda077372014-03-25 12:48:42 +0000119 EXPECT_EQ(276604u, bitrate_observer.last_bitrate_);
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000120
stefan@webrtc.org28a331e2013-09-17 07:49:56 +0000121 report_blocks.clear();
122 report_blocks.push_back(CreateReportBlock(1, 2, 0, 801));
123 bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, 4001);
andresp@webrtc.orgda077372014-03-25 12:48:42 +0000124 EXPECT_EQ(299732u, bitrate_observer.last_bitrate_);
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000125
andresp@webrtc.orgda077372014-03-25 12:48:42 +0000126 // Reach max cap.
stefan@webrtc.org28a331e2013-09-17 07:49:56 +0000127 report_blocks.clear();
128 report_blocks.push_back(CreateReportBlock(1, 2, 0, 101));
129 bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, 5001);
andresp@webrtc.orgda077372014-03-25 12:48:42 +0000130 EXPECT_EQ(300000u, bitrate_observer.last_bitrate_);
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000131
stefan@webrtc.org28a331e2013-09-17 07:49:56 +0000132 report_blocks.clear();
133 report_blocks.push_back(CreateReportBlock(1, 2, 0, 141));
134 bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, 7001);
andresp@webrtc.orgda077372014-03-25 12:48:42 +0000135 EXPECT_EQ(300000u, bitrate_observer.last_bitrate_);
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000136
137 // Test that a low REMB trigger immediately.
138 bandwidth_observer_->OnReceivedEstimatedBitrate(250000);
kjellander@webrtc.org2e84c112012-05-31 13:55:01 +0000139 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.org1cd11622012-04-19 12:13:52 +0000142
143 bandwidth_observer_->OnReceivedEstimatedBitrate(1000);
kjellander@webrtc.org2e84c112012-05-31 13:55:01 +0000144 EXPECT_EQ(100000u, bitrate_observer.last_bitrate_); // Min cap.
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000145 controller_->RemoveBitrateObserver(&bitrate_observer);
146}
147
148TEST_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);
andresp@webrtc.orgda077372014-03-25 12:48:42 +0000157 EXPECT_EQ(200000u, bitrate_observer.last_bitrate_);
158 EXPECT_EQ(0, bitrate_observer.last_fraction_loss_);
159 EXPECT_EQ(0u, bitrate_observer.last_rtt_);
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000160
161 // Test start bitrate.
stefan@webrtc.org28a331e2013-09-17 07:49:56 +0000162 webrtc::ReportBlockList report_blocks;
163 report_blocks.push_back(CreateReportBlock(1, 2, 0, 1));
164 bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, 1);
165 second_bandwidth_observer->OnReceivedRtcpReceiverReport(
166 report_blocks, 100, 1);
andresp@webrtc.orgda077372014-03-25 12:48:42 +0000167 EXPECT_EQ(217000u, bitrate_observer.last_bitrate_);
kjellander@webrtc.org2e84c112012-05-31 13:55:01 +0000168 EXPECT_EQ(0, bitrate_observer.last_fraction_loss_);
andresp@webrtc.org07bc7342014-03-21 16:51:01 +0000169 EXPECT_EQ(100u, bitrate_observer.last_rtt_);
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000170
171 // Test bitrate increase 8% per second.
stefan@webrtc.org28a331e2013-09-17 07:49:56 +0000172 report_blocks.clear();
173 report_blocks.push_back(CreateReportBlock(1, 2, 0, 21));
174 bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, 501);
175 second_bandwidth_observer->OnReceivedRtcpReceiverReport(report_blocks, 100,
176 1001);
andresp@webrtc.orgda077372014-03-25 12:48:42 +0000177 EXPECT_EQ(235360u, bitrate_observer.last_bitrate_);
kjellander@webrtc.org2e84c112012-05-31 13:55:01 +0000178 EXPECT_EQ(0, bitrate_observer.last_fraction_loss_);
179 EXPECT_EQ(100u, bitrate_observer.last_rtt_);
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000180
181 // Extra report should not change estimate.
stefan@webrtc.org28a331e2013-09-17 07:49:56 +0000182 report_blocks.clear();
183 report_blocks.push_back(CreateReportBlock(1, 2, 0, 31));
184 second_bandwidth_observer->OnReceivedRtcpReceiverReport(report_blocks, 100,
185 1501);
andresp@webrtc.orgda077372014-03-25 12:48:42 +0000186 EXPECT_EQ(235360u, bitrate_observer.last_bitrate_);
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000187
stefan@webrtc.org28a331e2013-09-17 07:49:56 +0000188 report_blocks.clear();
189 report_blocks.push_back(CreateReportBlock(1, 2, 0, 41));
190 bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, 2001);
andresp@webrtc.orgda077372014-03-25 12:48:42 +0000191 EXPECT_EQ(255189u, bitrate_observer.last_bitrate_);
pwestin@webrtc.orga2cd7322012-04-23 08:32:47 +0000192
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000193 // Second report should not change estimate.
stefan@webrtc.org28a331e2013-09-17 07:49:56 +0000194 report_blocks.clear();
195 report_blocks.push_back(CreateReportBlock(1, 2, 0, 41));
196 second_bandwidth_observer->OnReceivedRtcpReceiverReport(report_blocks, 100,
197 2001);
andresp@webrtc.orgda077372014-03-25 12:48:42 +0000198 EXPECT_EQ(255189u, bitrate_observer.last_bitrate_);
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000199
200 // Reports from only one bandwidth observer is ok.
stefan@webrtc.org28a331e2013-09-17 07:49:56 +0000201 report_blocks.clear();
202 report_blocks.push_back(CreateReportBlock(1, 2, 0, 61));
203 second_bandwidth_observer->OnReceivedRtcpReceiverReport(report_blocks, 50,
204 3001);
andresp@webrtc.orgda077372014-03-25 12:48:42 +0000205 EXPECT_EQ(276604u, bitrate_observer.last_bitrate_);
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000206
stefan@webrtc.org28a331e2013-09-17 07:49:56 +0000207 report_blocks.clear();
208 report_blocks.push_back(CreateReportBlock(1, 2, 0, 81));
209 second_bandwidth_observer->OnReceivedRtcpReceiverReport(report_blocks, 50,
210 4001);
kjellander@webrtc.org2e84c112012-05-31 13:55:01 +0000211 EXPECT_EQ(299732u, bitrate_observer.last_bitrate_);
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000212
andresp@webrtc.orgda077372014-03-25 12:48:42 +0000213 // Reach max cap.
stefan@webrtc.org28a331e2013-09-17 07:49:56 +0000214 report_blocks.clear();
215 report_blocks.push_back(CreateReportBlock(1, 2, 0, 121));
andresp@webrtc.orgda077372014-03-25 12:48:42 +0000216 second_bandwidth_observer->OnReceivedRtcpReceiverReport(
217 report_blocks, 50, 5001);
218 EXPECT_EQ(300000u, bitrate_observer.last_bitrate_);
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000219
stefan@webrtc.org28a331e2013-09-17 07:49:56 +0000220 report_blocks.clear();
221 report_blocks.push_back(CreateReportBlock(1, 2, 0, 141));
andresp@webrtc.orgda077372014-03-25 12:48:42 +0000222 second_bandwidth_observer->OnReceivedRtcpReceiverReport(
223 report_blocks, 50, 6001);
224 EXPECT_EQ(300000u, bitrate_observer.last_bitrate_);
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000225
226 // Test that a low REMB trigger immediately.
227 // We don't care which bandwidth observer that delivers the REMB.
228 second_bandwidth_observer->OnReceivedEstimatedBitrate(250000);
kjellander@webrtc.org2e84c112012-05-31 13:55:01 +0000229 EXPECT_EQ(250000u, bitrate_observer.last_bitrate_);
230 EXPECT_EQ(0, bitrate_observer.last_fraction_loss_);
231 EXPECT_EQ(50u, bitrate_observer.last_rtt_);
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000232
andresp@webrtc.orgda077372014-03-25 12:48:42 +0000233 // Min cap.
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000234 bandwidth_observer_->OnReceivedEstimatedBitrate(1000);
andresp@webrtc.orgda077372014-03-25 12:48:42 +0000235 EXPECT_EQ(100000u, bitrate_observer.last_bitrate_);
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000236 controller_->RemoveBitrateObserver(&bitrate_observer);
237 delete second_bandwidth_observer;
238}
239
stefan@webrtc.org28a331e2013-09-17 07:49:56 +0000240TEST_F(BitrateControllerTest, OneBitrateObserverMultipleReportBlocks) {
241 TestBitrateObserver bitrate_observer;
242 uint32_t sequence_number[2] = {0, 0xFF00};
243 const uint32_t kStartBitrate = 200000;
244 const uint32_t kMinBitrate = 100000;
245 const uint32_t kMaxBitrate = 300000;
246 controller_->SetBitrateObserver(&bitrate_observer, kStartBitrate, kMinBitrate,
247 kMaxBitrate);
248
249 // Receive a high REMB, test bitrate increase.
250 bandwidth_observer_->OnReceivedEstimatedBitrate(400000);
251
252 webrtc::ReportBlockList report_blocks;
253 int64_t time_ms = 1001;
254 uint32_t last_bitrate = 0;
255 // Ramp up to max bitrate.
256 for (int i = 0; i < 6; ++i) {
257 report_blocks.push_back(CreateReportBlock(1, 2, 0, sequence_number[0]));
258 report_blocks.push_back(CreateReportBlock(1, 3, 0, sequence_number[1]));
259 bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50,
260 time_ms);
261 EXPECT_GT(bitrate_observer.last_bitrate_, last_bitrate);
262 EXPECT_EQ(0, bitrate_observer.last_fraction_loss_);
263 EXPECT_EQ(50u, bitrate_observer.last_rtt_);
264 last_bitrate = bitrate_observer.last_bitrate_;
265 time_ms += 1000;
266 sequence_number[0] += 20;
267 sequence_number[1] += 1;
268 report_blocks.clear();
269 }
270
271 EXPECT_EQ(kMaxBitrate, bitrate_observer.last_bitrate_);
272
273 // Packet loss on the first stream. Verify that bitrate decreases.
274 report_blocks.push_back(CreateReportBlock(1, 2, 50, sequence_number[0]));
275 report_blocks.push_back(CreateReportBlock(1, 3, 0, sequence_number[1]));
276 bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, time_ms);
277 EXPECT_LT(bitrate_observer.last_bitrate_, last_bitrate);
278 EXPECT_EQ(WeightedLoss(20, 50, 1, 0), bitrate_observer.last_fraction_loss_);
279 EXPECT_EQ(50u, bitrate_observer.last_rtt_);
280 last_bitrate = bitrate_observer.last_bitrate_;
281 sequence_number[0] += 20;
282 sequence_number[1] += 20;
283 time_ms += 1000;
284 report_blocks.clear();
285
286 // Packet loss on the second stream. Verify that bitrate decreases.
287 report_blocks.push_back(CreateReportBlock(1, 2, 0, sequence_number[0]));
288 report_blocks.push_back(CreateReportBlock(1, 3, 75, sequence_number[1]));
289 bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, time_ms);
290 EXPECT_LT(bitrate_observer.last_bitrate_, last_bitrate);
291 EXPECT_EQ(WeightedLoss(20, 0, 20, 75), bitrate_observer.last_fraction_loss_);
292 EXPECT_EQ(50u, bitrate_observer.last_rtt_);
293 last_bitrate = bitrate_observer.last_bitrate_;
294 sequence_number[0] += 20;
295 sequence_number[1] += 1;
296 time_ms += 1000;
297 report_blocks.clear();
298
299 // All packets lost on stream with few packets, no back-off.
300 report_blocks.push_back(CreateReportBlock(1, 2, 1, sequence_number[0]));
301 report_blocks.push_back(CreateReportBlock(1, 3, 255, sequence_number[1]));
302 bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, time_ms);
303 EXPECT_EQ(bitrate_observer.last_bitrate_, last_bitrate);
304 EXPECT_EQ(WeightedLoss(20, 1, 1, 255), bitrate_observer.last_fraction_loss_);
305 EXPECT_EQ(50u, bitrate_observer.last_rtt_);
306 last_bitrate = bitrate_observer.last_bitrate_;
307 sequence_number[0] += 20;
308 sequence_number[1] += 1;
309 report_blocks.clear();
310}
311
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000312TEST_F(BitrateControllerTest, TwoBitrateObserversOneRtcpObserver) {
313 TestBitrateObserver bitrate_observer_1;
314 TestBitrateObserver bitrate_observer_2;
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000315 controller_->SetBitrateObserver(&bitrate_observer_2, 200000, 200000, 300000);
stefan@webrtc.org1281dc02012-08-13 16:13:09 +0000316 controller_->SetBitrateObserver(&bitrate_observer_1, 200000, 100000, 300000);
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000317
318 // Receive a high remb, test bitrate inc.
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000319 // Test too low start bitrate, hence lower than sum of min.
andresp@webrtc.orgda077372014-03-25 12:48:42 +0000320 bandwidth_observer_->OnReceivedEstimatedBitrate(400000);
321 EXPECT_EQ(100000u, bitrate_observer_1.last_bitrate_);
322 EXPECT_EQ(0, bitrate_observer_1.last_fraction_loss_);
323 EXPECT_EQ(0u, bitrate_observer_1.last_rtt_);
324
325 // Test bitrate increase 8% per second, distributed equally.
stefan@webrtc.org28a331e2013-09-17 07:49:56 +0000326 webrtc::ReportBlockList report_blocks;
327 report_blocks.push_back(CreateReportBlock(1, 2, 0, 1));
andresp@webrtc.orgda077372014-03-25 12:48:42 +0000328 bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, 1001);
329 EXPECT_EQ(112500u, bitrate_observer_1.last_bitrate_);
kjellander@webrtc.org2e84c112012-05-31 13:55:01 +0000330 EXPECT_EQ(0, bitrate_observer_1.last_fraction_loss_);
331 EXPECT_EQ(50u, bitrate_observer_1.last_rtt_);
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000332
andresp@webrtc.orgda077372014-03-25 12:48:42 +0000333 EXPECT_EQ(212500u, bitrate_observer_2.last_bitrate_);
kjellander@webrtc.org2e84c112012-05-31 13:55:01 +0000334 EXPECT_EQ(0, bitrate_observer_2.last_fraction_loss_);
335 EXPECT_EQ(50u, bitrate_observer_2.last_rtt_);
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000336
stefan@webrtc.org28a331e2013-09-17 07:49:56 +0000337 report_blocks.clear();
andresp@webrtc.org07bc7342014-03-21 16:51:01 +0000338 report_blocks.push_back(CreateReportBlock(1, 2, 0, 41));
339 bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, 2001);
kjellander@webrtc.org2e84c112012-05-31 13:55:01 +0000340 EXPECT_EQ(126000u, bitrate_observer_1.last_bitrate_);
341 EXPECT_EQ(226000u, bitrate_observer_2.last_bitrate_);
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000342
stefan@webrtc.org28a331e2013-09-17 07:49:56 +0000343 report_blocks.clear();
andresp@webrtc.org07bc7342014-03-21 16:51:01 +0000344 report_blocks.push_back(CreateReportBlock(1, 2, 0, 61));
345 bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, 3001);
kjellander@webrtc.org2e84c112012-05-31 13:55:01 +0000346 EXPECT_EQ(140580u, bitrate_observer_1.last_bitrate_);
347 EXPECT_EQ(240580u, bitrate_observer_2.last_bitrate_);
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000348
349 // Check that the bitrate sum honor our REMB.
stefan@webrtc.org28a331e2013-09-17 07:49:56 +0000350 report_blocks.clear();
351 report_blocks.push_back(CreateReportBlock(1, 2, 0, 101));
352 bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, 5001);
kjellander@webrtc.org2e84c112012-05-31 13:55:01 +0000353 EXPECT_EQ(150000u, bitrate_observer_1.last_bitrate_);
354 EXPECT_EQ(250000u, bitrate_observer_2.last_bitrate_);
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000355
356 // Remove REMB cap, higher than sum of max.
357 bandwidth_observer_->OnReceivedEstimatedBitrate(700000);
358
stefan@webrtc.org28a331e2013-09-17 07:49:56 +0000359 report_blocks.clear();
360 report_blocks.push_back(CreateReportBlock(1, 2, 0, 121));
361 bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, 6001);
kjellander@webrtc.org2e84c112012-05-31 13:55:01 +0000362 EXPECT_EQ(166500u, bitrate_observer_1.last_bitrate_);
363 EXPECT_EQ(266500u, bitrate_observer_2.last_bitrate_);
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000364
stefan@webrtc.org28a331e2013-09-17 07:49:56 +0000365 report_blocks.clear();
366 report_blocks.push_back(CreateReportBlock(1, 2, 0, 141));
367 bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, 7001);
kjellander@webrtc.org2e84c112012-05-31 13:55:01 +0000368 EXPECT_EQ(184320u, bitrate_observer_1.last_bitrate_);
369 EXPECT_EQ(284320u, bitrate_observer_2.last_bitrate_);
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000370
stefan@webrtc.org28a331e2013-09-17 07:49:56 +0000371 report_blocks.clear();
372 report_blocks.push_back(CreateReportBlock(1, 2, 0, 161));
373 bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, 8001);
kjellander@webrtc.org2e84c112012-05-31 13:55:01 +0000374 EXPECT_EQ(207130u, bitrate_observer_1.last_bitrate_);
375 EXPECT_EQ(300000u, bitrate_observer_2.last_bitrate_); // Max cap.
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000376
stefan@webrtc.org28a331e2013-09-17 07:49:56 +0000377 report_blocks.clear();
378 report_blocks.push_back(CreateReportBlock(1, 2, 0, 181));
379 bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, 9001);
kjellander@webrtc.org2e84c112012-05-31 13:55:01 +0000380 EXPECT_EQ(248700u, bitrate_observer_1.last_bitrate_);
381 EXPECT_EQ(300000u, bitrate_observer_2.last_bitrate_);
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000382
stefan@webrtc.org28a331e2013-09-17 07:49:56 +0000383 report_blocks.clear();
384 report_blocks.push_back(CreateReportBlock(1, 2, 0, 201));
385 bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, 10001);
kjellander@webrtc.org2e84c112012-05-31 13:55:01 +0000386 EXPECT_EQ(293596u, bitrate_observer_1.last_bitrate_);
387 EXPECT_EQ(300000u, bitrate_observer_2.last_bitrate_);
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000388
stefan@webrtc.org28a331e2013-09-17 07:49:56 +0000389 report_blocks.clear();
390 report_blocks.push_back(CreateReportBlock(1, 2, 0, 221));
391 bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, 11001);
kjellander@webrtc.org2e84c112012-05-31 13:55:01 +0000392 EXPECT_EQ(300000u, bitrate_observer_1.last_bitrate_); // Max cap.
393 EXPECT_EQ(300000u, bitrate_observer_2.last_bitrate_);
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000394
395 // Test that a low REMB trigger immediately.
396 bandwidth_observer_->OnReceivedEstimatedBitrate(350000);
kjellander@webrtc.org2e84c112012-05-31 13:55:01 +0000397 EXPECT_EQ(125000u, bitrate_observer_1.last_bitrate_);
398 EXPECT_EQ(0, bitrate_observer_1.last_fraction_loss_);
399 EXPECT_EQ(50u, bitrate_observer_1.last_rtt_);
400 EXPECT_EQ(225000u, bitrate_observer_2.last_bitrate_);
401 EXPECT_EQ(0, bitrate_observer_2.last_fraction_loss_);
402 EXPECT_EQ(50u, bitrate_observer_2.last_rtt_);
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000403
404 bandwidth_observer_->OnReceivedEstimatedBitrate(1000);
kjellander@webrtc.org2e84c112012-05-31 13:55:01 +0000405 EXPECT_EQ(100000u, bitrate_observer_1.last_bitrate_); // Min cap.
406 EXPECT_EQ(200000u, bitrate_observer_2.last_bitrate_); // Min cap.
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000407 controller_->RemoveBitrateObserver(&bitrate_observer_1);
408 controller_->RemoveBitrateObserver(&bitrate_observer_2);
409}
henrik.lundin@webrtc.org29dd0de2013-10-21 14:00:01 +0000410
411class BitrateControllerTestNoEnforceMin : public BitrateControllerTest {
412 protected:
413 BitrateControllerTestNoEnforceMin() : BitrateControllerTest() {
414 enforce_min_bitrate_ = false;
415 }
416};
417
418// The following three tests verify that the EnforceMinBitrate() method works
419// as intended.
420TEST_F(BitrateControllerTestNoEnforceMin, OneBitrateObserver) {
421 TestBitrateObserver bitrate_observer_1;
422 controller_->SetBitrateObserver(&bitrate_observer_1, 200000, 100000, 400000);
423
424 // High REMB.
425 bandwidth_observer_->OnReceivedEstimatedBitrate(150000);
426 EXPECT_EQ(150000u, bitrate_observer_1.last_bitrate_);
427
428 // Low REMB.
andresp@webrtc.org07bc7342014-03-21 16:51:01 +0000429 bandwidth_observer_->OnReceivedEstimatedBitrate(10000);
430 EXPECT_EQ(10000u, bitrate_observer_1.last_bitrate_);
431
432 // Keeps at least 10 kbps.
433 bandwidth_observer_->OnReceivedEstimatedBitrate(9000);
434 EXPECT_EQ(10000u, bitrate_observer_1.last_bitrate_);
henrik.lundin@webrtc.org29dd0de2013-10-21 14:00:01 +0000435
436 controller_->RemoveBitrateObserver(&bitrate_observer_1);
437}
438
439TEST_F(BitrateControllerTestNoEnforceMin, ThreeBitrateObservers) {
440 TestBitrateObserver bitrate_observer_1;
441 TestBitrateObserver bitrate_observer_2;
442 TestBitrateObserver bitrate_observer_3;
443 // Set up the observers with min bitrates at 100000, 200000, and 300000.
444 // Note: The start bitrate of bitrate_observer_1 (700000) is used as the
445 // overall start bitrate.
446 controller_->SetBitrateObserver(&bitrate_observer_1, 700000, 100000, 400000);
447 controller_->SetBitrateObserver(&bitrate_observer_2, 200000, 200000, 400000);
448 controller_->SetBitrateObserver(&bitrate_observer_3, 200000, 300000, 400000);
449
450 // High REMB. Make sure the controllers get a fair share of the surplus
451 // (i.e., what is left after each controller gets its min rate).
452 bandwidth_observer_->OnReceivedEstimatedBitrate(690000);
453 // Verify that each observer gets its min rate (sum of min rates is 600000),
454 // and that the remaining 90000 is divided equally among the three.
455 EXPECT_EQ(130000u, bitrate_observer_1.last_bitrate_);
456 EXPECT_EQ(230000u, bitrate_observer_2.last_bitrate_);
457 EXPECT_EQ(330000u, bitrate_observer_3.last_bitrate_);
458
459 // High REMB, but below the sum of min bitrates.
460 bandwidth_observer_->OnReceivedEstimatedBitrate(500000);
461 // Verify that the first and second observers get their min bitrates, and the
462 // third gets the remainder.
463 EXPECT_EQ(100000u, bitrate_observer_1.last_bitrate_); // Min bitrate.
464 EXPECT_EQ(200000u, bitrate_observer_2.last_bitrate_); // Min bitrate.
465 EXPECT_EQ(200000u, bitrate_observer_3.last_bitrate_); // Remainder.
466
467 // Low REMB.
andresp@webrtc.org07bc7342014-03-21 16:51:01 +0000468 bandwidth_observer_->OnReceivedEstimatedBitrate(10000);
henrik.lundin@webrtc.org29dd0de2013-10-21 14:00:01 +0000469 // Verify that the first observer gets all the rate, and the rest get zero.
andresp@webrtc.org07bc7342014-03-21 16:51:01 +0000470 EXPECT_EQ(10000u, bitrate_observer_1.last_bitrate_);
471 EXPECT_EQ(0u, bitrate_observer_2.last_bitrate_);
472 EXPECT_EQ(0u, bitrate_observer_3.last_bitrate_);
473
474 // Verify it keeps an estimate of at least 10kbps.
475 bandwidth_observer_->OnReceivedEstimatedBitrate(9000);
476 EXPECT_EQ(10000u, bitrate_observer_1.last_bitrate_);
henrik.lundin@webrtc.org29dd0de2013-10-21 14:00:01 +0000477 EXPECT_EQ(0u, bitrate_observer_2.last_bitrate_);
478 EXPECT_EQ(0u, bitrate_observer_3.last_bitrate_);
479
480 controller_->RemoveBitrateObserver(&bitrate_observer_1);
481 controller_->RemoveBitrateObserver(&bitrate_observer_2);
482 controller_->RemoveBitrateObserver(&bitrate_observer_3);
483}
484
485TEST_F(BitrateControllerTest, ThreeBitrateObserversLowRembEnforceMin) {
486 TestBitrateObserver bitrate_observer_1;
487 TestBitrateObserver bitrate_observer_2;
488 TestBitrateObserver bitrate_observer_3;
489 controller_->SetBitrateObserver(&bitrate_observer_1, 200000, 100000, 300000);
490 controller_->SetBitrateObserver(&bitrate_observer_2, 200000, 200000, 300000);
491 controller_->SetBitrateObserver(&bitrate_observer_3, 200000, 300000, 300000);
492
493 // Low REMB. Verify that all observers still get their respective min bitrate.
494 bandwidth_observer_->OnReceivedEstimatedBitrate(1000);
495 EXPECT_EQ(100000u, bitrate_observer_1.last_bitrate_); // Min cap.
496 EXPECT_EQ(200000u, bitrate_observer_2.last_bitrate_); // Min cap.
497 EXPECT_EQ(300000u, bitrate_observer_3.last_bitrate_); // Min cap.
498
499 controller_->RemoveBitrateObserver(&bitrate_observer_1);
500 controller_->RemoveBitrateObserver(&bitrate_observer_2);
501 controller_->RemoveBitrateObserver(&bitrate_observer_3);
502}