Index: media/renderers/renderer_impl_unittest.cc |
diff --git a/media/renderers/renderer_impl_unittest.cc b/media/renderers/renderer_impl_unittest.cc |
index 8f738b0bd8c4e06b0c2638c73f260e1e680742fd..30bad2fe27040a12da619aa263b819472278cadf 100644 |
--- a/media/renderers/renderer_impl_unittest.cc |
+++ b/media/renderers/renderer_impl_unittest.cc |
@@ -50,6 +50,7 @@ class RendererImplTest : public ::testing::Test { |
MOCK_METHOD1(OnUpdateStatistics, void(const PipelineStatistics&)); |
MOCK_METHOD1(OnBufferingStateChange, void(BufferingState)); |
MOCK_METHOD1(OnVideoFramePaint, void(const scoped_refptr<VideoFrame>&)); |
+ MOCK_METHOD0(OnWaitingForDecryptionKey, void()); |
private: |
DISALLOW_COPY_AND_ASSIGN(CallbackHelper); |
@@ -88,7 +89,7 @@ class RendererImplTest : public ::testing::Test { |
// Sets up expectations to allow the audio renderer to initialize. |
void SetAudioRendererInitializeExpectations(PipelineStatus status) { |
EXPECT_CALL(*audio_renderer_, |
- Initialize(audio_stream_.get(), _, _, _, _, _, _)) |
+ Initialize(audio_stream_.get(), _, _, _, _, _, _, _)) |
.WillOnce(DoAll(SaveArg<4>(&audio_buffering_state_cb_), |
SaveArg<5>(&audio_ended_cb_), |
SaveArg<6>(&audio_error_cb_), RunCallback<1>(status))); |
@@ -97,13 +98,14 @@ class RendererImplTest : public ::testing::Test { |
// Sets up expectations to allow the video renderer to initialize. |
void SetVideoRendererInitializeExpectations(PipelineStatus status) { |
EXPECT_CALL(*video_renderer_, |
- Initialize(video_stream_.get(), _, _, _, _, _, _, _, _)) |
+ Initialize(video_stream_.get(), _, _, _, _, _, _, _, _, _)) |
.WillOnce(DoAll(SaveArg<4>(&video_buffering_state_cb_), |
SaveArg<6>(&video_ended_cb_), RunCallback<1>(status))); |
} |
void InitializeAndExpect(PipelineStatus start_status) { |
EXPECT_CALL(callbacks_, OnInitialize(start_status)); |
+ EXPECT_CALL(callbacks_, OnWaitingForDecryptionKey()).Times(0); |
if (start_status == PIPELINE_OK && audio_stream_) { |
EXPECT_CALL(*audio_renderer_, GetTimeSource()) |
@@ -121,7 +123,9 @@ class RendererImplTest : public ::testing::Test { |
base::Bind(&CallbackHelper::OnVideoFramePaint, |
base::Unretained(&callbacks_)), |
base::Bind(&CallbackHelper::OnEnded, base::Unretained(&callbacks_)), |
- base::Bind(&CallbackHelper::OnError, base::Unretained(&callbacks_))); |
+ base::Bind(&CallbackHelper::OnError, base::Unretained(&callbacks_)), |
+ base::Bind(&CallbackHelper::OnWaitingForDecryptionKey, |
+ base::Unretained(&callbacks_))); |
base::RunLoop().RunUntilIdle(); |
} |
@@ -433,7 +437,7 @@ TEST_F(RendererImplTest, ErrorDuringInitialize) { |
// Force an audio error to occur during video renderer initialization. |
EXPECT_CALL(*video_renderer_, |
- Initialize(video_stream_.get(), _, _, _, _, _, _, _, _)) |
+ Initialize(video_stream_.get(), _, _, _, _, _, _, _, _, _)) |
.WillOnce(DoAll(AudioError(&audio_error_cb_, PIPELINE_ERROR_DECODE), |
SaveArg<4>(&video_buffering_state_cb_), |
SaveArg<6>(&video_ended_cb_), |