diff options
Diffstat (limited to 'media/video/ffmpeg_video_decode_engine_unittest.cc')
| -rw-r--r-- | media/video/ffmpeg_video_decode_engine_unittest.cc | 292 |
1 files changed, 0 insertions, 292 deletions
diff --git a/media/video/ffmpeg_video_decode_engine_unittest.cc b/media/video/ffmpeg_video_decode_engine_unittest.cc deleted file mode 100644 index 73ac917..0000000 --- a/media/video/ffmpeg_video_decode_engine_unittest.cc +++ /dev/null @@ -1,292 +0,0 @@ -// Copyright (c) 2010 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/scoped_ptr.h" -#include "media/base/data_buffer.h" -#include "media/base/mock_ffmpeg.h" -#include "media/base/mock_task.h" -#include "media/video/ffmpeg_video_decode_engine.h" -#include "testing/gmock/include/gmock/gmock.h" -#include "testing/gtest/include/gtest/gtest.h" - -using ::testing::_; -using ::testing::DoAll; -using ::testing::Return; -using ::testing::ReturnNull; -using ::testing::SetArgumentPointee; -using ::testing::StrictMock; - -namespace media { - -static const int kWidth = 320; -static const int kHeight = 240; -static const AVRational kTimeBase = { 1, 100 }; - -ACTION_P(SaveInitializeResult, engine) { - engine->info_ = arg0; -} - -class FFmpegVideoDecodeEngineTest : public testing::Test, - public VideoDecodeEngine::EventHandler { - protected: - FFmpegVideoDecodeEngineTest() { - // Setup FFmpeg structures. - frame_buffer_.reset(new uint8[kWidth * kHeight]); - memset(&yuv_frame_, 0, sizeof(yuv_frame_)); - - // DecodeFrame will check these pointers as non-NULL value. - yuv_frame_.data[0] = yuv_frame_.data[1] = yuv_frame_.data[2] - = frame_buffer_.get(); - yuv_frame_.linesize[0] = kWidth; - yuv_frame_.linesize[1] = yuv_frame_.linesize[2] = kWidth >> 1; - - memset(&codec_context_, 0, sizeof(codec_context_)); - codec_context_.width = kWidth; - codec_context_.height = kHeight; - codec_context_.time_base = kTimeBase; - - memset(&codec_, 0, sizeof(codec_)); - memset(&stream_, 0, sizeof(stream_)); - stream_.codec = &codec_context_; - stream_.r_frame_rate.num = kTimeBase.den; - stream_.r_frame_rate.den = kTimeBase.num; - - buffer_ = new DataBuffer(1); - - // Initialize MockFFmpeg. - MockFFmpeg::set(&mock_ffmpeg_); - - test_engine_ = new FFmpegVideoDecodeEngine(); - test_engine_->SetCodecContextForTest(&codec_context_); - - VideoFrame::CreateFrame(VideoFrame::YV12, - kWidth, - kHeight, - StreamSample::kInvalidTimestamp, - StreamSample::kInvalidTimestamp, - &video_frame_); - } - - ~FFmpegVideoDecodeEngineTest() { - test_engine_ = NULL; - MockFFmpeg::set(NULL); - } - - void Initialize() { - EXPECT_CALL(*MockFFmpeg::get(), AVCodecFindDecoder(CODEC_ID_NONE)) - .WillOnce(Return(&codec_)); - EXPECT_CALL(*MockFFmpeg::get(), AVCodecAllocFrame()) - .WillOnce(Return(&yuv_frame_)); - EXPECT_CALL(*MockFFmpeg::get(), AVCodecThreadInit(&codec_context_, 2)) - .WillOnce(Return(0)); - EXPECT_CALL(*MockFFmpeg::get(), AVCodecOpen(&codec_context_, &codec_)) - .WillOnce(Return(0)); - EXPECT_CALL(*MockFFmpeg::get(), AVFree(&yuv_frame_)) - .Times(1); - - config_.codec_ = kCodecH264; - config_.opaque_context_ = &stream_; - config_.width_ = kWidth; - config_.height_ = kHeight; - EXPECT_CALL(*this, OnInitializeComplete(_)) - .WillOnce(SaveInitializeResult(this)); - test_engine_->Initialize(MessageLoop::current(), this, config_); - EXPECT_TRUE(info_.success_); - } - - public: - MOCK_METHOD1(OnFillBufferCallback, - void(scoped_refptr<VideoFrame> video_frame)); - MOCK_METHOD1(OnEmptyBufferCallback, - void(scoped_refptr<Buffer> buffer)); - MOCK_METHOD1(OnInitializeComplete, - void(const VideoCodecInfo& info)); - MOCK_METHOD0(OnUninitializeComplete, void()); - MOCK_METHOD0(OnFlushComplete, void()); - MOCK_METHOD0(OnSeekComplete, void()); - MOCK_METHOD0(OnError, void()); - MOCK_METHOD1(OnFormatChange, void(VideoStreamInfo stream_info)); - - scoped_refptr<VideoFrame> video_frame_; - VideoCodecConfig config_; - VideoCodecInfo info_; - protected: - scoped_refptr<FFmpegVideoDecodeEngine> test_engine_; - scoped_array<uint8_t> frame_buffer_; - StrictMock<MockFFmpeg> mock_ffmpeg_; - - AVFrame yuv_frame_; - AVCodecContext codec_context_; - AVStream stream_; - AVCodec codec_; - scoped_refptr<DataBuffer> buffer_; - -}; - -TEST_F(FFmpegVideoDecodeEngineTest, Initialize_Normal) { - Initialize(); -} - -TEST_F(FFmpegVideoDecodeEngineTest, Initialize_FindDecoderFails) { - // Test avcodec_find_decoder() returning NULL. - EXPECT_CALL(*MockFFmpeg::get(), AVCodecFindDecoder(CODEC_ID_NONE)) - .WillOnce(ReturnNull()); - EXPECT_CALL(*MockFFmpeg::get(), AVCodecAllocFrame()) - .WillOnce(Return(&yuv_frame_)); - EXPECT_CALL(*MockFFmpeg::get(), AVFree(&yuv_frame_)) - .Times(1); - - config_.codec_ = kCodecH264; - config_.opaque_context_ = &stream_; - config_.width_ = kWidth; - config_.height_ = kHeight; - EXPECT_CALL(*this, OnInitializeComplete(_)) - .WillOnce(SaveInitializeResult(this)); - test_engine_->Initialize(MessageLoop::current(), this, config_); - EXPECT_FALSE(info_.success_); -} - -// Note There are 2 threads for FFmpeg-mt. -TEST_F(FFmpegVideoDecodeEngineTest, Initialize_InitThreadFails) { - // Test avcodec_thread_init() failing. - EXPECT_CALL(*MockFFmpeg::get(), AVCodecFindDecoder(CODEC_ID_NONE)) - .WillOnce(Return(&codec_)); - EXPECT_CALL(*MockFFmpeg::get(), AVCodecAllocFrame()) - .WillOnce(Return(&yuv_frame_)); - EXPECT_CALL(*MockFFmpeg::get(), AVCodecThreadInit(&codec_context_, 2)) - .WillOnce(Return(-1)); - EXPECT_CALL(*MockFFmpeg::get(), AVFree(&yuv_frame_)) - .Times(1); - - config_.codec_ = kCodecH264; - config_.opaque_context_ = &stream_; - config_.width_ = kWidth; - config_.height_ = kHeight; - EXPECT_CALL(*this, OnInitializeComplete(_)) - .WillOnce(SaveInitializeResult(this)); - test_engine_->Initialize(MessageLoop::current(), this, config_); - EXPECT_FALSE(info_.success_); -} - -TEST_F(FFmpegVideoDecodeEngineTest, Initialize_OpenDecoderFails) { - // Test avcodec_open() failing. - EXPECT_CALL(*MockFFmpeg::get(), AVCodecFindDecoder(CODEC_ID_NONE)) - .WillOnce(Return(&codec_)); - EXPECT_CALL(*MockFFmpeg::get(), AVCodecAllocFrame()) - .WillOnce(Return(&yuv_frame_)); - EXPECT_CALL(*MockFFmpeg::get(), AVCodecThreadInit(&codec_context_, 2)) - .WillOnce(Return(0)); - EXPECT_CALL(*MockFFmpeg::get(), AVCodecOpen(&codec_context_, &codec_)) - .WillOnce(Return(-1)); - EXPECT_CALL(*MockFFmpeg::get(), AVFree(&yuv_frame_)) - .Times(1); - - config_.codec_ = kCodecH264; - config_.opaque_context_ = &stream_; - config_.width_ = kWidth; - config_.height_ = kHeight; - EXPECT_CALL(*this, OnInitializeComplete(_)) - .WillOnce(SaveInitializeResult(this)); - test_engine_->Initialize(MessageLoop::current(), this, config_); - EXPECT_FALSE(info_.success_); -} - -ACTION_P2(DemuxComplete, engine, buffer) { - engine->EmptyThisBuffer(buffer); -} - -ACTION_P(DecodeComplete, decoder) { - decoder->video_frame_ = arg0; -} - -TEST_F(FFmpegVideoDecodeEngineTest, DecodeFrame_Normal) { - Initialize(); - - // We rely on FFmpeg for timestamp and duration reporting. The one tricky - // bit is calculating the duration when |repeat_pict| > 0. - const base::TimeDelta kTimestamp = base::TimeDelta::FromMicroseconds(123); - const base::TimeDelta kDuration = base::TimeDelta::FromMicroseconds(15000); - yuv_frame_.repeat_pict = 1; - yuv_frame_.reordered_opaque = kTimestamp.InMicroseconds(); - - // Expect a bunch of avcodec calls. - EXPECT_CALL(mock_ffmpeg_, AVInitPacket(_)); - EXPECT_CALL(mock_ffmpeg_, - AVCodecDecodeVideo2(&codec_context_, &yuv_frame_, _, _)) - .WillOnce(DoAll(SetArgumentPointee<2>(1), // Simulate 1 byte frame. - Return(0))); - - EXPECT_CALL(*this, OnEmptyBufferCallback(_)) - .WillOnce(DemuxComplete(test_engine_.get(), buffer_)); - EXPECT_CALL(*this, OnFillBufferCallback(_)) - .WillOnce(DecodeComplete(this)); - test_engine_->FillThisBuffer(video_frame_); - - // |video_frame_| timestamp is 0 because we set the timestamp based off - // the buffer timestamp. - EXPECT_EQ(0, video_frame_->GetTimestamp().ToInternalValue()); - EXPECT_EQ(kDuration.ToInternalValue(), - video_frame_->GetDuration().ToInternalValue()); -} - -TEST_F(FFmpegVideoDecodeEngineTest, DecodeFrame_0ByteFrame) { - Initialize(); - - // Expect a bunch of avcodec calls. - EXPECT_CALL(mock_ffmpeg_, AVInitPacket(_)) - .Times(2); - EXPECT_CALL(mock_ffmpeg_, - AVCodecDecodeVideo2(&codec_context_, &yuv_frame_, _, _)) - .WillOnce(DoAll(SetArgumentPointee<2>(0), // Simulate 0 byte frame. - Return(0))) - .WillOnce(DoAll(SetArgumentPointee<2>(1), // Simulate 1 byte frame. - Return(0))); - - EXPECT_CALL(*this, OnEmptyBufferCallback(_)) - .WillOnce(DemuxComplete(test_engine_.get(), buffer_)) - .WillOnce(DemuxComplete(test_engine_.get(), buffer_)); - EXPECT_CALL(*this, OnFillBufferCallback(_)) - .WillOnce(DecodeComplete(this)); - test_engine_->FillThisBuffer(video_frame_); - - EXPECT_TRUE(video_frame_.get()); -} - -TEST_F(FFmpegVideoDecodeEngineTest, DecodeFrame_DecodeError) { - Initialize(); - - // Expect a bunch of avcodec calls. - EXPECT_CALL(mock_ffmpeg_, AVInitPacket(_)); - EXPECT_CALL(mock_ffmpeg_, - AVCodecDecodeVideo2(&codec_context_, &yuv_frame_, _, _)) - .WillOnce(Return(-1)); - - EXPECT_CALL(*this, OnEmptyBufferCallback(_)) - .WillOnce(DemuxComplete(test_engine_.get(), buffer_)); - EXPECT_CALL(*this, OnFillBufferCallback(_)) - .WillOnce(DecodeComplete(this)); - test_engine_->FillThisBuffer(video_frame_); - - EXPECT_FALSE(video_frame_.get()); -} - -TEST_F(FFmpegVideoDecodeEngineTest, GetSurfaceFormat) { - // YV12 formats. - codec_context_.pix_fmt = PIX_FMT_YUV420P; - EXPECT_EQ(VideoFrame::YV12, test_engine_->GetSurfaceFormat()); - codec_context_.pix_fmt = PIX_FMT_YUVJ420P; - EXPECT_EQ(VideoFrame::YV12, test_engine_->GetSurfaceFormat()); - - // YV16 formats. - codec_context_.pix_fmt = PIX_FMT_YUV422P; - EXPECT_EQ(VideoFrame::YV16, test_engine_->GetSurfaceFormat()); - codec_context_.pix_fmt = PIX_FMT_YUVJ422P; - EXPECT_EQ(VideoFrame::YV16, test_engine_->GetSurfaceFormat()); - - // Invalid value. - codec_context_.pix_fmt = PIX_FMT_NONE; - EXPECT_EQ(VideoFrame::INVALID, test_engine_->GetSurfaceFormat()); -} - -} // namespace media |