Index: webrtc/modules/rtp_rtcp/source/rtp_sender_unittest.cc |
diff --git a/webrtc/modules/rtp_rtcp/source/rtp_sender_unittest.cc b/webrtc/modules/rtp_rtcp/source/rtp_sender_unittest.cc |
index 555f2274b01b5c954feebe8740fe02d20b58d327..d33b086eb25f39479c2740061bc170cba4d0a87a 100644 |
--- a/webrtc/modules/rtp_rtcp/source/rtp_sender_unittest.cc |
+++ b/webrtc/modules/rtp_rtcp/source/rtp_sender_unittest.cc |
@@ -12,11 +12,11 @@ |
* This file includes unit tests for the RTPSender. |
*/ |
+#include "testing/gmock/include/gmock/gmock.h" |
#include "testing/gtest/include/gtest/gtest.h" |
#include "webrtc/base/buffer.h" |
#include "webrtc/base/scoped_ptr.h" |
-#include "webrtc/modules/pacing/include/mock/mock_paced_sender.h" |
#include "webrtc/modules/rtp_rtcp/interface/rtp_cvo.h" |
#include "webrtc/modules/rtp_rtcp/interface/rtp_header_parser.h" |
#include "webrtc/modules/rtp_rtcp/interface/rtp_rtcp_defines.h" |
@@ -100,6 +100,20 @@ class LoopbackTransportTest : public webrtc::Transport { |
} // namespace |
+class MockRtpPacketSender : public RtpPacketSender { |
+ public: |
+ MockRtpPacketSender() {} |
+ virtual ~MockRtpPacketSender() {} |
+ |
+ MOCK_METHOD6(SendPacket, |
+ bool(Priority priority, |
+ uint32_t ssrc, |
+ uint16_t sequence_number, |
+ int64_t capture_time_ms, |
+ size_t bytes, |
+ bool retransmission)); |
+}; |
+ |
class RtpSenderTest : public ::testing::Test { |
protected: |
RtpSenderTest() |
@@ -121,7 +135,7 @@ class RtpSenderTest : public ::testing::Test { |
} |
SimulatedClock fake_clock_; |
- MockPacedSender mock_paced_sender_; |
+ MockRtpPacketSender mock_paced_sender_; |
rtc::scoped_ptr<RTPSender> rtp_sender_; |
int payload_; |
LoopbackTransportTest transport_; |
@@ -152,12 +166,9 @@ class RtpSenderTest : public ::testing::Test { |
ASSERT_GE(rtp_length, 0); |
// Packet should be stored in a send bucket. |
- EXPECT_EQ(0, rtp_sender_->SendToNetwork(packet_, |
- payload_length, |
- rtp_length, |
- capture_time_ms, |
- kAllowRetransmission, |
- PacedSender::kNormalPriority)); |
+ EXPECT_EQ(0, rtp_sender_->SendToNetwork( |
+ packet_, payload_length, rtp_length, capture_time_ms, |
+ kAllowRetransmission, RtpPacketSender::kNormalPriority)); |
} |
}; |
@@ -593,8 +604,8 @@ TEST_F(RtpSenderTest, BuildRTPPacketWithHeaderExtensions) { |
TEST_F(RtpSenderTest, TrafficSmoothingWithExtensions) { |
EXPECT_CALL(mock_paced_sender_, |
- SendPacket(PacedSender::kNormalPriority, _, kSeqNum, _, _, _)). |
- WillOnce(testing::Return(false)); |
+ SendPacket(RtpPacketSender::kNormalPriority, _, kSeqNum, _, _, _)) |
+ .WillOnce(testing::Return(false)); |
rtp_sender_->SetStorePacketsStatus(true, 10); |
EXPECT_EQ(0, rtp_sender_->RegisterRtpHeaderExtension( |
@@ -609,12 +620,9 @@ TEST_F(RtpSenderTest, TrafficSmoothingWithExtensions) { |
size_t rtp_length = static_cast<size_t>(rtp_length_int); |
// Packet should be stored in a send bucket. |
- EXPECT_EQ(0, rtp_sender_->SendToNetwork(packet_, |
- 0, |
- rtp_length, |
- capture_time_ms, |
- kAllowRetransmission, |
- PacedSender::kNormalPriority)); |
+ EXPECT_EQ(0, rtp_sender_->SendToNetwork(packet_, 0, rtp_length, |
+ capture_time_ms, kAllowRetransmission, |
+ RtpPacketSender::kNormalPriority)); |
EXPECT_EQ(0, transport_.packets_sent_); |
@@ -646,8 +654,8 @@ TEST_F(RtpSenderTest, TrafficSmoothingWithExtensions) { |
TEST_F(RtpSenderTest, TrafficSmoothingRetransmits) { |
EXPECT_CALL(mock_paced_sender_, |
- SendPacket(PacedSender::kNormalPriority, _, kSeqNum, _, _, _)). |
- WillOnce(testing::Return(false)); |
+ SendPacket(RtpPacketSender::kNormalPriority, _, kSeqNum, _, _, _)) |
+ .WillOnce(testing::Return(false)); |
rtp_sender_->SetStorePacketsStatus(true, 10); |
EXPECT_EQ(0, rtp_sender_->RegisterRtpHeaderExtension( |
@@ -662,18 +670,15 @@ TEST_F(RtpSenderTest, TrafficSmoothingRetransmits) { |
size_t rtp_length = static_cast<size_t>(rtp_length_int); |
// Packet should be stored in a send bucket. |
- EXPECT_EQ(0, rtp_sender_->SendToNetwork(packet_, |
- 0, |
- rtp_length, |
- capture_time_ms, |
- kAllowRetransmission, |
- PacedSender::kNormalPriority)); |
+ EXPECT_EQ(0, rtp_sender_->SendToNetwork(packet_, 0, rtp_length, |
+ capture_time_ms, kAllowRetransmission, |
+ RtpPacketSender::kNormalPriority)); |
EXPECT_EQ(0, transport_.packets_sent_); |
EXPECT_CALL(mock_paced_sender_, |
- SendPacket(PacedSender::kHighPriority, _, kSeqNum, _, _, _)). |
- WillOnce(testing::Return(false)); |
+ SendPacket(RtpPacketSender::kHighPriority, _, kSeqNum, _, _, _)) |
+ .WillOnce(testing::Return(false)); |
const int kStoredTimeInMs = 100; |
fake_clock_.AdvanceTimeMilliseconds(kStoredTimeInMs); |
@@ -710,8 +715,8 @@ TEST_F(RtpSenderTest, TrafficSmoothingRetransmits) { |
TEST_F(RtpSenderTest, SendPadding) { |
// Make all (non-padding) packets go to send queue. |
EXPECT_CALL(mock_paced_sender_, |
- SendPacket(PacedSender::kNormalPriority, _, _, _, _, _)). |
- WillRepeatedly(testing::Return(false)); |
+ SendPacket(RtpPacketSender::kNormalPriority, _, _, _, _, _)) |
+ .WillRepeatedly(testing::Return(false)); |
uint16_t seq_num = kSeqNum; |
uint32_t timestamp = kTimestamp; |
@@ -744,12 +749,9 @@ TEST_F(RtpSenderTest, SendPadding) { |
size_t rtp_length = static_cast<size_t>(rtp_length_int); |
// Packet should be stored in a send bucket. |
- EXPECT_EQ(0, rtp_sender_->SendToNetwork(packet_, |
- 0, |
- rtp_length, |
- capture_time_ms, |
- kAllowRetransmission, |
- PacedSender::kNormalPriority)); |
+ EXPECT_EQ(0, rtp_sender_->SendToNetwork(packet_, 0, rtp_length, |
+ capture_time_ms, kAllowRetransmission, |
+ RtpPacketSender::kNormalPriority)); |
int total_packets_sent = 0; |
EXPECT_EQ(total_packets_sent, transport_.packets_sent_); |
@@ -802,12 +804,9 @@ TEST_F(RtpSenderTest, SendPadding) { |
rtp_length = static_cast<size_t>(rtp_length_int); |
// Packet should be stored in a send bucket. |
- EXPECT_EQ(0, rtp_sender_->SendToNetwork(packet_, |
- 0, |
- rtp_length, |
- capture_time_ms, |
- kAllowRetransmission, |
- PacedSender::kNormalPriority)); |
+ EXPECT_EQ(0, rtp_sender_->SendToNetwork(packet_, 0, rtp_length, |
+ capture_time_ms, kAllowRetransmission, |
+ RtpPacketSender::kNormalPriority)); |
rtp_sender_->TimeToSendPacket(seq_num, capture_time_ms, false); |
// Process send bucket. |
@@ -836,8 +835,8 @@ TEST_F(RtpSenderTest, SendRedundantPayloads) { |
rtp_sender_->SetRtxPayloadType(kRtxPayload, kPayload); |
// Make all packets go through the pacer. |
EXPECT_CALL(mock_paced_sender_, |
- SendPacket(PacedSender::kNormalPriority, _, _, _, _, _)). |
- WillRepeatedly(testing::Return(false)); |
+ SendPacket(RtpPacketSender::kNormalPriority, _, _, _, _, _)) |
+ .WillRepeatedly(testing::Return(false)); |
uint16_t seq_num = kSeqNum; |
rtp_sender_->SetStorePacketsStatus(true, 10); |