src/media/filters/decrypting_demuxer_stream_unittest.cc - cobalt - Git at Google

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.

 #include <string>
 #include <vector>

 #include "base/bind.h"
 #include "base/callback_helpers.h"
 #include "base/message_loop.h"
 #include "media/base/decoder_buffer.h"
 #include "media/base/decrypt_config.h"
 #include "media/base/gmock_callback_support.h"
 #include "media/base/mock_filters.h"
 #include "media/base/test_helpers.h"
 #include "media/filters/decrypting_demuxer_stream.h"
 #include "testing/gmock/include/gmock/gmock.h"

 using ::testing::_;
 using ::testing::IsNull;
 using ::testing::Return;
 using ::testing::ReturnRef;
 using ::testing::SaveArg;
 using ::testing::StrictMock;

 namespace media {

 static const int kFakeBufferSize = 16;
 static const VideoFrame::Format kVideoFormat = VideoFrame::YV12;
 static const gfx::Size kCodedSize(320, 240);
 static const gfx::Rect kVisibleRect(320, 240);
 static const gfx::Size kNaturalSize(320, 240);
 static const uint8 kFakeKeyId[] = { 0x4b, 0x65, 0x79, 0x20, 0x49, 0x44 };
 static const uint8 kFakeIv[DecryptConfig::kDecryptionKeySize] = { 0 };

 // Create a fake non-empty encrypted buffer.
 static scoped_refptr<DecoderBuffer> CreateFakeEncryptedBuffer() {
   scoped_refptr<DecoderBuffer> buffer(new DecoderBuffer(kFakeBufferSize));
   buffer->SetDecryptConfig(scoped_ptr<DecryptConfig>(new DecryptConfig(
       std::string(reinterpret_cast<const char*>(kFakeKeyId),
                   arraysize(kFakeKeyId)),
       std::string(reinterpret_cast<const char*>(kFakeIv), arraysize(kFakeIv)),
       0,
       std::vector<SubsampleEntry>())));
   return buffer;
 }

 // Use anonymous namespace here to prevent the actions to be defined multiple
 // times across multiple test files. Sadly we can't use static for them.
 namespace {

 ACTION_P(ReturnBuffer, buffer) {
   arg0.Run(buffer ? DemuxerStream::kOk : DemuxerStream::kAborted, buffer);
 }

 ACTION_P(RunCallbackIfNotNull, param) {
   if (!arg0.is_null())
     arg0.Run(param);
 }

 ACTION_P2(ResetAndRunCallback, callback, param) {
   base::ResetAndReturn(callback).Run(param);
 }

 MATCHER(IsEndOfStream, "end of stream") {
   return (arg->IsEndOfStream());
 }

 }  // namespace

 class DecryptingDemuxerStreamTest : public testing::Test {
  public:
   DecryptingDemuxerStreamTest()
       : demuxer_stream_(new DecryptingDemuxerStream(
             message_loop_.message_loop_proxy(),
             base::Bind(
                 &DecryptingDemuxerStreamTest::RequestDecryptorNotification,
                 base::Unretained(this)))),
         decryptor_(new StrictMock<MockDecryptor>()),
         input_audio_stream_(new StrictMock<MockDemuxerStream>()),
         input_video_stream_(new StrictMock<MockDemuxerStream>()),
         encrypted_buffer_(CreateFakeEncryptedBuffer()),
         decrypted_buffer_(new DecoderBuffer(kFakeBufferSize)) {
   }

   void InitializeAudioAndExpectStatus(const AudioDecoderConfig& config,
                                       PipelineStatus status) {
     EXPECT_CALL(*input_audio_stream_, audio_decoder_config())
         .WillRepeatedly(ReturnRef(config));
     EXPECT_CALL(*input_audio_stream_, type())
         .WillRepeatedly(Return(DemuxerStream::AUDIO));

     demuxer_stream_->Initialize(input_audio_stream_,
                                 NewExpectedStatusCB(status));
     message_loop_.RunUntilIdle();
   }

   void InitializeVideoAndExpectStatus(const VideoDecoderConfig& config,
                                       PipelineStatus status) {
     EXPECT_CALL(*input_video_stream_, video_decoder_config())
         .WillRepeatedly(ReturnRef(config));
     EXPECT_CALL(*input_video_stream_, type())
         .WillRepeatedly(Return(DemuxerStream::VIDEO));

     demuxer_stream_->Initialize(input_video_stream_,
                                 NewExpectedStatusCB(status));
     message_loop_.RunUntilIdle();
   }

   // The following functions are used to test stream-type-neutral logic in
   // DecryptingDemuxerStream. Therefore, we don't specify audio or video in the
   // function names. But for testing purpose, they all use an audio input
   // demuxer stream.

   void Initialize() {
     EXPECT_CALL(*this, RequestDecryptorNotification(_))
         .WillOnce(RunCallbackIfNotNull(decryptor_.get()));
     EXPECT_CALL(*decryptor_, RegisterKeyAddedCB(Decryptor::kAudio, _))
         .WillOnce(SaveArg<1>(&key_added_cb_));

     AudioDecoderConfig input_config(
         kCodecVorbis, 16, CHANNEL_LAYOUT_STEREO, 44100, NULL, 0, true);
     InitializeAudioAndExpectStatus(input_config, PIPELINE_OK);

     const AudioDecoderConfig& output_config =
         demuxer_stream_->audio_decoder_config();
     EXPECT_EQ(DemuxerStream::AUDIO, demuxer_stream_->type());
     EXPECT_FALSE(output_config.is_encrypted());
     EXPECT_EQ(16, output_config.bits_per_channel());
     EXPECT_EQ(CHANNEL_LAYOUT_STEREO, output_config.channel_layout());
     EXPECT_EQ(44100, output_config.samples_per_second());
   }

   void ReadAndExpectBufferReadyWith(
       DemuxerStream::Status status,
       const scoped_refptr<DecoderBuffer>& decrypted_buffer) {
     if (status != DemuxerStream::kOk)
       EXPECT_CALL(*this, BufferReady(status, IsNull()));
     else if (decrypted_buffer->IsEndOfStream())
       EXPECT_CALL(*this, BufferReady(status, IsEndOfStream()));
     else
       EXPECT_CALL(*this, BufferReady(status, decrypted_buffer));

     demuxer_stream_->Read(base::Bind(&DecryptingDemuxerStreamTest::BufferReady,
                                      base::Unretained(this)));
     message_loop_.RunUntilIdle();
   }

   // Sets up expectations and actions to put DecryptingDemuxerStream in an
   // active normal reading state.
   void EnterNormalReadingState() {
     EXPECT_CALL(*input_audio_stream_, Read(_))
         .WillOnce(ReturnBuffer(encrypted_buffer_));
     EXPECT_CALL(*decryptor_, Decrypt(_, _, _))
         .WillOnce(RunCallback<2>(Decryptor::kSuccess, decrypted_buffer_));

     ReadAndExpectBufferReadyWith(DemuxerStream::kOk, decrypted_buffer_);
   }

   // Make the read callback pending by saving and not firing it.
   void EnterPendingReadState() {
     EXPECT_TRUE(pending_demuxer_read_cb_.is_null());
     EXPECT_CALL(*input_audio_stream_, Read(_))
         .WillOnce(SaveArg<0>(&pending_demuxer_read_cb_));
     demuxer_stream_->Read(base::Bind(&DecryptingDemuxerStreamTest::BufferReady,
                                      base::Unretained(this)));
     message_loop_.RunUntilIdle();
     // Make sure the Read() triggers a Read() on the input demuxer stream.
     EXPECT_FALSE(pending_demuxer_read_cb_.is_null());
   }

   // Make the decrypt callback pending by saving and not firing it.
   void EnterPendingDecryptState() {
     EXPECT_TRUE(pending_decrypt_cb_.is_null());
     EXPECT_CALL(*input_audio_stream_, Read(_))
         .WillRepeatedly(ReturnBuffer(encrypted_buffer_));
     EXPECT_CALL(*decryptor_, Decrypt(_, encrypted_buffer_, _))
         .WillOnce(SaveArg<2>(&pending_decrypt_cb_));

     demuxer_stream_->Read(base::Bind(&DecryptingDemuxerStreamTest::BufferReady,
                                      base::Unretained(this)));
     message_loop_.RunUntilIdle();
     // Make sure Read() triggers a Decrypt() on the decryptor.
     EXPECT_FALSE(pending_decrypt_cb_.is_null());
   }

   void EnterWaitingForKeyState() {
     EXPECT_CALL(*input_audio_stream_, Read(_))
         .WillRepeatedly(ReturnBuffer(encrypted_buffer_));
     EXPECT_CALL(*decryptor_, Decrypt(_, encrypted_buffer_, _))
         .WillRepeatedly(RunCallback<2>(Decryptor::kNoKey,
                                        scoped_refptr<DecoderBuffer>()));
     demuxer_stream_->Read(base::Bind(&DecryptingDemuxerStreamTest::BufferReady,
                                      base::Unretained(this)));
     message_loop_.RunUntilIdle();
   }

   void AbortPendingDecryptCB() {
     if (!pending_decrypt_cb_.is_null()) {
       base::ResetAndReturn(&pending_decrypt_cb_).Run(Decryptor::kSuccess, NULL);
     }
   }

   void Reset() {
     EXPECT_CALL(*decryptor_, CancelDecrypt(Decryptor::kAudio))
         .WillRepeatedly(InvokeWithoutArgs(
             this, &DecryptingDemuxerStreamTest::AbortPendingDecryptCB));

     demuxer_stream_->Reset(NewExpectedClosure());
     message_loop_.RunUntilIdle();
   }

   MOCK_METHOD1(RequestDecryptorNotification, void(const DecryptorReadyCB&));

   MOCK_METHOD2(BufferReady, void(DemuxerStream::Status,
                                  const scoped_refptr<DecoderBuffer>&));

   MessageLoop message_loop_;
   scoped_refptr<DecryptingDemuxerStream> demuxer_stream_;
   scoped_ptr<StrictMock<MockDecryptor> > decryptor_;
   scoped_refptr<StrictMock<MockDemuxerStream> > input_audio_stream_;
   scoped_refptr<StrictMock<MockDemuxerStream> > input_video_stream_;

   DemuxerStream::ReadCB pending_demuxer_read_cb_;
   Decryptor::KeyAddedCB key_added_cb_;
   Decryptor::DecryptCB pending_decrypt_cb_;

   // Constant buffers to be returned by the input demuxer streams and the
   // |decryptor_|.
   scoped_refptr<DecoderBuffer> encrypted_buffer_;
   scoped_refptr<DecoderBuffer> decrypted_buffer_;

  private:
   DISALLOW_COPY_AND_ASSIGN(DecryptingDemuxerStreamTest);
 };

 TEST_F(DecryptingDemuxerStreamTest, Initialize_NormalAudio) {
   Initialize();
 }

 TEST_F(DecryptingDemuxerStreamTest, Initialize_NormalVideo) {
   EXPECT_CALL(*this, RequestDecryptorNotification(_))
       .WillOnce(RunCallbackIfNotNull(decryptor_.get()));
   EXPECT_CALL(*decryptor_, RegisterKeyAddedCB(Decryptor::kVideo, _))
         .WillOnce(SaveArg<1>(&key_added_cb_));

   VideoDecoderConfig config(kCodecVP8, VIDEO_CODEC_PROFILE_UNKNOWN,
                             kVideoFormat,
                             kCodedSize, kVisibleRect, kNaturalSize,
                             NULL, 0, true);
   InitializeVideoAndExpectStatus(config, PIPELINE_OK);

   const VideoDecoderConfig& output_config =
       demuxer_stream_->video_decoder_config();
   EXPECT_EQ(DemuxerStream::VIDEO, demuxer_stream_->type());
   EXPECT_FALSE(output_config.is_encrypted());
   EXPECT_EQ(kCodecVP8, output_config.codec());
   EXPECT_EQ(kVideoFormat, output_config.format());
   EXPECT_EQ(VIDEO_CODEC_PROFILE_UNKNOWN, output_config.profile());
   EXPECT_EQ(kCodedSize, output_config.coded_size());
   EXPECT_EQ(kVisibleRect, output_config.visible_rect());
   EXPECT_EQ(kNaturalSize, output_config.natural_size());
   EXPECT_FALSE(output_config.extra_data());
   EXPECT_EQ(0u, output_config.extra_data_size());
 }

 TEST_F(DecryptingDemuxerStreamTest, Initialize_NullDecryptor) {
   EXPECT_CALL(*this, RequestDecryptorNotification(_))
       .WillRepeatedly(RunCallbackIfNotNull(static_cast<Decryptor*>(NULL)));

   AudioDecoderConfig input_config(kCodecVorbis, kSampleFormatPlanarF32,
                                   CHANNEL_LAYOUT_STEREO, 44100, NULL, 0, true);
   InitializeAudioAndExpectStatus(input_config, DECODER_ERROR_NOT_SUPPORTED);
 }

 // Test normal read case.
 TEST_F(DecryptingDemuxerStreamTest, Read_Normal) {
   Initialize();
   EnterNormalReadingState();
 }

 // Test the case where the decryptor returns error during read.
 TEST_F(DecryptingDemuxerStreamTest, Read_DecryptError) {
   Initialize();

   EXPECT_CALL(*input_audio_stream_, Read(_))
       .WillRepeatedly(ReturnBuffer(encrypted_buffer_));
   EXPECT_CALL(*decryptor_, Decrypt(_, encrypted_buffer_, _))
       .WillRepeatedly(RunCallback<2>(Decryptor::kError,
                                      scoped_refptr<DecoderBuffer>()));
   ReadAndExpectBufferReadyWith(DemuxerStream::kAborted, NULL);
 }

 // Test the case where the input is an end-of-stream buffer.
 TEST_F(DecryptingDemuxerStreamTest, Read_EndOfStream) {
   Initialize();
   EnterNormalReadingState();

   // No Decryptor::Decrypt() call is expected for EOS buffer.
   EXPECT_CALL(*input_audio_stream_, Read(_))
       .WillOnce(ReturnBuffer(DecoderBuffer::CreateEOSBuffer()));

   ReadAndExpectBufferReadyWith(DemuxerStream::kOk,
                                DecoderBuffer::CreateEOSBuffer());
 }

 // Test the case where the a key is added when the decryptor is in
 // kWaitingForKey state.
 TEST_F(DecryptingDemuxerStreamTest, KeyAdded_DuringWaitingForKey) {
   Initialize();
   EnterWaitingForKeyState();

   EXPECT_CALL(*decryptor_, Decrypt(_, encrypted_buffer_, _))
       .WillRepeatedly(RunCallback<2>(Decryptor::kSuccess, decrypted_buffer_));
   EXPECT_CALL(*this, BufferReady(DemuxerStream::kOk, decrypted_buffer_));
   key_added_cb_.Run();
   message_loop_.RunUntilIdle();
 }

 // Test the case where the a key is added when the decryptor is in
 // kPendingDecrypt state.
 TEST_F(DecryptingDemuxerStreamTest, KeyAdded_DruingPendingDecrypt) {
   Initialize();
   EnterPendingDecryptState();

   EXPECT_CALL(*decryptor_, Decrypt(_, encrypted_buffer_, _))
       .WillRepeatedly(RunCallback<2>(Decryptor::kSuccess, decrypted_buffer_));
   EXPECT_CALL(*this, BufferReady(DemuxerStream::kOk, decrypted_buffer_));
   // The decrypt callback is returned after the correct decryption key is added.
   key_added_cb_.Run();
   base::ResetAndReturn(&pending_decrypt_cb_).Run(Decryptor::kNoKey, NULL);
   message_loop_.RunUntilIdle();
 }

 // Test resetting when the DecryptingDemuxerStream is in kIdle state but has
 // not returned any buffer.
 TEST_F(DecryptingDemuxerStreamTest, Reset_DuringIdleAfterInitialization) {
   Initialize();
   Reset();
 }

 // Test resetting when the DecryptingDemuxerStream is in kIdle state after it
 // has returned one buffer.
 TEST_F(DecryptingDemuxerStreamTest, Reset_DuringIdleAfterReadOneBuffer) {
   Initialize();
   EnterNormalReadingState();
   Reset();
 }

 // Test resetting when DecryptingDemuxerStream is in kPendingDemuxerRead state.
 TEST_F(DecryptingDemuxerStreamTest, Reset_DuringPendingDemuxerRead) {
   Initialize();
   EnterPendingReadState();

   EXPECT_CALL(*this, BufferReady(DemuxerStream::kAborted, IsNull()));

   Reset();
   base::ResetAndReturn(&pending_demuxer_read_cb_).Run(DemuxerStream::kOk,
                                                       encrypted_buffer_);
   message_loop_.RunUntilIdle();
 }

 // Test resetting when the DecryptingDemuxerStream is in kPendingDecrypt state.
 TEST_F(DecryptingDemuxerStreamTest, Reset_DuringPendingDecrypt) {
   Initialize();
   EnterPendingDecryptState();

   EXPECT_CALL(*this, BufferReady(DemuxerStream::kAborted, IsNull()));

   Reset();
 }

 // Test resetting when the DecryptingDemuxerStream is in kWaitingForKey state.
 TEST_F(DecryptingDemuxerStreamTest, Reset_DuringWaitingForKey) {
   Initialize();
   EnterWaitingForKeyState();

   EXPECT_CALL(*this, BufferReady(DemuxerStream::kAborted, IsNull()));

   Reset();
 }

 // Test resetting after the DecryptingDemuxerStream has been reset.
 TEST_F(DecryptingDemuxerStreamTest, Reset_AfterReset) {
   Initialize();
   EnterNormalReadingState();
   Reset();
   Reset();
 }

 // Test aborted read on the demuxer stream.
 TEST_F(DecryptingDemuxerStreamTest, DemuxerRead_Aborted) {
   Initialize();

   // ReturnBuffer() with NULL triggers aborted demuxer read.
   EXPECT_CALL(*input_audio_stream_, Read(_))
       .WillOnce(ReturnBuffer(scoped_refptr<DecoderBuffer>()));

   ReadAndExpectBufferReadyWith(DemuxerStream::kAborted, NULL);
 }

 // Test aborted read on the input demuxer stream when the
 // DecryptingDemuxerStream is being reset.
 TEST_F(DecryptingDemuxerStreamTest, DemuxerRead_AbortedDuringReset) {
   Initialize();
   EnterPendingReadState();

   // Make sure we get a NULL audio frame returned.
   EXPECT_CALL(*this, BufferReady(DemuxerStream::kAborted, IsNull()));

   Reset();
   base::ResetAndReturn(&pending_demuxer_read_cb_).Run(DemuxerStream::kAborted,
                                                       NULL);
   message_loop_.RunUntilIdle();
 }

 // Test config change on the input demuxer stream.
 TEST_F(DecryptingDemuxerStreamTest, DemuxerRead_ConfigChanged) {
   Initialize();

   AudioDecoderConfig new_config(
       kCodecVorbis, 32, CHANNEL_LAYOUT_STEREO, 88200, NULL, 0, true);

   EXPECT_CALL(*input_audio_stream_, audio_decoder_config())
       .WillRepeatedly(ReturnRef(new_config));

   EXPECT_CALL(*input_audio_stream_, Read(_))
       .WillOnce(RunCallback<0>(DemuxerStream::kConfigChanged,
                                scoped_refptr<DecoderBuffer>()));

   ReadAndExpectBufferReadyWith(DemuxerStream::kConfigChanged, NULL);
 }

 }  // namespace media
	// Copyright (c) 2012 The Chromium Authors. All rights reserved.
	// Use of this source code is governed by a BSD-style license that can be
	// found in the LICENSE file.

	#include <string>
	#include <vector>

	#include "base/bind.h"
	#include "base/callback_helpers.h"
	#include "base/message_loop.h"
	#include "media/base/decoder_buffer.h"
	#include "media/base/decrypt_config.h"
	#include "media/base/gmock_callback_support.h"
	#include "media/base/mock_filters.h"
	#include "media/base/test_helpers.h"
	#include "media/filters/decrypting_demuxer_stream.h"
	#include "testing/gmock/include/gmock/gmock.h"

	using ::testing::_;
	using ::testing::IsNull;
	using ::testing::Return;
	using ::testing::ReturnRef;
	using ::testing::SaveArg;
	using ::testing::StrictMock;

	namespace media {

	static const int kFakeBufferSize = 16;
	static const VideoFrame::Format kVideoFormat = VideoFrame::YV12;
	static const gfx::Size kCodedSize(320, 240);
	static const gfx::Rect kVisibleRect(320, 240);
	static const gfx::Size kNaturalSize(320, 240);
	static const uint8 kFakeKeyId[] = { 0x4b, 0x65, 0x79, 0x20, 0x49, 0x44 };
	static const uint8 kFakeIv[DecryptConfig::kDecryptionKeySize] = { 0 };

	// Create a fake non-empty encrypted buffer.
	static scoped_refptr<DecoderBuffer> CreateFakeEncryptedBuffer() {
	scoped_refptr<DecoderBuffer> buffer(new DecoderBuffer(kFakeBufferSize));
	buffer->SetDecryptConfig(scoped_ptr<DecryptConfig>(new DecryptConfig(
	std::string(reinterpret_cast<const char*>(kFakeKeyId),
	arraysize(kFakeKeyId)),
	std::string(reinterpret_cast<const char*>(kFakeIv), arraysize(kFakeIv)),
	0,
	std::vector<SubsampleEntry>())));
	return buffer;
	}

	// Use anonymous namespace here to prevent the actions to be defined multiple
	// times across multiple test files. Sadly we can't use static for them.
	namespace {

	ACTION_P(ReturnBuffer, buffer) {
	arg0.Run(buffer ? DemuxerStream::kOk : DemuxerStream::kAborted, buffer);
	}

	ACTION_P(RunCallbackIfNotNull, param) {
	if (!arg0.is_null())
	arg0.Run(param);
	}

	ACTION_P2(ResetAndRunCallback, callback, param) {
	base::ResetAndReturn(callback).Run(param);
	}

	MATCHER(IsEndOfStream, "end of stream") {
	return (arg->IsEndOfStream());
	}

	} // namespace

	class DecryptingDemuxerStreamTest : public testing::Test {
	public:
	DecryptingDemuxerStreamTest()
	: demuxer_stream_(new DecryptingDemuxerStream(
	message_loop_.message_loop_proxy(),
	base::Bind(
	&DecryptingDemuxerStreamTest::RequestDecryptorNotification,
	base::Unretained(this)))),
	decryptor_(new StrictMock<MockDecryptor>()),
	input_audio_stream_(new StrictMock<MockDemuxerStream>()),
	input_video_stream_(new StrictMock<MockDemuxerStream>()),
	encrypted_buffer_(CreateFakeEncryptedBuffer()),
	decrypted_buffer_(new DecoderBuffer(kFakeBufferSize)) {
	}

	void InitializeAudioAndExpectStatus(const AudioDecoderConfig& config,
	PipelineStatus status) {
	EXPECT_CALL(*input_audio_stream_, audio_decoder_config())
	.WillRepeatedly(ReturnRef(config));
	EXPECT_CALL(*input_audio_stream_, type())
	.WillRepeatedly(Return(DemuxerStream::AUDIO));

	demuxer_stream_->Initialize(input_audio_stream_,
	NewExpectedStatusCB(status));
	message_loop_.RunUntilIdle();
	}

	void InitializeVideoAndExpectStatus(const VideoDecoderConfig& config,
	PipelineStatus status) {
	EXPECT_CALL(*input_video_stream_, video_decoder_config())
	.WillRepeatedly(ReturnRef(config));
	EXPECT_CALL(*input_video_stream_, type())
	.WillRepeatedly(Return(DemuxerStream::VIDEO));

	demuxer_stream_->Initialize(input_video_stream_,
	NewExpectedStatusCB(status));
	message_loop_.RunUntilIdle();
	}

	// The following functions are used to test stream-type-neutral logic in
	// DecryptingDemuxerStream. Therefore, we don't specify audio or video in the
	// function names. But for testing purpose, they all use an audio input
	// demuxer stream.

	void Initialize() {
	EXPECT_CALL(*this, RequestDecryptorNotification(_))
	.WillOnce(RunCallbackIfNotNull(decryptor_.get()));
	EXPECT_CALL(*decryptor_, RegisterKeyAddedCB(Decryptor::kAudio, _))
	.WillOnce(SaveArg<1>(&key_added_cb_));

	AudioDecoderConfig input_config(
	kCodecVorbis, 16, CHANNEL_LAYOUT_STEREO, 44100, NULL, 0, true);
	InitializeAudioAndExpectStatus(input_config, PIPELINE_OK);

	const AudioDecoderConfig& output_config =
	demuxer_stream_->audio_decoder_config();
	EXPECT_EQ(DemuxerStream::AUDIO, demuxer_stream_->type());
	EXPECT_FALSE(output_config.is_encrypted());
	EXPECT_EQ(16, output_config.bits_per_channel());
	EXPECT_EQ(CHANNEL_LAYOUT_STEREO, output_config.channel_layout());
	EXPECT_EQ(44100, output_config.samples_per_second());
	}

	void ReadAndExpectBufferReadyWith(
	DemuxerStream::Status status,
	const scoped_refptr<DecoderBuffer>& decrypted_buffer) {
	if (status != DemuxerStream::kOk)
	EXPECT_CALL(*this, BufferReady(status, IsNull()));
	else if (decrypted_buffer->IsEndOfStream())
	EXPECT_CALL(*this, BufferReady(status, IsEndOfStream()));
	else
	EXPECT_CALL(*this, BufferReady(status, decrypted_buffer));

	demuxer_stream_->Read(base::Bind(&DecryptingDemuxerStreamTest::BufferReady,
	base::Unretained(this)));
	message_loop_.RunUntilIdle();
	}

	// Sets up expectations and actions to put DecryptingDemuxerStream in an
	// active normal reading state.
	void EnterNormalReadingState() {
	EXPECT_CALL(*input_audio_stream_, Read(_))
	.WillOnce(ReturnBuffer(encrypted_buffer_));
	EXPECT_CALL(*decryptor_, Decrypt(_, _, _))
	.WillOnce(RunCallback<2>(Decryptor::kSuccess, decrypted_buffer_));

	ReadAndExpectBufferReadyWith(DemuxerStream::kOk, decrypted_buffer_);
	}

	// Make the read callback pending by saving and not firing it.
	void EnterPendingReadState() {
	EXPECT_TRUE(pending_demuxer_read_cb_.is_null());
	EXPECT_CALL(*input_audio_stream_, Read(_))
	.WillOnce(SaveArg<0>(&pending_demuxer_read_cb_));
	demuxer_stream_->Read(base::Bind(&DecryptingDemuxerStreamTest::BufferReady,
	base::Unretained(this)));
	message_loop_.RunUntilIdle();
	// Make sure the Read() triggers a Read() on the input demuxer stream.
	EXPECT_FALSE(pending_demuxer_read_cb_.is_null());
	}

	// Make the decrypt callback pending by saving and not firing it.
	void EnterPendingDecryptState() {
	EXPECT_TRUE(pending_decrypt_cb_.is_null());
	EXPECT_CALL(*input_audio_stream_, Read(_))
	.WillRepeatedly(ReturnBuffer(encrypted_buffer_));
	EXPECT_CALL(*decryptor_, Decrypt(_, encrypted_buffer_, _))
	.WillOnce(SaveArg<2>(&pending_decrypt_cb_));

	demuxer_stream_->Read(base::Bind(&DecryptingDemuxerStreamTest::BufferReady,
	base::Unretained(this)));
	message_loop_.RunUntilIdle();
	// Make sure Read() triggers a Decrypt() on the decryptor.
	EXPECT_FALSE(pending_decrypt_cb_.is_null());
	}

	void EnterWaitingForKeyState() {
	EXPECT_CALL(*input_audio_stream_, Read(_))
	.WillRepeatedly(ReturnBuffer(encrypted_buffer_));
	EXPECT_CALL(*decryptor_, Decrypt(_, encrypted_buffer_, _))
	.WillRepeatedly(RunCallback<2>(Decryptor::kNoKey,
	scoped_refptr<DecoderBuffer>()));
	demuxer_stream_->Read(base::Bind(&DecryptingDemuxerStreamTest::BufferReady,
	base::Unretained(this)));
	message_loop_.RunUntilIdle();
	}

	void AbortPendingDecryptCB() {
	if (!pending_decrypt_cb_.is_null()) {
	base::ResetAndReturn(&pending_decrypt_cb_).Run(Decryptor::kSuccess, NULL);
	}
	}

	void Reset() {
	EXPECT_CALL(*decryptor_, CancelDecrypt(Decryptor::kAudio))
	.WillRepeatedly(InvokeWithoutArgs(
	this, &DecryptingDemuxerStreamTest::AbortPendingDecryptCB));

	demuxer_stream_->Reset(NewExpectedClosure());
	message_loop_.RunUntilIdle();
	}

	MOCK_METHOD1(RequestDecryptorNotification, void(const DecryptorReadyCB&));

	MOCK_METHOD2(BufferReady, void(DemuxerStream::Status,
	const scoped_refptr<DecoderBuffer>&));

	MessageLoop message_loop_;
	scoped_refptr<DecryptingDemuxerStream> demuxer_stream_;
	scoped_ptr<StrictMock<MockDecryptor> > decryptor_;
	scoped_refptr<StrictMock<MockDemuxerStream> > input_audio_stream_;
	scoped_refptr<StrictMock<MockDemuxerStream> > input_video_stream_;

	DemuxerStream::ReadCB pending_demuxer_read_cb_;
	Decryptor::KeyAddedCB key_added_cb_;
	Decryptor::DecryptCB pending_decrypt_cb_;

	// Constant buffers to be returned by the input demuxer streams and the
	// \|decryptor_\|.
	scoped_refptr<DecoderBuffer> encrypted_buffer_;
	scoped_refptr<DecoderBuffer> decrypted_buffer_;

	private:
	DISALLOW_COPY_AND_ASSIGN(DecryptingDemuxerStreamTest);
	};

	TEST_F(DecryptingDemuxerStreamTest, Initialize_NormalAudio) {
	Initialize();
	}

	TEST_F(DecryptingDemuxerStreamTest, Initialize_NormalVideo) {
	EXPECT_CALL(*this, RequestDecryptorNotification(_))
	.WillOnce(RunCallbackIfNotNull(decryptor_.get()));
	EXPECT_CALL(*decryptor_, RegisterKeyAddedCB(Decryptor::kVideo, _))
	.WillOnce(SaveArg<1>(&key_added_cb_));

	VideoDecoderConfig config(kCodecVP8, VIDEO_CODEC_PROFILE_UNKNOWN,
	kVideoFormat,
	kCodedSize, kVisibleRect, kNaturalSize,
	NULL, 0, true);
	InitializeVideoAndExpectStatus(config, PIPELINE_OK);

	const VideoDecoderConfig& output_config =
	demuxer_stream_->video_decoder_config();
	EXPECT_EQ(DemuxerStream::VIDEO, demuxer_stream_->type());
	EXPECT_FALSE(output_config.is_encrypted());
	EXPECT_EQ(kCodecVP8, output_config.codec());
	EXPECT_EQ(kVideoFormat, output_config.format());
	EXPECT_EQ(VIDEO_CODEC_PROFILE_UNKNOWN, output_config.profile());
	EXPECT_EQ(kCodedSize, output_config.coded_size());
	EXPECT_EQ(kVisibleRect, output_config.visible_rect());
	EXPECT_EQ(kNaturalSize, output_config.natural_size());
	EXPECT_FALSE(output_config.extra_data());
	EXPECT_EQ(0u, output_config.extra_data_size());
	}

	TEST_F(DecryptingDemuxerStreamTest, Initialize_NullDecryptor) {
	EXPECT_CALL(*this, RequestDecryptorNotification(_))
	.WillRepeatedly(RunCallbackIfNotNull(static_cast<Decryptor*>(NULL)));

	AudioDecoderConfig input_config(kCodecVorbis, kSampleFormatPlanarF32,
	CHANNEL_LAYOUT_STEREO, 44100, NULL, 0, true);
	InitializeAudioAndExpectStatus(input_config, DECODER_ERROR_NOT_SUPPORTED);
	}

	// Test normal read case.
	TEST_F(DecryptingDemuxerStreamTest, Read_Normal) {
	Initialize();
	EnterNormalReadingState();
	}

	// Test the case where the decryptor returns error during read.
	TEST_F(DecryptingDemuxerStreamTest, Read_DecryptError) {
	Initialize();

	EXPECT_CALL(*input_audio_stream_, Read(_))
	.WillRepeatedly(ReturnBuffer(encrypted_buffer_));
	EXPECT_CALL(*decryptor_, Decrypt(_, encrypted_buffer_, _))
	.WillRepeatedly(RunCallback<2>(Decryptor::kError,
	scoped_refptr<DecoderBuffer>()));
	ReadAndExpectBufferReadyWith(DemuxerStream::kAborted, NULL);
	}

	// Test the case where the input is an end-of-stream buffer.
	TEST_F(DecryptingDemuxerStreamTest, Read_EndOfStream) {
	Initialize();
	EnterNormalReadingState();

	// No Decryptor::Decrypt() call is expected for EOS buffer.
	EXPECT_CALL(*input_audio_stream_, Read(_))
	.WillOnce(ReturnBuffer(DecoderBuffer::CreateEOSBuffer()));

	ReadAndExpectBufferReadyWith(DemuxerStream::kOk,
	DecoderBuffer::CreateEOSBuffer());
	}

	// Test the case where the a key is added when the decryptor is in
	// kWaitingForKey state.
	TEST_F(DecryptingDemuxerStreamTest, KeyAdded_DuringWaitingForKey) {
	Initialize();
	EnterWaitingForKeyState();

	EXPECT_CALL(*decryptor_, Decrypt(_, encrypted_buffer_, _))
	.WillRepeatedly(RunCallback<2>(Decryptor::kSuccess, decrypted_buffer_));
	EXPECT_CALL(*this, BufferReady(DemuxerStream::kOk, decrypted_buffer_));
	key_added_cb_.Run();
	message_loop_.RunUntilIdle();
	}

	// Test the case where the a key is added when the decryptor is in
	// kPendingDecrypt state.
	TEST_F(DecryptingDemuxerStreamTest, KeyAdded_DruingPendingDecrypt) {
	Initialize();
	EnterPendingDecryptState();

	EXPECT_CALL(*decryptor_, Decrypt(_, encrypted_buffer_, _))
	.WillRepeatedly(RunCallback<2>(Decryptor::kSuccess, decrypted_buffer_));
	EXPECT_CALL(*this, BufferReady(DemuxerStream::kOk, decrypted_buffer_));
	// The decrypt callback is returned after the correct decryption key is added.
	key_added_cb_.Run();
	base::ResetAndReturn(&pending_decrypt_cb_).Run(Decryptor::kNoKey, NULL);
	message_loop_.RunUntilIdle();
	}

	// Test resetting when the DecryptingDemuxerStream is in kIdle state but has
	// not returned any buffer.
	TEST_F(DecryptingDemuxerStreamTest, Reset_DuringIdleAfterInitialization) {
	Initialize();
	Reset();
	}

	// Test resetting when the DecryptingDemuxerStream is in kIdle state after it
	// has returned one buffer.
	TEST_F(DecryptingDemuxerStreamTest, Reset_DuringIdleAfterReadOneBuffer) {
	Initialize();
	EnterNormalReadingState();
	Reset();
	}

	// Test resetting when DecryptingDemuxerStream is in kPendingDemuxerRead state.
	TEST_F(DecryptingDemuxerStreamTest, Reset_DuringPendingDemuxerRead) {
	Initialize();
	EnterPendingReadState();

	EXPECT_CALL(*this, BufferReady(DemuxerStream::kAborted, IsNull()));

	Reset();
	base::ResetAndReturn(&pending_demuxer_read_cb_).Run(DemuxerStream::kOk,
	encrypted_buffer_);
	message_loop_.RunUntilIdle();
	}

	// Test resetting when the DecryptingDemuxerStream is in kPendingDecrypt state.
	TEST_F(DecryptingDemuxerStreamTest, Reset_DuringPendingDecrypt) {
	Initialize();
	EnterPendingDecryptState();

	EXPECT_CALL(*this, BufferReady(DemuxerStream::kAborted, IsNull()));

	Reset();
	}

	// Test resetting when the DecryptingDemuxerStream is in kWaitingForKey state.
	TEST_F(DecryptingDemuxerStreamTest, Reset_DuringWaitingForKey) {
	Initialize();
	EnterWaitingForKeyState();

	EXPECT_CALL(*this, BufferReady(DemuxerStream::kAborted, IsNull()));

	Reset();
	}

	// Test resetting after the DecryptingDemuxerStream has been reset.
	TEST_F(DecryptingDemuxerStreamTest, Reset_AfterReset) {
	Initialize();
	EnterNormalReadingState();
	Reset();
	Reset();
	}

	// Test aborted read on the demuxer stream.
	TEST_F(DecryptingDemuxerStreamTest, DemuxerRead_Aborted) {
	Initialize();

	// ReturnBuffer() with NULL triggers aborted demuxer read.
	EXPECT_CALL(*input_audio_stream_, Read(_))
	.WillOnce(ReturnBuffer(scoped_refptr<DecoderBuffer>()));

	ReadAndExpectBufferReadyWith(DemuxerStream::kAborted, NULL);
	}

	// Test aborted read on the input demuxer stream when the
	// DecryptingDemuxerStream is being reset.
	TEST_F(DecryptingDemuxerStreamTest, DemuxerRead_AbortedDuringReset) {
	Initialize();
	EnterPendingReadState();

	// Make sure we get a NULL audio frame returned.
	EXPECT_CALL(*this, BufferReady(DemuxerStream::kAborted, IsNull()));

	Reset();
	base::ResetAndReturn(&pending_demuxer_read_cb_).Run(DemuxerStream::kAborted,
	NULL);
	message_loop_.RunUntilIdle();
	}

	// Test config change on the input demuxer stream.
	TEST_F(DecryptingDemuxerStreamTest, DemuxerRead_ConfigChanged) {
	Initialize();

	AudioDecoderConfig new_config(
	kCodecVorbis, 32, CHANNEL_LAYOUT_STEREO, 88200, NULL, 0, true);

	EXPECT_CALL(*input_audio_stream_, audio_decoder_config())
	.WillRepeatedly(ReturnRef(new_config));

	EXPECT_CALL(*input_audio_stream_, Read(_))
	.WillOnce(RunCallback<0>(DemuxerStream::kConfigChanged,
	scoped_refptr<DecoderBuffer>()));

	ReadAndExpectBufferReadyWith(DemuxerStream::kConfigChanged, NULL);
	}

	} // namespace media