| Index: webrtc/modules/pacing/packet_router_unittest.cc
|
| diff --git a/webrtc/modules/pacing/packet_router_unittest.cc b/webrtc/modules/pacing/packet_router_unittest.cc
|
| index 31acf44b9b8c91418509e97675649a1aac025aeb..61237554c0ecf90e235bd547b8630c93ee339895 100644
|
| --- a/webrtc/modules/pacing/packet_router_unittest.cc
|
| +++ b/webrtc/modules/pacing/packet_router_unittest.cc
|
| @@ -16,6 +16,7 @@
|
| #include "webrtc/modules/pacing/packet_router.h"
|
| #include "webrtc/modules/rtp_rtcp/include/rtp_rtcp.h"
|
| #include "webrtc/modules/rtp_rtcp/mocks/mock_rtp_rtcp.h"
|
| +#include "webrtc/modules/rtp_rtcp/source/rtcp_packet/transport_feedback.h"
|
| #include "webrtc/base/scoped_ptr.h"
|
|
|
| using ::testing::_;
|
| @@ -35,8 +36,8 @@ class PacketRouterTest : public ::testing::Test {
|
| TEST_F(PacketRouterTest, TimeToSendPacket) {
|
| MockRtpRtcp rtp_1;
|
| MockRtpRtcp rtp_2;
|
| - packet_router_->AddRtpModule(&rtp_1);
|
| - packet_router_->AddRtpModule(&rtp_2);
|
| + packet_router_->AddRtpModule(&rtp_1, true);
|
| + packet_router_->AddRtpModule(&rtp_2, true);
|
|
|
| const uint16_t kSsrc1 = 1234;
|
| uint16_t sequence_number = 17;
|
| @@ -109,8 +110,8 @@ TEST_F(PacketRouterTest, TimeToSendPadding) {
|
| EXPECT_CALL(rtp_1, SSRC()).WillRepeatedly(Return(kSsrc1));
|
| MockRtpRtcp rtp_2;
|
| EXPECT_CALL(rtp_2, SSRC()).WillRepeatedly(Return(kSsrc2));
|
| - packet_router_->AddRtpModule(&rtp_1);
|
| - packet_router_->AddRtpModule(&rtp_2);
|
| + packet_router_->AddRtpModule(&rtp_1, true);
|
| + packet_router_->AddRtpModule(&rtp_2, true);
|
|
|
| // Default configuration, sending padding on all modules sending media,
|
| // ordered by SSRC.
|
| @@ -169,4 +170,19 @@ TEST_F(PacketRouterTest, AllocateSequenceNumbers) {
|
| EXPECT_EQ(static_cast<uint16_t>(expected_unwrapped_seq & 0xFFFF), seq);
|
| }
|
| }
|
| +
|
| +TEST_F(PacketRouterTest, SendFeedback) {
|
| + MockRtpRtcp rtp_1;
|
| + MockRtpRtcp rtp_2;
|
| + packet_router_->AddRtpModule(&rtp_1, false);
|
| + packet_router_->AddRtpModule(&rtp_2, true);
|
| +
|
| + rtcp::TransportFeedback feedback;
|
| + EXPECT_CALL(rtp_1, SendFeedbackPacket(_)).Times(1);
|
| + packet_router_->SendFeedback(&feedback);
|
| + packet_router_->RemoveRtpModule(&rtp_1);
|
| + EXPECT_CALL(rtp_2, SendFeedbackPacket(_)).Times(1);
|
| + packet_router_->SendFeedback(&feedback);
|
| + packet_router_->RemoveRtpModule(&rtp_2);
|
| +}
|
| } // namespace webrtc
|
|
|