mirror of
https://github.com/mollyim/webrtc.git
synced 2025-05-16 07:10:38 +01:00

Bug: webrtc:12338 Change-Id: Id3b5081cf73be31829d75d7ef34942c2259053da Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/227096 Reviewed-by: Harald Alvestrand <hta@webrtc.org> Commit-Queue: Artem Titov <titovartem@webrtc.org> Cr-Commit-Position: refs/heads/master@{#34674}
593 lines
24 KiB
C++
593 lines
24 KiB
C++
/*
|
|
* Copyright (c) 2019 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/pacing/task_queue_paced_sender.h"
|
|
|
|
#include <algorithm>
|
|
#include <list>
|
|
#include <memory>
|
|
#include <string>
|
|
#include <utility>
|
|
#include <vector>
|
|
|
|
#include "api/transport/network_types.h"
|
|
#include "modules/pacing/packet_router.h"
|
|
#include "modules/utility/include/mock/mock_process_thread.h"
|
|
#include "test/field_trial.h"
|
|
#include "test/gmock.h"
|
|
#include "test/gtest.h"
|
|
#include "test/time_controller/simulated_time_controller.h"
|
|
|
|
using ::testing::_;
|
|
using ::testing::AtLeast;
|
|
using ::testing::Return;
|
|
using ::testing::SaveArg;
|
|
|
|
namespace webrtc {
|
|
namespace {
|
|
constexpr uint32_t kAudioSsrc = 12345;
|
|
constexpr uint32_t kVideoSsrc = 234565;
|
|
constexpr uint32_t kVideoRtxSsrc = 34567;
|
|
constexpr uint32_t kFlexFecSsrc = 45678;
|
|
constexpr size_t kDefaultPacketSize = 1234;
|
|
|
|
class MockPacketRouter : public PacketRouter {
|
|
public:
|
|
MOCK_METHOD(void,
|
|
SendPacket,
|
|
(std::unique_ptr<RtpPacketToSend> packet,
|
|
const PacedPacketInfo& cluster_info),
|
|
(override));
|
|
MOCK_METHOD(std::vector<std::unique_ptr<RtpPacketToSend>>,
|
|
FetchFec,
|
|
(),
|
|
(override));
|
|
MOCK_METHOD(std::vector<std::unique_ptr<RtpPacketToSend>>,
|
|
GeneratePadding,
|
|
(DataSize target_size),
|
|
(override));
|
|
};
|
|
|
|
class StatsUpdateObserver {
|
|
public:
|
|
StatsUpdateObserver() = default;
|
|
virtual ~StatsUpdateObserver() = default;
|
|
|
|
virtual void OnStatsUpdated() = 0;
|
|
};
|
|
|
|
class TaskQueuePacedSenderForTest : public TaskQueuePacedSender {
|
|
public:
|
|
TaskQueuePacedSenderForTest(Clock* clock,
|
|
PacketRouter* packet_router,
|
|
RtcEventLog* event_log,
|
|
const WebRtcKeyValueConfig* field_trials,
|
|
TaskQueueFactory* task_queue_factory,
|
|
TimeDelta hold_back_window)
|
|
: TaskQueuePacedSender(clock,
|
|
packet_router,
|
|
event_log,
|
|
field_trials,
|
|
task_queue_factory,
|
|
hold_back_window) {}
|
|
|
|
void OnStatsUpdated(const Stats& stats) override {
|
|
++num_stats_updates_;
|
|
TaskQueuePacedSender::OnStatsUpdated(stats);
|
|
}
|
|
|
|
size_t num_stats_updates_ = 0;
|
|
};
|
|
|
|
std::vector<std::unique_ptr<RtpPacketToSend>> GeneratePadding(
|
|
DataSize target_size) {
|
|
// 224 bytes is the max padding size for plain padding packets generated by
|
|
// RTPSender::GeneratePadding().
|
|
const DataSize kMaxPaddingPacketSize = DataSize::Bytes(224);
|
|
DataSize padding_generated = DataSize::Zero();
|
|
std::vector<std::unique_ptr<RtpPacketToSend>> padding_packets;
|
|
while (padding_generated < target_size) {
|
|
DataSize packet_size =
|
|
std::min(target_size - padding_generated, kMaxPaddingPacketSize);
|
|
padding_generated += packet_size;
|
|
auto padding_packet =
|
|
std::make_unique<RtpPacketToSend>(/*extensions=*/nullptr);
|
|
padding_packet->set_packet_type(RtpPacketMediaType::kPadding);
|
|
padding_packet->SetPadding(packet_size.bytes());
|
|
padding_packets.push_back(std::move(padding_packet));
|
|
}
|
|
return padding_packets;
|
|
}
|
|
|
|
} // namespace
|
|
|
|
namespace test {
|
|
|
|
std::unique_ptr<RtpPacketToSend> BuildRtpPacket(RtpPacketMediaType type) {
|
|
auto packet = std::make_unique<RtpPacketToSend>(nullptr);
|
|
packet->set_packet_type(type);
|
|
switch (type) {
|
|
case RtpPacketMediaType::kAudio:
|
|
packet->SetSsrc(kAudioSsrc);
|
|
break;
|
|
case RtpPacketMediaType::kVideo:
|
|
packet->SetSsrc(kVideoSsrc);
|
|
break;
|
|
case RtpPacketMediaType::kRetransmission:
|
|
case RtpPacketMediaType::kPadding:
|
|
packet->SetSsrc(kVideoRtxSsrc);
|
|
break;
|
|
case RtpPacketMediaType::kForwardErrorCorrection:
|
|
packet->SetSsrc(kFlexFecSsrc);
|
|
break;
|
|
}
|
|
|
|
packet->SetPayloadSize(kDefaultPacketSize);
|
|
return packet;
|
|
}
|
|
|
|
std::vector<std::unique_ptr<RtpPacketToSend>> GeneratePackets(
|
|
RtpPacketMediaType type,
|
|
size_t num_packets) {
|
|
std::vector<std::unique_ptr<RtpPacketToSend>> packets;
|
|
for (size_t i = 0; i < num_packets; ++i) {
|
|
packets.push_back(BuildRtpPacket(type));
|
|
}
|
|
return packets;
|
|
}
|
|
|
|
TEST(TaskQueuePacedSenderTest, PacesPackets) {
|
|
GlobalSimulatedTimeController time_controller(Timestamp::Millis(1234));
|
|
MockPacketRouter packet_router;
|
|
TaskQueuePacedSenderForTest pacer(
|
|
time_controller.GetClock(), &packet_router,
|
|
/*event_log=*/nullptr,
|
|
/*field_trials=*/nullptr, time_controller.GetTaskQueueFactory(),
|
|
PacingController::kMinSleepTime);
|
|
|
|
// Insert a number of packets, covering one second.
|
|
static constexpr size_t kPacketsToSend = 42;
|
|
pacer.SetPacingRates(
|
|
DataRate::BitsPerSec(kDefaultPacketSize * 8 * kPacketsToSend),
|
|
DataRate::Zero());
|
|
pacer.EnsureStarted();
|
|
pacer.EnqueuePackets(
|
|
GeneratePackets(RtpPacketMediaType::kVideo, kPacketsToSend));
|
|
|
|
// Expect all of them to be sent.
|
|
size_t packets_sent = 0;
|
|
Timestamp end_time = Timestamp::PlusInfinity();
|
|
EXPECT_CALL(packet_router, SendPacket)
|
|
.WillRepeatedly([&](std::unique_ptr<RtpPacketToSend> packet,
|
|
const PacedPacketInfo& cluster_info) {
|
|
++packets_sent;
|
|
if (packets_sent == kPacketsToSend) {
|
|
end_time = time_controller.GetClock()->CurrentTime();
|
|
}
|
|
});
|
|
|
|
const Timestamp start_time = time_controller.GetClock()->CurrentTime();
|
|
|
|
// Packets should be sent over a period of close to 1s. Expect a little
|
|
// lower than this since initial probing is a bit quicker.
|
|
time_controller.AdvanceTime(TimeDelta::Seconds(1));
|
|
EXPECT_EQ(packets_sent, kPacketsToSend);
|
|
ASSERT_TRUE(end_time.IsFinite());
|
|
EXPECT_NEAR((end_time - start_time).ms<double>(), 1000.0, 50.0);
|
|
}
|
|
|
|
TEST(TaskQueuePacedSenderTest, ReschedulesProcessOnRateChange) {
|
|
GlobalSimulatedTimeController time_controller(Timestamp::Millis(1234));
|
|
MockPacketRouter packet_router;
|
|
TaskQueuePacedSenderForTest pacer(
|
|
time_controller.GetClock(), &packet_router,
|
|
/*event_log=*/nullptr,
|
|
/*field_trials=*/nullptr, time_controller.GetTaskQueueFactory(),
|
|
PacingController::kMinSleepTime);
|
|
|
|
// Insert a number of packets to be sent 200ms apart.
|
|
const size_t kPacketsPerSecond = 5;
|
|
const DataRate kPacingRate =
|
|
DataRate::BitsPerSec(kDefaultPacketSize * 8 * kPacketsPerSecond);
|
|
pacer.SetPacingRates(kPacingRate, DataRate::Zero());
|
|
pacer.EnsureStarted();
|
|
|
|
// Send some initial packets to be rid of any probes.
|
|
EXPECT_CALL(packet_router, SendPacket).Times(kPacketsPerSecond);
|
|
pacer.EnqueuePackets(
|
|
GeneratePackets(RtpPacketMediaType::kVideo, kPacketsPerSecond));
|
|
time_controller.AdvanceTime(TimeDelta::Seconds(1));
|
|
|
|
// Insert three packets, and record send time of each of them.
|
|
// After the second packet is sent, double the send rate so we can
|
|
// check the third packets is sent after half the wait time.
|
|
Timestamp first_packet_time = Timestamp::MinusInfinity();
|
|
Timestamp second_packet_time = Timestamp::MinusInfinity();
|
|
Timestamp third_packet_time = Timestamp::MinusInfinity();
|
|
|
|
EXPECT_CALL(packet_router, SendPacket)
|
|
.Times(3)
|
|
.WillRepeatedly([&](std::unique_ptr<RtpPacketToSend> packet,
|
|
const PacedPacketInfo& cluster_info) {
|
|
if (first_packet_time.IsInfinite()) {
|
|
first_packet_time = time_controller.GetClock()->CurrentTime();
|
|
} else if (second_packet_time.IsInfinite()) {
|
|
second_packet_time = time_controller.GetClock()->CurrentTime();
|
|
pacer.SetPacingRates(2 * kPacingRate, DataRate::Zero());
|
|
} else {
|
|
third_packet_time = time_controller.GetClock()->CurrentTime();
|
|
}
|
|
});
|
|
|
|
pacer.EnqueuePackets(GeneratePackets(RtpPacketMediaType::kVideo, 3));
|
|
time_controller.AdvanceTime(TimeDelta::Millis(500));
|
|
ASSERT_TRUE(third_packet_time.IsFinite());
|
|
EXPECT_NEAR((second_packet_time - first_packet_time).ms<double>(), 200.0,
|
|
1.0);
|
|
EXPECT_NEAR((third_packet_time - second_packet_time).ms<double>(), 100.0,
|
|
1.0);
|
|
}
|
|
|
|
TEST(TaskQueuePacedSenderTest, SendsAudioImmediately) {
|
|
GlobalSimulatedTimeController time_controller(Timestamp::Millis(1234));
|
|
MockPacketRouter packet_router;
|
|
TaskQueuePacedSenderForTest pacer(
|
|
time_controller.GetClock(), &packet_router,
|
|
/*event_log=*/nullptr,
|
|
/*field_trials=*/nullptr, time_controller.GetTaskQueueFactory(),
|
|
PacingController::kMinSleepTime);
|
|
|
|
const DataRate kPacingDataRate = DataRate::KilobitsPerSec(125);
|
|
const DataSize kPacketSize = DataSize::Bytes(kDefaultPacketSize);
|
|
const TimeDelta kPacketPacingTime = kPacketSize / kPacingDataRate;
|
|
|
|
pacer.SetPacingRates(kPacingDataRate, DataRate::Zero());
|
|
pacer.EnsureStarted();
|
|
|
|
// Add some initial video packets, only one should be sent.
|
|
EXPECT_CALL(packet_router, SendPacket);
|
|
pacer.EnqueuePackets(GeneratePackets(RtpPacketMediaType::kVideo, 10));
|
|
time_controller.AdvanceTime(TimeDelta::Zero());
|
|
::testing::Mock::VerifyAndClearExpectations(&packet_router);
|
|
|
|
// Advance time, but still before next packet should be sent.
|
|
time_controller.AdvanceTime(kPacketPacingTime / 2);
|
|
|
|
// Insert an audio packet, it should be sent immediately.
|
|
EXPECT_CALL(packet_router, SendPacket);
|
|
pacer.EnqueuePackets(GeneratePackets(RtpPacketMediaType::kAudio, 1));
|
|
time_controller.AdvanceTime(TimeDelta::Zero());
|
|
::testing::Mock::VerifyAndClearExpectations(&packet_router);
|
|
}
|
|
|
|
TEST(TaskQueuePacedSenderTest, SleepsDuringCoalscingWindow) {
|
|
const TimeDelta kCoalescingWindow = TimeDelta::Millis(5);
|
|
GlobalSimulatedTimeController time_controller(Timestamp::Millis(1234));
|
|
MockPacketRouter packet_router;
|
|
TaskQueuePacedSenderForTest pacer(
|
|
time_controller.GetClock(), &packet_router,
|
|
/*event_log=*/nullptr,
|
|
/*field_trials=*/nullptr, time_controller.GetTaskQueueFactory(),
|
|
kCoalescingWindow);
|
|
|
|
// Set rates so one packet adds one ms of buffer level.
|
|
const DataSize kPacketSize = DataSize::Bytes(kDefaultPacketSize);
|
|
const TimeDelta kPacketPacingTime = TimeDelta::Millis(1);
|
|
const DataRate kPacingDataRate = kPacketSize / kPacketPacingTime;
|
|
|
|
pacer.SetPacingRates(kPacingDataRate, DataRate::Zero());
|
|
pacer.EnsureStarted();
|
|
|
|
// Add 10 packets. The first should be sent immediately since the buffers
|
|
// are clear.
|
|
EXPECT_CALL(packet_router, SendPacket);
|
|
pacer.EnqueuePackets(GeneratePackets(RtpPacketMediaType::kVideo, 10));
|
|
time_controller.AdvanceTime(TimeDelta::Zero());
|
|
::testing::Mock::VerifyAndClearExpectations(&packet_router);
|
|
|
|
// Advance time to 1ms before the coalescing window ends. No packets should
|
|
// be sent.
|
|
EXPECT_CALL(packet_router, SendPacket).Times(0);
|
|
time_controller.AdvanceTime(kCoalescingWindow - TimeDelta::Millis(1));
|
|
|
|
// Advance time to where coalescing window ends. All packets that should
|
|
// have been sent up til now will be sent.
|
|
EXPECT_CALL(packet_router, SendPacket).Times(5);
|
|
time_controller.AdvanceTime(TimeDelta::Millis(1));
|
|
::testing::Mock::VerifyAndClearExpectations(&packet_router);
|
|
}
|
|
|
|
TEST(TaskQueuePacedSenderTest, ProbingOverridesCoalescingWindow) {
|
|
const TimeDelta kCoalescingWindow = TimeDelta::Millis(5);
|
|
GlobalSimulatedTimeController time_controller(Timestamp::Millis(1234));
|
|
MockPacketRouter packet_router;
|
|
TaskQueuePacedSenderForTest pacer(
|
|
time_controller.GetClock(), &packet_router,
|
|
/*event_log=*/nullptr,
|
|
/*field_trials=*/nullptr, time_controller.GetTaskQueueFactory(),
|
|
kCoalescingWindow);
|
|
|
|
// Set rates so one packet adds one ms of buffer level.
|
|
const DataSize kPacketSize = DataSize::Bytes(kDefaultPacketSize);
|
|
const TimeDelta kPacketPacingTime = TimeDelta::Millis(1);
|
|
const DataRate kPacingDataRate = kPacketSize / kPacketPacingTime;
|
|
|
|
pacer.SetPacingRates(kPacingDataRate, DataRate::Zero());
|
|
pacer.EnsureStarted();
|
|
|
|
// Add 10 packets. The first should be sent immediately since the buffers
|
|
// are clear. This will also trigger the probe to start.
|
|
EXPECT_CALL(packet_router, SendPacket).Times(AtLeast(1));
|
|
pacer.CreateProbeCluster(kPacingDataRate * 2, 17);
|
|
pacer.EnqueuePackets(GeneratePackets(RtpPacketMediaType::kVideo, 10));
|
|
time_controller.AdvanceTime(TimeDelta::Zero());
|
|
::testing::Mock::VerifyAndClearExpectations(&packet_router);
|
|
|
|
// Advance time to 1ms before the coalescing window ends. Packets should be
|
|
// flying.
|
|
EXPECT_CALL(packet_router, SendPacket).Times(AtLeast(1));
|
|
time_controller.AdvanceTime(kCoalescingWindow - TimeDelta::Millis(1));
|
|
}
|
|
|
|
TEST(TaskQueuePacedSenderTest, RespectedMinTimeBetweenStatsUpdates) {
|
|
const TimeDelta kCoalescingWindow = TimeDelta::Millis(5);
|
|
GlobalSimulatedTimeController time_controller(Timestamp::Millis(1234));
|
|
MockPacketRouter packet_router;
|
|
TaskQueuePacedSenderForTest pacer(
|
|
time_controller.GetClock(), &packet_router,
|
|
/*event_log=*/nullptr,
|
|
/*field_trials=*/nullptr, time_controller.GetTaskQueueFactory(),
|
|
kCoalescingWindow);
|
|
const DataRate kPacingDataRate = DataRate::KilobitsPerSec(300);
|
|
pacer.SetPacingRates(kPacingDataRate, DataRate::Zero());
|
|
pacer.EnsureStarted();
|
|
|
|
const TimeDelta kMinTimeBetweenStatsUpdates = TimeDelta::Millis(1);
|
|
|
|
// Nothing inserted, no stats updates yet.
|
|
EXPECT_EQ(pacer.num_stats_updates_, 0u);
|
|
|
|
// Insert one packet, stats should be updated.
|
|
pacer.EnqueuePackets(GeneratePackets(RtpPacketMediaType::kVideo, 1));
|
|
time_controller.AdvanceTime(TimeDelta::Zero());
|
|
EXPECT_EQ(pacer.num_stats_updates_, 1u);
|
|
|
|
// Advance time half of the min stats update interval, and trigger a
|
|
// refresh - stats should not be updated yet.
|
|
time_controller.AdvanceTime(kMinTimeBetweenStatsUpdates / 2);
|
|
pacer.EnqueuePackets({});
|
|
time_controller.AdvanceTime(TimeDelta::Zero());
|
|
EXPECT_EQ(pacer.num_stats_updates_, 1u);
|
|
|
|
// Advance time the next half, now stats update is triggered.
|
|
time_controller.AdvanceTime(kMinTimeBetweenStatsUpdates / 2);
|
|
pacer.EnqueuePackets({});
|
|
time_controller.AdvanceTime(TimeDelta::Zero());
|
|
EXPECT_EQ(pacer.num_stats_updates_, 2u);
|
|
}
|
|
|
|
TEST(TaskQueuePacedSenderTest, ThrottlesStatsUpdates) {
|
|
const TimeDelta kCoalescingWindow = TimeDelta::Millis(5);
|
|
GlobalSimulatedTimeController time_controller(Timestamp::Millis(1234));
|
|
MockPacketRouter packet_router;
|
|
TaskQueuePacedSenderForTest pacer(
|
|
time_controller.GetClock(), &packet_router,
|
|
/*event_log=*/nullptr,
|
|
/*field_trials=*/nullptr, time_controller.GetTaskQueueFactory(),
|
|
kCoalescingWindow);
|
|
|
|
// Set rates so one packet adds 10ms of buffer level.
|
|
const DataSize kPacketSize = DataSize::Bytes(kDefaultPacketSize);
|
|
const TimeDelta kPacketPacingTime = TimeDelta::Millis(10);
|
|
const DataRate kPacingDataRate = kPacketSize / kPacketPacingTime;
|
|
const TimeDelta kMinTimeBetweenStatsUpdates = TimeDelta::Millis(1);
|
|
const TimeDelta kMaxTimeBetweenStatsUpdates = TimeDelta::Millis(33);
|
|
|
|
// Nothing inserted, no stats updates yet.
|
|
size_t num_expected_stats_updates = 0;
|
|
EXPECT_EQ(pacer.num_stats_updates_, num_expected_stats_updates);
|
|
pacer.SetPacingRates(kPacingDataRate, DataRate::Zero());
|
|
pacer.EnsureStarted();
|
|
time_controller.AdvanceTime(kMinTimeBetweenStatsUpdates);
|
|
// Updating pacing rates refreshes stats.
|
|
EXPECT_EQ(pacer.num_stats_updates_, ++num_expected_stats_updates);
|
|
|
|
// Record time when we insert first packet, this triggers the scheduled
|
|
// stats updating.
|
|
Clock* const clock = time_controller.GetClock();
|
|
const Timestamp start_time = clock->CurrentTime();
|
|
|
|
while (clock->CurrentTime() - start_time <=
|
|
kMaxTimeBetweenStatsUpdates - kPacketPacingTime) {
|
|
// Enqueue packet, expect stats update.
|
|
pacer.EnqueuePackets(GeneratePackets(RtpPacketMediaType::kVideo, 1));
|
|
time_controller.AdvanceTime(TimeDelta::Zero());
|
|
EXPECT_EQ(pacer.num_stats_updates_, ++num_expected_stats_updates);
|
|
|
|
// Advance time to halfway through pacing time, expect another stats
|
|
// update.
|
|
time_controller.AdvanceTime(kPacketPacingTime / 2);
|
|
pacer.EnqueuePackets({});
|
|
time_controller.AdvanceTime(TimeDelta::Zero());
|
|
EXPECT_EQ(pacer.num_stats_updates_, ++num_expected_stats_updates);
|
|
|
|
// Advance time the rest of the way.
|
|
time_controller.AdvanceTime(kPacketPacingTime / 2);
|
|
}
|
|
|
|
// At this point, the pace queue is drained so there is no more intersting
|
|
// update to be made - but there is still as schduled task that should run
|
|
// `kMaxTimeBetweenStatsUpdates` after the first update.
|
|
time_controller.AdvanceTime(start_time + kMaxTimeBetweenStatsUpdates -
|
|
clock->CurrentTime());
|
|
EXPECT_EQ(pacer.num_stats_updates_, ++num_expected_stats_updates);
|
|
|
|
// Advance time a significant time - don't expect any more calls as stats
|
|
// updating does not happen when queue is drained.
|
|
time_controller.AdvanceTime(TimeDelta::Millis(400));
|
|
EXPECT_EQ(pacer.num_stats_updates_, num_expected_stats_updates);
|
|
}
|
|
|
|
TEST(TaskQueuePacedSenderTest, SchedulesProbeAtSetTime) {
|
|
ScopedFieldTrials trials("WebRTC-Bwe-ProbingBehavior/min_probe_delta:1ms/");
|
|
GlobalSimulatedTimeController time_controller(Timestamp::Millis(1234));
|
|
MockPacketRouter packet_router;
|
|
TaskQueuePacedSenderForTest pacer(
|
|
time_controller.GetClock(), &packet_router,
|
|
/*event_log=*/nullptr,
|
|
/*field_trials=*/nullptr, time_controller.GetTaskQueueFactory(),
|
|
PacingController::kMinSleepTime);
|
|
|
|
// Set rates so one packet adds 4ms of buffer level.
|
|
const DataSize kPacketSize = DataSize::Bytes(kDefaultPacketSize);
|
|
const TimeDelta kPacketPacingTime = TimeDelta::Millis(4);
|
|
const DataRate kPacingDataRate = kPacketSize / kPacketPacingTime;
|
|
pacer.SetPacingRates(kPacingDataRate, /*padding_rate=*/DataRate::Zero());
|
|
pacer.EnsureStarted();
|
|
EXPECT_CALL(packet_router, FetchFec).WillRepeatedly([]() {
|
|
return std::vector<std::unique_ptr<RtpPacketToSend>>();
|
|
});
|
|
EXPECT_CALL(packet_router, GeneratePadding(_))
|
|
.WillRepeatedly(
|
|
[](DataSize target_size) { return GeneratePadding(target_size); });
|
|
|
|
// Enqueue two packets, only the first is sent immediately and the next
|
|
// will be scheduled for sending in 4ms.
|
|
pacer.EnqueuePackets(GeneratePackets(RtpPacketMediaType::kVideo, 2));
|
|
const int kNotAProbe = PacedPacketInfo::kNotAProbe;
|
|
EXPECT_CALL(
|
|
packet_router,
|
|
SendPacket(_, ::testing::Field(&PacedPacketInfo::probe_cluster_id,
|
|
kNotAProbe)));
|
|
// Advance to less than 3ms before next packet send time.
|
|
time_controller.AdvanceTime(TimeDelta::Micros(1001));
|
|
|
|
// Trigger a probe at 4x the current pacing rate and insert the number of
|
|
// packets the probe needs.
|
|
const DataRate kProbeRate = 2 * kPacingDataRate;
|
|
const int kProbeClusterId = 1;
|
|
pacer.CreateProbeCluster(kProbeRate, kProbeClusterId);
|
|
|
|
// Expected size for each probe in a cluster is twice the expected bits
|
|
// sent during min_probe_delta.
|
|
// Expect one additional call since probe always starts with a small
|
|
const TimeDelta kProbeTimeDelta = TimeDelta::Millis(2);
|
|
const DataSize kProbeSize = kProbeRate * kProbeTimeDelta;
|
|
const size_t kNumPacketsInProbe =
|
|
(kProbeSize + kPacketSize - DataSize::Bytes(1)) / kPacketSize;
|
|
EXPECT_CALL(
|
|
packet_router,
|
|
SendPacket(_, ::testing::Field(&PacedPacketInfo::probe_cluster_id,
|
|
kProbeClusterId)))
|
|
.Times(kNumPacketsInProbe + 1);
|
|
|
|
pacer.EnqueuePackets(
|
|
GeneratePackets(RtpPacketMediaType::kVideo, kNumPacketsInProbe));
|
|
time_controller.AdvanceTime(TimeDelta::Zero());
|
|
|
|
// The pacer should have scheduled the next probe to be sent in
|
|
// kProbeTimeDelta. That there was existing scheduled call less than
|
|
// PacingController::kMinSleepTime before this should not matter.
|
|
|
|
EXPECT_CALL(
|
|
packet_router,
|
|
SendPacket(_, ::testing::Field(&PacedPacketInfo::probe_cluster_id,
|
|
kProbeClusterId)))
|
|
.Times(AtLeast(1));
|
|
time_controller.AdvanceTime(TimeDelta::Millis(2));
|
|
}
|
|
|
|
TEST(TaskQueuePacedSenderTest, NoMinSleepTimeWhenProbing) {
|
|
// Set min_probe_delta to be less than kMinSleepTime (1ms).
|
|
const TimeDelta kMinProbeDelta = TimeDelta::Micros(100);
|
|
ScopedFieldTrials trials(
|
|
"WebRTC-Bwe-ProbingBehavior/min_probe_delta:100us/");
|
|
GlobalSimulatedTimeController time_controller(Timestamp::Millis(1234));
|
|
MockPacketRouter packet_router;
|
|
TaskQueuePacedSenderForTest pacer(
|
|
time_controller.GetClock(), &packet_router,
|
|
/*event_log=*/nullptr,
|
|
/*field_trials=*/nullptr, time_controller.GetTaskQueueFactory(),
|
|
PacingController::kMinSleepTime);
|
|
|
|
// Set rates so one packet adds 4ms of buffer level.
|
|
const DataSize kPacketSize = DataSize::Bytes(kDefaultPacketSize);
|
|
const TimeDelta kPacketPacingTime = TimeDelta::Millis(4);
|
|
const DataRate kPacingDataRate = kPacketSize / kPacketPacingTime;
|
|
pacer.SetPacingRates(kPacingDataRate, /*padding_rate=*/DataRate::Zero());
|
|
pacer.EnsureStarted();
|
|
EXPECT_CALL(packet_router, FetchFec).WillRepeatedly([]() {
|
|
return std::vector<std::unique_ptr<RtpPacketToSend>>();
|
|
});
|
|
EXPECT_CALL(packet_router, GeneratePadding)
|
|
.WillRepeatedly(
|
|
[](DataSize target_size) { return GeneratePadding(target_size); });
|
|
|
|
// Set a high probe rate.
|
|
const int kProbeClusterId = 1;
|
|
DataRate kProbingRate = kPacingDataRate * 10;
|
|
pacer.CreateProbeCluster(kProbingRate, kProbeClusterId);
|
|
|
|
// Advance time less than PacingController::kMinSleepTime, probing packets
|
|
// for the first millisecond should be sent immediately. Min delta between
|
|
// probes is 2x 100us, meaning 4 times per ms we will get least one call to
|
|
// SendPacket().
|
|
DataSize data_sent = DataSize::Zero();
|
|
EXPECT_CALL(
|
|
packet_router,
|
|
SendPacket(_, ::testing::Field(&PacedPacketInfo::probe_cluster_id,
|
|
kProbeClusterId)))
|
|
.Times(AtLeast(4))
|
|
.WillRepeatedly([&](std::unique_ptr<RtpPacketToSend> packet,
|
|
const PacedPacketInfo&) {
|
|
data_sent +=
|
|
DataSize::Bytes(packet->payload_size() + packet->padding_size());
|
|
});
|
|
|
|
// Add one packet to kickstart probing, the rest will be padding packets.
|
|
pacer.EnqueuePackets(GeneratePackets(RtpPacketMediaType::kVideo, 1));
|
|
time_controller.AdvanceTime(kMinProbeDelta);
|
|
|
|
// Verify the amount of probing data sent.
|
|
// Probe always starts with a small (1 byte) padding packet that's not
|
|
// counted into the probe rate here.
|
|
EXPECT_EQ(data_sent,
|
|
kProbingRate * TimeDelta::Millis(1) + DataSize::Bytes(1));
|
|
}
|
|
|
|
TEST(TaskQueuePacedSenderTest, NoStatsUpdatesBeforeStart) {
|
|
const TimeDelta kCoalescingWindow = TimeDelta::Millis(5);
|
|
GlobalSimulatedTimeController time_controller(Timestamp::Millis(1234));
|
|
MockPacketRouter packet_router;
|
|
TaskQueuePacedSenderForTest pacer(
|
|
time_controller.GetClock(), &packet_router,
|
|
/*event_log=*/nullptr,
|
|
/*field_trials=*/nullptr, time_controller.GetTaskQueueFactory(),
|
|
kCoalescingWindow);
|
|
const DataRate kPacingDataRate = DataRate::KilobitsPerSec(300);
|
|
pacer.SetPacingRates(kPacingDataRate, DataRate::Zero());
|
|
|
|
const TimeDelta kMinTimeBetweenStatsUpdates = TimeDelta::Millis(1);
|
|
|
|
// Nothing inserted, no stats updates yet.
|
|
EXPECT_EQ(pacer.num_stats_updates_, 0u);
|
|
|
|
// Insert one packet, stats should not be updated.
|
|
pacer.EnqueuePackets(GeneratePackets(RtpPacketMediaType::kVideo, 1));
|
|
time_controller.AdvanceTime(TimeDelta::Zero());
|
|
EXPECT_EQ(pacer.num_stats_updates_, 0u);
|
|
|
|
// Advance time of the min stats update interval, and trigger a
|
|
// refresh - stats should not be updated still.
|
|
time_controller.AdvanceTime(kMinTimeBetweenStatsUpdates);
|
|
EXPECT_EQ(pacer.num_stats_updates_, 0u);
|
|
}
|
|
} // namespace test
|
|
} // namespace webrtc
|