Index: webrtc/modules/rtp_rtcp/source/flexfec_receiver_unittest.cc |
diff --git a/webrtc/modules/rtp_rtcp/source/flexfec_receiver_unittest.cc b/webrtc/modules/rtp_rtcp/source/flexfec_receiver_unittest.cc |
index 8bf0f3701fa2b3ba38d1eb84f82299cf075e6f54..01d9a59ceee9d860b19e0a9b9c27402dca255f2b 100644 |
--- a/webrtc/modules/rtp_rtcp/source/flexfec_receiver_unittest.cc |
+++ b/webrtc/modules/rtp_rtcp/source/flexfec_receiver_unittest.cc |
@@ -45,6 +45,18 @@ RtpPacketReceived ParsePacket(const Packet& packet) { |
} // namespace |
+class FlexfecReceiverForTest : public FlexfecReceiver { |
+ public: |
+ FlexfecReceiverForTest(uint32_t ssrc, |
+ uint32_t protected_media_ssrc, |
+ RecoveredPacketReceiver* recovered_packet_receiver) |
+ : FlexfecReceiver(ssrc, protected_media_ssrc, recovered_packet_receiver) { |
+ } |
+ // Expose methods for tests. |
+ using FlexfecReceiver::AddReceivedPacket; |
+ using FlexfecReceiver::ProcessReceivedPackets; |
+}; |
+ |
class FlexfecReceiverTest : public ::testing::Test { |
protected: |
FlexfecReceiverTest() |
@@ -61,7 +73,7 @@ class FlexfecReceiverTest : public ::testing::Test { |
std::list<Packet*> EncodeFec(const PacketList& media_packets, |
size_t num_fec_packets); |
- FlexfecReceiver receiver_; |
+ FlexfecReceiverForTest receiver_; |
std::unique_ptr<ForwardErrorCorrection> erasure_code_; |
FlexfecPacketGenerator packet_generator_; |
@@ -100,8 +112,8 @@ TEST_F(FlexfecReceiverTest, ReceivesMediaPacket) { |
std::unique_ptr<Packet> media_packet( |
packet_generator_.NextPacket(0, kPayloadLength)); |
- EXPECT_TRUE( |
- receiver_.AddAndProcessReceivedPacket(ParsePacket(*media_packet))); |
+ EXPECT_TRUE(receiver_.AddReceivedPacket(ParsePacket(*media_packet))); |
+ EXPECT_TRUE(receiver_.ProcessReceivedPackets()); |
} |
TEST_F(FlexfecReceiverTest, ReceivesMediaAndFecPackets) { |
@@ -114,9 +126,10 @@ TEST_F(FlexfecReceiverTest, ReceivesMediaAndFecPackets) { |
const auto& media_packet = media_packets.front(); |
auto fec_packet = packet_generator_.BuildFlexfecPacket(*fec_packets.front()); |
- EXPECT_TRUE( |
- receiver_.AddAndProcessReceivedPacket(ParsePacket(*media_packet))); |
- EXPECT_TRUE(receiver_.AddAndProcessReceivedPacket(ParsePacket(*fec_packet))); |
+ EXPECT_TRUE(receiver_.AddReceivedPacket(ParsePacket(*media_packet))); |
+ EXPECT_TRUE(receiver_.ProcessReceivedPackets()); |
+ EXPECT_TRUE(receiver_.AddReceivedPacket(ParsePacket(*fec_packet))); |
+ EXPECT_TRUE(receiver_.ProcessReceivedPackets()); |
} |
TEST_F(FlexfecReceiverTest, FailsOnTruncatedFecPacket) { |
@@ -131,9 +144,9 @@ TEST_F(FlexfecReceiverTest, FailsOnTruncatedFecPacket) { |
fec_packets.front()->length = 1; |
auto fec_packet = packet_generator_.BuildFlexfecPacket(*fec_packets.front()); |
- EXPECT_TRUE( |
- receiver_.AddAndProcessReceivedPacket(ParsePacket(*media_packet))); |
- EXPECT_FALSE(receiver_.AddAndProcessReceivedPacket(ParsePacket(*fec_packet))); |
+ EXPECT_TRUE(receiver_.AddReceivedPacket(ParsePacket(*media_packet))); |
+ EXPECT_TRUE(receiver_.ProcessReceivedPackets()); |
+ EXPECT_FALSE(receiver_.AddReceivedPacket(ParsePacket(*fec_packet))); |
} |
TEST_F(FlexfecReceiverTest, FailsOnUnknownMediaSsrc) { |
@@ -148,8 +161,7 @@ TEST_F(FlexfecReceiverTest, FailsOnUnknownMediaSsrc) { |
media_packet->data[10] = 2; |
media_packet->data[11] = 3; |
- EXPECT_FALSE( |
- receiver_.AddAndProcessReceivedPacket(ParsePacket(*media_packet))); |
+ EXPECT_FALSE(receiver_.AddReceivedPacket(ParsePacket(*media_packet))); |
} |
TEST_F(FlexfecReceiverTest, FailsOnUnknownFecSsrc) { |
@@ -167,9 +179,9 @@ TEST_F(FlexfecReceiverTest, FailsOnUnknownFecSsrc) { |
fec_packet->data[10] = 6; |
fec_packet->data[11] = 7; |
- EXPECT_TRUE( |
- receiver_.AddAndProcessReceivedPacket(ParsePacket(*media_packet))); |
- EXPECT_FALSE(receiver_.AddAndProcessReceivedPacket(ParsePacket(*fec_packet))); |
+ EXPECT_TRUE(receiver_.AddReceivedPacket(ParsePacket(*media_packet))); |
+ EXPECT_TRUE(receiver_.ProcessReceivedPackets()); |
+ EXPECT_FALSE(receiver_.AddReceivedPacket(ParsePacket(*fec_packet))); |
} |
TEST_F(FlexfecReceiverTest, ReceivesMultiplePackets) { |
@@ -182,16 +194,17 @@ TEST_F(FlexfecReceiverTest, ReceivesMultiplePackets) { |
// Receive all media packets. |
for (const auto& media_packet : media_packets) { |
- EXPECT_TRUE( |
- receiver_.AddAndProcessReceivedPacket(ParsePacket(*media_packet))); |
+ EXPECT_TRUE(receiver_.AddReceivedPacket(ParsePacket(*media_packet))); |
+ EXPECT_TRUE(receiver_.ProcessReceivedPackets()); |
} |
// Receive FEC packet. |
auto fec_packet = fec_packets.front(); |
std::unique_ptr<Packet> packet_with_rtp_header = |
packet_generator_.BuildFlexfecPacket(*fec_packet); |
- EXPECT_TRUE(receiver_.AddAndProcessReceivedPacket( |
- ParsePacket(*packet_with_rtp_header))); |
+ EXPECT_TRUE( |
+ receiver_.AddReceivedPacket(ParsePacket(*packet_with_rtp_header))); |
+ EXPECT_TRUE(receiver_.ProcessReceivedPackets()); |
} |
TEST_F(FlexfecReceiverTest, RecoversFromSingleMediaLoss) { |
@@ -204,7 +217,7 @@ TEST_F(FlexfecReceiverTest, RecoversFromSingleMediaLoss) { |
// Receive first media packet but drop second. |
auto media_it = media_packets.begin(); |
- EXPECT_TRUE(receiver_.AddAndProcessReceivedPacket(ParsePacket(**media_it))); |
+ receiver_.OnRtpPacket(ParsePacket(**media_it)); |
// Receive FEC packet and ensure recovery of lost media packet. |
auto fec_it = fec_packets.begin(); |
@@ -216,8 +229,7 @@ TEST_F(FlexfecReceiverTest, RecoversFromSingleMediaLoss) { |
.With( |
Args<0, 1>(ElementsAreArray((*media_it)->data, (*media_it)->length))) |
.WillOnce(Return(true)); |
- EXPECT_TRUE(receiver_.AddAndProcessReceivedPacket( |
- ParsePacket(*packet_with_rtp_header))); |
+ receiver_.OnRtpPacket(ParsePacket(*packet_with_rtp_header)); |
} |
TEST_F(FlexfecReceiverTest, RecoversFromDoubleMediaLoss) { |
@@ -240,8 +252,7 @@ TEST_F(FlexfecReceiverTest, RecoversFromDoubleMediaLoss) { |
.With( |
Args<0, 1>(ElementsAreArray((*media_it)->data, (*media_it)->length))) |
.WillOnce(Return(true)); |
- EXPECT_TRUE(receiver_.AddAndProcessReceivedPacket( |
- ParsePacket(*packet_with_rtp_header))); |
+ receiver_.OnRtpPacket(ParsePacket(*packet_with_rtp_header)); |
// Receive second FEC packet and recover second lost media packet. |
fec_it++; |
@@ -252,8 +263,7 @@ TEST_F(FlexfecReceiverTest, RecoversFromDoubleMediaLoss) { |
.With( |
Args<0, 1>(ElementsAreArray((*media_it)->data, (*media_it)->length))) |
.WillOnce(Return(true)); |
- EXPECT_TRUE(receiver_.AddAndProcessReceivedPacket( |
- ParsePacket(*packet_with_rtp_header))); |
+ receiver_.OnRtpPacket(ParsePacket(*packet_with_rtp_header)); |
} |
TEST_F(FlexfecReceiverTest, DoesNotRecoverFromMediaAndFecLoss) { |
@@ -266,7 +276,7 @@ TEST_F(FlexfecReceiverTest, DoesNotRecoverFromMediaAndFecLoss) { |
// Receive first media packet. |
auto media_it = media_packets.begin(); |
- EXPECT_TRUE(receiver_.AddAndProcessReceivedPacket(ParsePacket(**media_it))); |
+ receiver_.OnRtpPacket(ParsePacket(**media_it)); |
// Drop second media packet and FEC packet. Do not expect call back. |
} |
@@ -281,7 +291,7 @@ TEST_F(FlexfecReceiverTest, DoesNotCallbackTwice) { |
// Receive first media packet but drop second. |
auto media_it = media_packets.begin(); |
- EXPECT_TRUE(receiver_.AddAndProcessReceivedPacket(ParsePacket(**media_it))); |
+ receiver_.OnRtpPacket(ParsePacket(**media_it)); |
// Receive FEC packet and ensure recovery of lost media packet. |
auto fec_it = fec_packets.begin(); |
@@ -293,12 +303,10 @@ TEST_F(FlexfecReceiverTest, DoesNotCallbackTwice) { |
.With( |
Args<0, 1>(ElementsAreArray((*media_it)->data, (*media_it)->length))) |
.WillOnce(Return(true)); |
- EXPECT_TRUE(receiver_.AddAndProcessReceivedPacket( |
- ParsePacket(*packet_with_rtp_header))); |
+ receiver_.OnRtpPacket(ParsePacket(*packet_with_rtp_header)); |
// Receive FEC packet again. |
- EXPECT_TRUE(receiver_.AddAndProcessReceivedPacket( |
- ParsePacket(*packet_with_rtp_header))); |
+ receiver_.OnRtpPacket(ParsePacket(*packet_with_rtp_header)); |
// Do not call back again. |
} |
@@ -320,7 +328,7 @@ TEST_F(FlexfecReceiverTest, RecoversFrom50PercentLoss) { |
// Drop every second media packet. |
auto media_it = media_packets.begin(); |
while (media_it != media_packets.end()) { |
- EXPECT_TRUE(receiver_.AddAndProcessReceivedPacket(ParsePacket(**media_it))); |
+ receiver_.OnRtpPacket(ParsePacket(**media_it)); |
++media_it; |
if (media_it == media_packets.end()) { |
break; |
@@ -342,8 +350,7 @@ TEST_F(FlexfecReceiverTest, RecoversFrom50PercentLoss) { |
.With(Args<0, 1>( |
ElementsAreArray((*media_it)->data, (*media_it)->length))) |
.WillOnce(Return(true)); |
- EXPECT_TRUE(receiver_.AddAndProcessReceivedPacket( |
- ParsePacket(*fec_packet_with_rtp_header))); |
+ receiver_.OnRtpPacket(ParsePacket(*fec_packet_with_rtp_header)); |
++media_it; |
} |
} |
@@ -370,7 +377,7 @@ TEST_F(FlexfecReceiverTest, DelayedFecPacketDoesHelp) { |
// Receive all other media packets. |
while (media_it != media_packets.end()) { |
- EXPECT_TRUE(receiver_.AddAndProcessReceivedPacket(ParsePacket(**media_it))); |
+ receiver_.OnRtpPacket(ParsePacket(**media_it)); |
++media_it; |
} |
@@ -384,8 +391,7 @@ TEST_F(FlexfecReceiverTest, DelayedFecPacketDoesHelp) { |
.With( |
Args<0, 1>(ElementsAreArray((*media_it)->data, (*media_it)->length))) |
.WillOnce(Return(true)); |
- EXPECT_TRUE(receiver_.AddAndProcessReceivedPacket( |
- ParsePacket(*packet_with_rtp_header))); |
+ receiver_.OnRtpPacket(ParsePacket(*packet_with_rtp_header)); |
} |
TEST_F(FlexfecReceiverTest, TooDelayedFecPacketDoesNotHelp) { |
@@ -410,7 +416,7 @@ TEST_F(FlexfecReceiverTest, TooDelayedFecPacketDoesNotHelp) { |
// Receive all other media packets. |
while (media_it != media_packets.end()) { |
- EXPECT_TRUE(receiver_.AddAndProcessReceivedPacket(ParsePacket(**media_it))); |
+ receiver_.OnRtpPacket(ParsePacket(**media_it)); |
++media_it; |
} |
@@ -418,8 +424,7 @@ TEST_F(FlexfecReceiverTest, TooDelayedFecPacketDoesNotHelp) { |
auto fec_it = fec_packets.begin(); |
std::unique_ptr<Packet> packet_with_rtp_header = |
packet_generator_.BuildFlexfecPacket(**fec_it); |
- EXPECT_TRUE(receiver_.AddAndProcessReceivedPacket( |
- ParsePacket(*packet_with_rtp_header))); |
+ receiver_.OnRtpPacket(ParsePacket(*packet_with_rtp_header)); |
// Do not expect a call back. |
} |
@@ -440,14 +445,10 @@ TEST_F(FlexfecReceiverTest, RecoversWithMediaPacketsOutOfOrder) { |
auto media_packet3 = media_it++; |
auto media_packet4 = media_it++; |
auto media_packet5 = media_it++; |
- EXPECT_TRUE( |
- receiver_.AddAndProcessReceivedPacket(ParsePacket(**media_packet5))); |
- EXPECT_TRUE( |
- receiver_.AddAndProcessReceivedPacket(ParsePacket(**media_packet2))); |
- EXPECT_TRUE( |
- receiver_.AddAndProcessReceivedPacket(ParsePacket(**media_packet3))); |
- EXPECT_TRUE( |
- receiver_.AddAndProcessReceivedPacket(ParsePacket(**media_packet0))); |
+ receiver_.OnRtpPacket(ParsePacket(**media_packet5)); |
+ receiver_.OnRtpPacket(ParsePacket(**media_packet2)); |
+ receiver_.OnRtpPacket(ParsePacket(**media_packet3)); |
+ receiver_.OnRtpPacket(ParsePacket(**media_packet0)); |
// Expect to recover lost media packets. |
EXPECT_CALL(recovered_packet_receiver_, |
@@ -466,8 +467,7 @@ TEST_F(FlexfecReceiverTest, RecoversWithMediaPacketsOutOfOrder) { |
std::unique_ptr<Packet> packet_with_rtp_header; |
while (fec_it != fec_packets.end()) { |
packet_with_rtp_header = packet_generator_.BuildFlexfecPacket(**fec_it); |
- EXPECT_TRUE(receiver_.AddAndProcessReceivedPacket( |
- ParsePacket(*packet_with_rtp_header))); |
+ receiver_.OnRtpPacket(ParsePacket(*packet_with_rtp_header)); |
++fec_it; |
} |
} |
@@ -482,7 +482,7 @@ TEST_F(FlexfecReceiverTest, CalculatesNumberOfPackets) { |
// Receive first media packet but drop second. |
auto media_it = media_packets.begin(); |
- EXPECT_TRUE(receiver_.AddAndProcessReceivedPacket(ParsePacket(**media_it))); |
+ receiver_.OnRtpPacket(ParsePacket(**media_it)); |
// Receive FEC packet and ensure recovery of lost media packet. |
auto fec_it = fec_packets.begin(); |
@@ -494,8 +494,7 @@ TEST_F(FlexfecReceiverTest, CalculatesNumberOfPackets) { |
.With( |
Args<0, 1>(ElementsAreArray((*media_it)->data, (*media_it)->length))) |
.WillOnce(Return(true)); |
- EXPECT_TRUE(receiver_.AddAndProcessReceivedPacket( |
- ParsePacket(*packet_with_rtp_header))); |
+ receiver_.OnRtpPacket(ParsePacket(*packet_with_rtp_header)); |
// Check stats calculations. |
FecPacketCounter packet_counter = receiver_.GetPacketCounter(); |