aboutsummaryrefslogtreecommitdiff
path: root/webrtc/modules/bitrate_controller/bitrate_controller_unittest.cc
diff options
context:
space:
mode:
authorChih-hung Hsieh <chh@google.com>2015-12-01 17:07:48 +0000
committerandroid-build-merger <android-build-merger@google.com>2015-12-01 17:07:48 +0000
commita4acd9d6bc9b3b033d7d274316e75ee067df8d20 (patch)
tree672a185b294789cf991f385c3e395dd63bea9063 /webrtc/modules/bitrate_controller/bitrate_controller_unittest.cc
parent3681b90ba4fe7a27232dd3e27897d5d7ed9d651c (diff)
parentfe8b4a657979b49e1701bd92f6d5814a99e0b2be (diff)
downloadwebrtc-a4acd9d6bc9b3b033d7d274316e75ee067df8d20.tar.gz
Merge changes I7bbf776e,I1b827825
am: fe8b4a6579 * commit 'fe8b4a657979b49e1701bd92f6d5814a99e0b2be': (7237 commits) WIP: Changes after merge commit 'cb3f9bd' Make the nonlinear beamformer steerable Utilize bitrate above codec max to protect video. Enable VP9 internal resize by default. Filter overlapping RTP header extensions. Make VCMEncodedFrameCallback const. MediaCodecVideoEncoder: Add number of quality resolution downscales to Encoded callback. Remove redudant encoder rate calls. Create isolate files for nonparallel tests. Register header extensions in RtpRtcpObserver to avoid log spam. Make an enum class out of NetEqDecoder, and hide the neteq_decoders_ table ACM: Move NACK functionality inside NetEq Fix chromium-style warnings in webrtc/sound/. Create a 'webrtc_nonparallel_tests' target. Update scalability structure data according to updates in the RTP payload profile. audio_coding: rename interface -> include Rewrote perform_action_on_all_files to be parallell. Update reference indices according to updates in the RTP payload profile. Disable P2PTransport...TestFailoverControlledSide on Memcheck pass clangcl compile options to ignore warnings in gflags.cc ...
Diffstat (limited to 'webrtc/modules/bitrate_controller/bitrate_controller_unittest.cc')
-rw-r--r--webrtc/modules/bitrate_controller/bitrate_controller_unittest.cc397
1 files changed, 397 insertions, 0 deletions
diff --git a/webrtc/modules/bitrate_controller/bitrate_controller_unittest.cc b/webrtc/modules/bitrate_controller/bitrate_controller_unittest.cc
new file mode 100644
index 0000000000..72831c78d6
--- /dev/null
+++ b/webrtc/modules/bitrate_controller/bitrate_controller_unittest.cc
@@ -0,0 +1,397 @@
+/*
+ * Copyright (c) 2012 The WebRTC project authors. All Rights Reserved.
+ *
+ * Use of this source code is governed by a BSD-style license
+ * that can be found in the LICENSE file in the root of the source
+ * tree. An additional intellectual property rights grant can be found
+ * in the file PATENTS. All contributing project authors may
+ * be found in the AUTHORS file in the root of the source tree.
+ */
+
+#include "testing/gtest/include/gtest/gtest.h"
+
+#include <algorithm>
+#include <vector>
+
+#include "webrtc/modules/bitrate_controller/include/bitrate_controller.h"
+#include "webrtc/modules/rtp_rtcp/interface/rtp_rtcp_defines.h"
+
+using webrtc::RtcpBandwidthObserver;
+using webrtc::BitrateObserver;
+using webrtc::BitrateController;
+
+uint8_t WeightedLoss(int num_packets1, uint8_t fraction_loss1,
+ int num_packets2, uint8_t fraction_loss2) {
+ int weighted_sum = num_packets1 * fraction_loss1 +
+ num_packets2 * fraction_loss2;
+ int total_num_packets = num_packets1 + num_packets2;
+ return (weighted_sum + total_num_packets / 2) / total_num_packets;
+}
+
+webrtc::RTCPReportBlock CreateReportBlock(
+ uint32_t remote_ssrc, uint32_t source_ssrc,
+ uint8_t fraction_lost, uint32_t extended_high_sequence_number) {
+ return webrtc::RTCPReportBlock(remote_ssrc, source_ssrc, fraction_lost, 0,
+ extended_high_sequence_number, 0, 0, 0);
+}
+
+class TestBitrateObserver: public BitrateObserver {
+ public:
+ TestBitrateObserver()
+ : last_bitrate_(0),
+ last_fraction_loss_(0),
+ last_rtt_(0) {
+ }
+
+ virtual void OnNetworkChanged(uint32_t bitrate,
+ uint8_t fraction_loss,
+ int64_t rtt) {
+ last_bitrate_ = static_cast<int>(bitrate);
+ last_fraction_loss_ = fraction_loss;
+ last_rtt_ = rtt;
+ }
+ int last_bitrate_;
+ uint8_t last_fraction_loss_;
+ int64_t last_rtt_;
+};
+
+class BitrateControllerTest : public ::testing::Test {
+ protected:
+ BitrateControllerTest() : clock_(0) {}
+ ~BitrateControllerTest() {}
+
+ virtual void SetUp() {
+ controller_ =
+ BitrateController::CreateBitrateController(&clock_, &bitrate_observer_);
+ 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();
+ }
+
+ virtual void TearDown() {
+ delete bandwidth_observer_;
+ delete controller_;
+ }
+
+ const int kMinBitrateBps = 100000;
+ const int kStartBitrateBps = 200000;
+ const int kMaxBitrateBps = 300000;
+
+ const int kDefaultMinBitrateBps = 10000;
+ const int kDefaultMaxBitrateBps = 1000000000;
+
+ webrtc::SimulatedClock clock_;
+ TestBitrateObserver bitrate_observer_;
+ BitrateController* controller_;
+ RtcpBandwidthObserver* bandwidth_observer_;
+};
+
+TEST_F(BitrateControllerTest, DefaultMinMaxBitrate) {
+ // Receive successively lower REMBs, verify the reserved bitrate is deducted.
+ controller_->SetMinMaxBitrate(0, 0);
+ EXPECT_EQ(kStartBitrateBps, bitrate_observer_.last_bitrate_);
+ bandwidth_observer_->OnReceivedEstimatedBitrate(kDefaultMinBitrateBps / 2);
+ EXPECT_EQ(kDefaultMinBitrateBps, bitrate_observer_.last_bitrate_);
+ bandwidth_observer_->OnReceivedEstimatedBitrate(2 * kDefaultMaxBitrateBps);
+ clock_.AdvanceTimeMilliseconds(1000);
+ controller_->Process();
+ EXPECT_EQ(kDefaultMaxBitrateBps, bitrate_observer_.last_bitrate_);
+}
+
+TEST_F(BitrateControllerTest, OneBitrateObserverOneRtcpObserver) {
+ // First REMB applies immediately.
+ int64_t time_ms = 1001;
+ webrtc::ReportBlockList report_blocks;
+ report_blocks.push_back(CreateReportBlock(1, 2, 0, 1));
+ bandwidth_observer_->OnReceivedEstimatedBitrate(200000);
+ EXPECT_EQ(200000, bitrate_observer_.last_bitrate_);
+ EXPECT_EQ(0, bitrate_observer_.last_fraction_loss_);
+ EXPECT_EQ(0, bitrate_observer_.last_rtt_);
+ bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, time_ms);
+ report_blocks.clear();
+ time_ms += 2000;
+
+ // Receive a high remb, test bitrate inc.
+ bandwidth_observer_->OnReceivedEstimatedBitrate(400000);
+
+ // Test bitrate increase 8% per second.
+ report_blocks.push_back(CreateReportBlock(1, 2, 0, 21));
+ bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, time_ms);
+ EXPECT_EQ(217000, bitrate_observer_.last_bitrate_);
+ EXPECT_EQ(0, bitrate_observer_.last_fraction_loss_);
+ EXPECT_EQ(50, bitrate_observer_.last_rtt_);
+ time_ms += 1000;
+
+ report_blocks.clear();
+ report_blocks.push_back(CreateReportBlock(1, 2, 0, 41));
+ bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, time_ms);
+ EXPECT_EQ(235360, bitrate_observer_.last_bitrate_);
+ EXPECT_EQ(0, bitrate_observer_.last_fraction_loss_);
+ EXPECT_EQ(50, bitrate_observer_.last_rtt_);
+ time_ms += 1000;
+
+ report_blocks.clear();
+ report_blocks.push_back(CreateReportBlock(1, 2, 0, 61));
+ bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, time_ms);
+ EXPECT_EQ(255189, bitrate_observer_.last_bitrate_);
+ time_ms += 1000;
+
+ report_blocks.clear();
+ report_blocks.push_back(CreateReportBlock(1, 2, 0, 81));
+ bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, time_ms);
+ EXPECT_EQ(276604, bitrate_observer_.last_bitrate_);
+ time_ms += 1000;
+
+ report_blocks.clear();
+ report_blocks.push_back(CreateReportBlock(1, 2, 0, 801));
+ bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, time_ms);
+ EXPECT_EQ(299732, bitrate_observer_.last_bitrate_);
+ time_ms += 1000;
+
+ // Reach max cap.
+ report_blocks.clear();
+ report_blocks.push_back(CreateReportBlock(1, 2, 0, 101));
+ bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, time_ms);
+ EXPECT_EQ(300000, bitrate_observer_.last_bitrate_);
+ time_ms += 1000;
+
+ report_blocks.clear();
+ report_blocks.push_back(CreateReportBlock(1, 2, 0, 141));
+ bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, time_ms);
+ EXPECT_EQ(300000, bitrate_observer_.last_bitrate_);
+
+ // Test that a low REMB trigger immediately.
+ bandwidth_observer_->OnReceivedEstimatedBitrate(250000);
+ EXPECT_EQ(250000, bitrate_observer_.last_bitrate_);
+ EXPECT_EQ(0, bitrate_observer_.last_fraction_loss_);
+ EXPECT_EQ(50, bitrate_observer_.last_rtt_);
+
+ bandwidth_observer_->OnReceivedEstimatedBitrate(1000);
+ EXPECT_EQ(100000, bitrate_observer_.last_bitrate_); // Min cap.
+}
+
+TEST_F(BitrateControllerTest, OneBitrateObserverTwoRtcpObservers) {
+ // 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));
+ bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, time_ms);
+ report_blocks.clear();
+ time_ms += 500;
+
+ RtcpBandwidthObserver* second_bandwidth_observer =
+ controller_->CreateRtcpBandwidthObserver();
+
+ // Test start bitrate.
+ report_blocks.push_back(CreateReportBlock(1, 2, 0, 21));
+ second_bandwidth_observer->OnReceivedRtcpReceiverReport(
+ report_blocks, 100, 1);
+ 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;
+
+ // Test bitrate increase 8% per second.
+ report_blocks.clear();
+ report_blocks.push_back(CreateReportBlock(1, 2, 0, 21));
+ bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, time_ms);
+ time_ms += 500;
+ second_bandwidth_observer->OnReceivedRtcpReceiverReport(
+ report_blocks, 100, time_ms);
+ EXPECT_EQ(235360, 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));
+ second_bandwidth_observer->OnReceivedRtcpReceiverReport(
+ report_blocks, 100, time_ms);
+ EXPECT_EQ(235360, bitrate_observer_.last_bitrate_);
+ time_ms += 500;
+
+ report_blocks.clear();
+ report_blocks.push_back(CreateReportBlock(1, 2, 0, 41));
+ bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, time_ms);
+ EXPECT_EQ(255189, bitrate_observer_.last_bitrate_);
+
+ // Second report should not change estimate.
+ report_blocks.clear();
+ report_blocks.push_back(CreateReportBlock(1, 2, 0, 41));
+ second_bandwidth_observer->OnReceivedRtcpReceiverReport(
+ report_blocks, 100, time_ms);
+ EXPECT_EQ(255189, 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));
+ second_bandwidth_observer->OnReceivedRtcpReceiverReport(
+ report_blocks, 50, time_ms);
+ EXPECT_EQ(276604, bitrate_observer_.last_bitrate_);
+ time_ms += 1000;
+
+ report_blocks.clear();
+ report_blocks.push_back(CreateReportBlock(1, 2, 0, 81));
+ second_bandwidth_observer->OnReceivedRtcpReceiverReport(
+ report_blocks, 50, time_ms);
+ EXPECT_EQ(299732, bitrate_observer_.last_bitrate_);
+ time_ms += 1000;
+
+ // Reach max cap.
+ report_blocks.clear();
+ report_blocks.push_back(CreateReportBlock(1, 2, 0, 121));
+ second_bandwidth_observer->OnReceivedRtcpReceiverReport(
+ report_blocks, 50, time_ms);
+ EXPECT_EQ(300000, bitrate_observer_.last_bitrate_);
+ time_ms += 1000;
+
+ report_blocks.clear();
+ report_blocks.push_back(CreateReportBlock(1, 2, 0, 141));
+ second_bandwidth_observer->OnReceivedRtcpReceiverReport(
+ report_blocks, 50, time_ms);
+ EXPECT_EQ(300000, bitrate_observer_.last_bitrate_);
+
+ // Test that a low REMB trigger immediately.
+ // We don't care which bandwidth observer that delivers the REMB.
+ second_bandwidth_observer->OnReceivedEstimatedBitrate(250000);
+ EXPECT_EQ(250000, bitrate_observer_.last_bitrate_);
+ EXPECT_EQ(0, bitrate_observer_.last_fraction_loss_);
+ EXPECT_EQ(50, bitrate_observer_.last_rtt_);
+
+ // Min cap.
+ bandwidth_observer_->OnReceivedEstimatedBitrate(1000);
+ EXPECT_EQ(100000, bitrate_observer_.last_bitrate_);
+ delete second_bandwidth_observer;
+}
+
+TEST_F(BitrateControllerTest, OneBitrateObserverMultipleReportBlocks) {
+ uint32_t sequence_number[2] = {0, 0xFF00};
+ const int kStartBitrate = 200000;
+ const int kMinBitrate = 100000;
+ const int kMaxBitrate = 300000;
+ controller_->SetStartBitrate(kStartBitrate);
+ controller_->SetMinMaxBitrate(kMinBitrate, kMaxBitrate);
+
+ // REMBs during the first 2 seconds apply immediately.
+ int64_t time_ms = 1001;
+ webrtc::ReportBlockList report_blocks;
+ report_blocks.push_back(CreateReportBlock(1, 2, 0, sequence_number[0]));
+ bandwidth_observer_->OnReceivedEstimatedBitrate(kStartBitrate);
+ bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, time_ms);
+ report_blocks.clear();
+ time_ms += 2000;
+
+ // Receive a high REMB, test bitrate increase.
+ bandwidth_observer_->OnReceivedEstimatedBitrate(400000);
+
+ int last_bitrate = 0;
+ // Ramp up to max bitrate.
+ for (int i = 0; i < 6; ++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,
+ time_ms);
+ EXPECT_GT(bitrate_observer_.last_bitrate_, last_bitrate);
+ EXPECT_EQ(0, bitrate_observer_.last_fraction_loss_);
+ EXPECT_EQ(50, bitrate_observer_.last_rtt_);
+ last_bitrate = bitrate_observer_.last_bitrate_;
+ time_ms += 1000;
+ sequence_number[0] += 20;
+ sequence_number[1] += 1;
+ report_blocks.clear();
+ }
+
+ EXPECT_EQ(kMaxBitrate, bitrate_observer_.last_bitrate_);
+
+ // Packet loss on the first stream. Verify that bitrate decreases.
+ report_blocks.push_back(CreateReportBlock(1, 2, 50, sequence_number[0]));
+ report_blocks.push_back(CreateReportBlock(1, 3, 0, sequence_number[1]));
+ bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, time_ms);
+ EXPECT_LT(bitrate_observer_.last_bitrate_, last_bitrate);
+ EXPECT_EQ(WeightedLoss(20, 50, 1, 0), bitrate_observer_.last_fraction_loss_);
+ EXPECT_EQ(50, bitrate_observer_.last_rtt_);
+ last_bitrate = bitrate_observer_.last_bitrate_;
+ sequence_number[0] += 20;
+ sequence_number[1] += 20;
+ time_ms += 1000;
+ report_blocks.clear();
+
+ // Packet loss on the second stream. Verify that bitrate decreases.
+ report_blocks.push_back(CreateReportBlock(1, 2, 0, sequence_number[0]));
+ report_blocks.push_back(CreateReportBlock(1, 3, 75, sequence_number[1]));
+ bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, time_ms);
+ EXPECT_LT(bitrate_observer_.last_bitrate_, last_bitrate);
+ EXPECT_EQ(WeightedLoss(20, 0, 20, 75), bitrate_observer_.last_fraction_loss_);
+ EXPECT_EQ(50, bitrate_observer_.last_rtt_);
+ last_bitrate = bitrate_observer_.last_bitrate_;
+ sequence_number[0] += 20;
+ sequence_number[1] += 1;
+ time_ms += 1000;
+ report_blocks.clear();
+
+ // All packets lost on stream with few packets, no back-off.
+ report_blocks.push_back(CreateReportBlock(1, 2, 1, sequence_number[0]));
+ report_blocks.push_back(CreateReportBlock(1, 3, 255, sequence_number[1]));
+ bandwidth_observer_->OnReceivedRtcpReceiverReport(report_blocks, 50, time_ms);
+ EXPECT_EQ(bitrate_observer_.last_bitrate_, last_bitrate);
+ EXPECT_EQ(WeightedLoss(20, 1, 1, 255), bitrate_observer_.last_fraction_loss_);
+ EXPECT_EQ(50, bitrate_observer_.last_rtt_);
+ last_bitrate = bitrate_observer_.last_bitrate_;
+ sequence_number[0] += 20;
+ sequence_number[1] += 1;
+ report_blocks.clear();
+}
+
+TEST_F(BitrateControllerTest, SetReservedBitrate) {
+ // Receive successively lower REMBs, verify the reserved bitrate is deducted.
+ controller_->SetReservedBitrate(0);
+ bandwidth_observer_->OnReceivedEstimatedBitrate(400000);
+ EXPECT_EQ(200000, bitrate_observer_.last_bitrate_);
+ controller_->SetReservedBitrate(50000);
+ bandwidth_observer_->OnReceivedEstimatedBitrate(400000);
+ EXPECT_EQ(150000, bitrate_observer_.last_bitrate_);
+
+ controller_->SetReservedBitrate(0);
+ bandwidth_observer_->OnReceivedEstimatedBitrate(250000);
+ EXPECT_EQ(200000, bitrate_observer_.last_bitrate_);
+ controller_->SetReservedBitrate(50000);
+ bandwidth_observer_->OnReceivedEstimatedBitrate(250000);
+ EXPECT_EQ(150000, bitrate_observer_.last_bitrate_);
+
+ controller_->SetReservedBitrate(0);
+ bandwidth_observer_->OnReceivedEstimatedBitrate(200000);
+ EXPECT_EQ(200000, bitrate_observer_.last_bitrate_);
+ controller_->SetReservedBitrate(30000);
+ bandwidth_observer_->OnReceivedEstimatedBitrate(200000);
+ EXPECT_EQ(170000, bitrate_observer_.last_bitrate_);
+
+ controller_->SetReservedBitrate(0);
+ bandwidth_observer_->OnReceivedEstimatedBitrate(160000);
+ EXPECT_EQ(160000, bitrate_observer_.last_bitrate_);
+ controller_->SetReservedBitrate(30000);
+ bandwidth_observer_->OnReceivedEstimatedBitrate(160000);
+ EXPECT_EQ(130000, bitrate_observer_.last_bitrate_);
+
+ controller_->SetReservedBitrate(0);
+ bandwidth_observer_->OnReceivedEstimatedBitrate(120000);
+ EXPECT_EQ(120000, bitrate_observer_.last_bitrate_);
+ controller_->SetReservedBitrate(10000);
+ bandwidth_observer_->OnReceivedEstimatedBitrate(120000);
+ EXPECT_EQ(110000, bitrate_observer_.last_bitrate_);
+
+ controller_->SetReservedBitrate(0);
+ bandwidth_observer_->OnReceivedEstimatedBitrate(120000);
+ EXPECT_EQ(120000, bitrate_observer_.last_bitrate_);
+ controller_->SetReservedBitrate(50000);
+ bandwidth_observer_->OnReceivedEstimatedBitrate(120000);
+ // Limited by min bitrate.
+ EXPECT_EQ(100000, bitrate_observer_.last_bitrate_);
+
+ controller_->SetReservedBitrate(10000);
+ bandwidth_observer_->OnReceivedEstimatedBitrate(1);
+ EXPECT_EQ(100000, bitrate_observer_.last_bitrate_);
+}