233c4ba4fd
The ProbingCalculator class calculates and validates the results from probing attempts. BUG=webrtc:5859 Review-Url: https://codereview.webrtc.org/2121183002 Cr-Commit-Position: refs/heads/master@{#13589}
128 lines
3.7 KiB
C++
128 lines
3.7 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 "webrtc/modules/congestion_controller/probe_bitrate_estimator.h"
|
|
|
|
#include <vector>
|
|
#include <utility>
|
|
|
|
#include "testing/gmock/include/gmock/gmock.h"
|
|
#include "testing/gtest/include/gtest/gtest.h"
|
|
#include "webrtc/modules/remote_bitrate_estimator/aimd_rate_control.h"
|
|
|
|
namespace webrtc {
|
|
|
|
class TestProbeBitrateEstimator : public ::testing::Test {
|
|
public:
|
|
TestProbeBitrateEstimator() : probe_bitrate_estimator_() {}
|
|
|
|
void AddPacketFeedback(int probe_cluster_id,
|
|
size_t size,
|
|
int64_t send_time_ms,
|
|
int64_t arrival_time_ms) {
|
|
PacketInfo info(arrival_time_ms, send_time_ms, 0, size, probe_cluster_id);
|
|
ProbingResult res = probe_bitrate_estimator_.PacketFeedback(info);
|
|
if (res.bps != ProbingResult::kNoEstimate)
|
|
results_.emplace_back(res.bps, res.timestamp);
|
|
}
|
|
|
|
void CheckResult(size_t index, int bps, int max_diff, int64_t timestamp) {
|
|
ASSERT_GT(results_.size(), index);
|
|
EXPECT_NEAR(results_[index].first, bps, max_diff);
|
|
EXPECT_EQ(results_[index].second, timestamp);
|
|
}
|
|
|
|
protected:
|
|
std::vector<std::pair<int, int64_t>> results_;
|
|
ProbeBitrateEstimator probe_bitrate_estimator_;
|
|
};
|
|
|
|
TEST_F(TestProbeBitrateEstimator, OneCluster) {
|
|
AddPacketFeedback(0, 1000, 0, 10);
|
|
AddPacketFeedback(0, 1000, 10, 20);
|
|
AddPacketFeedback(0, 1000, 20, 30);
|
|
AddPacketFeedback(0, 1000, 40, 50);
|
|
|
|
CheckResult(0, 100000, 10, 50);
|
|
}
|
|
|
|
TEST_F(TestProbeBitrateEstimator, FastReceive) {
|
|
AddPacketFeedback(0, 1000, 0, 15);
|
|
AddPacketFeedback(0, 1000, 10, 30);
|
|
AddPacketFeedback(0, 1000, 20, 40);
|
|
AddPacketFeedback(0, 1000, 40, 50);
|
|
|
|
CheckResult(0, 100000, 10, 50);
|
|
}
|
|
|
|
TEST_F(TestProbeBitrateEstimator, TooFastReceive) {
|
|
AddPacketFeedback(0, 1000, 0, 19);
|
|
AddPacketFeedback(0, 1000, 10, 30);
|
|
AddPacketFeedback(0, 1000, 20, 40);
|
|
AddPacketFeedback(0, 1000, 40, 50);
|
|
|
|
EXPECT_TRUE(results_.empty());
|
|
}
|
|
|
|
TEST_F(TestProbeBitrateEstimator, SlowReceive) {
|
|
AddPacketFeedback(0, 1000, 0, 10);
|
|
AddPacketFeedback(0, 1000, 10, 40);
|
|
AddPacketFeedback(0, 1000, 20, 70);
|
|
AddPacketFeedback(0, 1000, 40, 110);
|
|
|
|
CheckResult(0, 40000, 10, 110);
|
|
}
|
|
|
|
TEST_F(TestProbeBitrateEstimator, BurstReceive) {
|
|
AddPacketFeedback(0, 1000, 0, 50);
|
|
AddPacketFeedback(0, 1000, 10, 50);
|
|
AddPacketFeedback(0, 1000, 20, 50);
|
|
AddPacketFeedback(0, 1000, 40, 50);
|
|
|
|
EXPECT_TRUE(results_.empty());
|
|
}
|
|
|
|
TEST_F(TestProbeBitrateEstimator, MultipleClusters) {
|
|
AddPacketFeedback(0, 1000, 0, 10);
|
|
AddPacketFeedback(0, 1000, 10, 20);
|
|
AddPacketFeedback(0, 1000, 20, 30);
|
|
AddPacketFeedback(0, 1000, 40, 60);
|
|
AddPacketFeedback(0, 1000, 50, 60);
|
|
|
|
CheckResult(0, 80000, 10, 60);
|
|
CheckResult(1, 100000, 10, 60);
|
|
|
|
AddPacketFeedback(1, 1000, 60, 70);
|
|
AddPacketFeedback(1, 1000, 65, 77);
|
|
AddPacketFeedback(1, 1000, 70, 84);
|
|
AddPacketFeedback(1, 1000, 75, 90);
|
|
|
|
CheckResult(2, 200000, 10, 90);
|
|
}
|
|
|
|
TEST_F(TestProbeBitrateEstimator, OldProbe) {
|
|
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);
|
|
|
|
CheckResult(0, 200000, 10, 90);
|
|
|
|
AddPacketFeedback(0, 1000, 40, 60);
|
|
|
|
EXPECT_EQ(1ul, results_.size());
|
|
}
|
|
|
|
} // namespace webrtc
|