mirror of
https://github.com/mollyim/webrtc.git
synced 2025-05-20 09:07:52 +01:00
ea55b0872f
Bug: webrtc:10809 Change-Id: Ib2f7ce9d14ee2ce808ab745ff20baf2761811cfb Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/155367 Reviewed-by: Sebastian Jansson <srte@webrtc.org> Reviewed-by: Oskar Sundbom <ossu@webrtc.org> Commit-Queue: Erik Språng <sprang@webrtc.org> Cr-Commit-Position: refs/heads/master@{#29378}
119 lines
3.8 KiB
C++
119 lines
3.8 KiB
C++
/*
|
|
* Copyright (c) 2012 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/paced_sender.h"
|
|
|
|
#include <list>
|
|
#include <memory>
|
|
#include <string>
|
|
#include <utility>
|
|
#include <vector>
|
|
|
|
#include "modules/pacing/packet_router.h"
|
|
#include "modules/utility/include/mock/mock_process_thread.h"
|
|
#include "system_wrappers/include/clock.h"
|
|
#include "system_wrappers/include/field_trial.h"
|
|
#include "test/field_trial.h"
|
|
#include "test/gmock.h"
|
|
#include "test/gtest.h"
|
|
|
|
using ::testing::_;
|
|
using ::testing::Return;
|
|
using ::testing::SaveArg;
|
|
|
|
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 = 234;
|
|
} // namespace
|
|
|
|
namespace webrtc {
|
|
namespace test {
|
|
|
|
|
|
// Mock callback implementing the raw api.
|
|
class MockCallback : public PacketRouter {
|
|
public:
|
|
MOCK_METHOD2(SendPacket,
|
|
void(std::unique_ptr<RtpPacketToSend> packet,
|
|
const PacedPacketInfo& cluster_info));
|
|
MOCK_METHOD1(
|
|
GeneratePadding,
|
|
std::vector<std::unique_ptr<RtpPacketToSend>>(size_t target_size_bytes));
|
|
};
|
|
|
|
std::unique_ptr<RtpPacketToSend> BuildRtpPacket(RtpPacketToSend::Type type) {
|
|
auto packet = std::make_unique<RtpPacketToSend>(nullptr);
|
|
packet->set_packet_type(type);
|
|
switch (type) {
|
|
case RtpPacketToSend::Type::kAudio:
|
|
packet->SetSsrc(kAudioSsrc);
|
|
break;
|
|
case RtpPacketToSend::Type::kVideo:
|
|
packet->SetSsrc(kVideoSsrc);
|
|
break;
|
|
case RtpPacketToSend::Type::kRetransmission:
|
|
case RtpPacketToSend::Type::kPadding:
|
|
packet->SetSsrc(kVideoRtxSsrc);
|
|
break;
|
|
case RtpPacketToSend::Type::kForwardErrorCorrection:
|
|
packet->SetSsrc(kFlexFecSsrc);
|
|
break;
|
|
}
|
|
|
|
packet->SetPayloadSize(kDefaultPacketSize);
|
|
return packet;
|
|
}
|
|
|
|
TEST(PacedSenderTest, PacesPackets) {
|
|
SimulatedClock clock(0);
|
|
MockCallback callback;
|
|
MockProcessThread process_thread;
|
|
Module* paced_module = nullptr;
|
|
EXPECT_CALL(process_thread, RegisterModule(_, _))
|
|
.WillOnce(SaveArg<0>(&paced_module));
|
|
PacedSender pacer(&clock, &callback, nullptr, nullptr, &process_thread);
|
|
EXPECT_CALL(process_thread, DeRegisterModule(paced_module)).Times(1);
|
|
|
|
// Insert a number of packets, covering one second.
|
|
static constexpr size_t kPacketsToSend = 42;
|
|
pacer.SetPacingRates(DataRate::bps(kDefaultPacketSize * 8 * kPacketsToSend),
|
|
DataRate::Zero());
|
|
std::vector<std::unique_ptr<RtpPacketToSend>> packets;
|
|
for (size_t i = 0; i < kPacketsToSend; ++i) {
|
|
packets.emplace_back(BuildRtpPacket(RtpPacketToSend::Type::kVideo));
|
|
}
|
|
pacer.EnqueuePackets(std::move(packets));
|
|
|
|
// Expect all of them to be sent.
|
|
size_t packets_sent = 0;
|
|
clock.AdvanceTimeMilliseconds(paced_module->TimeUntilNextProcess());
|
|
EXPECT_CALL(callback, SendPacket)
|
|
.WillRepeatedly(
|
|
[&](std::unique_ptr<RtpPacketToSend> packet,
|
|
const PacedPacketInfo& cluster_info) { ++packets_sent; });
|
|
|
|
const Timestamp start_time = clock.CurrentTime();
|
|
|
|
while (packets_sent < kPacketsToSend) {
|
|
clock.AdvanceTimeMilliseconds(paced_module->TimeUntilNextProcess());
|
|
paced_module->Process();
|
|
}
|
|
|
|
// Packets should be sent over a period of close to 1s. Expect a little lower
|
|
// than this since initial probing is a bit quicker.
|
|
TimeDelta duration = clock.CurrentTime() - start_time;
|
|
EXPECT_GT(duration, TimeDelta::ms(900));
|
|
}
|
|
|
|
} // namespace test
|
|
} // namespace webrtc
|