Chromium Code Reviews| Index: chromecast/renderer/media/cma_renderer_unittest.cc |
| diff --git a/chromecast/renderer/media/cma_renderer_unittest.cc b/chromecast/renderer/media/cma_renderer_unittest.cc |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..edfa8347c8463ad5029019d9fd5dd4ef490d7686 |
| --- /dev/null |
| +++ b/chromecast/renderer/media/cma_renderer_unittest.cc |
| @@ -0,0 +1,142 @@ |
| +// Copyright 2015 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 "base/bind.h" |
| +#include "base/macros.h" |
| +#include "base/memory/ref_counted.h" |
| +#include "base/memory/scoped_ptr.h" |
| +#include "base/memory/shared_memory.h" |
| +#include "base/message_loop/message_loop.h" |
| +#include "base/time/time.h" |
| +#include "chromecast/common/media/cma_messages.h" |
| +#include "chromecast/common/media/shared_memory_chunk.h" |
| +#include "chromecast/media/cma/base/buffering_defs.h" |
| +#include "chromecast/media/cma/filters/cma_renderer.h" |
| +#include "chromecast/media/cma/ipc/media_memory_chunk.h" |
| +#include "chromecast/media/cma/ipc/media_message_fifo.h" |
| +#include "chromecast/media/cma/pipeline/media_pipeline.h" |
| +#include "chromecast/renderer/media/cma_message_filter_proxy.h" |
| +#include "chromecast/renderer/media/media_pipeline_proxy.h" |
| +#include "ipc/ipc_test_sink.h" |
| +#include "media/base/demuxer_stream_provider.h" |
| +#include "media/base/fake_demuxer_stream.h" |
| +#include "media/base/null_video_sink.h" |
| +#include "testing/gmock/include/gmock/gmock.h" |
| +#include "testing/gtest/include/gtest/gtest.h" |
| + |
| +namespace chromecast { |
| +namespace media { |
| + |
| +namespace { |
| + |
| +class FakeDemuxerStreamProvider : public ::media::DemuxerStreamProvider { |
|
servolk
2015/05/14 20:54:15
Looks like this might be potentially useful for ot
gunsch
2015/05/14 22:01:15
Done. xhwang@, are you okay with this change?
|
| + public: |
| + FakeDemuxerStreamProvider() : fake_stream_(1, 1, false) {} |
| + ~FakeDemuxerStreamProvider() override {} |
| + |
| + ::media::DemuxerStream* GetStream( |
| + ::media::DemuxerStream::Type type) override { |
| + // Note: FakeDemuxerStream only supports video |
| + if (type == ::media::DemuxerStream::Type::AUDIO) |
| + return nullptr; |
| + return &fake_stream_; |
| + } |
| + |
| + private: |
| + ::media::FakeDemuxerStream fake_stream_; |
| + |
| + DISALLOW_COPY_AND_ASSIGN(FakeDemuxerStreamProvider); |
| +}; |
| + |
| +class CmaRendererTest : public testing::Test { |
| + public: |
| + CmaRendererTest() { |
| + null_sink_.reset(new ::media::NullVideoSink( |
| + false, |
| + base::TimeDelta::FromSecondsD(1.0 / 60), |
| + base::Bind(&MockCB::OnFrameReceived, base::Unretained(&mock_)), |
| + message_loop_.task_runner())); |
| + |
| + cma_proxy_ = new CmaMessageFilterProxy(message_loop_.message_loop_proxy()); |
| + cma_proxy_->OnFilterAdded(&ipc_sink_); |
| + |
| + renderer_.reset(new CmaRenderer( |
| + scoped_ptr<MediaPipelineProxy>( |
| + new MediaPipelineProxy(0, message_loop_.message_loop_proxy(), |
| + LoadType::kLoadTypeMediaSource)), |
| + null_sink_.get())); |
| + } |
| + |
| + ~CmaRendererTest() override { |
| + cma_proxy_->OnFilterRemoved(); |
| + } |
| + |
| + protected: |
| + base::MessageLoop message_loop_; |
| + FakeDemuxerStreamProvider demuxer_stream_provider_; |
| + IPC::TestSink ipc_sink_; |
| + scoped_refptr<CmaMessageFilterProxy> cma_proxy_; |
| + scoped_ptr<CmaRenderer> renderer_; |
| + |
| + class MockCB { |
| + public: |
| + MOCK_METHOD1(OnInitialized, void(::media::PipelineStatus)); |
| + MOCK_METHOD1(OnFrameReceived, |
| + void(const scoped_refptr<::media::VideoFrame>&)); |
| + MOCK_METHOD1(OnStatistics, void(const ::media::PipelineStatistics&)); |
| + MOCK_METHOD1(OnBufferingState, void(::media::BufferingState)); |
| + MOCK_METHOD0(OnEnded, void()); |
| + MOCK_METHOD1(OnError, void(::media::PipelineStatus)); |
| + MOCK_METHOD0(OnWaitingForDecryptionKey, void()); |
| + }; |
| + MockCB mock_; |
| + |
| + private: |
| + scoped_ptr<::media::NullVideoSink> null_sink_; |
| + |
| + DISALLOW_COPY_AND_ASSIGN(CmaRendererTest); |
| +}; |
| + |
| +} // namespace |
| + |
| +TEST_F(CmaRendererTest, TestInitialization) { |
| + renderer_->Initialize( |
| + &demuxer_stream_provider_, |
| + base::Bind(&MockCB::OnInitialized, base::Unretained(&mock_)), |
| + base::Bind(&MockCB::OnStatistics, base::Unretained(&mock_)), |
| + base::Bind(&MockCB::OnBufferingState, base::Unretained(&mock_)), |
| + base::Bind(&MockCB::OnEnded, base::Unretained(&mock_)), |
| + base::Bind(&MockCB::OnError, base::Unretained(&mock_)), |
| + base::Bind(&MockCB::OnWaitingForDecryptionKey, base::Unretained(&mock_))); |
| + message_loop_.RunUntilIdle(); |
| + |
| + EXPECT_TRUE(ipc_sink_.GetUniqueMessageMatching(CmaHostMsg_CreateAvPipe::ID)); |
| + base::SharedMemory* shared_memory = new base::SharedMemory(); |
| + EXPECT_TRUE(shared_memory->CreateAndMapAnonymous(kAppVideoBufferSize)); |
| + |
| + // Renderer MediaMessageFifo instance expects the first bytes of the shared |
| + // memory blob to describe how much space is available for writing. |
| + *(static_cast<size_t*>(shared_memory->memory())) = |
| + shared_memory->requested_size() - MediaMessageFifo::kDescriptorSize; |
| + |
| + base::FileDescriptor foreign_socket_handle; |
| + cma_proxy_->OnMessageReceived(CmaMsg_AvPipeCreated( |
| + 1, kVideoTrackId, true, |
| + shared_memory->handle(), foreign_socket_handle)); |
| + message_loop_.RunUntilIdle(); |
| + |
| + EXPECT_FALSE( |
| + ipc_sink_.GetUniqueMessageMatching(CmaHostMsg_AudioInitialize::ID)); |
| + EXPECT_TRUE( |
| + ipc_sink_.GetUniqueMessageMatching(CmaHostMsg_VideoInitialize::ID)); |
| + |
| + EXPECT_CALL(mock_, OnInitialized(::media::PIPELINE_OK)); |
| + cma_proxy_->OnMessageReceived( |
| + CmaMsg_TrackStateChanged(1, kVideoTrackId, ::media::PIPELINE_OK)); |
| + message_loop_.RunUntilIdle(); |
| +} |
| + |
| + |
| +} // namespace media |
| +} // namespace chromecast |