Index: webrtc/modules/rtp_rtcp/source/producer_fec_unittest.cc |
diff --git a/webrtc/modules/rtp_rtcp/source/producer_fec_unittest.cc b/webrtc/modules/rtp_rtcp/source/producer_fec_unittest.cc |
index ea59dc1b4da2a3878028e758ca6754820a75c2d8..75ca2e5534a9ef5962d4075a509d8a022b55427e 100644 |
--- a/webrtc/modules/rtp_rtcp/source/producer_fec_unittest.cc |
+++ b/webrtc/modules/rtp_rtcp/source/producer_fec_unittest.cc |
@@ -22,8 +22,8 @@ namespace webrtc { |
void VerifyHeader(uint16_t seq_num, |
uint32_t timestamp, |
- int red_pltype, |
- int fec_pltype, |
+ int red_payload_type, |
+ int fec_payload_type, |
RedPacket* packet, |
bool marker_bit) { |
EXPECT_GT(packet->length(), kRtpHeaderSize); |
@@ -31,30 +31,25 @@ void VerifyHeader(uint16_t seq_num, |
uint8_t* data = packet->data(); |
// Marker bit not set. |
EXPECT_EQ(marker_bit ? 0x80 : 0, data[1] & 0x80); |
- EXPECT_EQ(red_pltype, data[1] & 0x7F); |
+ EXPECT_EQ(red_payload_type, data[1] & 0x7F); |
EXPECT_EQ(seq_num, (data[2] << 8) + data[3]); |
uint32_t parsed_timestamp = (data[4] << 24) + (data[5] << 16) + |
(data[6] << 8) + data[7]; |
EXPECT_EQ(timestamp, parsed_timestamp); |
- EXPECT_EQ(static_cast<uint8_t>(fec_pltype), data[kRtpHeaderSize]); |
+ EXPECT_EQ(static_cast<uint8_t>(fec_payload_type), data[kRtpHeaderSize]); |
} |
class ProducerFecTest : public ::testing::Test { |
protected: |
virtual void SetUp() { |
- fec_ = new ForwardErrorCorrection(); |
- producer_ = new ProducerFec(fec_); |
- generator_ = new FrameGenerator(); |
+ fec_.reset(new ForwardErrorCorrection()); |
+ producer_.reset(new ProducerFec(fec_.get())); |
+ generator_.reset(new FrameGenerator()); |
} |
- virtual void TearDown() { |
- delete producer_; |
- delete fec_; |
- delete generator_; |
- } |
- ForwardErrorCorrection* fec_; |
- ProducerFec* producer_; |
- FrameGenerator* generator_; |
+ std::unique_ptr<ForwardErrorCorrection> fec_; |
danilchap
2016/07/19 14:55:32
may be use classes directly instead of smart point
brandtr
2016/07/21 09:03:58
Done.
|
+ std::unique_ptr<ProducerFec> producer_; |
+ std::unique_ptr<FrameGenerator> generator_; |
}; |
// Verifies bug found via fuzzing, where a gap in the packet sequence caused us |
@@ -92,15 +87,14 @@ TEST_F(ProducerFecTest, NoEmptyFecWithSeqNumGaps) { |
producer_->AddRtpPacketAndGenerateFec(packet, p.payload_size, |
p.header_size); |
uint16_t num_fec_packets = producer_->NumAvailableFecPackets(); |
- std::vector<RedPacket*> fec_packets; |
if (num_fec_packets > 0) { |
- fec_packets = |
- producer_->GetFecPackets(kRedPayloadType, 99, 100, p.header_size); |
+ std::vector<std::unique_ptr<RedPacket>> fec_packets = |
+ producer_->GetFecPacketsAsRed(kRedPayloadType, |
+ kFecPayloadType, |
+ 100, |
+ p.header_size); |
EXPECT_EQ(num_fec_packets, fec_packets.size()); |
} |
- for (RedPacket* fec_packet : fec_packets) { |
- delete fec_packet; |
- } |
} |
} |
@@ -127,19 +121,19 @@ TEST_F(ProducerFecTest, OneFrameFec) { |
} |
EXPECT_TRUE(producer_->FecAvailable()); |
uint16_t seq_num = generator_->NextSeqNum(); |
- std::vector<RedPacket*> packets = producer_->GetFecPackets(kRedPayloadType, |
- kFecPayloadType, |
- seq_num, |
- kRtpHeaderSize); |
+ std::vector<std::unique_ptr<RedPacket>> packets = |
+ producer_->GetFecPacketsAsRed(kRedPayloadType, |
+ kFecPayloadType, |
+ seq_num, |
+ kRtpHeaderSize); |
EXPECT_FALSE(producer_->FecAvailable()); |
ASSERT_EQ(1u, packets.size()); |
VerifyHeader(seq_num, last_timestamp, |
- kRedPayloadType, kFecPayloadType, packets.front(), false); |
+ kRedPayloadType, kFecPayloadType, packets.front().get(), false); |
while (!rtp_packets.empty()) { |
delete rtp_packets.front(); |
rtp_packets.pop_front(); |
} |
- delete packets.front(); |
} |
TEST_F(ProducerFecTest, TwoFrameFec) { |
@@ -171,27 +165,27 @@ TEST_F(ProducerFecTest, TwoFrameFec) { |
} |
EXPECT_TRUE(producer_->FecAvailable()); |
uint16_t seq_num = generator_->NextSeqNum(); |
- std::vector<RedPacket*> packets = producer_->GetFecPackets(kRedPayloadType, |
- kFecPayloadType, |
- seq_num, |
- kRtpHeaderSize); |
+ std::vector<std::unique_ptr<RedPacket>> packets = |
+ producer_->GetFecPacketsAsRed(kRedPayloadType, |
+ kFecPayloadType, |
+ seq_num, |
+ kRtpHeaderSize); |
EXPECT_FALSE(producer_->FecAvailable()); |
ASSERT_EQ(1u, packets.size()); |
VerifyHeader(seq_num, last_timestamp, kRedPayloadType, kFecPayloadType, |
- packets.front(), false); |
+ packets.front().get(), false); |
while (!rtp_packets.empty()) { |
delete rtp_packets.front(); |
rtp_packets.pop_front(); |
} |
- delete packets.front(); |
} |
TEST_F(ProducerFecTest, BuildRedPacket) { |
generator_->NewFrame(1); |
test::RawRtpPacket* packet = generator_->NextPacket(0, 10); |
- std::unique_ptr<RedPacket> red_packet(producer_->BuildRedPacket( |
+ std::unique_ptr<RedPacket> red_packet = ProducerFec::BuildRedPacket( |
packet->data, packet->length - kRtpHeaderSize, kRtpHeaderSize, |
- kRedPayloadType)); |
+ kRedPayloadType); |
EXPECT_EQ(packet->length + 1, red_packet->length()); |
VerifyHeader(packet->header.header.sequenceNumber, |
packet->header.header.timestamp, |