// Copyright (c) 2006-2009 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 "base/waitable_event.h"
#include "media/base/pipeline_impl.h"
#include "media/base/media_format.h"
#include "media/base/filters.h"
#include "media/base/factory.h"
#include "media/base/filter_host.h"
#include "media/base/mock_media_filters.h"
#include "testing/gtest/include/gtest/gtest.h"

using media::FilterFactory;
using media::FilterFactoryCollection;
using media::FilterHost;
using media::InitializationHelper;
using media::MediaFormat;
using media::MockAudioDecoder;
using media::MockAudioRenderer;
using media::MockDataSource;
using media::MockDemuxer;
using media::MockFilterConfig;
using media::MockVideoDecoder;
using media::MockVideoRenderer;
using media::PipelineImpl;

namespace {

void AddAllMockFilters(FilterFactoryCollection* factories,
                       const MockFilterConfig* config) {
  factories->AddFactory(MockDataSource::CreateFactory(config));
  factories->AddFactory(MockDemuxer::CreateFactory(config));
  factories->AddFactory(MockAudioDecoder::CreateFactory(config));
  factories->AddFactory(MockAudioRenderer::CreateFactory(config));
  factories->AddFactory(MockVideoDecoder::CreateFactory(config));
  factories->AddFactory(MockVideoRenderer::CreateFactory(config));
}

}  // namespace

// TODO(ralphl): Get rid of single character variable names in these tests.
TEST(PipelineImplTest, Initialization) {
  std::string u("");

  // This test hangs during initialization of the data source (it never
  // calls InitializationComplete).  Make sure we tear down the pipeline
  // propertly.
  PipelineImpl p;
  InitializationHelper h;
  MockFilterConfig config;
  config.data_source_behavior = media::MOCK_DATA_SOURCE_NEVER_INIT;
  h.Start(&p, MockDataSource::CreateFactory(&config), u,
          media::PIPELINE_OK, true);
  p.Stop();
  EXPECT_FALSE(h.waiting_for_callback());
  EXPECT_FALSE(h.callback_success_status());
  EXPECT_TRUE(media::PIPELINE_OK == p.GetError());

  // This test should not hang.  Should return an error indicating that we are
  // missing a requried filter.
  config.data_source_behavior = media::MOCK_DATA_SOURCE_TASK_INIT;
  h.Start(&p, MockDataSource::CreateFactory(&config), u,
          media::PIPELINE_ERROR_REQUIRED_FILTER_MISSING);
  p.Stop();

  // This test should return a specific error from the mock data source.
  config.data_source_behavior = media::MOCK_DATA_SOURCE_URL_ERROR_IN_INIT;
  h.Start(&p, MockDataSource::CreateFactory(&config), u,
          media::PIPELINE_ERROR_URL_NOT_FOUND);
  p.Stop();
}

TEST(PipelineImplTest, MockAudioPipeline) {
  std::string url("");
  PipelineImpl p;
  MockFilterConfig config;
  config.has_video = false;
  scoped_refptr<FilterFactoryCollection> c = new FilterFactoryCollection();
  AddAllMockFilters(c, &config);
  InitializationHelper h;
  h.Start(&p, c, url);
  size_t width, height;
  p.GetVideoSize(&width, &height);
  EXPECT_TRUE(p.IsRendered(media::mime_type::kMajorTypeAudio));
  EXPECT_FALSE(p.IsRendered(media::mime_type::kMajorTypeVideo));
  EXPECT_EQ(0u, width);
  EXPECT_EQ(0u, height);
  p.SetPlaybackRate(1.0f);
  p.SetVolume(0.5f);
  p.Stop();
  EXPECT_FALSE(p.IsInitialized());
}

TEST(PipelineImplTest, MockVideoPipeline) {
  std::string url("");
  PipelineImpl p;
  scoped_refptr<FilterFactoryCollection> c = new FilterFactoryCollection();
  MockFilterConfig config;
  AddAllMockFilters(c, &config);
  InitializationHelper h;
  h.Start(&p, c, url);
  size_t width, height;
  p.GetVideoSize(&width, &height);
  EXPECT_EQ(config.video_width, width);
  EXPECT_EQ(config.video_height, height);
  EXPECT_TRUE(p.IsRendered(media::mime_type::kMajorTypeAudio));
  EXPECT_TRUE(p.IsRendered(media::mime_type::kMajorTypeVideo));
  p.SetPlaybackRate(1.0f);
  p.SetVolume(0.5f);
  p.Stop();
  EXPECT_FALSE(p.IsInitialized());
}

TEST(PipelineImplTest, MockVideoOnlyPipeline) {
  std::string url("");
  PipelineImpl p;
  scoped_refptr<FilterFactoryCollection> c = new FilterFactoryCollection();
  MockFilterConfig config;
  config.has_audio = false;
  AddAllMockFilters(c, &config);
  InitializationHelper h;
  h.Start(&p, c, url);
  size_t width, height;
  p.GetVideoSize(&width, &height);
  EXPECT_EQ(config.video_width, width);
  EXPECT_EQ(config.video_height, height);
  EXPECT_FALSE(p.IsRendered(media::mime_type::kMajorTypeAudio));
  EXPECT_TRUE(p.IsRendered(media::mime_type::kMajorTypeVideo));
  p.SetPlaybackRate(1.0f);
  p.Stop();
  EXPECT_FALSE(p.IsInitialized());
}

TEST(PipelineImplTest, MockNothingToRenderPipeline) {
  std::string url("");
  PipelineImpl p;
  scoped_refptr<FilterFactoryCollection> c = new FilterFactoryCollection();
  MockFilterConfig config;
  config.has_audio = false;
  config.has_video = false;
  AddAllMockFilters(c, &config);
  InitializationHelper h;
  h.Start(&p, c, url, media::PIPELINE_ERROR_COULD_NOT_RENDER);
  p.Stop();
}

// TODO(ralphl): Add a unit test that makes sure that the mock audio filter
// is actually called on a SetVolume() call to the pipeline.  I almost checked
// in code that broke this, but all unit tests were passing.