| Index: media/filters/decrypting_demuxer_stream_unittest.cc
|
| diff --git a/media/filters/decrypting_demuxer_stream_unittest.cc b/media/filters/decrypting_demuxer_stream_unittest.cc
|
| index fa2a36cf947a69408c06dced673665c4fa41ff18..dc72c1c3d34a2e741239662de43f1e6d048e7f15 100644
|
| --- a/media/filters/decrypting_demuxer_stream_unittest.cc
|
| +++ b/media/filters/decrypting_demuxer_stream_unittest.cc
|
| @@ -99,16 +99,20 @@ class DecryptingDemuxerStreamTest : public testing::Test {
|
| void InitializeAudioAndExpectStatus(const AudioDecoderConfig& config,
|
| PipelineStatus status) {
|
| input_audio_stream_->set_audio_decoder_config(config);
|
| - demuxer_stream_->Initialize(input_audio_stream_.get(),
|
| - NewExpectedStatusCB(status));
|
| + demuxer_stream_->Initialize(
|
| + input_audio_stream_.get(), NewExpectedStatusCB(status),
|
| + base::Bind(&DecryptingDemuxerStreamTest::OnWaitingForEncryptionKey,
|
| + base::Unretained(this)));
|
| message_loop_.RunUntilIdle();
|
| }
|
|
|
| void InitializeVideoAndExpectStatus(const VideoDecoderConfig& config,
|
| PipelineStatus status) {
|
| input_video_stream_->set_video_decoder_config(config);
|
| - demuxer_stream_->Initialize(input_video_stream_.get(),
|
| - NewExpectedStatusCB(status));
|
| + demuxer_stream_->Initialize(
|
| + input_video_stream_.get(), NewExpectedStatusCB(status),
|
| + base::Bind(&DecryptingDemuxerStreamTest::OnWaitingForEncryptionKey,
|
| + base::Unretained(this)));
|
| message_loop_.RunUntilIdle();
|
| }
|
|
|
| @@ -225,6 +229,7 @@ class DecryptingDemuxerStreamTest : public testing::Test {
|
| EXPECT_CALL(*decryptor_, Decrypt(_, encrypted_buffer_, _))
|
| .WillRepeatedly(RunCallback<2>(Decryptor::kNoKey,
|
| scoped_refptr<DecoderBuffer>()));
|
| + EXPECT_CALL(*this, OnWaitingForEncryptionKey());
|
| demuxer_stream_->Read(base::Bind(&DecryptingDemuxerStreamTest::BufferReady,
|
| base::Unretained(this)));
|
| message_loop_.RunUntilIdle();
|
| @@ -260,6 +265,8 @@ class DecryptingDemuxerStreamTest : public testing::Test {
|
|
|
| MOCK_METHOD1(DecryptorSet, void(bool));
|
|
|
| + MOCK_METHOD0(OnWaitingForEncryptionKey, void(void));
|
| +
|
| base::MessageLoop message_loop_;
|
| scoped_ptr<DecryptingDemuxerStream> demuxer_stream_;
|
| scoped_ptr<StrictMock<MockDecryptor> > decryptor_;
|
|
|