mirror of
https://github.com/mollyim/webrtc.git
synced 2025-05-16 07:10:38 +01:00

Bug: webrtc:7990 Change-Id: I967d2065b85d6a2ca938ac0e83035cb92b45a907 Reviewed-on: https://webrtc-review.googlesource.com/98160 Reviewed-by: Björn Terelius <terelius@webrtc.org> Reviewed-by: Steve Anton <steveanton@webrtc.org> Reviewed-by: Erik Språng <sprang@webrtc.org> Reviewed-by: Danil Chapovalov <danilchap@webrtc.org> Commit-Queue: Johannes Kron <kron@webrtc.org> Cr-Commit-Position: refs/heads/master@{#24881}
120 lines
3.9 KiB
C++
120 lines
3.9 KiB
C++
/*
|
|
* Copyright (c) 2012 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 "modules/rtp_rtcp/include/rtp_header_extension_map.h"
|
|
|
|
#include <vector>
|
|
|
|
#include "modules/rtp_rtcp/include/rtp_rtcp_defines.h"
|
|
#include "modules/rtp_rtcp/source/rtp_header_extensions.h"
|
|
#include "test/gtest.h"
|
|
|
|
namespace webrtc {
|
|
|
|
TEST(RtpHeaderExtensionTest, RegisterByType) {
|
|
RtpHeaderExtensionMap map;
|
|
EXPECT_FALSE(map.IsRegistered(TransmissionOffset::kId));
|
|
|
|
EXPECT_TRUE(map.RegisterByType(3, TransmissionOffset::kId));
|
|
|
|
EXPECT_TRUE(map.IsRegistered(TransmissionOffset::kId));
|
|
EXPECT_EQ(3, map.GetId(TransmissionOffset::kId));
|
|
EXPECT_EQ(TransmissionOffset::kId, map.GetType(3));
|
|
}
|
|
|
|
TEST(RtpHeaderExtensionTest, RegisterByUri) {
|
|
RtpHeaderExtensionMap map;
|
|
|
|
EXPECT_TRUE(map.RegisterByUri(3, TransmissionOffset::kUri));
|
|
|
|
EXPECT_TRUE(map.IsRegistered(TransmissionOffset::kId));
|
|
EXPECT_EQ(3, map.GetId(TransmissionOffset::kId));
|
|
EXPECT_EQ(TransmissionOffset::kId, map.GetType(3));
|
|
}
|
|
|
|
TEST(RtpHeaderExtensionTest, RegisterWithTrait) {
|
|
RtpHeaderExtensionMap map;
|
|
|
|
EXPECT_TRUE(map.Register<TransmissionOffset>(3));
|
|
|
|
EXPECT_TRUE(map.IsRegistered(TransmissionOffset::kId));
|
|
EXPECT_EQ(3, map.GetId(TransmissionOffset::kId));
|
|
EXPECT_EQ(TransmissionOffset::kId, map.GetType(3));
|
|
}
|
|
|
|
TEST(RtpHeaderExtensionTest, RegisterDuringContruction) {
|
|
const std::vector<RtpExtension> config = {{TransmissionOffset::kUri, 1},
|
|
{AbsoluteSendTime::kUri, 3}};
|
|
const RtpHeaderExtensionMap map(config);
|
|
|
|
EXPECT_EQ(1, map.GetId(TransmissionOffset::kId));
|
|
EXPECT_EQ(3, map.GetId(AbsoluteSendTime::kId));
|
|
}
|
|
|
|
TEST(RtpHeaderExtensionTest, RegisterTwoByteHeaderExtensions) {
|
|
RtpHeaderExtensionMap map;
|
|
// Two-byte header extension needed for id: [15-255].
|
|
EXPECT_TRUE(map.Register<TransmissionOffset>(18));
|
|
EXPECT_TRUE(map.Register<AbsoluteSendTime>(255));
|
|
}
|
|
|
|
TEST(RtpHeaderExtensionTest, RegisterIllegalArg) {
|
|
RtpHeaderExtensionMap map;
|
|
// Valid range for id: [1-255].
|
|
EXPECT_FALSE(map.Register<TransmissionOffset>(0));
|
|
EXPECT_FALSE(map.Register<TransmissionOffset>(256));
|
|
}
|
|
|
|
TEST(RtpHeaderExtensionTest, Idempotent) {
|
|
RtpHeaderExtensionMap map;
|
|
|
|
EXPECT_TRUE(map.Register<TransmissionOffset>(3));
|
|
EXPECT_TRUE(map.Register<TransmissionOffset>(3));
|
|
|
|
map.Deregister(TransmissionOffset::kId);
|
|
map.Deregister(TransmissionOffset::kId);
|
|
}
|
|
|
|
TEST(RtpHeaderExtensionTest, NonUniqueId) {
|
|
RtpHeaderExtensionMap map;
|
|
EXPECT_TRUE(map.Register<TransmissionOffset>(3));
|
|
|
|
EXPECT_FALSE(map.Register<AudioLevel>(3));
|
|
EXPECT_TRUE(map.Register<AudioLevel>(4));
|
|
}
|
|
|
|
TEST(RtpHeaderExtensionTest, GetTotalLength) {
|
|
RtpHeaderExtensionMap map;
|
|
constexpr RtpExtensionSize kExtensionSizes[] = {
|
|
{TransmissionOffset::kId, TransmissionOffset::kValueSizeBytes}};
|
|
EXPECT_EQ(0u, map.GetTotalLengthInBytes(kExtensionSizes));
|
|
EXPECT_TRUE(map.Register<TransmissionOffset>(3));
|
|
static constexpr size_t kRtpOneByteHeaderLength = 4;
|
|
EXPECT_EQ(kRtpOneByteHeaderLength + (TransmissionOffset::kValueSizeBytes + 1),
|
|
map.GetTotalLengthInBytes(kExtensionSizes));
|
|
}
|
|
|
|
TEST(RtpHeaderExtensionTest, GetType) {
|
|
RtpHeaderExtensionMap map;
|
|
EXPECT_EQ(RtpHeaderExtensionMap::kInvalidType, map.GetType(3));
|
|
EXPECT_TRUE(map.Register<TransmissionOffset>(3));
|
|
|
|
EXPECT_EQ(TransmissionOffset::kId, map.GetType(3));
|
|
}
|
|
|
|
TEST(RtpHeaderExtensionTest, GetId) {
|
|
RtpHeaderExtensionMap map;
|
|
EXPECT_EQ(RtpHeaderExtensionMap::kInvalidId,
|
|
map.GetId(TransmissionOffset::kId));
|
|
EXPECT_TRUE(map.Register<TransmissionOffset>(3));
|
|
|
|
EXPECT_EQ(3, map.GetId(TransmissionOffset::kId));
|
|
}
|
|
|
|
} // namespace webrtc
|