Allow setContentDecryptionModule() to get called when setting is done.

media/filters/decrypting_demuxer_stream_unittest.cc

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 fec56248e0bc97e5063ed3b8eaa244e3e0b8a584..d91a4a7d92032acd8e6e7899b9298b4eed5fc71d 100644
--- a/media/filters/decrypting_demuxer_stream_unittest.cc
+++ b/media/filters/decrypting_demuxer_stream_unittest.cc
@@ -17,6 +17,7 @@
 #include "testing/gmock/include/gmock/gmock.h"
 
 using ::testing::_;
+using ::testing::Eq;
 using ::testing::IsNull;
 using ::testing::Return;
 using ::testing::SaveArg;
@@ -53,9 +54,9 @@ ACTION_P(ReturnBuffer, 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) {
+ACTION_P3(SetDecryptorIfNotNull, decryptor, done_cb, is_decryptor_set) {
   if (!arg0.is_null())
-    arg0.Run(decryptor);
+    arg0.Run(decryptor, done_cb);
 
   *is_decryptor_set = !arg0.is_null() && decryptor;
 }
@@ -119,7 +120,12 @@ class DecryptingDemuxerStreamTest : public testing::Test {
 
   void Initialize() {
     EXPECT_CALL(*this, RequestDecryptorNotification(_))
-        .WillOnce(SetDecryptorIfNotNull(decryptor_.get(), &is_decryptor_set_));
+        .WillOnce(SetDecryptorIfNotNull(
+            decryptor_.get(),
+            base::Bind(&DecryptingDemuxerStreamTest::DecryptorSet,
+                       base::Unretained(this)),
+            &is_decryptor_set_));
+    EXPECT_CALL(*this, DecryptorSet(Eq(true)));
     EXPECT_CALL(*decryptor_, RegisterNewKeyCB(Decryptor::kAudio, _))
         .WillOnce(SaveArg<1>(&key_added_cb_));
 
@@ -249,6 +255,8 @@ class DecryptingDemuxerStreamTest : public testing::Test {
   MOCK_METHOD2(BufferReady, void(DemuxerStream::Status,
                                  const scoped_refptr<DecoderBuffer>&));
 
+  MOCK_METHOD1(DecryptorSet, void(bool));
+
   base::MessageLoop message_loop_;
   scoped_ptr<DecryptingDemuxerStream> demuxer_stream_;
   scoped_ptr<StrictMock<MockDecryptor> > decryptor_;
@@ -277,7 +285,12 @@ TEST_F(DecryptingDemuxerStreamTest, Initialize_NormalAudio) {
 
 TEST_F(DecryptingDemuxerStreamTest, Initialize_NormalVideo) {
   EXPECT_CALL(*this, RequestDecryptorNotification(_))
-      .WillOnce(SetDecryptorIfNotNull(decryptor_.get(), &is_decryptor_set_));
+      .WillOnce(SetDecryptorIfNotNull(
+          decryptor_.get(),
+          base::Bind(&DecryptingDemuxerStreamTest::DecryptorSet,
+                     base::Unretained(this)),
+          &is_decryptor_set_));
+  EXPECT_CALL(*this, DecryptorSet(Eq(true)));
   EXPECT_CALL(*decryptor_, RegisterNewKeyCB(Decryptor::kVideo, _))
       .WillOnce(SaveArg<1>(&key_added_cb_));
 
@@ -304,8 +317,12 @@ TEST_F(DecryptingDemuxerStreamTest, Initialize_NormalVideo) {
 
 TEST_F(DecryptingDemuxerStreamTest, Initialize_NullDecryptor) {
   EXPECT_CALL(*this, RequestDecryptorNotification(_))
-      .WillRepeatedly(SetDecryptorIfNotNull(static_cast<Decryptor*>(NULL),
-                                            &is_decryptor_set_));
+      .WillRepeatedly(SetDecryptorIfNotNull(
+          static_cast<Decryptor*>(NULL),
+          base::Bind(&DecryptingDemuxerStreamTest::DecryptorSet,
+                     base::Unretained(this)),
+          &is_decryptor_set_));
+  EXPECT_CALL(*this, DecryptorSet(Eq(false)));
 
   AudioDecoderConfig input_config(kCodecVorbis, kSampleFormatPlanarF32,
                                   CHANNEL_LAYOUT_STEREO, 44100, NULL, 0, true);