Index: webrtc/video/video_send_stream_tests.cc |
diff --git a/webrtc/video/video_send_stream_tests.cc b/webrtc/video/video_send_stream_tests.cc |
index 894b84047c5d218bc14d16fcc37002920596ad00..e24cb41aac3ac07ce895cd33acaaaee45cbf7aca 100644 |
--- a/webrtc/video/video_send_stream_tests.cc |
+++ b/webrtc/video/video_send_stream_tests.cc |
@@ -291,6 +291,43 @@ TEST_F(VideoSendStreamTest, SupportsVideoRotation) { |
RunBaseTest(&test); |
} |
+TEST_F(VideoSendStreamTest, SupportsVideoContentType) { |
+ class VideoRotationObserver : public test::SendTest { |
+ public: |
+ VideoRotationObserver() : SendTest(kDefaultTimeoutMs) { |
+ EXPECT_TRUE(parser_->RegisterRtpHeaderExtension( |
+ kRtpExtensionVideoContentType, test::kVideoContentTypeExtensionId)); |
+ } |
+ |
+ Action OnSendRtp(const uint8_t* packet, size_t length) override { |
+ RTPHeader header; |
+ EXPECT_TRUE(parser_->Parse(packet, length, &header)); |
+ EXPECT_TRUE(header.extension.hasVideoContentType); |
+ EXPECT_EQ(VideoContentType::SCREENSHARE, |
+ header.extension.videoContentType); |
+ observation_complete_.Set(); |
+ return SEND_PACKET; |
+ } |
+ |
+ void ModifyVideoConfigs( |
+ VideoSendStream::Config* send_config, |
+ std::vector<VideoReceiveStream::Config>* receive_configs, |
+ VideoEncoderConfig* encoder_config) override { |
+ send_config->rtp.extensions.clear(); |
+ send_config->rtp.extensions.push_back( |
+ RtpExtension(RtpExtension::kVideoContentTypeUri, |
+ test::kVideoContentTypeExtensionId)); |
+ encoder_config->content_type = VideoEncoderConfig::ContentType::kScreen; |
+ } |
+ |
+ void PerformTest() override { |
+ EXPECT_TRUE(Wait()) << "Timed out while waiting for single RTP packet."; |
+ } |
+ } test; |
+ |
+ RunBaseTest(&test); |
+} |
+ |
class FakeReceiveStatistics : public NullReceiveStatistics { |
public: |
FakeReceiveStatistics(uint32_t send_ssrc, |