Implement WebRTC in Chrome for TV

content/renderer/media/rtc_video_decoder_factory_tv_unittest.cc

+// Copyright (c) 2013 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/synchronization/waitable_event.h"
+#include "base/task_runner_util.h"
+#include "base/threading/thread.h"
+#include "content/renderer/media/rtc_video_decoder_factory_tv.h"
+#include "media/base/decoder_buffer.h"
+#include "media/base/video_decoder_config.h"
+#include "testing/gmock/include/gmock/gmock.h"
+#include "testing/gtest/include/gtest/gtest.h"
+#include "third_party/webrtc/modules/video_coding/codecs/interface/mock/mock_video_codec_interface.h"
+#include "ui/gfx/rect.h"
+
+using ::testing::_;
+using ::testing::Return;
+
+namespace content {
+
+class RTCVideoDecoderFactoryTvTest : public ::testing::Test {
+ public:
+  RTCVideoDecoderFactoryTvTest()
+      : factory_(new RTCVideoDecoderFactoryTv),
+        decoder_(NULL),
+        is_demuxer_acquired_(false),
+        video_stream_(NULL),
+        size_(1280, 720),
+        input_image_(&data_, sizeof(data_), sizeof(data_)),
+        data_('a'),
+        read_event_(false, false),
+        decoder_thread_("Test decoeder thread"),

[acolwell GONE FROM CHROMIUM, 2013/05/14 18:17:47]

nit:s/decoeder/decoder

[wonsik, 2013/05/20 14:02:24]

Done.

+        decoder_thread_event_(false, false) {}
+
+  void ReadCallback(media::DemuxerStream::Status status,
+                    const scoped_refptr<media::DecoderBuffer>& decoder_buffer) {
+    switch (status) {
+      case media::DemuxerStream::kOk:
+        EXPECT_TRUE(decoder_buffer);
+        break;
+      case media::DemuxerStream::kConfigChanged:
+      case media::DemuxerStream::kAborted:
+        EXPECT_FALSE(decoder_buffer);
+        break;
+    }
+    last_decoder_buffer_ = decoder_buffer;
+    read_event_.Signal();
+  }
+
+  void ExpectEqualsAndSignal(int32_t expected, int32_t actual) {
+    EXPECT_EQ(expected, actual);
+    decoder_thread_event_.Signal();
+  }
+
+  void ExpectNotEqualsAndSignal(int32_t unexpected, int32_t actual) {
+    EXPECT_NE(unexpected, actual);
+    decoder_thread_event_.Signal();
+  }
+
+ protected:
+  virtual void SetUp() OVERRIDE {
+    memset(&codec_, 0, sizeof(codec_));

[acolwell GONE FROM CHROMIUM, 2013/05/14 18:17:47]

I think you can just move all the logic in this method into the constructor.

[wonsik, 2013/05/20 14:02:24]

Done.

+    message_loop_proxy_ = base::MessageLoopProxy::current();
+    input_image_._frameType = webrtc::kKeyFrame;
+    input_image_._encodedWidth = size_.width();
+    input_image_._encodedHeight = size_.height();
+    input_image_._completeFrame = true;
+    decoder_thread_.Start();
+  }
+
+  virtual void TearDown() OVERRIDE {
+    if (is_demuxer_acquired_) {

[acolwell GONE FROM CHROMIUM, 2013/05/14 18:17:47]

I believe you can move all of the logic in this method into the destructor.

[wonsik, 2013/05/20 14:02:24]

Done.

+      factory_->ReleaseDemuxer();
+      is_demuxer_acquired_ = false;
+    }
+    if (decoder_) {
+      factory_->DestroyVideoDecoder(decoder_);
+      decoder_ = NULL;
+    }
+
+    decoder_thread_.Stop();
+  }
+
+  base::Callback<void(int32_t)> BindExpectEquals(int32_t expected) {
+    return base::Bind(&RTCVideoDecoderFactoryTvTest::ExpectEqualsAndSignal,
+                      base::Unretained(this),
+                      expected);
+  }
+
+  base::Callback<void(int32_t)> BindExpectNotEquals(int32_t unexpected) {
+    return base::Bind(&RTCVideoDecoderFactoryTvTest::ExpectNotEqualsAndSignal,
+                      base::Unretained(this),
+                      unexpected);
+  }
+
+  base::Callback<int32_t(void)> BindInitDecode(const webrtc::VideoCodec* codec,
+                                               int32_t num_cores) {
+    return base::Bind(&webrtc::VideoDecoder::InitDecode,
+                      base::Unretained(decoder_),
+                      codec,
+                      num_cores);
+  }
+
+  base::Callback<int32_t(void)> BindDecode(
+      const webrtc::EncodedImage& input_image,
+      bool missing_frames,
+      const webrtc::RTPFragmentationHeader* fragmentation,
+      const webrtc::CodecSpecificInfo* info = NULL,
+      int64_t render_time_ms = -1) {
+    return base::Bind(&webrtc::VideoDecoder::Decode,
+                      base::Unretained(decoder_),
+                      input_image,
+                      missing_frames,
+                      fragmentation,
+                      info,
+                      render_time_ms);
+  }
+
+  void CreateDecoderAndAcquireDemuxer() {
+    decoder_ = factory_->CreateVideoDecoder(webrtc::kVideoCodecVP8);
+    ASSERT_TRUE(decoder_);
+    ASSERT_TRUE(factory_->AcquireDemuxer());
+    is_demuxer_acquired_ = true;
+  }
+
+  void InitDecode() {
+    codec_.codecType = webrtc::kVideoCodecVP8;
+    codec_.width = size_.width();
+    codec_.height = size_.height();
+    base::PostTaskAndReplyWithResult(decoder_thread_.message_loop_proxy(),
+                                     FROM_HERE,
+                                     BindInitDecode(&codec_, 1),
+                                     BindExpectEquals(WEBRTC_VIDEO_CODEC_OK));
+    decoder_thread_event_.Wait();
+    base::PostTaskAndReplyWithResult(
+        decoder_thread_.message_loop_proxy(),
+        FROM_HERE,
+        base::Bind(&webrtc::VideoDecoder::RegisterDecodeCompleteCallback,
+                   base::Unretained(decoder_),
+                   &decode_complete_callback_),
+        BindExpectEquals(WEBRTC_VIDEO_CODEC_OK));
+    decoder_thread_event_.Wait();
+  }
+
+  void GetVideoStream() {
+    video_stream_ = factory_->GetStream(media::DemuxerStream::VIDEO);
+    ASSERT_TRUE(video_stream_);
+    EXPECT_EQ(media::kCodecVP8, video_stream_->video_decoder_config().codec());
+    EXPECT_EQ(size_, video_stream_->video_decoder_config().coded_size());
+    EXPECT_EQ(gfx::Rect(size_),
+              video_stream_->video_decoder_config().visible_rect());
+    EXPECT_EQ(size_, video_stream_->video_decoder_config().natural_size());
+  }
+
+  void PostDecodeAndWait(int32_t expected,
+                         const webrtc::EncodedImage& input_image,
+                         bool missing_frames,
+                         const webrtc::RTPFragmentationHeader* fragmentation,
+                         const webrtc::CodecSpecificInfo* info = NULL,
+                         int64_t render_time_ms = -1) {
+    base::PostTaskAndReplyWithResult(
+        decoder_thread_.message_loop_proxy(),
+        FROM_HERE,
+        BindDecode(
+            input_image, missing_frames, fragmentation, info, render_time_ms),
+        BindExpectEquals(expected));
+    decoder_thread_event_.Wait();
+  }
+
+  RTCVideoDecoderFactoryTv* factory_;
+  webrtc::VideoDecoder* decoder_;
+  bool is_demuxer_acquired_;
+  base::MessageLoopProxy* message_loop_proxy_;
+  media::DemuxerStream* video_stream_;
+  webrtc::VideoCodec codec_;
+  gfx::Size size_;
+  webrtc::EncodedImage input_image_;
+  unsigned char data_;
+  webrtc::MockDecodedImageCallback decode_complete_callback_;
+  base::WaitableEvent read_event_;
+  base::Thread decoder_thread_;
+  base::WaitableEvent decoder_thread_event_;
+  scoped_refptr<media::DecoderBuffer> last_decoder_buffer_;
+};
+
+TEST_F(RTCVideoDecoderFactoryTvTest, CreateAndDestroyDecoder) {
+  // Only VP8 decoder is supported.
+  ASSERT_FALSE(factory_->CreateVideoDecoder(webrtc::kVideoCodecI420));
+  decoder_ = factory_->CreateVideoDecoder(webrtc::kVideoCodecVP8);
+  ASSERT_TRUE(decoder_);
+  // Only one decoder at a time will be created.
+  ASSERT_FALSE(factory_->CreateVideoDecoder(webrtc::kVideoCodecVP8));
+  factory_->DestroyVideoDecoder(decoder_);
+}
+
+TEST_F(RTCVideoDecoderFactoryTvTest, AcquireDemuxerAfterCreateDecoder) {
+  decoder_ = factory_->CreateVideoDecoder(webrtc::kVideoCodecVP8);
+  ASSERT_TRUE(decoder_);
+  ASSERT_TRUE(factory_->AcquireDemuxer());
+  is_demuxer_acquired_ = true;
+  // Demuxer can be acquired only once.
+  ASSERT_FALSE(factory_->AcquireDemuxer());
+}
+
+TEST_F(RTCVideoDecoderFactoryTvTest, AcquireDemuxerBeforeCreateDecoder) {
+  ASSERT_TRUE(factory_->AcquireDemuxer());
+  is_demuxer_acquired_ = true;
+  decoder_ = factory_->CreateVideoDecoder(webrtc::kVideoCodecVP8);
+  ASSERT_TRUE(decoder_);
+}
+
+TEST_F(RTCVideoDecoderFactoryTvTest, InitDecodeReturnsErrorOnNonVP8Codec) {
+  CreateDecoderAndAcquireDemuxer();
+  codec_.codecType = webrtc::kVideoCodecI420;
+  base::PostTaskAndReplyWithResult(decoder_thread_.message_loop_proxy(),
+                                   FROM_HERE,
+                                   BindInitDecode(&codec_, 1),
+                                   BindExpectNotEquals(WEBRTC_VIDEO_CODEC_OK));
+  decoder_thread_event_.Wait();
+}
+
+TEST_F(RTCVideoDecoderFactoryTvTest, InitDecodeReturnsErrorOnFeedbackMode) {
+  CreateDecoderAndAcquireDemuxer();
+  codec_.codecType = webrtc::kVideoCodecVP8;
+  codec_.codecSpecific.VP8.feedbackModeOn = true;
+  base::PostTaskAndReplyWithResult(decoder_thread_.message_loop_proxy(),
+                                   FROM_HERE,
+                                   BindInitDecode(&codec_, 1),
+                                   BindExpectNotEquals(WEBRTC_VIDEO_CODEC_OK));
+  decoder_thread_event_.Wait();
+}
+
+TEST_F(RTCVideoDecoderFactoryTvTest, DecodeReturnsErrorBeforeInitDecode) {
+  CreateDecoderAndAcquireDemuxer();
+  PostDecodeAndWait(
+      WEBRTC_VIDEO_CODEC_UNINITIALIZED, input_image_, false, NULL, NULL, 0);
+}
+
+TEST_F(RTCVideoDecoderFactoryTvTest, DecodeReturnsErrorOnDamagedBitstream) {
+  CreateDecoderAndAcquireDemuxer();
+  InitDecode();
+  input_image_._completeFrame = false;
+  PostDecodeAndWait(
+      WEBRTC_VIDEO_CODEC_ERROR, input_image_, false, NULL, NULL, 0);
+}
+
+TEST_F(RTCVideoDecoderFactoryTvTest, DecodeReturnsErrorOnMissingFrames) {
+  CreateDecoderAndAcquireDemuxer();
+  InitDecode();
+  PostDecodeAndWait(
+      WEBRTC_VIDEO_CODEC_ERROR, input_image_, true, NULL, NULL, 0);
+}
+
+TEST_F(RTCVideoDecoderFactoryTvTest, GetNonVideoStreamFails) {
+  CreateDecoderAndAcquireDemuxer();
+  InitDecode();
+  EXPECT_FALSE(factory_->GetStream(media::DemuxerStream::AUDIO));
+  EXPECT_FALSE(factory_->GetStream(media::DemuxerStream::UNKNOWN));
+}
+
+TEST_F(RTCVideoDecoderFactoryTvTest, GetVideoStreamSucceeds) {
+  CreateDecoderAndAcquireDemuxer();
+  InitDecode();
+  GetVideoStream();
+}
+
+TEST_F(RTCVideoDecoderFactoryTvTest, DecodeReturnsErrorOnNonKeyFrameAtFirst) {
+  CreateDecoderAndAcquireDemuxer();
+  InitDecode();
+  GetVideoStream();
+  input_image_._frameType = webrtc::kDeltaFrame;
+  PostDecodeAndWait(
+      WEBRTC_VIDEO_CODEC_ERROR, input_image_, false, NULL, NULL, 0);
+}
+
+TEST_F(RTCVideoDecoderFactoryTvTest, DecodeUpdatesVideoSizeOnKeyFrame) {
+  CreateDecoderAndAcquireDemuxer();
+  InitDecode();
+  GetVideoStream();
+  gfx::Size new_size(320, 240);
+  input_image_._encodedWidth = new_size.width();
+  input_image_._encodedHeight = new_size.height();
+  PostDecodeAndWait(WEBRTC_VIDEO_CODEC_OK, input_image_, false, NULL, NULL, 0);
+  EXPECT_EQ(new_size, video_stream_->video_decoder_config().coded_size());
+  EXPECT_EQ(gfx::Rect(new_size),
+            video_stream_->video_decoder_config().visible_rect());
+  EXPECT_EQ(new_size, video_stream_->video_decoder_config().natural_size());
+}
+
+TEST_F(RTCVideoDecoderFactoryTvTest, DecodeAdjustsTimestampFromZero) {
+  CreateDecoderAndAcquireDemuxer();
+  InitDecode();
+  GetVideoStream();
+  PostDecodeAndWait(
+      WEBRTC_VIDEO_CODEC_OK, input_image_, false, NULL, NULL, 10000);
+  video_stream_->Read(base::Bind(&RTCVideoDecoderFactoryTvTest::ReadCallback,
+                                 base::Unretained(this)));
+  read_event_.Wait();
+  EXPECT_EQ(base::TimeDelta::FromMilliseconds(0),
+            last_decoder_buffer_->GetTimestamp());
+  PostDecodeAndWait(
+      WEBRTC_VIDEO_CODEC_OK, input_image_, false, NULL, NULL, 10033);
+  video_stream_->Read(base::Bind(&RTCVideoDecoderFactoryTvTest::ReadCallback,
+                                 base::Unretained(this)));
+  read_event_.Wait();
+  EXPECT_EQ(base::TimeDelta::FromMilliseconds(33),
+            last_decoder_buffer_->GetTimestamp());
+}
+
+TEST_F(RTCVideoDecoderFactoryTvTest, DecodePassesDataCorrectly) {
+  CreateDecoderAndAcquireDemuxer();
+  InitDecode();
+  GetVideoStream();
+  video_stream_->Read(base::Bind(&RTCVideoDecoderFactoryTvTest::ReadCallback,
+                                 base::Unretained(this)));
+  PostDecodeAndWait(WEBRTC_VIDEO_CODEC_OK, input_image_, false, NULL, NULL, 0);
+  read_event_.Wait();
+  EXPECT_EQ(static_cast<int>(sizeof(data_)),
+            last_decoder_buffer_->GetDataSize());
+  EXPECT_EQ(data_, last_decoder_buffer_->GetData()[0]);
+}
+
+TEST_F(RTCVideoDecoderFactoryTvTest, NextReadTriggersDecodeCompleteCallback) {
+  EXPECT_CALL(decode_complete_callback_, Decoded(_))
+      .WillOnce(Return(WEBRTC_VIDEO_CODEC_OK));
+
+  CreateDecoderAndAcquireDemuxer();
+  InitDecode();
+  GetVideoStream();
+  video_stream_->Read(base::Bind(&RTCVideoDecoderFactoryTvTest::ReadCallback,
+                                 base::Unretained(this)));
+  PostDecodeAndWait(WEBRTC_VIDEO_CODEC_OK, input_image_, false, NULL, NULL, 0);
+  read_event_.Wait();
+  video_stream_->Read(base::Bind(&RTCVideoDecoderFactoryTvTest::ReadCallback,
+                                 base::Unretained(this)));
+}
+
+TEST_F(RTCVideoDecoderFactoryTvTest, ResetReturnsOk) {
+  CreateDecoderAndAcquireDemuxer();
+  InitDecode();
+  EXPECT_EQ(WEBRTC_VIDEO_CODEC_OK, decoder_->Reset());
+}
+
+TEST_F(RTCVideoDecoderFactoryTvTest, ReleaseReturnsOk) {
+  CreateDecoderAndAcquireDemuxer();
+  InitDecode();
+  EXPECT_EQ(WEBRTC_VIDEO_CODEC_OK, decoder_->Release());
+}
+
+}  // content