chromecast/media/cma/pipeline/audio_video_pipeline_impl_unittest.cc

   1 // Copyright 2014 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include <vector>
   6
   7 #include "base/basictypes.h"
   8 #include "base/bind.h"
   9 #include "base/memory/ref_counted.h"
  10 #include "base/memory/scoped_ptr.h"
  11 #include "base/message_loop/message_loop.h"
  12 #include "base/single_thread_task_runner.h"
  13 #include "base/thread_task_runner_handle.h"
  14 #include "base/time/time.h"
  15 #include "chromecast/media/cma/backend/audio_pipeline_device.h"
  16 #include "chromecast/media/cma/backend/media_clock_device.h"
  17 #include "chromecast/media/cma/backend/media_pipeline_device.h"
  18 #include "chromecast/media/cma/backend/media_pipeline_device_factory_default.h"
  19 #include "chromecast/media/cma/base/buffering_controller.h"
  20 #include "chromecast/media/cma/base/decoder_buffer_base.h"
  21 #include "chromecast/media/cma/pipeline/audio_pipeline_impl.h"
  22 #include "chromecast/media/cma/pipeline/av_pipeline_client.h"
  23 #include "chromecast/media/cma/pipeline/media_pipeline_impl.h"
  24 #include "chromecast/media/cma/pipeline/video_pipeline_client.h"
  25 #include "chromecast/media/cma/pipeline/video_pipeline_impl.h"
  26 #include "chromecast/media/cma/test/frame_generator_for_test.h"
  27 #include "chromecast/media/cma/test/mock_frame_provider.h"
  28 #include "media/base/audio_decoder_config.h"
  29 #include "media/base/buffers.h"
  30 #include "media/base/decoder_buffer.h"
  31 #include "media/base/video_decoder_config.h"
  32 #include "testing/gtest/include/gtest/gtest.h"
  33
  34 namespace chromecast {
  35 namespace media {
  36
  37 class AudioVideoPipelineImplTest : public testing::Test {
  38  public:
  39   AudioVideoPipelineImplTest();
  40   ~AudioVideoPipelineImplTest() override;
  41
  42   void Initialize(const base::Closure& done_cb,
  43                   ::media::PipelineStatus status,
  44                   bool is_audio);
  45   void StartPlaying(const base::Closure& done_cb,
  46                     ::media::PipelineStatus status);
  47
  48   void Flush(const base::Closure& done_cb, ::media::PipelineStatus status);
  49   void Stop(const base::Closure& done_cb, ::media::PipelineStatus status);
  50
  51   void OnEos();
  52
  53   base::Closure task_after_eos_cb_;
  54
  55  private:
  56   scoped_ptr<MediaPipelineImpl> media_pipeline_;
  57
  58   DISALLOW_COPY_AND_ASSIGN(AudioVideoPipelineImplTest);
  59 };
  60
  61 AudioVideoPipelineImplTest::AudioVideoPipelineImplTest()
  62   : media_pipeline_(new MediaPipelineImpl()) {
  63   scoped_ptr<MediaPipelineDeviceFactory> factory =
  64       make_scoped_ptr(new MediaPipelineDeviceFactoryDefault());
  65   scoped_ptr<MediaPipelineDevice> media_pipeline_device =
  66       make_scoped_ptr(new MediaPipelineDevice(factory.Pass()));
  67   media_pipeline_->Initialize(kLoadTypeURL, media_pipeline_device.Pass());
  68   media_pipeline_->SetPlaybackRate(1.0);
  69 }
  70
  71 AudioVideoPipelineImplTest::~AudioVideoPipelineImplTest() {
  72 }
  73
  74 void AudioVideoPipelineImplTest::Initialize(
  75     const base::Closure& done_cb,
  76     ::media::PipelineStatus status,
  77     bool is_audio) {
  78   if (is_audio) {
  79     AvPipelineClient client;
  80     client.eos_cb =
  81         base::Bind(&AudioVideoPipelineImplTest::OnEos, base::Unretained(this));
  82     media_pipeline_->GetAudioPipeline()->SetClient(client);
  83   } else {
  84     VideoPipelineClient client;
  85     client.av_pipeline_client.eos_cb =
  86         base::Bind(&AudioVideoPipelineImplTest::OnEos, base::Unretained(this));
  87     media_pipeline_->GetVideoPipeline()->SetClient(client);
  88   }
  89
  90   ::media::AudioDecoderConfig audio_config(
  91       ::media::kCodecMP3,
  92       ::media::kSampleFormatS16,
  93       ::media::CHANNEL_LAYOUT_STEREO,
  94       44100,
  95       NULL, 0, false);
  96   std::vector<::media::VideoDecoderConfig> video_configs;
  97   video_configs.push_back(::media::VideoDecoderConfig(
  98       ::media::kCodecH264,
  99       ::media::H264PROFILE_MAIN,
 100       ::media::VideoFrame::I420,
 101       gfx::Size(640, 480),
 102       gfx::Rect(0, 0, 640, 480),
 103       gfx::Size(640, 480),
 104       NULL, 0, false));
 105
 106   // Frame generation on the producer side.
 107   std::vector<FrameGeneratorForTest::FrameSpec> frame_specs;
 108   frame_specs.resize(100);
 109   for (size_t k = 0; k < frame_specs.size() - 1; k++) {
 110     frame_specs[k].has_config = (k == 0);
 111     frame_specs[k].timestamp = base::TimeDelta::FromMilliseconds(40) * k;
 112     frame_specs[k].size = 512;
 113     frame_specs[k].has_decrypt_config = false;
 114   }
 115   frame_specs[frame_specs.size() - 1].is_eos = true;
 116
 117   scoped_ptr<FrameGeneratorForTest> frame_generator_provider(
 118       new FrameGeneratorForTest(frame_specs));
 119   bool provider_delayed_pattern[] = { true, false };
 120   scoped_ptr<MockFrameProvider> frame_provider(new MockFrameProvider());
 121   frame_provider->Configure(
 122       std::vector<bool>(
 123           provider_delayed_pattern,
 124           provider_delayed_pattern + arraysize(provider_delayed_pattern)),
 125       frame_generator_provider.Pass());
 126
 127   ::media::PipelineStatusCB next_task =
 128       base::Bind(&AudioVideoPipelineImplTest::StartPlaying,
 129                  base::Unretained(this),
 130                  done_cb);
 131
 132   scoped_ptr<CodedFrameProvider> frame_provider_base(frame_provider.release());
 133   base::Closure task = is_audio ?
 134       base::Bind(&MediaPipeline::InitializeAudio,
 135                  base::Unretained(media_pipeline_.get()),
 136                  audio_config,
 137                  base::Passed(&frame_provider_base),
 138                  next_task) :
 139       base::Bind(&MediaPipeline::InitializeVideo,
 140                  base::Unretained(media_pipeline_.get()),
 141                  video_configs,
 142                  base::Passed(&frame_provider_base),
 143                  next_task);
 144
 145   base::ThreadTaskRunnerHandle::Get()->PostTask(FROM_HERE, task);
 146 }
 147
 148 void AudioVideoPipelineImplTest::StartPlaying(
 149     const base::Closure& done_cb, ::media::PipelineStatus status) {
 150   base::TimeDelta start_time = base::TimeDelta::FromMilliseconds(0);
 151
 152   media_pipeline_->StartPlayingFrom(start_time);
 153   if (!done_cb.is_null())
 154     done_cb.Run();
 155 }
 156
 157 void AudioVideoPipelineImplTest::OnEos() {
 158   task_after_eos_cb_.Run();
 159 }
 160
 161 void AudioVideoPipelineImplTest::Flush(
 162     const base::Closure& done_cb, ::media::PipelineStatus status) {
 163   ::media::PipelineStatusCB next_task =
 164       base::Bind(&AudioVideoPipelineImplTest::Stop, base::Unretained(this),
 165                  done_cb);
 166   base::ThreadTaskRunnerHandle::Get()->PostTask(
 167       FROM_HERE,
 168       base::Bind(&MediaPipeline::Flush, base::Unretained(media_pipeline_.get()),
 169                  next_task));
 170 }
 171
 172 void AudioVideoPipelineImplTest::Stop(
 173     const base::Closure& done_cb, ::media::PipelineStatus status) {
 174   media_pipeline_->Stop();
 175   if (!done_cb.is_null())
 176     done_cb.Run();
 177   base::MessageLoop::current()->QuitWhenIdle();
 178 }
 179
 180
 181 TEST_F(AudioVideoPipelineImplTest, AudioFullCycleInitToStop) {
 182   bool is_audio = true;
 183   task_after_eos_cb_ = base::Bind(
 184       &AudioVideoPipelineImplTest::Flush, base::Unretained(this),
 185       base::Closure(), ::media::PIPELINE_OK);
 186
 187   scoped_ptr<base::MessageLoop> message_loop(new base::MessageLoop());
 188   message_loop->PostTask(
 189       FROM_HERE,
 190       base::Bind(&AudioVideoPipelineImplTest::Initialize,
 191                  base::Unretained(this),
 192                  base::Closure(),
 193                  ::media::PIPELINE_OK, is_audio));
 194   message_loop->Run();
 195 };
 196
 197 TEST_F(AudioVideoPipelineImplTest, VideoFullCycleInitToStop) {
 198   bool is_audio = false;
 199   task_after_eos_cb_ = base::Bind(
 200       &AudioVideoPipelineImplTest::Flush, base::Unretained(this),
 201       base::Closure(), ::media::PIPELINE_OK);
 202
 203   scoped_ptr<base::MessageLoop> message_loop(new base::MessageLoop());
 204   message_loop->PostTask(
 205       FROM_HERE,
 206       base::Bind(&AudioVideoPipelineImplTest::Initialize,
 207                  base::Unretained(this),
 208                  base::Closure(),
 209                  ::media::PIPELINE_OK, is_audio));
 210   message_loop->Run();
 211 };
 212
 213 }  // namespace media
 214 }  // namespace chromecast