| /* |
| * Copyright (c) 2014 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 <algorithm> |
| #include <vector> |
| |
| #include "testing/gtest/include/gtest/gtest.h" |
| #include "webrtc/modules/bitrate_controller/send_side_bandwidth_estimation.h" |
| |
| namespace webrtc { |
| |
| class TestBandwidthEstimation : public SendSideBandwidthEstimation { |
| public: |
| explicit TestBandwidthEstimation(bool in_experiment) |
| : in_experiment_(in_experiment) {} |
| |
| virtual bool ProbingExperimentIsEnabled() const OVERRIDE { |
| return in_experiment_; |
| } |
| |
| private: |
| bool in_experiment_; |
| }; |
| |
| TEST(SendSideBweTest, InitialRembWithProbing) { |
| TestBandwidthEstimation bwe(true); |
| bwe.SetMinMaxBitrate(100000, 1500000); |
| bwe.SetSendBitrate(200000); |
| |
| const uint32_t kRemb = 1000000u; |
| const uint32_t kSecondRemb = kRemb + 500000u; |
| int64_t now_ms = 0; |
| |
| bwe.UpdateReceiverBlock(0, 50, 1, now_ms); |
| |
| // Initial REMB applies immediately. |
| bwe.UpdateReceiverEstimate(kRemb); |
| bwe.UpdateEstimate(now_ms); |
| uint32_t bitrate; |
| uint8_t fraction_loss; |
| int64_t rtt; |
| bwe.CurrentEstimate(&bitrate, &fraction_loss, &rtt); |
| EXPECT_EQ(kRemb, bitrate); |
| |
| // Second REMB doesn't apply immediately. |
| now_ms += 2001; |
| bwe.UpdateReceiverEstimate(kSecondRemb); |
| bwe.UpdateEstimate(now_ms); |
| bitrate = 0; |
| bwe.CurrentEstimate(&bitrate, &fraction_loss, &rtt); |
| EXPECT_EQ(kRemb, bitrate); |
| } |
| |
| TEST(SendSideBweTest, InitialRembWithoutProbing) { |
| TestBandwidthEstimation bwe(false); |
| bwe.SetMinMaxBitrate(100000, 1500000); |
| const uint32_t kStartBitrate = 200000; |
| bwe.SetSendBitrate(kStartBitrate); |
| |
| int64_t now_ms = 0; |
| bwe.UpdateReceiverBlock(0, 50, 1, now_ms); |
| |
| // Initial REMB doesn't apply immediately. |
| const uint32_t kRemb = 1000000u; |
| bwe.UpdateReceiverEstimate(kRemb); |
| bwe.UpdateEstimate(now_ms); |
| uint32_t bitrate; |
| uint8_t fraction_loss; |
| int64_t rtt; |
| bwe.CurrentEstimate(&bitrate, &fraction_loss, &rtt); |
| EXPECT_EQ(kStartBitrate, bitrate); |
| } |
| } // namespace webrtc |