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 1cc6acecc6e657c3a6007fcb0b3ec5caf9c96ccc..bfb221997dd8508cf91670f625e864f187ce2899 100644 |
--- a/media/filters/decrypting_demuxer_stream_unittest.cc |
+++ b/media/filters/decrypting_demuxer_stream_unittest.cc |
@@ -50,6 +50,9 @@ ACTION_P(ReturnBuffer, buffer) { |
arg0.Run(buffer.get() ? DemuxerStream::kOk : DemuxerStream::kAborted, buffer); |
} |
+// Sets the |decryptor| if the DecryptorReadyCB (arg0) is not null. Sets |
+// |is_decryptor_set| to true if a non-NULL |decryptor| has been set through the |
+// callback. |
ACTION_P2(SetDecryptorIfNotNull, decryptor, is_decryptor_set) { |
if (!arg0.is_null()) |
arg0.Run(decryptor); |
@@ -110,6 +113,7 @@ class DecryptingDemuxerStreamTest : public testing::Test { |
void Initialize() { |
EXPECT_CALL(*this, RequestDecryptorNotification(_)) |
.WillOnce(SetDecryptorIfNotNull(decryptor_.get(), &is_decryptor_set_)); |
+ // TODO(xhwang): Use WillOnce after we remove the cancellation of NewKeyCB. |
EXPECT_CALL(*decryptor_, RegisterNewKeyCB(Decryptor::kAudio, _)) |
.WillRepeatedly(SaveArg<1>(&key_added_cb_)); |
@@ -251,7 +255,7 @@ class DecryptingDemuxerStreamTest : public testing::Test { |
base::MessageLoop message_loop_; |
scoped_ptr<DecryptingDemuxerStream> demuxer_stream_; |
scoped_ptr<StrictMock<MockDecryptor> > decryptor_; |
- // Whether a valid Decryptor is set to the |demuxer_stream_|. |
+ // Whether a valid Decryptor has been set in the |demuxer_stream_|. |
bool is_decryptor_set_; |
scoped_ptr<StrictMock<MockDemuxerStream> > input_audio_stream_; |
scoped_ptr<StrictMock<MockDemuxerStream> > input_video_stream_; |