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 5e5a61389a932bb5537e336160ed862c430e88a4..eb4bd2573e500554e9dcad7412708f9c52a17dc8 100644 |
--- a/webrtc/video/video_send_stream_tests.cc |
+++ b/webrtc/video/video_send_stream_tests.cc |
@@ -281,6 +281,43 @@ |
void OnFrameGeneratorCapturerCreated( |
test::FrameGeneratorCapturer* frame_generator_capturer) override { |
frame_generator_capturer->SetFakeRotation(kVideoRotation_90); |
+ } |
+ |
+ void PerformTest() override { |
+ EXPECT_TRUE(Wait()) << "Timed out while waiting for single RTP packet."; |
+ } |
+ } test; |
+ |
+ 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 { |