Index: webrtc/modules/bitrate_controller/bitrate_controller_unittest.cc |
diff --git a/webrtc/modules/bitrate_controller/bitrate_controller_unittest.cc b/webrtc/modules/bitrate_controller/bitrate_controller_unittest.cc |
index bfc44ea709d96e8c46eee848f3bb7a1dedeaef06..43060e2d1adbb5951d78544f08fd5121c3dfdb89 100644 |
--- a/webrtc/modules/bitrate_controller/bitrate_controller_unittest.cc |
+++ b/webrtc/modules/bitrate_controller/bitrate_controller_unittest.cc |
@@ -17,6 +17,7 @@ |
#include "webrtc/modules/bitrate_controller/include/bitrate_controller.h" |
#include "webrtc/modules/pacing/mock/mock_paced_sender.h" |
#include "webrtc/modules/rtp_rtcp/include/rtp_rtcp_defines.h" |
+#include "webrtc/test/field_trial.h" |
using ::testing::Exactly; |
using ::testing::Return; |
@@ -67,18 +68,16 @@ class BitrateControllerTest : public ::testing::Test { |
~BitrateControllerTest() {} |
virtual void SetUp() { |
- controller_ = BitrateController::CreateBitrateController( |
- &clock_, &bitrate_observer_, &event_log_); |
+ controller_.reset(BitrateController::CreateBitrateController( |
+ &clock_, &bitrate_observer_, &event_log_)); |
controller_->SetStartBitrate(kStartBitrateBps); |
EXPECT_EQ(kStartBitrateBps, bitrate_observer_.last_bitrate_); |
controller_->SetMinMaxBitrate(kMinBitrateBps, kMaxBitrateBps); |
EXPECT_EQ(kStartBitrateBps, bitrate_observer_.last_bitrate_); |
- bandwidth_observer_ = controller_->CreateRtcpBandwidthObserver(); |
+ bandwidth_observer_.reset(controller_->CreateRtcpBandwidthObserver()); |
} |
virtual void TearDown() { |
- delete bandwidth_observer_; |
- delete controller_; |
} |
const int kMinBitrateBps = 100000; |
@@ -90,8 +89,8 @@ class BitrateControllerTest : public ::testing::Test { |
webrtc::SimulatedClock clock_; |
TestBitrateObserver bitrate_observer_; |
- BitrateController* controller_; |
- RtcpBandwidthObserver* bandwidth_observer_; |
+ std::unique_ptr<BitrateController> controller_; |
+ std::unique_ptr<RtcpBandwidthObserver> bandwidth_observer_; |
testing::NiceMock<webrtc::MockRtcEventLog> event_log_; |
}; |
@@ -184,10 +183,12 @@ TEST_F(BitrateControllerTest, OneBitrateObserverOneRtcpObserver) { |
} |
TEST_F(BitrateControllerTest, OneBitrateObserverTwoRtcpObservers) { |
+ const uint32_t kSsrc1 = 1; |
+ const uint32_t kSsrc2 = 2; |
// REMBs during the first 2 seconds apply immediately. |
int64_t time_ms = 1; |
webrtc::ReportBlockList report_blocks; |
- report_blocks.push_back(CreateReportBlock(1, 2, 0, 1)); |
+ report_blocks.push_back(CreateReportBlock(kSsrc1, 2, 0, 1)); |
bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, time_ms); |
report_blocks.clear(); |
time_ms += 500; |
@@ -196,72 +197,73 @@ TEST_F(BitrateControllerTest, OneBitrateObserverTwoRtcpObservers) { |
controller_->CreateRtcpBandwidthObserver(); |
// Test start bitrate. |
- report_blocks.push_back(CreateReportBlock(1, 2, 0, 21)); |
+ report_blocks.push_back(CreateReportBlock(2, 2, 0, 21)); |
second_bandwidth_observer->OnReceivedRtcpReceiverReport( |
report_blocks, 100, 1); |
- EXPECT_EQ(217000, bitrate_observer_.last_bitrate_); |
+ EXPECT_EQ(200000, bitrate_observer_.last_bitrate_); |
EXPECT_EQ(0, bitrate_observer_.last_fraction_loss_); |
EXPECT_EQ(100, bitrate_observer_.last_rtt_); |
time_ms += 500; |
// Test bitrate increase 8% per second. |
report_blocks.clear(); |
- report_blocks.push_back(CreateReportBlock(1, 2, 0, 21)); |
+ report_blocks.push_back(CreateReportBlock(kSsrc1, 2, 0, 21)); |
bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, time_ms); |
time_ms += 500; |
+ report_blocks.front().remoteSSRC = kSsrc2; |
second_bandwidth_observer->OnReceivedRtcpReceiverReport( |
report_blocks, 100, time_ms); |
- EXPECT_EQ(235360, bitrate_observer_.last_bitrate_); |
+ EXPECT_EQ(217000, bitrate_observer_.last_bitrate_); |
EXPECT_EQ(0, bitrate_observer_.last_fraction_loss_); |
EXPECT_EQ(100, bitrate_observer_.last_rtt_); |
time_ms += 500; |
// Extra report should not change estimate. |
report_blocks.clear(); |
- report_blocks.push_back(CreateReportBlock(1, 2, 0, 31)); |
+ report_blocks.push_back(CreateReportBlock(kSsrc2, 2, 0, 31)); |
second_bandwidth_observer->OnReceivedRtcpReceiverReport( |
report_blocks, 100, time_ms); |
- EXPECT_EQ(235360, bitrate_observer_.last_bitrate_); |
+ EXPECT_EQ(217000, bitrate_observer_.last_bitrate_); |
time_ms += 500; |
report_blocks.clear(); |
- report_blocks.push_back(CreateReportBlock(1, 2, 0, 41)); |
+ report_blocks.push_back(CreateReportBlock(kSsrc1, 2, 0, 41)); |
bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, time_ms); |
- EXPECT_EQ(255189, bitrate_observer_.last_bitrate_); |
+ EXPECT_EQ(235360, bitrate_observer_.last_bitrate_); |
// Second report should not change estimate. |
report_blocks.clear(); |
- report_blocks.push_back(CreateReportBlock(1, 2, 0, 41)); |
+ report_blocks.push_back(CreateReportBlock(kSsrc2, 2, 0, 41)); |
second_bandwidth_observer->OnReceivedRtcpReceiverReport( |
report_blocks, 100, time_ms); |
- EXPECT_EQ(255189, bitrate_observer_.last_bitrate_); |
+ EXPECT_EQ(235360, bitrate_observer_.last_bitrate_); |
time_ms += 1000; |
// Reports from only one bandwidth observer is ok. |
report_blocks.clear(); |
- report_blocks.push_back(CreateReportBlock(1, 2, 0, 61)); |
+ report_blocks.push_back(CreateReportBlock(kSsrc2, 2, 0, 61)); |
second_bandwidth_observer->OnReceivedRtcpReceiverReport( |
report_blocks, 50, time_ms); |
- EXPECT_EQ(276604, bitrate_observer_.last_bitrate_); |
+ EXPECT_EQ(255189, bitrate_observer_.last_bitrate_); |
time_ms += 1000; |
report_blocks.clear(); |
- report_blocks.push_back(CreateReportBlock(1, 2, 0, 81)); |
+ report_blocks.push_back(CreateReportBlock(kSsrc2, 2, 0, 81)); |
second_bandwidth_observer->OnReceivedRtcpReceiverReport( |
report_blocks, 50, time_ms); |
- EXPECT_EQ(299732, bitrate_observer_.last_bitrate_); |
+ EXPECT_EQ(276604, bitrate_observer_.last_bitrate_); |
time_ms += 1000; |
- // Reach max cap. |
report_blocks.clear(); |
- report_blocks.push_back(CreateReportBlock(1, 2, 0, 121)); |
+ report_blocks.push_back(CreateReportBlock(kSsrc2, 2, 0, 121)); |
second_bandwidth_observer->OnReceivedRtcpReceiverReport( |
report_blocks, 50, time_ms); |
- EXPECT_EQ(300000, bitrate_observer_.last_bitrate_); |
+ EXPECT_EQ(299732, bitrate_observer_.last_bitrate_); |
time_ms += 1000; |
+ // Reach max cap. |
report_blocks.clear(); |
- report_blocks.push_back(CreateReportBlock(1, 2, 0, 141)); |
+ report_blocks.push_back(CreateReportBlock(kSsrc2, 2, 0, 141)); |
second_bandwidth_observer->OnReceivedRtcpReceiverReport( |
report_blocks, 50, time_ms); |
EXPECT_EQ(300000, bitrate_observer_.last_bitrate_); |
@@ -301,7 +303,7 @@ TEST_F(BitrateControllerTest, OneBitrateObserverMultipleReportBlocks) { |
int last_bitrate = 0; |
// Ramp up to max bitrate. |
- for (int i = 0; i < 6; ++i) { |
+ for (int i = 0; i < 7; ++i) { |
report_blocks.push_back(CreateReportBlock(1, 2, 0, sequence_number[0])); |
report_blocks.push_back(CreateReportBlock(1, 3, 0, sequence_number[1])); |
bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, |
@@ -406,3 +408,107 @@ TEST_F(BitrateControllerTest, SetReservedBitrate) { |
bandwidth_observer_->OnReceivedEstimatedBitrate(1); |
EXPECT_EQ(100000, bitrate_observer_.last_bitrate_); |
} |
+ |
+TEST_F(BitrateControllerTest, TimeoutsWithoutFeedback) { |
+ { |
+ webrtc::test::ScopedFieldTrials override_field_trials( |
+ "WebRTC-SendSideBwe/Enabled/"); |
+ SetUp(); |
+ int expected_bitrate_bps = 300000; |
+ controller_->SetBitrates(300000, kDefaultMinBitrateBps, |
+ kDefaultMaxBitrateBps); |
+ |
+ webrtc::ReportBlockList report_blocks; |
+ report_blocks.push_back(CreateReportBlock(1, 2, 0, 1)); |
+ bandwidth_observer_->OnReceivedRtcpReceiverReport( |
+ report_blocks, 50, clock_.TimeInMilliseconds()); |
+ EXPECT_EQ(expected_bitrate_bps, bitrate_observer_.last_bitrate_); |
+ clock_.AdvanceTimeMilliseconds(500); |
+ |
+ report_blocks.push_back(CreateReportBlock(1, 2, 0, 21)); |
+ bandwidth_observer_->OnReceivedRtcpReceiverReport( |
+ report_blocks, 50, clock_.TimeInMilliseconds()); |
+ report_blocks.clear(); |
+ expected_bitrate_bps = expected_bitrate_bps * 1.08 + 1000; |
+ EXPECT_EQ(expected_bitrate_bps, bitrate_observer_.last_bitrate_); |
+ clock_.AdvanceTimeMilliseconds(1500); |
+ |
+ report_blocks.push_back(CreateReportBlock(1, 2, 0, 41)); |
+ bandwidth_observer_->OnReceivedRtcpReceiverReport( |
+ report_blocks, 50, clock_.TimeInMilliseconds()); |
+ expected_bitrate_bps = expected_bitrate_bps * 1.08 + 1000; |
+ EXPECT_EQ(expected_bitrate_bps, bitrate_observer_.last_bitrate_); |
+ clock_.AdvanceTimeMilliseconds(1000); |
+ |
+ // 1 seconds since feedback, expect increase. |
+ controller_->Process(); |
+ expected_bitrate_bps = expected_bitrate_bps * 1.08 + 1000; |
+ EXPECT_EQ(expected_bitrate_bps, bitrate_observer_.last_bitrate_); |
+ clock_.AdvanceTimeMilliseconds(800); |
+ |
+ // 1.8 seconds since feedback, expect no increase. |
+ controller_->Process(); |
+ EXPECT_EQ(expected_bitrate_bps, bitrate_observer_.last_bitrate_); |
+ clock_.AdvanceTimeMilliseconds(3701); |
+ |
+ // More than 4.5 seconds since feedback, expect decrease. |
+ controller_->Process(); |
+ expected_bitrate_bps *= 0.8; |
+ EXPECT_EQ(expected_bitrate_bps, bitrate_observer_.last_bitrate_); |
+ clock_.AdvanceTimeMilliseconds(500); |
+ |
+ // Only one timeout every second. |
+ controller_->Process(); |
+ EXPECT_EQ(expected_bitrate_bps, bitrate_observer_.last_bitrate_); |
+ clock_.AdvanceTimeMilliseconds(501); |
+ |
+ // New timeout allowed. |
+ controller_->Process(); |
+ expected_bitrate_bps *= 0.8; |
+ EXPECT_EQ(expected_bitrate_bps, bitrate_observer_.last_bitrate_); |
+ } |
+} |
+ |
+TEST_F(BitrateControllerTest, StopIncreaseWithoutPacketReports) { |
+ int expected_bitrate_bps = 300000; |
+ controller_->SetBitrates(300000, kDefaultMinBitrateBps, |
+ kDefaultMaxBitrateBps); |
+ |
+ webrtc::ReportBlockList report_blocks; |
+ report_blocks.push_back(CreateReportBlock(1, 2, 0, 1)); |
+ bandwidth_observer_->OnReceivedRtcpReceiverReport( |
+ report_blocks, 50, clock_.TimeInMilliseconds()); |
+ EXPECT_EQ(expected_bitrate_bps, bitrate_observer_.last_bitrate_); |
+ clock_.AdvanceTimeMilliseconds(500); |
+ |
+ report_blocks.push_back(CreateReportBlock(1, 2, 0, 21)); |
+ bandwidth_observer_->OnReceivedRtcpReceiverReport( |
+ report_blocks, 50, clock_.TimeInMilliseconds()); |
+ report_blocks.clear(); |
+ expected_bitrate_bps = expected_bitrate_bps * 1.08 + 1000; |
+ EXPECT_EQ(expected_bitrate_bps, bitrate_observer_.last_bitrate_); |
+ clock_.AdvanceTimeMilliseconds(1500); |
+ |
+ // 1.2 seconds without packets reported as received, no increase. |
+ report_blocks.push_back(CreateReportBlock(1, 2, 0, 21)); |
+ bandwidth_observer_->OnReceivedRtcpReceiverReport( |
+ report_blocks, 50, clock_.TimeInMilliseconds()); |
+ EXPECT_EQ(expected_bitrate_bps, bitrate_observer_.last_bitrate_); |
+ clock_.AdvanceTimeMilliseconds(1000); |
+ |
+ // 5 packets reported as received since last, too few, no increase. |
+ report_blocks.push_back(CreateReportBlock(1, 2, 0, 26)); |
+ bandwidth_observer_->OnReceivedRtcpReceiverReport( |
+ report_blocks, 50, clock_.TimeInMilliseconds()); |
+ report_blocks.clear(); |
+ EXPECT_EQ(expected_bitrate_bps, bitrate_observer_.last_bitrate_); |
+ clock_.AdvanceTimeMilliseconds(100); |
+ |
+ // 15 packets reported as received since last, enough to increase. |
+ report_blocks.push_back(CreateReportBlock(1, 2, 0, 41)); |
+ bandwidth_observer_->OnReceivedRtcpReceiverReport( |
+ report_blocks, 50, clock_.TimeInMilliseconds()); |
+ expected_bitrate_bps = expected_bitrate_bps * 1.08 + 1000; |
+ EXPECT_EQ(expected_bitrate_bps, bitrate_observer_.last_bitrate_); |
+ clock_.AdvanceTimeMilliseconds(1000); |
+} |