Index: media/cast/test/end2end_unittest.cc |
diff --git a/media/cast/test/end2end_unittest.cc b/media/cast/test/end2end_unittest.cc |
index 7145ba369a3a9868427ce91042aa783e3da2f062..bd5b49ec63fbb97d6616fd81c43f7c74e494a96c 100644 |
--- a/media/cast/test/end2end_unittest.cc |
+++ b/media/cast/test/end2end_unittest.cc |
@@ -1275,5 +1275,45 @@ TEST_F(End2EndTest, TestSetPlayoutDelay) { |
EXPECT_LT(jump, 120u); |
} |
+namespace { |
+ |
+void OnEncodedVideoFrame(scoped_ptr<EncodedFrame> encoded_frame) { |
+ const uint32_t frame_id = encoded_frame->frame_id; |
+ if ((frame_id == 0) || (frame_id == 3)) |
+ EXPECT_EQ(encoded_frame->dependency, EncodedFrame::KEY); |
+ else |
+ EXPECT_EQ(encoded_frame->dependency, EncodedFrame::DEPENDENT); |
+} |
+ |
+} // namespace |
+ |
+TEST_F(End2EndTest, TestPliGeneratingKeyFrames) { |
+ Configure(CODEC_VIDEO_VP8, CODEC_AUDIO_PCM16); |
+ Create(); |
+ |
+ // Enqueue request of 4 frames from the encoder. |
+ for (int i = 0; i < 4; ++i) { |
+ cast_receiver_->RequestEncodedVideoFrame(base::Bind(&OnEncodedVideoFrame)); |
+ } |
+ |
+ // Encode 3 frames. |
+ for (int frame_id = 0; frame_id < 3; ++frame_id) { |
+ SendVideoFrame(frame_id, testing_clock_.NowTicks()); |
+ RunTasks(33); |
+ } |
+ |
+ // Send Pli from receiver. |
+ RtcpCastMessage cast_message_(video_sender_config_.ssrc); |
+ cast_message_.picture_loss_indicator = true; |
+ transport_receiver_->SendRtcpFromRtpReceiver( |
+ video_sender_config_.receiver_ssrc, video_sender_config_.ssrc, |
+ media::cast::RtcpTimeData(), &cast_message_, base::TimeDelta(), nullptr, |
+ nullptr); |
+ // Encode the 4th frame. |
+ SendVideoFrame( |
+ 3, testing_clock_.NowTicks() + base::TimeDelta::FromMilliseconds(500)); |
+ RunTasks(33); |
+} |
+ |
} // namespace cast |
} // namespace media |