MediaRecorderHandler (video part) and unittests

content/renderer/media/media_recorder_handler_unittest.cc

+// Copyright 2015 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/run_loop.h"
+#include "base/strings/utf_string_conversions.h"
+#include "content/child/child_process.h"
+#include "content/renderer/media/media_recorder_handler.h"
+#include "content/renderer/media/mock_media_stream_registry.h"
+#include "media/base/video_frame.h"
+#include "testing/gmock/include/gmock/gmock.h"
+#include "testing/gtest/include/gtest/gtest.h"
+#include "third_party/WebKit/public/platform/WebString.h"
+
+using ::testing::_;
+using ::testing::AtLeast;
+using ::testing::InSequence;
+using ::testing::Lt;
+using ::testing::Mock;
+
+namespace content {
+
+ACTION_P(RunClosure, closure) {
+  closure.Run();
+}
+
+static const std::string kTestStreamUrl = "stream_url";
+static const std::string kTestVideoTrackId = "video_track_id";
+
+class MediaRecorderHandlerTest
+    : public testing::Test
+    , public MediaRecorderHandlerClient {
+ public:
+  MediaRecorderHandlerTest()
+      : media_recorder_handler_(new MediaRecorderHandler()) {
+    EXPECT_FALSE(media_recorder_handler_->recording_);
+
+    registry_.Init(kTestStreamUrl);
+    registry_.AddVideoTrack(kTestVideoTrackId);
+  }
+
+  MOCK_METHOD3(writeData, void(const char*, size_t, bool));
+
+  bool recording() const { return media_recorder_handler_->recording_; }
+  bool hasVideoRecorders() const {
+    return !media_recorder_handler_->video_recorders_.empty();
+  }
+
+  void OnVideoFrameForTesting(const scoped_refptr<media::VideoFrame>& frame) {
+    media_recorder_handler_->OnVideoFrameForTesting(frame,
+                                                    base::TimeTicks::Now());
+  }
+
+  // The Class under test. Needs to be scoped_ptr to force its destruction.
+  scoped_ptr<MediaRecorderHandler> media_recorder_handler_;
+
+  // A ChildProcess and a MessageLoopForUI are both needed to fool the Tracks
+  // and Sources in |registry_| into believing they are on the right threads.
+  const base::MessageLoopForUI message_loop_;
+  const ChildProcess child_process_;
+  MockMediaStreamRegistry registry_;
+
+ private:
+  DISALLOW_COPY_AND_ASSIGN(MediaRecorderHandlerTest);
+};
+
+// Checks that canSupportMimeType() works as expected.
+// TODO(mcasas): revisit this when canSupportMimeType() is fully implemented.
+TEST_F(MediaRecorderHandlerTest, CanSupportMimeType) {
+  const blink::WebString mime_type(base::UTF8ToUTF16("video/vp8"));
+  EXPECT_FALSE(media_recorder_handler_->canSupportMimeType(mime_type));
+}
+
+// Checks that the initialization-destruction sequence works fine.
+TEST_F(MediaRecorderHandlerTest, InitializeStartStop) {
+  const blink::WebString mime_type(base::UTF8ToUTF16("video/vp8"));
+  EXPECT_TRUE(media_recorder_handler_->initialize(this,
+                                                 registry_.test_stream(),
+                                                 mime_type));
+  EXPECT_FALSE(recording());
+  EXPECT_FALSE(hasVideoRecorders());
+
+  EXPECT_TRUE(media_recorder_handler_->start());
+  EXPECT_TRUE(recording());
+  EXPECT_TRUE(hasVideoRecorders());
+
+  media_recorder_handler_->stop();
+  EXPECT_FALSE(recording());
+  EXPECT_FALSE(hasVideoRecorders());
+
+  // Expect a last call on destruction.
+  EXPECT_CALL(*this, writeData(_, _, true)).Times(1);
+  media_recorder_handler_.reset();
+}
+
+// Sends 2 frames and expect them as WebM contained encoded data in writeData().
+TEST_F(MediaRecorderHandlerTest, EncodeVideoFrames) {
+  const blink::WebString mime_type(base::UTF8ToUTF16("video/vp8"));
+  EXPECT_TRUE(media_recorder_handler_->initialize(this, registry_.test_stream(),
+                                                  mime_type));
+  EXPECT_TRUE(media_recorder_handler_->start());
+
+  InSequence s;
+  const scoped_refptr<media::VideoFrame> video_frame =
+      media::VideoFrame::CreateBlackFrame(gfx::Size(160, 80));
+
+  {
+    base::RunLoop run_loop;
+    base::Closure quit_closure = run_loop.QuitClosure();
+    // writeData() is pinged a number of times as the WebM header is written;
+    // the last time it is called it has the encoded data.
+    const size_t kEncodedDataSize = 52;
+    EXPECT_CALL(*this, writeData(_, Lt(kEncodedDataSize), false))
+        .Times(AtLeast(1));
+    EXPECT_CALL(*this, writeData(_, kEncodedDataSize, false))
+        .Times(1)
+        .WillOnce(RunClosure(quit_closure));
+
+    OnVideoFrameForTesting(video_frame);
+    run_loop.Run();
+  }
+
+  {
+    base::RunLoop run_loop;
+    base::Closure quit_closure = run_loop.QuitClosure();
+    // The second time around writeData() is called a number of times to write
+    // the WebM frame header, and then is pinged with the encoded data.
+    const size_t kSecondEncodedDataSize = 32;
+    EXPECT_CALL(*this, writeData(_, Lt(kSecondEncodedDataSize), false))
+        .Times(AtLeast(1));
+    EXPECT_CALL(*this, writeData(_, kSecondEncodedDataSize, false))
+        .Times(1)
+        .WillOnce(RunClosure(quit_closure));
+
+    OnVideoFrameForTesting(video_frame);
+    run_loop.Run();
+  }
+
+  media_recorder_handler_->stop();
+
+  // Expect a last call on destruction, with size 0 and |lastInSlice| true.
+  EXPECT_CALL(*this, writeData(nullptr, 0, true)).Times(1);
+  media_recorder_handler_.reset();
+}
+
+}  // namespace content