Refactor VideoDestinationHandler to implement MediaStreamVideoSource.

content/renderer/media/media_stream_video_track_unittest.cc

 // Copyright 2014 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/strings/utf_string_conversions.h"
 #include "content/renderer/media/media_stream_video_track.h"
 #include "content/renderer/media/mock_media_stream_dependency_factory.h"
+#include "content/renderer/media/mock_media_stream_video_sink.h"
 #include "content/renderer/media/mock_media_stream_video_source.h"
 #include "media/base/video_frame.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace content {
 
-class MockVideoSink : public MediaStreamVideoSink {
- public:
-  MockVideoSink()
-      : number_of_frames_(0), enabled_(true),
-        state_(blink::WebMediaStreamSource::ReadyStateLive) {
-  }
-
-  virtual void OnVideoFrame(
-      const scoped_refptr<media::VideoFrame>& frame) OVERRIDE {
-    ++number_of_frames_;
-  }
-
-  virtual void OnReadyStateChanged(
-        blink::WebMediaStreamSource::ReadyState state) OVERRIDE {
-    state_ = state;
-  }
-
-  virtual void OnEnabledChanged(bool enabled) OVERRIDE {
-    enabled_ = enabled;
-  }
-
-  int number_of_frames() const { return number_of_frames_; }
-  bool enabled() const { return enabled_; }
-  blink::WebMediaStreamSource::ReadyState state() const { return state_; }
-
- private:
-  int number_of_frames_;
-  bool enabled_;
-  blink::WebMediaStreamSource::ReadyState state_;
-};
-
 class MediaStreamVideoTrackTest : public ::testing::Test {
  public:
   MediaStreamVideoTrackTest()
       : mock_source_(new MockMediaStreamVideoSource(&factory_, false)) {
     webkit_source_.initialize(base::UTF8ToUTF16("dummy_source_id"),
                               blink::WebMediaStreamSource::TypeVideo,
                               base::UTF8ToUTF16("dummy_source_name"));
     webkit_source_.setExtraData(mock_source_);
   }
 
@@ skipping 22 matching lines @@
 };
 
 TEST_F(MediaStreamVideoTrackTest, GetAdapter) {
   blink::WebMediaStreamTrack track = CreateTrack();
   MediaStreamVideoTrack* video_track =
       MediaStreamVideoTrack::GetVideoTrack(track);
   EXPECT_TRUE(video_track->GetVideoAdapter() != NULL);
 }
 
 TEST_F(MediaStreamVideoTrackTest, AddAndRemoveSink) {
-  MockVideoSink sink;
+  MockMediaStreamVideoSink sink;
   blink::WebMediaStreamTrack track = CreateTrack();
   MediaStreamVideoSink::AddToVideoTrack(&sink, track);
 
   MediaStreamVideoTrack* video_track =
         MediaStreamVideoTrack::GetVideoTrack(track);
   scoped_refptr<media::VideoFrame> frame =
       media::VideoFrame::CreateBlackFrame(
           gfx::Size(MediaStreamVideoSource::kDefaultWidth,
                     MediaStreamVideoSource::kDefaultHeight));
   video_track->OnVideoFrame(frame);
   EXPECT_EQ(1, sink.number_of_frames());
   video_track->OnVideoFrame(frame);
   EXPECT_EQ(2, sink.number_of_frames());
 
   MediaStreamVideoSink::RemoveFromVideoTrack(&sink, track);
   video_track->OnVideoFrame(frame);
   EXPECT_EQ(2, sink.number_of_frames());
 }
 
 TEST_F(MediaStreamVideoTrackTest, SetEnabled) {
-  MockVideoSink sink;
+  MockMediaStreamVideoSink sink;
   blink::WebMediaStreamTrack track = CreateTrack();
   MediaStreamVideoSink::AddToVideoTrack(&sink, track);
 
   MediaStreamVideoTrack* video_track =
       MediaStreamVideoTrack::GetVideoTrack(track);
   scoped_refptr<media::VideoFrame> frame =
       media::VideoFrame::CreateBlackFrame(
           gfx::Size(MediaStreamVideoSource::kDefaultWidth,
                     MediaStreamVideoSource::kDefaultHeight));
   video_track->OnVideoFrame(frame);
   EXPECT_EQ(1, sink.number_of_frames());
 
   video_track->SetEnabled(false);
   EXPECT_FALSE(sink.enabled());
   video_track->OnVideoFrame(frame);
   EXPECT_EQ(1, sink.number_of_frames());
 
   video_track->SetEnabled(true);
   EXPECT_TRUE(sink.enabled());
   video_track->OnVideoFrame(frame);
   EXPECT_EQ(2, sink.number_of_frames());
   MediaStreamVideoSink::RemoveFromVideoTrack(&sink, track);
 }
 
 TEST_F(MediaStreamVideoTrackTest, SourceStopped) {
-  MockVideoSink sink;
+  MockMediaStreamVideoSink sink;
   blink::WebMediaStreamTrack track = CreateTrack();
   MediaStreamVideoSink::AddToVideoTrack(&sink, track);
   EXPECT_EQ(blink::WebMediaStreamSource::ReadyStateLive, sink.state());
 
   mock_source()->StopSource();
   EXPECT_EQ(blink::WebMediaStreamSource::ReadyStateEnded, sink.state());
   MediaStreamVideoSink::RemoveFromVideoTrack(&sink, track);
 }
 
 }  // namespace content