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 |