Refactor VideoDestinationHandler to implement MediaStreamVideoSource.

content/renderer/media/media_stream_video_track_unittest.cc

Index: content/renderer/media/media_stream_video_track_unittest.cc
diff --git a/content/renderer/media/media_stream_video_track_unittest.cc b/content/renderer/media/media_stream_video_track_unittest.cc
index 242b9cefff9929ab62fefe09306fef6dd0f0f60c..d2e8482f2a4556a7cc6595a67bcecf8aa0ae9d32 100644
--- a/content/renderer/media/media_stream_video_track_unittest.cc
+++ b/content/renderer/media/media_stream_video_track_unittest.cc
@@ -5,43 +5,13 @@
 #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()
@@ -84,7 +54,7 @@ TEST_F(MediaStreamVideoTrackTest, GetAdapter) {
 }
 
 TEST_F(MediaStreamVideoTrackTest, AddAndRemoveSink) {
-  MockVideoSink sink;
+  MockMediaStreamVideoSink sink;
   blink::WebMediaStreamTrack track = CreateTrack();
   MediaStreamVideoSink::AddToVideoTrack(&sink, track);
 
@@ -105,7 +75,7 @@ TEST_F(MediaStreamVideoTrackTest, AddAndRemoveSink) {
 }
 
 TEST_F(MediaStreamVideoTrackTest, SetEnabled) {
-  MockVideoSink sink;
+  MockMediaStreamVideoSink sink;
   blink::WebMediaStreamTrack track = CreateTrack();
   MediaStreamVideoSink::AddToVideoTrack(&sink, track);
 
@@ -131,7 +101,7 @@ TEST_F(MediaStreamVideoTrackTest, SetEnabled) {
 }
 
 TEST_F(MediaStreamVideoTrackTest, SourceStopped) {
-  MockVideoSink sink;
+  MockMediaStreamVideoSink sink;
   blink::WebMediaStreamTrack track = CreateTrack();
   MediaStreamVideoSink::AddToVideoTrack(&sink, track);
   EXPECT_EQ(blink::WebMediaStreamSource::ReadyStateLive, sink.state());