blob: ab9928fc840cde0711c6a6f15067041a074d64da [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
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +000011#include <algorithm>
12#include <vector>
13
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020014#include "logging/rtc_event_log/mock/mock_rtc_event_log.h"
15#include "modules/bitrate_controller/include/bitrate_controller.h"
16#include "modules/pacing/mock/mock_paced_sender.h"
17#include "modules/remote_bitrate_estimator/include/bwe_defines.h"
18#include "test/field_trial.h"
19#include "test/gtest.h"
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +000020
perkjec81bcd2016-05-11 06:01:13 -070021using ::testing::Exactly;
22using ::testing::Return;
23
perkje30c2722016-05-09 04:57:11 -070024using webrtc::BitrateController;
perkjec81bcd2016-05-11 06:01:13 -070025using webrtc::BitrateObserver;
26using webrtc::PacedSender;
27using webrtc::RtcpBandwidthObserver;
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +000028
stefan@webrtc.org28a331e2013-09-17 07:49:56 +000029uint8_t WeightedLoss(int num_packets1, uint8_t fraction_loss1,
30 int num_packets2, uint8_t fraction_loss2) {
31 int weighted_sum = num_packets1 * fraction_loss1 +
32 num_packets2 * fraction_loss2;
33 int total_num_packets = num_packets1 + num_packets2;
34 return (weighted_sum + total_num_packets / 2) / total_num_packets;
35}
36
37webrtc::RTCPReportBlock CreateReportBlock(
38 uint32_t remote_ssrc, uint32_t source_ssrc,
39 uint8_t fraction_lost, uint32_t extended_high_sequence_number) {
40 return webrtc::RTCPReportBlock(remote_ssrc, source_ssrc, fraction_lost, 0,
41 extended_high_sequence_number, 0, 0, 0);
42}
43
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +000044class TestBitrateObserver: public BitrateObserver {
45 public:
46 TestBitrateObserver()
kjellander@webrtc.org2e84c112012-05-31 13:55:01 +000047 : last_bitrate_(0),
48 last_fraction_loss_(0),
49 last_rtt_(0) {
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +000050 }
51
stefan@webrtc.orgedeea912014-12-08 19:46:23 +000052 virtual void OnNetworkChanged(uint32_t bitrate,
53 uint8_t fraction_loss,
pkasting@chromium.org16825b12015-01-12 21:51:21 +000054 int64_t rtt) {
Stefan Holmere5904162015-03-26 11:11:06 +010055 last_bitrate_ = static_cast<int>(bitrate);
kjellander@webrtc.org2e84c112012-05-31 13:55:01 +000056 last_fraction_loss_ = fraction_loss;
57 last_rtt_ = rtt;
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +000058 }
Stefan Holmere5904162015-03-26 11:11:06 +010059 int last_bitrate_;
kjellander@webrtc.org2e84c112012-05-31 13:55:01 +000060 uint8_t last_fraction_loss_;
pkasting@chromium.org16825b12015-01-12 21:51:21 +000061 int64_t last_rtt_;
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +000062};
63
64class BitrateControllerTest : public ::testing::Test {
65 protected:
stefan@webrtc.org792f1a12015-03-04 12:24:26 +000066 BitrateControllerTest() : clock_(0) {}
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +000067 ~BitrateControllerTest() {}
68
69 virtual void SetUp() {
Stefan Holmer52200d02016-09-20 14:14:23 +020070 controller_.reset(BitrateController::CreateBitrateController(
71 &clock_, &bitrate_observer_, &event_log_));
Stefan Holmere5904162015-03-26 11:11:06 +010072 controller_->SetStartBitrate(kStartBitrateBps);
73 EXPECT_EQ(kStartBitrateBps, bitrate_observer_.last_bitrate_);
74 controller_->SetMinMaxBitrate(kMinBitrateBps, kMaxBitrateBps);
75 EXPECT_EQ(kStartBitrateBps, bitrate_observer_.last_bitrate_);
Stefan Holmer52200d02016-09-20 14:14:23 +020076 bandwidth_observer_.reset(controller_->CreateRtcpBandwidthObserver());
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +000077 }
78
79 virtual void TearDown() {
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +000080 }
andresp@webrtc.org44caf012014-03-26 21:00:21 +000081
Stefan Holmere5904162015-03-26 11:11:06 +010082 const int kMinBitrateBps = 100000;
83 const int kStartBitrateBps = 200000;
84 const int kMaxBitrateBps = 300000;
85
86 const int kDefaultMinBitrateBps = 10000;
87 const int kDefaultMaxBitrateBps = 1000000000;
88
andresp@webrtc.org44caf012014-03-26 21:00:21 +000089 webrtc::SimulatedClock clock_;
stefan@webrtc.org792f1a12015-03-04 12:24:26 +000090 TestBitrateObserver bitrate_observer_;
Stefan Holmer52200d02016-09-20 14:14:23 +020091 std::unique_ptr<BitrateController> controller_;
92 std::unique_ptr<RtcpBandwidthObserver> bandwidth_observer_;
Ivo Creusenfa1d5682016-07-06 09:12:06 +020093 testing::NiceMock<webrtc::MockRtcEventLog> event_log_;
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +000094};
95
Stefan Holmere5904162015-03-26 11:11:06 +010096TEST_F(BitrateControllerTest, DefaultMinMaxBitrate) {
97 // Receive successively lower REMBs, verify the reserved bitrate is deducted.
98 controller_->SetMinMaxBitrate(0, 0);
99 EXPECT_EQ(kStartBitrateBps, bitrate_observer_.last_bitrate_);
100 bandwidth_observer_->OnReceivedEstimatedBitrate(kDefaultMinBitrateBps / 2);
michaeltf082c2a2016-11-07 04:17:14 -0800101 EXPECT_EQ(webrtc::congestion_controller::GetMinBitrateBps(),
102 bitrate_observer_.last_bitrate_);
Stefan Holmere5904162015-03-26 11:11:06 +0100103 bandwidth_observer_->OnReceivedEstimatedBitrate(2 * kDefaultMaxBitrateBps);
104 clock_.AdvanceTimeMilliseconds(1000);
105 controller_->Process();
106 EXPECT_EQ(kDefaultMaxBitrateBps, bitrate_observer_.last_bitrate_);
107}
108
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000109TEST_F(BitrateControllerTest, OneBitrateObserverOneRtcpObserver) {
stefan@webrtc.org82462aa2014-10-23 11:57:05 +0000110 // First REMB applies immediately.
111 int64_t time_ms = 1001;
112 webrtc::ReportBlockList report_blocks;
113 report_blocks.push_back(CreateReportBlock(1, 2, 0, 1));
114 bandwidth_observer_->OnReceivedEstimatedBitrate(200000);
Stefan Holmere5904162015-03-26 11:11:06 +0100115 EXPECT_EQ(200000, bitrate_observer_.last_bitrate_);
stefan@webrtc.org792f1a12015-03-04 12:24:26 +0000116 EXPECT_EQ(0, bitrate_observer_.last_fraction_loss_);
117 EXPECT_EQ(0, bitrate_observer_.last_rtt_);
stefan@webrtc.org82462aa2014-10-23 11:57:05 +0000118 bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, time_ms);
119 report_blocks.clear();
120 time_ms += 2000;
121
122 // Receive a high remb, test bitrate inc.
123 bandwidth_observer_->OnReceivedEstimatedBitrate(400000);
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000124
andresp@webrtc.org44caf012014-03-26 21:00:21 +0000125 // Test bitrate increase 8% per second.
stefan@webrtc.org82462aa2014-10-23 11:57:05 +0000126 report_blocks.push_back(CreateReportBlock(1, 2, 0, 21));
127 bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, time_ms);
Stefan Holmere5904162015-03-26 11:11:06 +0100128 EXPECT_EQ(217000, bitrate_observer_.last_bitrate_);
stefan@webrtc.org792f1a12015-03-04 12:24:26 +0000129 EXPECT_EQ(0, bitrate_observer_.last_fraction_loss_);
130 EXPECT_EQ(50, bitrate_observer_.last_rtt_);
stefan@webrtc.org82462aa2014-10-23 11:57:05 +0000131 time_ms += 1000;
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000132
stefan@webrtc.org28a331e2013-09-17 07:49:56 +0000133 report_blocks.clear();
134 report_blocks.push_back(CreateReportBlock(1, 2, 0, 41));
stefan@webrtc.org82462aa2014-10-23 11:57:05 +0000135 bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, time_ms);
Stefan Holmere5904162015-03-26 11:11:06 +0100136 EXPECT_EQ(235360, bitrate_observer_.last_bitrate_);
stefan@webrtc.org792f1a12015-03-04 12:24:26 +0000137 EXPECT_EQ(0, bitrate_observer_.last_fraction_loss_);
138 EXPECT_EQ(50, bitrate_observer_.last_rtt_);
stefan@webrtc.org82462aa2014-10-23 11:57:05 +0000139 time_ms += 1000;
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000140
stefan@webrtc.org28a331e2013-09-17 07:49:56 +0000141 report_blocks.clear();
142 report_blocks.push_back(CreateReportBlock(1, 2, 0, 61));
stefan@webrtc.org82462aa2014-10-23 11:57:05 +0000143 bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, time_ms);
Stefan Holmere5904162015-03-26 11:11:06 +0100144 EXPECT_EQ(255189, bitrate_observer_.last_bitrate_);
stefan@webrtc.org82462aa2014-10-23 11:57:05 +0000145 time_ms += 1000;
146
147 report_blocks.clear();
148 report_blocks.push_back(CreateReportBlock(1, 2, 0, 81));
149 bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, time_ms);
Stefan Holmere5904162015-03-26 11:11:06 +0100150 EXPECT_EQ(276604, bitrate_observer_.last_bitrate_);
stefan@webrtc.org82462aa2014-10-23 11:57:05 +0000151 time_ms += 1000;
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000152
stefan@webrtc.org28a331e2013-09-17 07:49:56 +0000153 report_blocks.clear();
stefand48717b2016-08-22 08:50:31 -0700154 report_blocks.push_back(CreateReportBlock(1, 2, 0, 101));
stefan@webrtc.org82462aa2014-10-23 11:57:05 +0000155 bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, time_ms);
Stefan Holmere5904162015-03-26 11:11:06 +0100156 EXPECT_EQ(299732, bitrate_observer_.last_bitrate_);
stefan@webrtc.org82462aa2014-10-23 11:57:05 +0000157 time_ms += 1000;
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000158
andresp@webrtc.org44caf012014-03-26 21:00:21 +0000159 // Reach max cap.
stefan@webrtc.org28a331e2013-09-17 07:49:56 +0000160 report_blocks.clear();
stefand48717b2016-08-22 08:50:31 -0700161 report_blocks.push_back(CreateReportBlock(1, 2, 0, 121));
stefan@webrtc.org82462aa2014-10-23 11:57:05 +0000162 bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, time_ms);
Stefan Holmere5904162015-03-26 11:11:06 +0100163 EXPECT_EQ(300000, bitrate_observer_.last_bitrate_);
stefan@webrtc.org82462aa2014-10-23 11:57:05 +0000164 time_ms += 1000;
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000165
stefan@webrtc.org28a331e2013-09-17 07:49:56 +0000166 report_blocks.clear();
167 report_blocks.push_back(CreateReportBlock(1, 2, 0, 141));
stefan@webrtc.org82462aa2014-10-23 11:57:05 +0000168 bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, time_ms);
Stefan Holmere5904162015-03-26 11:11:06 +0100169 EXPECT_EQ(300000, bitrate_observer_.last_bitrate_);
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000170
stefan32f81542016-01-20 07:13:58 -0800171 // Test that a low delay-based estimate limits the combined estimate.
Stefan Holmer280de9e2016-09-30 10:06:51 +0200172 webrtc::DelayBasedBwe::Result result(false, 280000);
173 controller_->OnDelayBasedBweResult(result);
stefan32f81542016-01-20 07:13:58 -0800174 EXPECT_EQ(280000, bitrate_observer_.last_bitrate_);
175
176 // Test that a low REMB limits the combined estimate.
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000177 bandwidth_observer_->OnReceivedEstimatedBitrate(250000);
Stefan Holmere5904162015-03-26 11:11:06 +0100178 EXPECT_EQ(250000, bitrate_observer_.last_bitrate_);
stefan@webrtc.org792f1a12015-03-04 12:24:26 +0000179 EXPECT_EQ(0, bitrate_observer_.last_fraction_loss_);
180 EXPECT_EQ(50, bitrate_observer_.last_rtt_);
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000181
182 bandwidth_observer_->OnReceivedEstimatedBitrate(1000);
stefan32f81542016-01-20 07:13:58 -0800183 EXPECT_EQ(100000, bitrate_observer_.last_bitrate_);
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000184}
185
186TEST_F(BitrateControllerTest, OneBitrateObserverTwoRtcpObservers) {
Danil Chapovalov7f8369a2017-06-08 17:22:51 +0200187 const uint32_t kSenderSsrc1 = 1;
188 const uint32_t kSenderSsrc2 = 2;
189 const uint32_t kMediaSsrc1 = 3;
190 const uint32_t kMediaSsrc2 = 4;
191
stefan@webrtc.org82462aa2014-10-23 11:57:05 +0000192 int64_t time_ms = 1;
193 webrtc::ReportBlockList report_blocks;
Danil Chapovalov7f8369a2017-06-08 17:22:51 +0200194 report_blocks = {CreateReportBlock(kSenderSsrc1, kMediaSsrc1, 0, 1)};
stefan@webrtc.org82462aa2014-10-23 11:57:05 +0000195 bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, time_ms);
stefan@webrtc.org82462aa2014-10-23 11:57:05 +0000196 time_ms += 500;
197
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000198 RtcpBandwidthObserver* second_bandwidth_observer =
199 controller_->CreateRtcpBandwidthObserver();
Danil Chapovalov7f8369a2017-06-08 17:22:51 +0200200 report_blocks = {CreateReportBlock(kSenderSsrc2, kMediaSsrc2, 0, 21)};
201 second_bandwidth_observer->OnReceivedRtcpReceiverReport(
202 report_blocks, 100, time_ms);
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000203
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000204 // Test start bitrate.
Stefan Holmer52200d02016-09-20 14:14:23 +0200205 EXPECT_EQ(200000, bitrate_observer_.last_bitrate_);
stefan@webrtc.org792f1a12015-03-04 12:24:26 +0000206 EXPECT_EQ(0, bitrate_observer_.last_fraction_loss_);
207 EXPECT_EQ(100, bitrate_observer_.last_rtt_);
stefan@webrtc.org82462aa2014-10-23 11:57:05 +0000208 time_ms += 500;
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000209
210 // Test bitrate increase 8% per second.
Danil Chapovalov7f8369a2017-06-08 17:22:51 +0200211 report_blocks = {CreateReportBlock(kSenderSsrc1, kMediaSsrc1, 0, 21)};
stefan@webrtc.org82462aa2014-10-23 11:57:05 +0000212 bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, time_ms);
213 time_ms += 500;
Danil Chapovalov7f8369a2017-06-08 17:22:51 +0200214
215 report_blocks = {CreateReportBlock(kSenderSsrc2, kMediaSsrc2, 0, 21)};
stefan@webrtc.org82462aa2014-10-23 11:57:05 +0000216 second_bandwidth_observer->OnReceivedRtcpReceiverReport(
217 report_blocks, 100, time_ms);
Stefan Holmer52200d02016-09-20 14:14:23 +0200218 EXPECT_EQ(217000, bitrate_observer_.last_bitrate_);
stefan@webrtc.org792f1a12015-03-04 12:24:26 +0000219 EXPECT_EQ(0, bitrate_observer_.last_fraction_loss_);
220 EXPECT_EQ(100, bitrate_observer_.last_rtt_);
stefan@webrtc.org82462aa2014-10-23 11:57:05 +0000221 time_ms += 500;
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000222
223 // Extra report should not change estimate.
Danil Chapovalov7f8369a2017-06-08 17:22:51 +0200224 report_blocks = {CreateReportBlock(kSenderSsrc2, kMediaSsrc2, 0, 31)};
stefan@webrtc.org82462aa2014-10-23 11:57:05 +0000225 second_bandwidth_observer->OnReceivedRtcpReceiverReport(
226 report_blocks, 100, time_ms);
Stefan Holmer52200d02016-09-20 14:14:23 +0200227 EXPECT_EQ(217000, bitrate_observer_.last_bitrate_);
stefan@webrtc.org82462aa2014-10-23 11:57:05 +0000228 time_ms += 500;
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000229
Danil Chapovalov7f8369a2017-06-08 17:22:51 +0200230 report_blocks = {CreateReportBlock(kSenderSsrc1, kMediaSsrc1, 0, 41)};
stefan@webrtc.org82462aa2014-10-23 11:57:05 +0000231 bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, time_ms);
Stefan Holmer52200d02016-09-20 14:14:23 +0200232 EXPECT_EQ(235360, bitrate_observer_.last_bitrate_);
pwestin@webrtc.orga2cd7322012-04-23 08:32:47 +0000233
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000234 // Second report should not change estimate.
Danil Chapovalov7f8369a2017-06-08 17:22:51 +0200235 report_blocks = {CreateReportBlock(kSenderSsrc2, kMediaSsrc2, 0, 41)};
stefan@webrtc.org82462aa2014-10-23 11:57:05 +0000236 second_bandwidth_observer->OnReceivedRtcpReceiverReport(
237 report_blocks, 100, time_ms);
Stefan Holmer52200d02016-09-20 14:14:23 +0200238 EXPECT_EQ(235360, bitrate_observer_.last_bitrate_);
stefan@webrtc.org82462aa2014-10-23 11:57:05 +0000239 time_ms += 1000;
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000240
241 // Reports from only one bandwidth observer is ok.
Danil Chapovalov7f8369a2017-06-08 17:22:51 +0200242 report_blocks = {CreateReportBlock(kSenderSsrc2, kMediaSsrc2, 0, 61)};
Stefan Holmer52200d02016-09-20 14:14:23 +0200243 second_bandwidth_observer->OnReceivedRtcpReceiverReport(
244 report_blocks, 50, time_ms);
245 EXPECT_EQ(255189, bitrate_observer_.last_bitrate_);
246 time_ms += 1000;
247
Danil Chapovalov7f8369a2017-06-08 17:22:51 +0200248 report_blocks = {CreateReportBlock(kSenderSsrc2, kMediaSsrc2, 0, 81)};
stefan@webrtc.org82462aa2014-10-23 11:57:05 +0000249 second_bandwidth_observer->OnReceivedRtcpReceiverReport(
250 report_blocks, 50, time_ms);
Stefan Holmere5904162015-03-26 11:11:06 +0100251 EXPECT_EQ(276604, bitrate_observer_.last_bitrate_);
stefan@webrtc.org82462aa2014-10-23 11:57:05 +0000252 time_ms += 1000;
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000253
Danil Chapovalov7f8369a2017-06-08 17:22:51 +0200254 report_blocks = {CreateReportBlock(kSenderSsrc2, kMediaSsrc2, 0, 121)};
stefan@webrtc.org82462aa2014-10-23 11:57:05 +0000255 second_bandwidth_observer->OnReceivedRtcpReceiverReport(
256 report_blocks, 50, time_ms);
Stefan Holmere5904162015-03-26 11:11:06 +0100257 EXPECT_EQ(299732, bitrate_observer_.last_bitrate_);
stefan@webrtc.org82462aa2014-10-23 11:57:05 +0000258 time_ms += 1000;
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000259
andresp@webrtc.org44caf012014-03-26 21:00:21 +0000260 // Reach max cap.
Danil Chapovalov7f8369a2017-06-08 17:22:51 +0200261 report_blocks = {CreateReportBlock(kSenderSsrc2, kMediaSsrc2, 0, 141)};
andresp@webrtc.org44caf012014-03-26 21:00:21 +0000262 second_bandwidth_observer->OnReceivedRtcpReceiverReport(
stefan@webrtc.org82462aa2014-10-23 11:57:05 +0000263 report_blocks, 50, time_ms);
Stefan Holmere5904162015-03-26 11:11:06 +0100264 EXPECT_EQ(300000, bitrate_observer_.last_bitrate_);
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000265
266 // Test that a low REMB trigger immediately.
267 // We don't care which bandwidth observer that delivers the REMB.
268 second_bandwidth_observer->OnReceivedEstimatedBitrate(250000);
Stefan Holmere5904162015-03-26 11:11:06 +0100269 EXPECT_EQ(250000, bitrate_observer_.last_bitrate_);
stefan@webrtc.org792f1a12015-03-04 12:24:26 +0000270 EXPECT_EQ(0, bitrate_observer_.last_fraction_loss_);
271 EXPECT_EQ(50, bitrate_observer_.last_rtt_);
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000272
andresp@webrtc.org44caf012014-03-26 21:00:21 +0000273 // Min cap.
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000274 bandwidth_observer_->OnReceivedEstimatedBitrate(1000);
Stefan Holmere5904162015-03-26 11:11:06 +0100275 EXPECT_EQ(100000, bitrate_observer_.last_bitrate_);
pwestin@webrtc.org1cd11622012-04-19 12:13:52 +0000276 delete second_bandwidth_observer;
277}
278
stefan@webrtc.org28a331e2013-09-17 07:49:56 +0000279TEST_F(BitrateControllerTest, OneBitrateObserverMultipleReportBlocks) {
stefan@webrtc.org28a331e2013-09-17 07:49:56 +0000280 uint32_t sequence_number[2] = {0, 0xFF00};
Stefan Holmere5904162015-03-26 11:11:06 +0100281 const int kStartBitrate = 200000;
282 const int kMinBitrate = 100000;
283 const int kMaxBitrate = 300000;
stefan@webrtc.org792f1a12015-03-04 12:24:26 +0000284 controller_->SetStartBitrate(kStartBitrate);
285 controller_->SetMinMaxBitrate(kMinBitrate, kMaxBitrate);
stefan@webrtc.org28a331e2013-09-17 07:49:56 +0000286
stefan@webrtc.org82462aa2014-10-23 11:57:05 +0000287 // REMBs during the first 2 seconds apply immediately.
288 int64_t time_ms = 1001;
289 webrtc::ReportBlockList report_blocks;
290 report_blocks.push_back(CreateReportBlock(1, 2, 0, sequence_number[0]));
291 bandwidth_observer_->OnReceivedEstimatedBitrate(kStartBitrate);
292 bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, time_ms);
293 report_blocks.clear();
294 time_ms += 2000;
295
stefan@webrtc.org28a331e2013-09-17 07:49:56 +0000296 // Receive a high REMB, test bitrate increase.
297 bandwidth_observer_->OnReceivedEstimatedBitrate(400000);
298
Stefan Holmere5904162015-03-26 11:11:06 +0100299 int last_bitrate = 0;
stefan@webrtc.org28a331e2013-09-17 07:49:56 +0000300 // Ramp up to max bitrate.
Stefan Holmer52200d02016-09-20 14:14:23 +0200301 for (int i = 0; i < 7; ++i) {
stefan@webrtc.org28a331e2013-09-17 07:49:56 +0000302 report_blocks.push_back(CreateReportBlock(1, 2, 0, sequence_number[0]));
303 report_blocks.push_back(CreateReportBlock(1, 3, 0, sequence_number[1]));
304 bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50,
305 time_ms);
stefan@webrtc.org792f1a12015-03-04 12:24:26 +0000306 EXPECT_GT(bitrate_observer_.last_bitrate_, last_bitrate);
307 EXPECT_EQ(0, bitrate_observer_.last_fraction_loss_);
308 EXPECT_EQ(50, bitrate_observer_.last_rtt_);
309 last_bitrate = bitrate_observer_.last_bitrate_;
stefan@webrtc.org28a331e2013-09-17 07:49:56 +0000310 time_ms += 1000;
311 sequence_number[0] += 20;
312 sequence_number[1] += 1;
313 report_blocks.clear();
314 }
315
stefan@webrtc.org792f1a12015-03-04 12:24:26 +0000316 EXPECT_EQ(kMaxBitrate, bitrate_observer_.last_bitrate_);
stefan@webrtc.org28a331e2013-09-17 07:49:56 +0000317
318 // Packet loss on the first stream. Verify that bitrate decreases.
319 report_blocks.push_back(CreateReportBlock(1, 2, 50, sequence_number[0]));
320 report_blocks.push_back(CreateReportBlock(1, 3, 0, sequence_number[1]));
321 bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, time_ms);
stefan@webrtc.org792f1a12015-03-04 12:24:26 +0000322 EXPECT_LT(bitrate_observer_.last_bitrate_, last_bitrate);
323 EXPECT_EQ(WeightedLoss(20, 50, 1, 0), bitrate_observer_.last_fraction_loss_);
324 EXPECT_EQ(50, bitrate_observer_.last_rtt_);
325 last_bitrate = bitrate_observer_.last_bitrate_;
stefan@webrtc.org28a331e2013-09-17 07:49:56 +0000326 sequence_number[0] += 20;
327 sequence_number[1] += 20;
328 time_ms += 1000;
329 report_blocks.clear();
330
331 // Packet loss on the second stream. Verify that bitrate decreases.
332 report_blocks.push_back(CreateReportBlock(1, 2, 0, sequence_number[0]));
333 report_blocks.push_back(CreateReportBlock(1, 3, 75, sequence_number[1]));
334 bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, time_ms);
stefan@webrtc.org792f1a12015-03-04 12:24:26 +0000335 EXPECT_LT(bitrate_observer_.last_bitrate_, last_bitrate);
336 EXPECT_EQ(WeightedLoss(20, 0, 20, 75), bitrate_observer_.last_fraction_loss_);
337 EXPECT_EQ(50, bitrate_observer_.last_rtt_);
338 last_bitrate = bitrate_observer_.last_bitrate_;
stefan@webrtc.org28a331e2013-09-17 07:49:56 +0000339 sequence_number[0] += 20;
340 sequence_number[1] += 1;
341 time_ms += 1000;
342 report_blocks.clear();
343
344 // All packets lost on stream with few packets, no back-off.
345 report_blocks.push_back(CreateReportBlock(1, 2, 1, sequence_number[0]));
346 report_blocks.push_back(CreateReportBlock(1, 3, 255, sequence_number[1]));
347 bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, time_ms);
stefan@webrtc.org792f1a12015-03-04 12:24:26 +0000348 EXPECT_EQ(bitrate_observer_.last_bitrate_, last_bitrate);
349 EXPECT_EQ(WeightedLoss(20, 1, 1, 255), bitrate_observer_.last_fraction_loss_);
350 EXPECT_EQ(50, bitrate_observer_.last_rtt_);
351 last_bitrate = bitrate_observer_.last_bitrate_;
stefan@webrtc.org28a331e2013-09-17 07:49:56 +0000352 sequence_number[0] += 20;
353 sequence_number[1] += 1;
354 report_blocks.clear();
355}
356
solenberg@webrtc.org4e656022014-03-26 14:32:47 +0000357TEST_F(BitrateControllerTest, SetReservedBitrate) {
solenberg@webrtc.org4e656022014-03-26 14:32:47 +0000358 // Receive successively lower REMBs, verify the reserved bitrate is deducted.
solenberg@webrtc.org4e656022014-03-26 14:32:47 +0000359 controller_->SetReservedBitrate(0);
360 bandwidth_observer_->OnReceivedEstimatedBitrate(400000);
Stefan Holmere5904162015-03-26 11:11:06 +0100361 EXPECT_EQ(200000, bitrate_observer_.last_bitrate_);
solenberg@webrtc.org4e656022014-03-26 14:32:47 +0000362 controller_->SetReservedBitrate(50000);
363 bandwidth_observer_->OnReceivedEstimatedBitrate(400000);
Stefan Holmere5904162015-03-26 11:11:06 +0100364 EXPECT_EQ(150000, bitrate_observer_.last_bitrate_);
solenberg@webrtc.org4e656022014-03-26 14:32:47 +0000365
366 controller_->SetReservedBitrate(0);
367 bandwidth_observer_->OnReceivedEstimatedBitrate(250000);
Stefan Holmere5904162015-03-26 11:11:06 +0100368 EXPECT_EQ(200000, bitrate_observer_.last_bitrate_);
solenberg@webrtc.org4e656022014-03-26 14:32:47 +0000369 controller_->SetReservedBitrate(50000);
370 bandwidth_observer_->OnReceivedEstimatedBitrate(250000);
Stefan Holmere5904162015-03-26 11:11:06 +0100371 EXPECT_EQ(150000, bitrate_observer_.last_bitrate_);
solenberg@webrtc.org4e656022014-03-26 14:32:47 +0000372
373 controller_->SetReservedBitrate(0);
374 bandwidth_observer_->OnReceivedEstimatedBitrate(200000);
Stefan Holmere5904162015-03-26 11:11:06 +0100375 EXPECT_EQ(200000, bitrate_observer_.last_bitrate_);
solenberg@webrtc.org4e656022014-03-26 14:32:47 +0000376 controller_->SetReservedBitrate(30000);
377 bandwidth_observer_->OnReceivedEstimatedBitrate(200000);
Stefan Holmere5904162015-03-26 11:11:06 +0100378 EXPECT_EQ(170000, bitrate_observer_.last_bitrate_);
solenberg@webrtc.org4e656022014-03-26 14:32:47 +0000379
380 controller_->SetReservedBitrate(0);
381 bandwidth_observer_->OnReceivedEstimatedBitrate(160000);
Stefan Holmere5904162015-03-26 11:11:06 +0100382 EXPECT_EQ(160000, bitrate_observer_.last_bitrate_);
solenberg@webrtc.org4e656022014-03-26 14:32:47 +0000383 controller_->SetReservedBitrate(30000);
384 bandwidth_observer_->OnReceivedEstimatedBitrate(160000);
Stefan Holmere5904162015-03-26 11:11:06 +0100385 EXPECT_EQ(130000, bitrate_observer_.last_bitrate_);
solenberg@webrtc.org4e656022014-03-26 14:32:47 +0000386
387 controller_->SetReservedBitrate(0);
388 bandwidth_observer_->OnReceivedEstimatedBitrate(120000);
Stefan Holmere5904162015-03-26 11:11:06 +0100389 EXPECT_EQ(120000, bitrate_observer_.last_bitrate_);
solenberg@webrtc.org4e656022014-03-26 14:32:47 +0000390 controller_->SetReservedBitrate(10000);
391 bandwidth_observer_->OnReceivedEstimatedBitrate(120000);
Stefan Holmere5904162015-03-26 11:11:06 +0100392 EXPECT_EQ(110000, bitrate_observer_.last_bitrate_);
solenberg@webrtc.org4e656022014-03-26 14:32:47 +0000393
394 controller_->SetReservedBitrate(0);
395 bandwidth_observer_->OnReceivedEstimatedBitrate(120000);
Stefan Holmere5904162015-03-26 11:11:06 +0100396 EXPECT_EQ(120000, bitrate_observer_.last_bitrate_);
solenberg@webrtc.org4e656022014-03-26 14:32:47 +0000397 controller_->SetReservedBitrate(50000);
398 bandwidth_observer_->OnReceivedEstimatedBitrate(120000);
stefan@webrtc.org792f1a12015-03-04 12:24:26 +0000399 // Limited by min bitrate.
Stefan Holmere5904162015-03-26 11:11:06 +0100400 EXPECT_EQ(100000, bitrate_observer_.last_bitrate_);
solenberg@webrtc.org4e656022014-03-26 14:32:47 +0000401
402 controller_->SetReservedBitrate(10000);
stefan@webrtc.org792f1a12015-03-04 12:24:26 +0000403 bandwidth_observer_->OnReceivedEstimatedBitrate(1);
Stefan Holmere5904162015-03-26 11:11:06 +0100404 EXPECT_EQ(100000, bitrate_observer_.last_bitrate_);
henrik.lundin@webrtc.org29dd0de2013-10-21 14:00:01 +0000405}
Stefan Holmer52200d02016-09-20 14:14:23 +0200406
407TEST_F(BitrateControllerTest, TimeoutsWithoutFeedback) {
408 {
409 webrtc::test::ScopedFieldTrials override_field_trials(
Stefan Holmerdb158f92016-10-13 02:11:51 +0200410 "WebRTC-FeedbackTimeout/Enabled/");
Stefan Holmer52200d02016-09-20 14:14:23 +0200411 SetUp();
412 int expected_bitrate_bps = 300000;
413 controller_->SetBitrates(300000, kDefaultMinBitrateBps,
414 kDefaultMaxBitrateBps);
415
416 webrtc::ReportBlockList report_blocks;
417 report_blocks.push_back(CreateReportBlock(1, 2, 0, 1));
418 bandwidth_observer_->OnReceivedRtcpReceiverReport(
419 report_blocks, 50, clock_.TimeInMilliseconds());
420 EXPECT_EQ(expected_bitrate_bps, bitrate_observer_.last_bitrate_);
421 clock_.AdvanceTimeMilliseconds(500);
422
423 report_blocks.push_back(CreateReportBlock(1, 2, 0, 21));
424 bandwidth_observer_->OnReceivedRtcpReceiverReport(
425 report_blocks, 50, clock_.TimeInMilliseconds());
426 report_blocks.clear();
427 expected_bitrate_bps = expected_bitrate_bps * 1.08 + 1000;
428 EXPECT_EQ(expected_bitrate_bps, bitrate_observer_.last_bitrate_);
429 clock_.AdvanceTimeMilliseconds(1500);
430
431 report_blocks.push_back(CreateReportBlock(1, 2, 0, 41));
432 bandwidth_observer_->OnReceivedRtcpReceiverReport(
433 report_blocks, 50, clock_.TimeInMilliseconds());
434 expected_bitrate_bps = expected_bitrate_bps * 1.08 + 1000;
435 EXPECT_EQ(expected_bitrate_bps, bitrate_observer_.last_bitrate_);
stefan5cb19822017-06-16 07:47:00 -0700436 clock_.AdvanceTimeMilliseconds(4000);
Stefan Holmer52200d02016-09-20 14:14:23 +0200437
stefan5cb19822017-06-16 07:47:00 -0700438 // 4 seconds since feedback, expect increase.
Stefan Holmer52200d02016-09-20 14:14:23 +0200439 controller_->Process();
440 expected_bitrate_bps = expected_bitrate_bps * 1.08 + 1000;
441 EXPECT_EQ(expected_bitrate_bps, bitrate_observer_.last_bitrate_);
stefan5cb19822017-06-16 07:47:00 -0700442 clock_.AdvanceTimeMilliseconds(2000);
Stefan Holmer52200d02016-09-20 14:14:23 +0200443
stefan5cb19822017-06-16 07:47:00 -0700444 // 6 seconds since feedback, expect no increase.
Stefan Holmer52200d02016-09-20 14:14:23 +0200445 controller_->Process();
446 EXPECT_EQ(expected_bitrate_bps, bitrate_observer_.last_bitrate_);
stefan5cb19822017-06-16 07:47:00 -0700447 clock_.AdvanceTimeMilliseconds(9001);
Stefan Holmer52200d02016-09-20 14:14:23 +0200448
stefan5cb19822017-06-16 07:47:00 -0700449 // More than 15 seconds since feedback, expect decrease.
Stefan Holmer52200d02016-09-20 14:14:23 +0200450 controller_->Process();
451 expected_bitrate_bps *= 0.8;
452 EXPECT_EQ(expected_bitrate_bps, bitrate_observer_.last_bitrate_);
453 clock_.AdvanceTimeMilliseconds(500);
454
455 // Only one timeout every second.
456 controller_->Process();
457 EXPECT_EQ(expected_bitrate_bps, bitrate_observer_.last_bitrate_);
458 clock_.AdvanceTimeMilliseconds(501);
459
460 // New timeout allowed.
461 controller_->Process();
462 expected_bitrate_bps *= 0.8;
463 EXPECT_EQ(expected_bitrate_bps, bitrate_observer_.last_bitrate_);
464 }
465}
466
467TEST_F(BitrateControllerTest, StopIncreaseWithoutPacketReports) {
468 int expected_bitrate_bps = 300000;
469 controller_->SetBitrates(300000, kDefaultMinBitrateBps,
470 kDefaultMaxBitrateBps);
471
472 webrtc::ReportBlockList report_blocks;
473 report_blocks.push_back(CreateReportBlock(1, 2, 0, 1));
474 bandwidth_observer_->OnReceivedRtcpReceiverReport(
475 report_blocks, 50, clock_.TimeInMilliseconds());
476 EXPECT_EQ(expected_bitrate_bps, bitrate_observer_.last_bitrate_);
477 clock_.AdvanceTimeMilliseconds(500);
478
479 report_blocks.push_back(CreateReportBlock(1, 2, 0, 21));
480 bandwidth_observer_->OnReceivedRtcpReceiverReport(
481 report_blocks, 50, clock_.TimeInMilliseconds());
482 report_blocks.clear();
483 expected_bitrate_bps = expected_bitrate_bps * 1.08 + 1000;
484 EXPECT_EQ(expected_bitrate_bps, bitrate_observer_.last_bitrate_);
485 clock_.AdvanceTimeMilliseconds(1500);
486
487 // 1.2 seconds without packets reported as received, no increase.
488 report_blocks.push_back(CreateReportBlock(1, 2, 0, 21));
489 bandwidth_observer_->OnReceivedRtcpReceiverReport(
490 report_blocks, 50, clock_.TimeInMilliseconds());
491 EXPECT_EQ(expected_bitrate_bps, bitrate_observer_.last_bitrate_);
492 clock_.AdvanceTimeMilliseconds(1000);
493
494 // 5 packets reported as received since last, too few, no increase.
495 report_blocks.push_back(CreateReportBlock(1, 2, 0, 26));
496 bandwidth_observer_->OnReceivedRtcpReceiverReport(
497 report_blocks, 50, clock_.TimeInMilliseconds());
498 report_blocks.clear();
499 EXPECT_EQ(expected_bitrate_bps, bitrate_observer_.last_bitrate_);
500 clock_.AdvanceTimeMilliseconds(100);
501
502 // 15 packets reported as received since last, enough to increase.
503 report_blocks.push_back(CreateReportBlock(1, 2, 0, 41));
504 bandwidth_observer_->OnReceivedRtcpReceiverReport(
505 report_blocks, 50, clock_.TimeInMilliseconds());
506 expected_bitrate_bps = expected_bitrate_bps * 1.08 + 1000;
507 EXPECT_EQ(expected_bitrate_bps, bitrate_observer_.last_bitrate_);
508 clock_.AdvanceTimeMilliseconds(1000);
509}