1 /*
2 * Copyright (c) 2022 The WebRTC project authors. All Rights Reserved.
3 *
4 * Use of this source code is governed by a BSD-style license
5 * that can be found in the LICENSE file in the root of the source
6 * tree. An additional intellectual property rights grant can be found
7 * in the file PATENTS. All contributing project authors may
8 * be found in the AUTHORS file in the root of the source tree.
9 */
10
11 #include "modules/rtp_rtcp/source/rtp_video_header.h"
12
13 #include "api/video/video_frame_metadata.h"
14 #include "api/video/video_frame_type.h"
15 #include "test/gmock.h"
16 #include "test/gtest.h"
17
18 namespace webrtc {
19 namespace {
20
21 using ::testing::ElementsAre;
22 using ::testing::IsEmpty;
23
TEST(RTPVideoHeaderTest,GetAsMetadataGetFrameType)24 TEST(RTPVideoHeaderTest, GetAsMetadataGetFrameType) {
25 RTPVideoHeader video_header;
26 video_header.frame_type = VideoFrameType::kVideoFrameKey;
27 VideoFrameMetadata metadata = video_header.GetAsMetadata();
28 EXPECT_EQ(metadata.GetFrameType(), VideoFrameType::kVideoFrameKey);
29 }
30
TEST(RTPVideoHeaderTest,GetAsMetadataGetWidth)31 TEST(RTPVideoHeaderTest, GetAsMetadataGetWidth) {
32 RTPVideoHeader video_header;
33 video_header.width = 1280u;
34 VideoFrameMetadata metadata = video_header.GetAsMetadata();
35 EXPECT_EQ(metadata.GetWidth(), video_header.width);
36 }
37
TEST(RTPVideoHeaderTest,GetAsMetadataGetHeight)38 TEST(RTPVideoHeaderTest, GetAsMetadataGetHeight) {
39 RTPVideoHeader video_header;
40 video_header.height = 720u;
41 VideoFrameMetadata metadata = video_header.GetAsMetadata();
42 EXPECT_EQ(metadata.GetHeight(), video_header.height);
43 }
44
TEST(RTPVideoHeaderTest,GetAsMetadataGetRotation)45 TEST(RTPVideoHeaderTest, GetAsMetadataGetRotation) {
46 RTPVideoHeader video_header;
47 video_header.rotation = VideoRotation::kVideoRotation_90;
48 VideoFrameMetadata metadata = video_header.GetAsMetadata();
49 EXPECT_EQ(metadata.GetRotation(), VideoRotation::kVideoRotation_90);
50 }
51
TEST(RTPVideoHeaderTest,GetAsMetadataGetContentType)52 TEST(RTPVideoHeaderTest, GetAsMetadataGetContentType) {
53 RTPVideoHeader video_header;
54 video_header.content_type = VideoContentType::SCREENSHARE;
55 VideoFrameMetadata metadata = video_header.GetAsMetadata();
56 EXPECT_EQ(metadata.GetContentType(), VideoContentType::SCREENSHARE);
57 }
58
TEST(RTPVideoHeaderTest,GetAsMetadataGetFrameId)59 TEST(RTPVideoHeaderTest, GetAsMetadataGetFrameId) {
60 RTPVideoHeader video_header;
61 RTPVideoHeader::GenericDescriptorInfo& generic =
62 video_header.generic.emplace();
63 generic.frame_id = 10;
64 VideoFrameMetadata metadata = video_header.GetAsMetadata();
65 EXPECT_EQ(metadata.GetFrameId().value(), 10);
66 }
67
TEST(RTPVideoHeaderTest,GetAsMetadataHasNoFrameIdForHeaderWithoutGeneric)68 TEST(RTPVideoHeaderTest, GetAsMetadataHasNoFrameIdForHeaderWithoutGeneric) {
69 RTPVideoHeader video_header;
70 VideoFrameMetadata metadata = video_header.GetAsMetadata();
71 ASSERT_FALSE(video_header.generic);
72 EXPECT_FALSE(metadata.GetFrameId().has_value());
73 }
74
TEST(RTPVideoHeaderTest,GetAsMetadataGetSpatialIndex)75 TEST(RTPVideoHeaderTest, GetAsMetadataGetSpatialIndex) {
76 RTPVideoHeader video_header;
77 RTPVideoHeader::GenericDescriptorInfo& generic =
78 video_header.generic.emplace();
79 generic.spatial_index = 2;
80 VideoFrameMetadata metadata = video_header.GetAsMetadata();
81 EXPECT_EQ(metadata.GetSpatialIndex(), 2);
82 }
83
TEST(RTPVideoHeaderTest,GetAsMetadataSpatialIndexIsZeroForHeaderWithoutGeneric)84 TEST(RTPVideoHeaderTest,
85 GetAsMetadataSpatialIndexIsZeroForHeaderWithoutGeneric) {
86 RTPVideoHeader video_header;
87 VideoFrameMetadata metadata = video_header.GetAsMetadata();
88 ASSERT_FALSE(video_header.generic);
89 EXPECT_EQ(metadata.GetSpatialIndex(), 0);
90 }
91
TEST(RTPVideoHeaderTest,GetAsMetadataGetTemporalIndex)92 TEST(RTPVideoHeaderTest, GetAsMetadataGetTemporalIndex) {
93 RTPVideoHeader video_header;
94 RTPVideoHeader::GenericDescriptorInfo& generic =
95 video_header.generic.emplace();
96 generic.temporal_index = 3;
97 VideoFrameMetadata metadata = video_header.GetAsMetadata();
98 EXPECT_EQ(metadata.GetTemporalIndex(), 3);
99 }
100
TEST(RTPVideoHeaderTest,GetAsMetadataTemporalIndexIsZeroForHeaderWithoutGeneric)101 TEST(RTPVideoHeaderTest,
102 GetAsMetadataTemporalIndexIsZeroForHeaderWithoutGeneric) {
103 RTPVideoHeader video_header;
104 VideoFrameMetadata metadata = video_header.GetAsMetadata();
105 ASSERT_FALSE(video_header.generic);
106 EXPECT_EQ(metadata.GetTemporalIndex(), 0);
107 }
108
TEST(RTPVideoHeaderTest,GetAsMetadataGetFrameDependencies)109 TEST(RTPVideoHeaderTest, GetAsMetadataGetFrameDependencies) {
110 RTPVideoHeader video_header;
111 RTPVideoHeader::GenericDescriptorInfo& generic =
112 video_header.generic.emplace();
113 generic.dependencies = {5, 6, 7};
114 VideoFrameMetadata metadata = video_header.GetAsMetadata();
115 EXPECT_THAT(metadata.GetFrameDependencies(), ElementsAre(5, 6, 7));
116 }
117
TEST(RTPVideoHeaderTest,GetAsMetadataFrameDependencyIsEmptyForHeaderWithoutGeneric)118 TEST(RTPVideoHeaderTest,
119 GetAsMetadataFrameDependencyIsEmptyForHeaderWithoutGeneric) {
120 RTPVideoHeader video_header;
121 VideoFrameMetadata metadata = video_header.GetAsMetadata();
122 ASSERT_FALSE(video_header.generic);
123 EXPECT_THAT(metadata.GetFrameDependencies(), IsEmpty());
124 }
125
TEST(RTPVideoHeaderTest,GetAsMetadataGetDecodeTargetIndications)126 TEST(RTPVideoHeaderTest, GetAsMetadataGetDecodeTargetIndications) {
127 RTPVideoHeader video_header;
128 RTPVideoHeader::GenericDescriptorInfo& generic =
129 video_header.generic.emplace();
130 generic.decode_target_indications = {DecodeTargetIndication::kSwitch};
131 VideoFrameMetadata metadata = video_header.GetAsMetadata();
132 EXPECT_THAT(metadata.GetDecodeTargetIndications(),
133 ElementsAre(DecodeTargetIndication::kSwitch));
134 }
135
TEST(RTPVideoHeaderTest,GetAsMetadataGetDecodeTargetIndicationsIsEmptyForHeaderWithoutGeneric)136 TEST(RTPVideoHeaderTest,
137 GetAsMetadataGetDecodeTargetIndicationsIsEmptyForHeaderWithoutGeneric) {
138 RTPVideoHeader video_header;
139 VideoFrameMetadata metadata = video_header.GetAsMetadata();
140 ASSERT_FALSE(video_header.generic);
141 EXPECT_THAT(metadata.GetDecodeTargetIndications(), IsEmpty());
142 }
143
TEST(RTPVideoHeaderTest,GetAsMetadataGetIsLastFrameInPicture)144 TEST(RTPVideoHeaderTest, GetAsMetadataGetIsLastFrameInPicture) {
145 RTPVideoHeader video_header;
146 video_header.is_last_frame_in_picture = false;
147 VideoFrameMetadata metadata = video_header.GetAsMetadata();
148 EXPECT_FALSE(metadata.GetIsLastFrameInPicture());
149 }
150
TEST(RTPVideoHeaderTest,GetAsMetadataGetSimulcastIdx)151 TEST(RTPVideoHeaderTest, GetAsMetadataGetSimulcastIdx) {
152 RTPVideoHeader video_header;
153 video_header.simulcastIdx = 123;
154 VideoFrameMetadata metadata = video_header.GetAsMetadata();
155 EXPECT_EQ(metadata.GetSimulcastIdx(), 123);
156 }
157
TEST(RTPVideoHeaderTest,GetAsMetadataGetCodec)158 TEST(RTPVideoHeaderTest, GetAsMetadataGetCodec) {
159 RTPVideoHeader video_header;
160 video_header.codec = VideoCodecType::kVideoCodecVP9;
161 VideoFrameMetadata metadata = video_header.GetAsMetadata();
162 EXPECT_EQ(metadata.GetCodec(), VideoCodecType::kVideoCodecVP9);
163 }
164
165 } // namespace
166 } // namespace webrtc
167