Index: webrtc/modules/pacing/paced_sender_unittest.cc |
diff --git a/webrtc/modules/pacing/paced_sender_unittest.cc b/webrtc/modules/pacing/paced_sender_unittest.cc |
index 941c81335b6cf073c7a5bac225c84fecbeb5784d..bebfbf79b315363f17e3015d3192f34d127910da 100644 |
--- a/webrtc/modules/pacing/paced_sender_unittest.cc |
+++ b/webrtc/modules/pacing/paced_sender_unittest.cc |
@@ -27,11 +27,12 @@ static const float kPaceMultiplier = 1.5f; |
class MockPacedSenderCallback : public PacedSender::Callback { |
public: |
- MOCK_METHOD4(TimeToSendPacket, |
+ MOCK_METHOD5(TimeToSendPacket, |
bool(uint32_t ssrc, |
uint16_t sequence_number, |
int64_t capture_time_ms, |
- bool retransmission)); |
+ bool retransmission, |
+ int cluster_id)); |
MOCK_METHOD1(TimeToSendPadding, |
size_t(size_t bytes)); |
}; |
@@ -43,7 +44,8 @@ class PacedSenderPadding : public PacedSender::Callback { |
bool TimeToSendPacket(uint32_t ssrc, |
uint16_t sequence_number, |
int64_t capture_time_ms, |
- bool retransmission) { |
+ bool retransmission, |
+ int cluster_id) override { |
return true; |
} |
@@ -71,7 +73,8 @@ class PacedSenderProbing : public PacedSender::Callback { |
bool TimeToSendPacket(uint32_t ssrc, |
uint16_t sequence_number, |
int64_t capture_time_ms, |
- bool retransmission) { |
+ bool retransmission, |
+ int cluster_id) override { |
ExpectAndCountPacket(); |
return true; |
} |
@@ -127,8 +130,8 @@ class PacedSenderTest : public ::testing::Test { |
bool retransmission) { |
send_bucket_->InsertPacket(priority, ssrc, sequence_number, capture_time_ms, |
size, retransmission); |
- EXPECT_CALL(callback_, |
- TimeToSendPacket(ssrc, sequence_number, capture_time_ms, false)) |
+ EXPECT_CALL(callback_, TimeToSendPacket(ssrc, sequence_number, |
+ capture_time_ms, false, _)) |
.Times(1) |
.WillRepeatedly(Return(true)); |
} |
@@ -171,9 +174,8 @@ TEST_F(PacedSenderTest, QueuePacket) { |
EXPECT_EQ(1, send_bucket_->TimeUntilNextProcess()); |
clock_.AdvanceTimeMilliseconds(1); |
EXPECT_EQ(0, send_bucket_->TimeUntilNextProcess()); |
- EXPECT_CALL( |
- callback_, |
- TimeToSendPacket(ssrc, sequence_number++, queued_packet_timestamp, false)) |
+ EXPECT_CALL(callback_, TimeToSendPacket(ssrc, sequence_number++, |
+ queued_packet_timestamp, false, _)) |
.Times(1) |
.WillRepeatedly(Return(true)); |
send_bucket_->Process(); |
@@ -219,7 +221,7 @@ TEST_F(PacedSenderTest, PaceQueuedPackets) { |
for (int k = 0; k < 10; ++k) { |
EXPECT_EQ(5, send_bucket_->TimeUntilNextProcess()); |
clock_.AdvanceTimeMilliseconds(5); |
- EXPECT_CALL(callback_, TimeToSendPacket(ssrc, _, _, false)) |
+ EXPECT_CALL(callback_, TimeToSendPacket(ssrc, _, _, false, _)) |
.Times(3) |
.WillRepeatedly(Return(true)); |
EXPECT_EQ(0, send_bucket_->TimeUntilNextProcess()); |
@@ -286,7 +288,7 @@ TEST_F(PacedSenderTest, PaceQueuedPacketsWithDuplicates) { |
for (int i = 0; i < 3; ++i) { |
EXPECT_CALL(callback_, |
- TimeToSendPacket(ssrc, queued_sequence_number++, _, false)) |
+ TimeToSendPacket(ssrc, queued_sequence_number++, _, false, _)) |
.Times(1) |
.WillRepeatedly(Return(true)); |
} |
@@ -480,7 +482,7 @@ TEST_F(PacedSenderTest, Priority) { |
// Expect all high and normal priority to be sent out first. |
EXPECT_CALL(callback_, TimeToSendPadding(_)).Times(0); |
- EXPECT_CALL(callback_, TimeToSendPacket(ssrc, _, capture_time_ms, false)) |
+ EXPECT_CALL(callback_, TimeToSendPacket(ssrc, _, capture_time_ms, false, _)) |
.Times(4) |
.WillRepeatedly(Return(true)); |
@@ -490,8 +492,8 @@ TEST_F(PacedSenderTest, Priority) { |
send_bucket_->Process(); |
EXPECT_CALL(callback_, |
- TimeToSendPacket( |
- ssrc_low_priority, _, capture_time_ms_low_priority, false)) |
+ TimeToSendPacket(ssrc_low_priority, _, |
+ capture_time_ms_low_priority, false, _)) |
.Times(1) |
.WillRepeatedly(Return(true)); |
@@ -524,8 +526,8 @@ TEST_F(PacedSenderTest, HighPrioDoesntAffectBudget) { |
EXPECT_EQ(5, send_bucket_->TimeUntilNextProcess()); |
clock_.AdvanceTimeMilliseconds(5); |
send_bucket_->Process(); |
- EXPECT_CALL(callback_, |
- TimeToSendPacket(ssrc, sequence_number++, capture_time_ms, false)) |
+ EXPECT_CALL(callback_, TimeToSendPacket(ssrc, sequence_number++, |
+ capture_time_ms, false, _)) |
.Times(1); |
EXPECT_EQ(5, send_bucket_->TimeUntilNextProcess()); |
clock_.AdvanceTimeMilliseconds(5); |
@@ -583,7 +585,7 @@ TEST_F(PacedSenderTest, Pause) { |
// Expect no packet to come out while paused. |
EXPECT_CALL(callback_, TimeToSendPadding(_)).Times(0); |
- EXPECT_CALL(callback_, TimeToSendPacket(_, _, _, _)).Times(0); |
+ EXPECT_CALL(callback_, TimeToSendPacket(_, _, _, _, _)).Times(0); |
for (int i = 0; i < 10; ++i) { |
clock_.AdvanceTimeMilliseconds(5); |
@@ -592,10 +594,11 @@ TEST_F(PacedSenderTest, Pause) { |
} |
// Expect high prio packets to come out first followed by all packets in the |
// way they were added. |
- EXPECT_CALL(callback_, TimeToSendPacket(_, _, capture_time_ms, false)) |
+ EXPECT_CALL(callback_, TimeToSendPacket(_, _, capture_time_ms, false, _)) |
.Times(3) |
.WillRepeatedly(Return(true)); |
- EXPECT_CALL(callback_, TimeToSendPacket(_, _, second_capture_time_ms, false)) |
+ EXPECT_CALL(callback_, |
+ TimeToSendPacket(_, _, second_capture_time_ms, false, _)) |
.Times(1) |
.WillRepeatedly(Return(true)); |
send_bucket_->Resume(); |
@@ -624,8 +627,8 @@ TEST_F(PacedSenderTest, ResendPacket) { |
EXPECT_EQ(clock_.TimeInMilliseconds() - capture_time_ms, |
send_bucket_->QueueInMs()); |
// Fails to send first packet so only one call. |
- EXPECT_CALL(callback_, |
- TimeToSendPacket(ssrc, sequence_number, capture_time_ms, false)) |
+ EXPECT_CALL(callback_, TimeToSendPacket(ssrc, sequence_number, |
+ capture_time_ms, false, _)) |
.Times(1) |
.WillOnce(Return(false)); |
clock_.AdvanceTimeMilliseconds(10000); |
@@ -636,13 +639,12 @@ TEST_F(PacedSenderTest, ResendPacket) { |
send_bucket_->QueueInMs()); |
// Fails to send second packet. |
- EXPECT_CALL(callback_, |
- TimeToSendPacket(ssrc, sequence_number, capture_time_ms, false)) |
+ EXPECT_CALL(callback_, TimeToSendPacket(ssrc, sequence_number, |
+ capture_time_ms, false, _)) |
.Times(1) |
.WillOnce(Return(true)); |
- EXPECT_CALL( |
- callback_, |
- TimeToSendPacket(ssrc, sequence_number + 1, capture_time_ms + 1, false)) |
+ EXPECT_CALL(callback_, TimeToSendPacket(ssrc, sequence_number + 1, |
+ capture_time_ms + 1, false, _)) |
.Times(1) |
.WillOnce(Return(false)); |
clock_.AdvanceTimeMilliseconds(10000); |
@@ -653,9 +655,8 @@ TEST_F(PacedSenderTest, ResendPacket) { |
send_bucket_->QueueInMs()); |
// Send second packet and queue becomes empty. |
- EXPECT_CALL( |
- callback_, |
- TimeToSendPacket(ssrc, sequence_number + 1, capture_time_ms + 1, false)) |
+ EXPECT_CALL(callback_, TimeToSendPacket(ssrc, sequence_number + 1, |
+ capture_time_ms + 1, false, _)) |
.Times(1) |
.WillOnce(Return(true)); |
clock_.AdvanceTimeMilliseconds(10000); |
@@ -810,18 +811,22 @@ TEST_F(PacedSenderTest, PriorityInversion) { |
// Packets from earlier frames should be sent first. |
{ |
::testing::InSequence sequence; |
- EXPECT_CALL(callback_, TimeToSendPacket(ssrc, sequence_number, |
- clock_.TimeInMilliseconds(), true)) |
+ EXPECT_CALL(callback_, |
+ TimeToSendPacket(ssrc, sequence_number, |
+ clock_.TimeInMilliseconds(), true, _)) |
+ .WillOnce(Return(true)); |
+ EXPECT_CALL(callback_, |
+ TimeToSendPacket(ssrc, sequence_number + 1, |
+ clock_.TimeInMilliseconds(), true, _)) |
+ .WillOnce(Return(true)); |
+ EXPECT_CALL(callback_, |
+ TimeToSendPacket(ssrc, sequence_number + 3, |
+ clock_.TimeInMilliseconds() + 33, true, _)) |
.WillOnce(Return(true)); |
- EXPECT_CALL(callback_, TimeToSendPacket(ssrc, sequence_number + 1, |
- clock_.TimeInMilliseconds(), true)) |
+ EXPECT_CALL(callback_, |
+ TimeToSendPacket(ssrc, sequence_number + 2, |
+ clock_.TimeInMilliseconds() + 33, true, _)) |
.WillOnce(Return(true)); |
- EXPECT_CALL(callback_, TimeToSendPacket(ssrc, sequence_number + 3, |
- clock_.TimeInMilliseconds() + 33, |
- true)).WillOnce(Return(true)); |
- EXPECT_CALL(callback_, TimeToSendPacket(ssrc, sequence_number + 2, |
- clock_.TimeInMilliseconds() + 33, |
- true)).WillOnce(Return(true)); |
while (send_bucket_->QueueSizePackets() > 0) { |
int time_until_process = send_bucket_->TimeUntilNextProcess(); |
@@ -885,7 +890,7 @@ TEST_F(PacedSenderTest, AverageQueueTime) { |
// Only first packet (queued for 20ms) should be removed, leave the second |
// packet (queued for 10ms) alone in the queue. |
EXPECT_CALL(callback_, TimeToSendPacket(ssrc, sequence_number, |
- first_capture_time, false)) |
+ first_capture_time, false, _)) |
.Times(1) |
.WillRepeatedly(Return(true)); |
send_bucket_->Process(); |
@@ -894,7 +899,7 @@ TEST_F(PacedSenderTest, AverageQueueTime) { |
clock_.AdvanceTimeMilliseconds(10); |
EXPECT_CALL(callback_, TimeToSendPacket(ssrc, sequence_number + 1, |
- first_capture_time + 10, false)) |
+ first_capture_time + 10, false, _)) |
.Times(1) |
.WillRepeatedly(Return(true)); |
for (int i = 0; i < 3; ++i) { |