mirror of
https://github.com/mollyim/webrtc.git
synced 2025-05-15 06:40:43 +01:00
56da2f7868
This CL activates unit tests for the task queue based send side congestion controller that will replace the current one in the future. To be able to have the tests run side by side with the tests of the old congestion controller, the old tests have been prefixed with "Legacy". This CL also contains some minor fixes to the new congestion controller code. Bug: webrtc:8415 Change-Id: I5e7474d42f17fcbfef402e26f638846fa3424695 Reviewed-on: https://webrtc-review.googlesource.com/55381 Reviewed-by: Philip Eliasson <philipel@webrtc.org> Reviewed-by: Björn Terelius <terelius@webrtc.org> Commit-Queue: Sebastian Jansson <srte@webrtc.org> Cr-Commit-Position: refs/heads/master@{#22229}
221 lines
7.5 KiB
C++
221 lines
7.5 KiB
C++
/*
|
|
* Copyright (c) 2016 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 "modules/congestion_controller/probe_bitrate_estimator.h"
|
|
|
|
#include <vector>
|
|
#include <utility>
|
|
|
|
#include "modules/remote_bitrate_estimator/aimd_rate_control.h"
|
|
#include "test/gmock.h"
|
|
#include "test/gtest.h"
|
|
|
|
namespace webrtc {
|
|
|
|
namespace {
|
|
constexpr int kInvalidBitrate = -1;
|
|
constexpr int kDefaultMinProbes = 5;
|
|
constexpr int kDefaultMinBytes = 5000;
|
|
constexpr float kTargetUtilizationFraction = 0.95f;
|
|
} // anonymous namespace
|
|
|
|
class LegacyTestProbeBitrateEstimator : public ::testing::Test {
|
|
public:
|
|
LegacyTestProbeBitrateEstimator() : probe_bitrate_estimator_(nullptr) {}
|
|
|
|
// TODO(philipel): Use PacedPacketInfo when ProbeBitrateEstimator is rewritten
|
|
// to use that information.
|
|
void AddPacketFeedback(int probe_cluster_id,
|
|
size_t size_bytes,
|
|
int64_t send_time_ms,
|
|
int64_t arrival_time_ms,
|
|
int min_probes = kDefaultMinProbes,
|
|
int min_bytes = kDefaultMinBytes) {
|
|
PacedPacketInfo pacing_info(probe_cluster_id, min_probes, min_bytes);
|
|
PacketFeedback packet_feedback(arrival_time_ms, send_time_ms, 0, size_bytes,
|
|
pacing_info);
|
|
measured_bps_ =
|
|
probe_bitrate_estimator_.HandleProbeAndEstimateBitrate(packet_feedback);
|
|
}
|
|
|
|
protected:
|
|
int measured_bps_ = kInvalidBitrate;
|
|
ProbeBitrateEstimator probe_bitrate_estimator_;
|
|
};
|
|
|
|
TEST_F(LegacyTestProbeBitrateEstimator, OneCluster) {
|
|
AddPacketFeedback(0, 1000, 0, 10);
|
|
AddPacketFeedback(0, 1000, 10, 20);
|
|
AddPacketFeedback(0, 1000, 20, 30);
|
|
AddPacketFeedback(0, 1000, 30, 40);
|
|
|
|
EXPECT_NEAR(measured_bps_, 800000, 10);
|
|
}
|
|
|
|
TEST_F(LegacyTestProbeBitrateEstimator, OneClusterTooFewProbes) {
|
|
AddPacketFeedback(0, 2000, 0, 10);
|
|
AddPacketFeedback(0, 2000, 10, 20);
|
|
AddPacketFeedback(0, 2000, 20, 30);
|
|
|
|
EXPECT_EQ(kInvalidBitrate, measured_bps_);
|
|
}
|
|
|
|
TEST_F(LegacyTestProbeBitrateEstimator, OneClusterTooFewBytes) {
|
|
const int kMinBytes = 6000;
|
|
AddPacketFeedback(0, 800, 0, 10, kDefaultMinProbes, kMinBytes);
|
|
AddPacketFeedback(0, 800, 10, 20, kDefaultMinProbes, kMinBytes);
|
|
AddPacketFeedback(0, 800, 20, 30, kDefaultMinProbes, kMinBytes);
|
|
AddPacketFeedback(0, 800, 30, 40, kDefaultMinProbes, kMinBytes);
|
|
AddPacketFeedback(0, 800, 40, 50, kDefaultMinProbes, kMinBytes);
|
|
|
|
EXPECT_EQ(kInvalidBitrate, measured_bps_);
|
|
}
|
|
|
|
TEST_F(LegacyTestProbeBitrateEstimator, SmallCluster) {
|
|
const int kMinBytes = 1000;
|
|
AddPacketFeedback(0, 150, 0, 10, kDefaultMinProbes, kMinBytes);
|
|
AddPacketFeedback(0, 150, 10, 20, kDefaultMinProbes, kMinBytes);
|
|
AddPacketFeedback(0, 150, 20, 30, kDefaultMinProbes, kMinBytes);
|
|
AddPacketFeedback(0, 150, 30, 40, kDefaultMinProbes, kMinBytes);
|
|
AddPacketFeedback(0, 150, 40, 50, kDefaultMinProbes, kMinBytes);
|
|
AddPacketFeedback(0, 150, 50, 60, kDefaultMinProbes, kMinBytes);
|
|
EXPECT_NEAR(measured_bps_, 120000, 10);
|
|
}
|
|
|
|
TEST_F(LegacyTestProbeBitrateEstimator, LargeCluster) {
|
|
const int kMinProbes = 30;
|
|
const int kMinBytes = 312500;
|
|
|
|
int64_t send_time = 0;
|
|
int64_t receive_time = 5;
|
|
for (int i = 0; i < 25; ++i) {
|
|
AddPacketFeedback(0, 12500, send_time, receive_time, kMinProbes, kMinBytes);
|
|
++send_time;
|
|
++receive_time;
|
|
}
|
|
EXPECT_NEAR(measured_bps_, 100000000, 10);
|
|
}
|
|
|
|
TEST_F(LegacyTestProbeBitrateEstimator, FastReceive) {
|
|
AddPacketFeedback(0, 1000, 0, 15);
|
|
AddPacketFeedback(0, 1000, 10, 30);
|
|
AddPacketFeedback(0, 1000, 20, 35);
|
|
AddPacketFeedback(0, 1000, 30, 40);
|
|
|
|
EXPECT_NEAR(measured_bps_, 800000, 10);
|
|
}
|
|
|
|
TEST_F(LegacyTestProbeBitrateEstimator, TooFastReceive) {
|
|
AddPacketFeedback(0, 1000, 0, 19);
|
|
AddPacketFeedback(0, 1000, 10, 22);
|
|
AddPacketFeedback(0, 1000, 20, 25);
|
|
AddPacketFeedback(0, 1000, 40, 27);
|
|
|
|
EXPECT_EQ(measured_bps_, kInvalidBitrate);
|
|
}
|
|
|
|
TEST_F(LegacyTestProbeBitrateEstimator, SlowReceive) {
|
|
AddPacketFeedback(0, 1000, 0, 10);
|
|
AddPacketFeedback(0, 1000, 10, 40);
|
|
AddPacketFeedback(0, 1000, 20, 70);
|
|
AddPacketFeedback(0, 1000, 30, 85);
|
|
// Expected send rate = 800 kbps, expected receive rate = 320 kbps.
|
|
|
|
EXPECT_NEAR(measured_bps_, kTargetUtilizationFraction * 320000, 10);
|
|
}
|
|
|
|
TEST_F(LegacyTestProbeBitrateEstimator, BurstReceive) {
|
|
AddPacketFeedback(0, 1000, 0, 50);
|
|
AddPacketFeedback(0, 1000, 10, 50);
|
|
AddPacketFeedback(0, 1000, 20, 50);
|
|
AddPacketFeedback(0, 1000, 40, 50);
|
|
|
|
EXPECT_EQ(measured_bps_, kInvalidBitrate);
|
|
}
|
|
|
|
TEST_F(LegacyTestProbeBitrateEstimator, MultipleClusters) {
|
|
AddPacketFeedback(0, 1000, 0, 10);
|
|
AddPacketFeedback(0, 1000, 10, 20);
|
|
AddPacketFeedback(0, 1000, 20, 30);
|
|
AddPacketFeedback(0, 1000, 40, 60);
|
|
// Expected send rate = 600 kbps, expected receive rate = 480 kbps.
|
|
EXPECT_NEAR(measured_bps_, kTargetUtilizationFraction * 480000, 10);
|
|
|
|
AddPacketFeedback(0, 1000, 50, 60);
|
|
// Expected send rate = 640 kbps, expected receive rate = 640 kbps.
|
|
EXPECT_NEAR(measured_bps_, 640000, 10);
|
|
|
|
AddPacketFeedback(1, 1000, 60, 70);
|
|
AddPacketFeedback(1, 1000, 65, 77);
|
|
AddPacketFeedback(1, 1000, 70, 84);
|
|
AddPacketFeedback(1, 1000, 75, 90);
|
|
// Expected send rate = 1600 kbps, expected receive rate = 1200 kbps.
|
|
|
|
EXPECT_NEAR(measured_bps_, kTargetUtilizationFraction * 1200000, 10);
|
|
}
|
|
|
|
TEST_F(LegacyTestProbeBitrateEstimator, IgnoreOldClusters) {
|
|
AddPacketFeedback(0, 1000, 0, 10);
|
|
AddPacketFeedback(0, 1000, 10, 20);
|
|
AddPacketFeedback(0, 1000, 20, 30);
|
|
|
|
AddPacketFeedback(1, 1000, 60, 70);
|
|
AddPacketFeedback(1, 1000, 65, 77);
|
|
AddPacketFeedback(1, 1000, 70, 84);
|
|
AddPacketFeedback(1, 1000, 75, 90);
|
|
// Expected send rate = 1600 kbps, expected receive rate = 1200 kbps.
|
|
|
|
EXPECT_NEAR(measured_bps_, kTargetUtilizationFraction * 1200000, 10);
|
|
|
|
// Coming in 6s later
|
|
AddPacketFeedback(0, 1000, 40 + 6000, 60 + 6000);
|
|
|
|
EXPECT_EQ(measured_bps_, kInvalidBitrate);
|
|
}
|
|
|
|
TEST_F(LegacyTestProbeBitrateEstimator, IgnoreSizeLastSendPacket) {
|
|
AddPacketFeedback(0, 1000, 0, 10);
|
|
AddPacketFeedback(0, 1000, 10, 20);
|
|
AddPacketFeedback(0, 1000, 20, 30);
|
|
AddPacketFeedback(0, 1000, 30, 40);
|
|
AddPacketFeedback(0, 1500, 40, 50);
|
|
// Expected send rate = 800 kbps, expected receive rate = 900 kbps.
|
|
|
|
EXPECT_NEAR(measured_bps_, 800000, 10);
|
|
}
|
|
|
|
TEST_F(LegacyTestProbeBitrateEstimator, IgnoreSizeFirstReceivePacket) {
|
|
AddPacketFeedback(0, 1500, 0, 10);
|
|
AddPacketFeedback(0, 1000, 10, 20);
|
|
AddPacketFeedback(0, 1000, 20, 30);
|
|
AddPacketFeedback(0, 1000, 30, 40);
|
|
// Expected send rate = 933 kbps, expected receive rate = 800 kbps.
|
|
|
|
EXPECT_NEAR(measured_bps_, kTargetUtilizationFraction * 800000, 10);
|
|
}
|
|
|
|
TEST_F(LegacyTestProbeBitrateEstimator, NoLastEstimatedBitrateBps) {
|
|
EXPECT_FALSE(probe_bitrate_estimator_.FetchAndResetLastEstimatedBitrateBps());
|
|
}
|
|
|
|
TEST_F(LegacyTestProbeBitrateEstimator, FetchLastEstimatedBitrateBps) {
|
|
AddPacketFeedback(0, 1000, 0, 10);
|
|
AddPacketFeedback(0, 1000, 10, 20);
|
|
AddPacketFeedback(0, 1000, 20, 30);
|
|
AddPacketFeedback(0, 1000, 30, 40);
|
|
|
|
auto estimated_bitrate_bps =
|
|
probe_bitrate_estimator_.FetchAndResetLastEstimatedBitrateBps();
|
|
EXPECT_TRUE(estimated_bitrate_bps);
|
|
EXPECT_NEAR(*estimated_bitrate_bps, 800000, 10);
|
|
EXPECT_FALSE(probe_bitrate_estimator_.FetchAndResetLastEstimatedBitrateBps());
|
|
}
|
|
|
|
} // namespace webrtc
|