| Index: webrtc/modules/rtp_rtcp/source/rtcp_packet/target_bitrate_unittest.cc
|
| diff --git a/webrtc/modules/rtp_rtcp/source/rtcp_packet/target_bitrate_unittest.cc b/webrtc/modules/rtp_rtcp/source/rtcp_packet/target_bitrate_unittest.cc
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..1c8f39de61d58541084f12346cf5a6b18d25f23f
|
| --- /dev/null
|
| +++ b/webrtc/modules/rtp_rtcp/source/rtcp_packet/target_bitrate_unittest.cc
|
| @@ -0,0 +1,89 @@
|
| +/*
|
| + * 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/target_bitrate.h"
|
| +
|
| +#include "webrtc/base/buffer.h"
|
| +#include "webrtc/test/gtest.h"
|
| +#include "webrtc/test/rtcp_packet_parser.h"
|
| +#include "webrtc/modules/rtp_rtcp/source/byte_io.h"
|
| +#include "webrtc/modules/rtp_rtcp/source/rtcp_packet/extended_reports.h"
|
| +
|
| +namespace webrtc {
|
| +namespace {
|
| +using BitrateItem = rtcp::TargetBitrate::BitrateItem;
|
| +using rtcp::TargetBitrate;
|
| +using test::ParseSinglePacket;
|
| +
|
| +constexpr uint32_t kSsrc = 0x12345678;
|
| +
|
| +// clang-format off
|
| +const uint8_t kPacket[] = { TargetBitrate::kBlockType, // Block ID.
|
| + 0x00, // Reserved.
|
| + 0x00, 0x04, // Length = 4 words.
|
| + 0x00, 0x01, 0x02, 0x03, // S0T0 0x010203 kbps.
|
| + 0x01, 0x02, 0x03, 0x04, // S0T1 0x020304 kbps.
|
| + 0x10, 0x03, 0x04, 0x05, // S1T0 0x030405 kbps.
|
| + 0x11, 0x04, 0x05, 0x06 }; // S1T1 0x040506 kbps.
|
| +constexpr size_t kPacketLengthBlocks = ((sizeof(kPacket) + 3) / 4) - 1;
|
| +// clang-format on
|
| +
|
| +void ExpectBirateItemEquals(const BitrateItem& expected,
|
| + const BitrateItem& actual) {
|
| + EXPECT_EQ(expected.spatial_layer, actual.spatial_layer);
|
| + EXPECT_EQ(expected.temporal_layer, actual.temporal_layer);
|
| + EXPECT_EQ(expected.target_bitrate_kbps, actual.target_bitrate_kbps);
|
| +}
|
| +
|
| +void CheckBitrateItems(const std::vector<BitrateItem>& bitrates) {
|
| + EXPECT_EQ(4U, bitrates.size());
|
| + ExpectBirateItemEquals(BitrateItem(0, 0, 0x010203), bitrates[0]);
|
| + ExpectBirateItemEquals(BitrateItem(0, 1, 0x020304), bitrates[1]);
|
| + ExpectBirateItemEquals(BitrateItem(1, 0, 0x030405), bitrates[2]);
|
| + ExpectBirateItemEquals(BitrateItem(1, 1, 0x040506), bitrates[3]);
|
| +}
|
| +
|
| +} // namespace
|
| +
|
| +TEST(TargetBitrateTest, Parse) {
|
| + TargetBitrate target_bitrate;
|
| + EXPECT_TRUE(target_bitrate.Parse(kPacket, kPacketLengthBlocks));
|
| + CheckBitrateItems(target_bitrate.GetTargetBitrates());
|
| +}
|
| +
|
| +TEST(TargetBitrateTest, FullPacket) {
|
| + const size_t kXRHeaderSize = 8; // RTCP header (4) + SSRC (4).
|
| + const size_t kTotalSize = kXRHeaderSize + sizeof(kPacket);
|
| + uint8_t kRtcpPacket[kTotalSize] = {2 << 6, 207, 0x00, (kTotalSize / 4) - 1,
|
| + 0x12, 0x34, 0x56, 0x78}; // SSRC.
|
| + memcpy(&kRtcpPacket[kXRHeaderSize], kPacket, sizeof(kPacket));
|
| + rtcp::ExtendedReports xr;
|
| + EXPECT_TRUE(ParseSinglePacket(kRtcpPacket, &xr));
|
| + EXPECT_EQ(kSsrc, xr.sender_ssrc());
|
| + const rtc::Optional<TargetBitrate>& target_bitrate = xr.target_bitrate();
|
| + ASSERT_TRUE(static_cast<bool>(target_bitrate));
|
| + CheckBitrateItems(target_bitrate->GetTargetBitrates());
|
| +}
|
| +
|
| +TEST(TargetBitrateTest, Create) {
|
| + TargetBitrate target_bitrate;
|
| + target_bitrate.AddTargetBitrate(0, 0, 0x010203);
|
| + target_bitrate.AddTargetBitrate(0, 1, 0x020304);
|
| + target_bitrate.AddTargetBitrate(1, 0, 0x030405);
|
| + target_bitrate.AddTargetBitrate(1, 1, 0x040506);
|
| +
|
| + uint8_t buffer[sizeof(kPacket)] = {};
|
| + ASSERT_EQ(sizeof(kPacket), target_bitrate.BlockLength());
|
| + target_bitrate.Create(buffer);
|
| +
|
| + EXPECT_EQ(0, memcmp(kPacket, buffer, sizeof(kPacket)));
|
| +}
|
| +
|
| +} // namespace webrtc
|
|
|