| 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 5c1654e6d03a0207d04d35a8825380bcbde04a82..006b9f2bf48c2ed803e41a5e0e6ee615975dbd5b 100644
|
| --- a/webrtc/modules/pacing/packet_router_unittest.cc
|
| +++ b/webrtc/modules/pacing/packet_router_unittest.cc
|
| @@ -48,12 +48,12 @@
|
| EXPECT_CALL(rtp_1, SendingMedia()).Times(1).WillOnce(Return(true));
|
| EXPECT_CALL(rtp_1, SSRC()).Times(1).WillOnce(Return(kSsrc1));
|
| EXPECT_CALL(rtp_1, TimeToSendPacket(kSsrc1, sequence_number, timestamp,
|
| - retransmission, 1))
|
| + retransmission))
|
| .Times(1)
|
| .WillOnce(Return(true));
|
| - EXPECT_CALL(rtp_2, TimeToSendPacket(_, _, _, _, _)).Times(0);
|
| + EXPECT_CALL(rtp_2, TimeToSendPacket(_, _, _, _)).Times(0);
|
| EXPECT_TRUE(packet_router_->TimeToSendPacket(kSsrc1, sequence_number,
|
| - timestamp, retransmission, 1));
|
| + timestamp, retransmission, -1));
|
|
|
| // Send on the second module by letting rtp_2 be sending, but not rtp_1.
|
| ++sequence_number;
|
| @@ -63,31 +63,31 @@
|
| EXPECT_CALL(rtp_1, SendingMedia()).Times(1).WillOnce(Return(false));
|
| EXPECT_CALL(rtp_2, SendingMedia()).Times(1).WillOnce(Return(true));
|
| EXPECT_CALL(rtp_2, SSRC()).Times(1).WillOnce(Return(kSsrc2));
|
| - EXPECT_CALL(rtp_1, TimeToSendPacket(_, _, _, _, _)).Times(0);
|
| + EXPECT_CALL(rtp_1, TimeToSendPacket(_, _, _, _)).Times(0);
|
| EXPECT_CALL(rtp_2, TimeToSendPacket(kSsrc2, sequence_number, timestamp,
|
| - retransmission, 2))
|
| + retransmission))
|
| .Times(1)
|
| .WillOnce(Return(true));
|
| EXPECT_TRUE(packet_router_->TimeToSendPacket(kSsrc2, sequence_number,
|
| - timestamp, retransmission, 2));
|
| + timestamp, retransmission, -1));
|
|
|
| // No module is sending, hence no packet should be sent.
|
| EXPECT_CALL(rtp_1, SendingMedia()).Times(1).WillOnce(Return(false));
|
| - EXPECT_CALL(rtp_1, TimeToSendPacket(_, _, _, _, _)).Times(0);
|
| + EXPECT_CALL(rtp_1, TimeToSendPacket(_, _, _, _)).Times(0);
|
| EXPECT_CALL(rtp_2, SendingMedia()).Times(1).WillOnce(Return(false));
|
| - EXPECT_CALL(rtp_2, TimeToSendPacket(_, _, _, _, _)).Times(0);
|
| + EXPECT_CALL(rtp_2, TimeToSendPacket(_, _, _, _)).Times(0);
|
| EXPECT_TRUE(packet_router_->TimeToSendPacket(kSsrc1, sequence_number,
|
| - timestamp, retransmission, 1));
|
| + timestamp, retransmission, -1));
|
|
|
| // Add a packet with incorrect ssrc and test it's dropped in the router.
|
| EXPECT_CALL(rtp_1, SendingMedia()).Times(1).WillOnce(Return(true));
|
| EXPECT_CALL(rtp_1, SSRC()).Times(1).WillOnce(Return(kSsrc1));
|
| EXPECT_CALL(rtp_2, SendingMedia()).Times(1).WillOnce(Return(true));
|
| EXPECT_CALL(rtp_2, SSRC()).Times(1).WillOnce(Return(kSsrc2));
|
| - EXPECT_CALL(rtp_1, TimeToSendPacket(_, _, _, _, _)).Times(0);
|
| - EXPECT_CALL(rtp_2, TimeToSendPacket(_, _, _, _, _)).Times(0);
|
| + EXPECT_CALL(rtp_1, TimeToSendPacket(_, _, _, _)).Times(0);
|
| + EXPECT_CALL(rtp_2, TimeToSendPacket(_, _, _, _)).Times(0);
|
| EXPECT_TRUE(packet_router_->TimeToSendPacket(kSsrc1 + kSsrc2, sequence_number,
|
| - timestamp, retransmission, 1));
|
| + timestamp, retransmission, -1));
|
|
|
| packet_router_->RemoveRtpModule(&rtp_1);
|
|
|
| @@ -95,10 +95,9 @@
|
| // it is dropped as expected by not expecting any calls to rtp_1.
|
| EXPECT_CALL(rtp_2, SendingMedia()).Times(1).WillOnce(Return(true));
|
| EXPECT_CALL(rtp_2, SSRC()).Times(1).WillOnce(Return(kSsrc2));
|
| - EXPECT_CALL(rtp_2, TimeToSendPacket(_, _, _, _, _)).Times(0);
|
| + EXPECT_CALL(rtp_2, TimeToSendPacket(_, _, _, _)).Times(0);
|
| EXPECT_TRUE(packet_router_->TimeToSendPacket(kSsrc1, sequence_number,
|
| - timestamp, retransmission,
|
| - PacketInfo::kNotAProbe));
|
| + timestamp, retransmission, -1));
|
|
|
| packet_router_->RemoveRtpModule(&rtp_2);
|
| }
|
| @@ -119,45 +118,42 @@
|
| const size_t requested_padding_bytes = 1000;
|
| const size_t sent_padding_bytes = 890;
|
| EXPECT_CALL(rtp_1, SendingMedia()).Times(1).WillOnce(Return(true));
|
| - EXPECT_CALL(rtp_1, TimeToSendPadding(requested_padding_bytes, 111))
|
| + EXPECT_CALL(rtp_1, TimeToSendPadding(requested_padding_bytes))
|
| .Times(1)
|
| .WillOnce(Return(sent_padding_bytes));
|
| EXPECT_CALL(rtp_2, SendingMedia()).Times(1).WillOnce(Return(true));
|
| - EXPECT_CALL(rtp_2, TimeToSendPadding(
|
| - requested_padding_bytes - sent_padding_bytes, 111))
|
| + EXPECT_CALL(rtp_2,
|
| + TimeToSendPadding(requested_padding_bytes - sent_padding_bytes))
|
| .Times(1)
|
| .WillOnce(Return(requested_padding_bytes - sent_padding_bytes));
|
| EXPECT_EQ(requested_padding_bytes,
|
| - packet_router_->TimeToSendPadding(requested_padding_bytes, 111));
|
| + packet_router_->TimeToSendPadding(requested_padding_bytes));
|
|
|
| // Let only the second module be sending and verify the padding request is
|
| // routed there.
|
| EXPECT_CALL(rtp_1, SendingMedia()).Times(1).WillOnce(Return(false));
|
| - EXPECT_CALL(rtp_1, TimeToSendPadding(requested_padding_bytes, _)).Times(0);
|
| - EXPECT_CALL(rtp_2, SendingMedia()).Times(1).WillOnce(Return(true));
|
| - EXPECT_CALL(rtp_2, TimeToSendPadding(_, _))
|
| + EXPECT_CALL(rtp_1, TimeToSendPadding(requested_padding_bytes)).Times(0);
|
| + EXPECT_CALL(rtp_2, SendingMedia()).Times(1).WillOnce(Return(true));
|
| + EXPECT_CALL(rtp_2, TimeToSendPadding(_))
|
| .Times(1)
|
| .WillOnce(Return(sent_padding_bytes));
|
| EXPECT_EQ(sent_padding_bytes,
|
| - packet_router_->TimeToSendPadding(requested_padding_bytes,
|
| - PacketInfo::kNotAProbe));
|
| + packet_router_->TimeToSendPadding(requested_padding_bytes));
|
|
|
| // No sending module at all.
|
| EXPECT_CALL(rtp_1, SendingMedia()).Times(1).WillOnce(Return(false));
|
| - EXPECT_CALL(rtp_1, TimeToSendPadding(requested_padding_bytes, _)).Times(0);
|
| + EXPECT_CALL(rtp_1, TimeToSendPadding(requested_padding_bytes)).Times(0);
|
| EXPECT_CALL(rtp_2, SendingMedia()).Times(1).WillOnce(Return(false));
|
| - EXPECT_CALL(rtp_2, TimeToSendPadding(_, _)).Times(0);
|
| - EXPECT_EQ(0u, packet_router_->TimeToSendPadding(requested_padding_bytes,
|
| - PacketInfo::kNotAProbe));
|
| + EXPECT_CALL(rtp_2, TimeToSendPadding(_)).Times(0);
|
| + EXPECT_EQ(0u, packet_router_->TimeToSendPadding(requested_padding_bytes));
|
|
|
| packet_router_->RemoveRtpModule(&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.
|
| EXPECT_CALL(rtp_2, SendingMedia()).Times(1).WillOnce(Return(true));
|
| - EXPECT_CALL(rtp_2, TimeToSendPadding(requested_padding_bytes, _)).Times(1);
|
| - EXPECT_EQ(0u, packet_router_->TimeToSendPadding(requested_padding_bytes,
|
| - PacketInfo::kNotAProbe));
|
| + EXPECT_CALL(rtp_2, TimeToSendPadding(requested_padding_bytes)).Times(1);
|
| + EXPECT_EQ(0u, packet_router_->TimeToSendPadding(requested_padding_bytes));
|
|
|
| packet_router_->RemoveRtpModule(&rtp_2);
|
| }
|
| @@ -170,12 +166,11 @@
|
| EXPECT_CALL(rtp, SendingMedia()).WillRepeatedly(Return(false));
|
|
|
| // Verify that TimeToSendPacket does not end up in a receiver.
|
| - EXPECT_CALL(rtp, TimeToSendPacket(_, _, _, _, _)).Times(0);
|
| - EXPECT_TRUE(packet_router_->TimeToSendPacket(kSsrc, 1, 1, false,
|
| - PacketInfo::kNotAProbe));
|
| + EXPECT_CALL(rtp, TimeToSendPacket(_, _, _, _)).Times(0);
|
| + EXPECT_TRUE(packet_router_->TimeToSendPacket(kSsrc, 1, 1, false, -1));
|
| // Verify that TimeToSendPadding does not end up in a receiver.
|
| - EXPECT_CALL(rtp, TimeToSendPadding(_, _)).Times(0);
|
| - EXPECT_EQ(0u, packet_router_->TimeToSendPadding(200, PacketInfo::kNotAProbe));
|
| + EXPECT_CALL(rtp, TimeToSendPadding(_)).Times(0);
|
| + EXPECT_EQ(0u, packet_router_->TimeToSendPadding(200));
|
|
|
| packet_router_->RemoveRtpModule(&rtp);
|
| }
|
|
|