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_; |