Chrome MediaStream VideoTrack implementation.

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_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_);
+  }
+
+ protected:
+  // Create a track that's associated with |webkit_source_|.
+  blink::WebMediaStreamTrack CreateTrack() {
+    blink::WebMediaStreamTrack track;
+    track.initialize(base::UTF8ToUTF16("track_id"), webkit_source_);
+    track.setExtraData(new MediaStreamVideoTrack(mock_source_, &factory_));
+
+    MediaStreamVideoSource* source =
+        static_cast<MediaStreamVideoSource*>(track.source().extraData());
+    blink::WebMediaConstraints constraints;
+    constraints.initialize();
+    source->AddTrack(track,
+                     constraints,
+                     MediaStreamSource::ConstraintsCallback());
+    mock_source_->StartMockedSource();
+
+    return track;
+  }
+
+  MockMediaStreamVideoSource* mock_source() { return mock_source_; }
+
+ private:
+  MockMediaStreamDependencyFactory factory_;
+  blink::WebMediaStreamSource webkit_source_;
+  // |mock_source_| is owned by |webkit_source_|.
+  MockMediaStreamVideoSource* mock_source_;
+};
+
+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;
+  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;
+  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;
+  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