|  | /* | 
|  | *  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 |