| Index: webrtc/modules/rtp_rtcp/source/rtcp_packet/compound_packet_unittest.cc
|
| diff --git a/webrtc/modules/rtp_rtcp/source/rtcp_packet/compound_packet_unittest.cc b/webrtc/modules/rtp_rtcp/source/rtcp_packet/compound_packet_unittest.cc
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..83dc5f6ed3ea434ab0435ac929c67fb1f2763544
|
| --- /dev/null
|
| +++ b/webrtc/modules/rtp_rtcp/source/rtcp_packet/compound_packet_unittest.cc
|
| @@ -0,0 +1,157 @@
|
| +/*
|
| + * Copyright (c) 2016 The WebRTC project authors. All Rights Reserved.
|
| + *
|
| + * Use of this source code is governed by a BSD-style license
|
| + * that can be found in the LICENSE file in the root of the source
|
| + * tree. An additional intellectual property rights grant can be found
|
| + * in the file PATENTS. All contributing project authors may
|
| + * be found in the AUTHORS file in the root of the source tree.
|
| + */
|
| +
|
| +#include "webrtc/modules/rtp_rtcp/source/rtcp_packet/compound_packet.h"
|
| +
|
| +#include "testing/gtest/include/gtest/gtest.h"
|
| +#include "webrtc/modules/rtp_rtcp/source/rtcp_packet.h"
|
| +#include "webrtc/modules/rtp_rtcp/source/rtcp_packet/bye.h"
|
| +#include "webrtc/modules/rtp_rtcp/source/rtcp_packet/receiver_report.h"
|
| +#include "webrtc/test/rtcp_packet_parser.h"
|
| +
|
| +using webrtc::rtcp::Bye;
|
| +using webrtc::rtcp::CompoundPacket;
|
| +using webrtc::rtcp::Fir;
|
| +using webrtc::rtcp::RawPacket;
|
| +using webrtc::rtcp::ReceiverReport;
|
| +using webrtc::rtcp::ReportBlock;
|
| +using webrtc::rtcp::SenderReport;
|
| +using webrtc::test::RtcpPacketParser;
|
| +
|
| +namespace webrtc {
|
| +
|
| +const uint32_t kSenderSsrc = 0x12345678;
|
| +
|
| +TEST(RtcpCompoundPacketTest, AppendPacket) {
|
| + Fir fir;
|
| + ReportBlock rb;
|
| + ReceiverReport rr;
|
| + rr.From(kSenderSsrc);
|
| + EXPECT_TRUE(rr.WithReportBlock(rb));
|
| + rr.Append(&fir);
|
| +
|
| + rtc::scoped_ptr<RawPacket> packet(rr.Build());
|
| + RtcpPacketParser parser;
|
| + parser.Parse(packet->Buffer(), packet->Length());
|
| + EXPECT_EQ(1, parser.receiver_report()->num_packets());
|
| + EXPECT_EQ(kSenderSsrc, parser.receiver_report()->Ssrc());
|
| + EXPECT_EQ(1, parser.report_block()->num_packets());
|
| + EXPECT_EQ(1, parser.fir()->num_packets());
|
| +}
|
| +
|
| +TEST(RtcpCompoundPacketTest, AppendPacketOnEmpty) {
|
| + CompoundPacket empty;
|
| + ReceiverReport rr;
|
| + rr.From(kSenderSsrc);
|
| + empty.Append(&rr);
|
| +
|
| + rtc::scoped_ptr<RawPacket> packet(empty.Build());
|
| + RtcpPacketParser parser;
|
| + parser.Parse(packet->Buffer(), packet->Length());
|
| + EXPECT_EQ(1, parser.receiver_report()->num_packets());
|
| + EXPECT_EQ(0, parser.report_block()->num_packets());
|
| +}
|
| +
|
| +TEST(RtcpCompoundPacketTest, AppendPacketWithOwnAppendedPacket) {
|
| + Fir fir;
|
| + Bye bye;
|
| + ReportBlock rb;
|
| +
|
| + ReceiverReport rr;
|
| + EXPECT_TRUE(rr.WithReportBlock(rb));
|
| + rr.Append(&fir);
|
| +
|
| + SenderReport sr;
|
| + sr.Append(&bye);
|
| + sr.Append(&rr);
|
| +
|
| + rtc::scoped_ptr<RawPacket> packet(sr.Build());
|
| + RtcpPacketParser parser;
|
| + parser.Parse(packet->Buffer(), packet->Length());
|
| + EXPECT_EQ(1, parser.sender_report()->num_packets());
|
| + EXPECT_EQ(1, parser.receiver_report()->num_packets());
|
| + EXPECT_EQ(1, parser.report_block()->num_packets());
|
| + EXPECT_EQ(1, parser.bye()->num_packets());
|
| + EXPECT_EQ(1, parser.fir()->num_packets());
|
| +}
|
| +
|
| +TEST(RtcpCompoundPacketTest, BuildWithInputBuffer) {
|
| + Fir fir;
|
| + ReportBlock rb;
|
| + ReceiverReport rr;
|
| + rr.From(kSenderSsrc);
|
| + EXPECT_TRUE(rr.WithReportBlock(rb));
|
| + rr.Append(&fir);
|
| +
|
| + const size_t kRrLength = 8;
|
| + const size_t kReportBlockLength = 24;
|
| + const size_t kFirLength = 20;
|
| +
|
| + class Verifier : public rtcp::RtcpPacket::PacketReadyCallback {
|
| + public:
|
| + void OnPacketReady(uint8_t* data, size_t length) override {
|
| + RtcpPacketParser parser;
|
| + parser.Parse(data, length);
|
| + EXPECT_EQ(1, parser.receiver_report()->num_packets());
|
| + EXPECT_EQ(1, parser.report_block()->num_packets());
|
| + EXPECT_EQ(1, parser.fir()->num_packets());
|
| + ++packets_created_;
|
| + }
|
| +
|
| + int packets_created_ = 0;
|
| + } verifier;
|
| + const size_t kBufferSize = kRrLength + kReportBlockLength + kFirLength;
|
| + uint8_t buffer[kBufferSize];
|
| + EXPECT_TRUE(rr.BuildExternalBuffer(buffer, kBufferSize, &verifier));
|
| + EXPECT_EQ(1, verifier.packets_created_);
|
| +}
|
| +
|
| +TEST(RtcpCompoundPacketTest, BuildWithTooSmallBuffer_FragmentedSend) {
|
| + Fir fir;
|
| + ReportBlock rb;
|
| + ReceiverReport rr;
|
| + rr.From(kSenderSsrc);
|
| + EXPECT_TRUE(rr.WithReportBlock(rb));
|
| + rr.Append(&fir);
|
| +
|
| + const size_t kRrLength = 8;
|
| + const size_t kReportBlockLength = 24;
|
| +
|
| + class Verifier : public rtcp::RtcpPacket::PacketReadyCallback {
|
| + public:
|
| + void OnPacketReady(uint8_t* data, size_t length) override {
|
| + RtcpPacketParser parser;
|
| + parser.Parse(data, length);
|
| + switch (packets_created_++) {
|
| + case 0:
|
| + EXPECT_EQ(1, parser.receiver_report()->num_packets());
|
| + EXPECT_EQ(1, parser.report_block()->num_packets());
|
| + EXPECT_EQ(0, parser.fir()->num_packets());
|
| + break;
|
| + case 1:
|
| + EXPECT_EQ(0, parser.receiver_report()->num_packets());
|
| + EXPECT_EQ(0, parser.report_block()->num_packets());
|
| + EXPECT_EQ(1, parser.fir()->num_packets());
|
| + break;
|
| + default:
|
| + ADD_FAILURE() << "OnPacketReady not expected to be called "
|
| + << packets_created_ << " times.";
|
| + }
|
| + }
|
| +
|
| + int packets_created_ = 0;
|
| + } verifier;
|
| + const size_t kBufferSize = kRrLength + kReportBlockLength;
|
| + uint8_t buffer[kBufferSize];
|
| + EXPECT_TRUE(rr.BuildExternalBuffer(buffer, kBufferSize, &verifier));
|
| + EXPECT_EQ(2, verifier.packets_created_);
|
| +}
|
| +
|
| +} // namespace webrtc
|
|
|