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 967fdccae0d7b61fdf489c57b69a622fb84a0edc..be28bf1dbe65bb705036d96a25a5eb6a7058832b 100644 |
--- a/webrtc/modules/pacing/packet_router_unittest.cc |
+++ b/webrtc/modules/pacing/packet_router_unittest.cc |
@@ -39,8 +39,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_->AddSendRtpModule(&rtp_1); |
+ packet_router_->AddSendRtpModule(&rtp_2); |
const uint16_t kSsrc1 = 1234; |
uint16_t sequence_number = 17; |
@@ -98,7 +98,7 @@ TEST_F(PacketRouterTest, TimeToSendPacket) { |
kSsrc1 + kSsrc2, sequence_number, timestamp, retransmission, |
PacedPacketInfo(1, kProbeMinProbes, kProbeMinBytes))); |
- packet_router_->RemoveRtpModule(&rtp_1); |
+ packet_router_->RemoveSendRtpModule(&rtp_1); |
// rtp_1 has been removed, try sending a packet on that ssrc and make sure |
// it is dropped as expected by not expecting any calls to rtp_1. |
@@ -110,7 +110,7 @@ TEST_F(PacketRouterTest, TimeToSendPacket) { |
PacedPacketInfo(PacedPacketInfo::kNotAProbe, kProbeMinBytes, |
kProbeMinBytes))); |
- packet_router_->RemoveRtpModule(&rtp_2); |
+ packet_router_->RemoveSendRtpModule(&rtp_2); |
} |
TEST_F(PacketRouterTest, TimeToSendPadding) { |
@@ -124,8 +124,8 @@ TEST_F(PacketRouterTest, TimeToSendPadding) { |
// rtp_2 will be prioritized for padding. |
EXPECT_CALL(rtp_2, RtxSendStatus()).WillOnce(Return(kRtxRedundantPayloads)); |
EXPECT_CALL(rtp_2, SSRC()).WillRepeatedly(Return(kSsrc2)); |
- packet_router_->AddRtpModule(&rtp_1); |
- packet_router_->AddRtpModule(&rtp_2); |
+ packet_router_->AddSendRtpModule(&rtp_1); |
+ packet_router_->AddSendRtpModule(&rtp_2); |
// Default configuration, sending padding on all modules sending media, |
// ordered by priority (based on rtx mode). |
@@ -191,7 +191,7 @@ TEST_F(PacketRouterTest, TimeToSendPadding) { |
PacedPacketInfo(PacedPacketInfo::kNotAProbe, kProbeMinBytes, |
kProbeMinBytes))); |
- packet_router_->RemoveRtpModule(&rtp_1); |
+ packet_router_->RemoveSendRtpModule(&rtp_1); |
// rtp_1 has been removed, try sending padding and make sure rtp_1 isn't asked |
// to send by not expecting any calls. Instead verify rtp_2 is called. |
@@ -204,12 +204,12 @@ TEST_F(PacketRouterTest, TimeToSendPadding) { |
PacedPacketInfo(PacedPacketInfo::kNotAProbe, kProbeMinBytes, |
kProbeMinBytes))); |
- packet_router_->RemoveRtpModule(&rtp_2); |
+ packet_router_->RemoveSendRtpModule(&rtp_2); |
} |
TEST_F(PacketRouterTest, SenderOnlyFunctionsRespectSendingMedia) { |
MockRtpRtcp rtp; |
- packet_router_->AddRtpModule(&rtp); |
+ packet_router_->AddSendRtpModule(&rtp); |
static const uint16_t kSsrc = 1234; |
EXPECT_CALL(rtp, SSRC()).WillRepeatedly(Return(kSsrc)); |
EXPECT_CALL(rtp, SendingMedia()).WillRepeatedly(Return(false)); |
@@ -226,7 +226,7 @@ TEST_F(PacketRouterTest, SenderOnlyFunctionsRespectSendingMedia) { |
200, PacedPacketInfo(PacedPacketInfo::kNotAProbe, |
kProbeMinBytes, kProbeMinBytes))); |
- packet_router_->RemoveRtpModule(&rtp); |
+ packet_router_->RemoveSendRtpModule(&rtp); |
} |
TEST_F(PacketRouterTest, AllocateSequenceNumbers) { |
@@ -245,15 +245,15 @@ TEST_F(PacketRouterTest, AllocateSequenceNumbers) { |
TEST_F(PacketRouterTest, SendFeedback) { |
MockRtpRtcp rtp_1; |
MockRtpRtcp rtp_2; |
- packet_router_->AddRtpModule(&rtp_1); |
- packet_router_->AddRtpModule(&rtp_2); |
+ packet_router_->AddSendRtpModule(&rtp_1); |
+ packet_router_->AddReceiveRtpModule(&rtp_2); |
rtcp::TransportFeedback feedback; |
EXPECT_CALL(rtp_1, SendFeedbackPacket(_)).Times(1); |
packet_router_->SendFeedback(&feedback); |
- packet_router_->RemoveRtpModule(&rtp_1); |
+ packet_router_->RemoveSendRtpModule(&rtp_1); |
EXPECT_CALL(rtp_2, SendFeedbackPacket(_)).Times(1); |
packet_router_->SendFeedback(&feedback); |
- packet_router_->RemoveRtpModule(&rtp_2); |
+ packet_router_->RemoveReceiveRtpModule(&rtp_2); |
} |
} // namespace webrtc |