Chromium Code Reviews| 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 006b9f2bf48c2ed803e41a5e0e6ee615975dbd5b..a9cd71804595e33f1b6b9bdb33c93d3df93df03a 100644 |
| --- a/webrtc/modules/pacing/packet_router_unittest.cc |
| +++ b/webrtc/modules/pacing/packet_router_unittest.cc |
| @@ -48,12 +48,13 @@ TEST_F(PacketRouterTest, TimeToSendPacket) { |
| 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)) |
| + retransmission, PacketInfo::kNotAProbe)) |
|
danilchap
2016/05/31 13:06:49
may be instead of constant 'not a probe' use some
|
| .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, |
| + PacketInfo::kNotAProbe)); |
| // Send on the second module by letting rtp_2 be sending, but not rtp_1. |
| ++sequence_number; |
| @@ -63,31 +64,34 @@ TEST_F(PacketRouterTest, TimeToSendPacket) { |
| 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)) |
| + retransmission, PacketInfo::kNotAProbe)) |
| .Times(1) |
| .WillOnce(Return(true)); |
| EXPECT_TRUE(packet_router_->TimeToSendPacket(kSsrc2, sequence_number, |
| - timestamp, retransmission, -1)); |
| + timestamp, retransmission, |
| + PacketInfo::kNotAProbe)); |
| // 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, |
| + PacketInfo::kNotAProbe)); |
| // 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, |
| + PacketInfo::kNotAProbe)); |
| packet_router_->RemoveRtpModule(&rtp_1); |
| @@ -95,9 +99,10 @@ TEST_F(PacketRouterTest, TimeToSendPacket) { |
| // 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, -1)); |
| + timestamp, retransmission, |
| + PacketInfo::kNotAProbe)); |
| packet_router_->RemoveRtpModule(&rtp_2); |
| } |
| @@ -118,42 +123,46 @@ TEST_F(PacketRouterTest, TimeToSendPadding) { |
| 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)) |
| + 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)) |
| + 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)); |
| + packet_router_->TimeToSendPadding(requested_padding_bytes, |
| + PacketInfo::kNotAProbe)); |
| // 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_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_2, TimeToSendPadding(_, _)) |
| .Times(1) |
| .WillOnce(Return(sent_padding_bytes)); |
| EXPECT_EQ(sent_padding_bytes, |
| - packet_router_->TimeToSendPadding(requested_padding_bytes)); |
| + packet_router_->TimeToSendPadding(requested_padding_bytes, |
| + PacketInfo::kNotAProbe)); |
| // 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)); |
| + EXPECT_CALL(rtp_2, TimeToSendPadding(_, _)).Times(0); |
| + EXPECT_EQ(0u, packet_router_->TimeToSendPadding(requested_padding_bytes, |
| + PacketInfo::kNotAProbe)); |
| 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)); |
| + EXPECT_CALL(rtp_2, TimeToSendPadding(requested_padding_bytes, _)).Times(1); |
| + EXPECT_EQ(0u, packet_router_->TimeToSendPadding(requested_padding_bytes, |
| + PacketInfo::kNotAProbe)); |
| packet_router_->RemoveRtpModule(&rtp_2); |
| } |
| @@ -166,11 +175,12 @@ TEST_F(PacketRouterTest, SenderOnlyFunctionsRespectSendingMedia) { |
| 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, -1)); |
| + EXPECT_CALL(rtp, TimeToSendPacket(_, _, _, _, _)).Times(0); |
| + EXPECT_TRUE(packet_router_->TimeToSendPacket(kSsrc, 1, 1, false, |
| + PacketInfo::kNotAProbe)); |
| // Verify that TimeToSendPadding does not end up in a receiver. |
| - EXPECT_CALL(rtp, TimeToSendPadding(_)).Times(0); |
| - EXPECT_EQ(0u, packet_router_->TimeToSendPadding(200)); |
| + EXPECT_CALL(rtp, TimeToSendPadding(_, _)).Times(0); |
| + EXPECT_EQ(0u, packet_router_->TimeToSendPadding(200, PacketInfo::kNotAProbe)); |
| packet_router_->RemoveRtpModule(&rtp); |
| } |