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}
135 lines
4.7 KiB
C++
135 lines
4.7 KiB
C++
/*
|
|
* Copyright (c) 2017 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/acknowledged_bitrate_estimator.h"
|
|
|
|
#include <utility>
|
|
|
|
#include "modules/rtp_rtcp/include/rtp_rtcp_defines.h"
|
|
#include "rtc_base/fakeclock.h"
|
|
#include "rtc_base/ptr_util.h"
|
|
#include "test/gmock.h"
|
|
#include "test/gtest.h"
|
|
|
|
using testing::_;
|
|
using testing::NiceMock;
|
|
using testing::InSequence;
|
|
using testing::Return;
|
|
|
|
namespace webrtc {
|
|
|
|
namespace {
|
|
|
|
constexpr int64_t kFirstArrivalTimeMs = 10;
|
|
constexpr int64_t kFirstSendTimeMs = 10;
|
|
constexpr uint16_t kSequenceNumber = 1;
|
|
constexpr size_t kPayloadSize = 10;
|
|
|
|
class MockBitrateEstimator : public BitrateEstimator {
|
|
public:
|
|
MOCK_METHOD2(Update, void(int64_t now_ms, int bytes));
|
|
MOCK_CONST_METHOD0(bitrate_bps, rtc::Optional<uint32_t>());
|
|
MOCK_METHOD0(ExpectFastRateChange, void());
|
|
};
|
|
|
|
struct AcknowledgedBitrateEstimatorTestStates {
|
|
std::unique_ptr<AcknowledgedBitrateEstimator> acknowledged_bitrate_estimator;
|
|
MockBitrateEstimator* mock_bitrate_estimator;
|
|
};
|
|
|
|
AcknowledgedBitrateEstimatorTestStates CreateTestStates() {
|
|
AcknowledgedBitrateEstimatorTestStates states;
|
|
auto mock_bitrate_estimator = rtc::MakeUnique<MockBitrateEstimator>();
|
|
states.mock_bitrate_estimator = mock_bitrate_estimator.get();
|
|
states.acknowledged_bitrate_estimator =
|
|
rtc::MakeUnique<AcknowledgedBitrateEstimator>(
|
|
std::move(mock_bitrate_estimator));
|
|
return states;
|
|
}
|
|
|
|
std::vector<PacketFeedback> CreateFeedbackVector() {
|
|
std::vector<PacketFeedback> packet_feedback_vector;
|
|
const PacedPacketInfo pacing_info;
|
|
packet_feedback_vector.push_back(
|
|
PacketFeedback(kFirstArrivalTimeMs, kFirstSendTimeMs, kSequenceNumber,
|
|
kPayloadSize, pacing_info));
|
|
packet_feedback_vector.push_back(
|
|
PacketFeedback(kFirstArrivalTimeMs + 10, kFirstSendTimeMs + 10,
|
|
kSequenceNumber, kPayloadSize + 10, pacing_info));
|
|
return packet_feedback_vector;
|
|
}
|
|
|
|
} // anonymous namespace
|
|
|
|
TEST(LegacyTestAcknowledgedBitrateEstimator,
|
|
DontAddPacketsWhichAreNotInSendHistory) {
|
|
auto states = CreateTestStates();
|
|
std::vector<PacketFeedback> packet_feedback_vector;
|
|
packet_feedback_vector.push_back(
|
|
PacketFeedback(kFirstArrivalTimeMs, kSequenceNumber));
|
|
EXPECT_CALL(*states.mock_bitrate_estimator, Update(_, _)).Times(0);
|
|
states.acknowledged_bitrate_estimator->IncomingPacketFeedbackVector(
|
|
packet_feedback_vector);
|
|
}
|
|
|
|
TEST(LegacyTestAcknowledgedBitrateEstimator, UpdateBandwith) {
|
|
auto states = CreateTestStates();
|
|
auto packet_feedback_vector = CreateFeedbackVector();
|
|
{
|
|
InSequence dummy;
|
|
EXPECT_CALL(
|
|
*states.mock_bitrate_estimator,
|
|
Update(packet_feedback_vector[0].arrival_time_ms,
|
|
static_cast<int>(packet_feedback_vector[0].payload_size)))
|
|
.Times(1);
|
|
EXPECT_CALL(
|
|
*states.mock_bitrate_estimator,
|
|
Update(packet_feedback_vector[1].arrival_time_ms,
|
|
static_cast<int>(packet_feedback_vector[1].payload_size)))
|
|
.Times(1);
|
|
}
|
|
states.acknowledged_bitrate_estimator->IncomingPacketFeedbackVector(
|
|
packet_feedback_vector);
|
|
}
|
|
|
|
TEST(LegacyTestAcknowledgedBitrateEstimator, ExpectFastRateChangeWhenLeftAlr) {
|
|
auto states = CreateTestStates();
|
|
auto packet_feedback_vector = CreateFeedbackVector();
|
|
{
|
|
InSequence dummy;
|
|
EXPECT_CALL(
|
|
*states.mock_bitrate_estimator,
|
|
Update(packet_feedback_vector[0].arrival_time_ms,
|
|
static_cast<int>(packet_feedback_vector[0].payload_size)))
|
|
.Times(1);
|
|
EXPECT_CALL(*states.mock_bitrate_estimator, ExpectFastRateChange())
|
|
.Times(1);
|
|
EXPECT_CALL(
|
|
*states.mock_bitrate_estimator,
|
|
Update(packet_feedback_vector[1].arrival_time_ms,
|
|
static_cast<int>(packet_feedback_vector[1].payload_size)))
|
|
.Times(1);
|
|
}
|
|
states.acknowledged_bitrate_estimator->SetAlrEndedTimeMs(kFirstArrivalTimeMs +
|
|
1);
|
|
states.acknowledged_bitrate_estimator->IncomingPacketFeedbackVector(
|
|
packet_feedback_vector);
|
|
}
|
|
|
|
TEST(LegacyTestAcknowledgedBitrateEstimator, ReturnBitrate) {
|
|
auto states = CreateTestStates();
|
|
rtc::Optional<uint32_t> return_value(42);
|
|
EXPECT_CALL(*states.mock_bitrate_estimator, bitrate_bps())
|
|
.Times(1)
|
|
.WillOnce(Return(return_value));
|
|
EXPECT_EQ(return_value, states.acknowledged_bitrate_estimator->bitrate_bps());
|
|
}
|
|
|
|
} // namespace webrtc*/
|