Cleanup VideoFramePump.

remoting/host/mouse_shape_pump_unittest.cc

Index: remoting/host/mouse_shape_pump_unittest.cc
diff --git a/remoting/host/mouse_shape_pump_unittest.cc b/remoting/host/mouse_shape_pump_unittest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..acbb7f26670957846afb4317fa84f9dab52335c6
--- /dev/null
+++ b/remoting/host/mouse_shape_pump_unittest.cc
@@ -0,0 +1,195 @@
+// 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 "remoting/host/mouse_shape_pump.h"
+
+#include "base/bind.h"
+#include "base/message_loop/message_loop.h"
+#include "base/run_loop.h"
+#include "base/single_thread_task_runner.h"
+#include "remoting/base/auto_thread.h"
+#include "remoting/base/auto_thread_task_runner.h"
+#include "remoting/host/fake_mouse_cursor_monitor.h"
+#include "remoting/host/host_mock_objects.h"
+#include "remoting/proto/control.pb.h"
+#include "remoting/proto/video.pb.h"
+#include "remoting/protocol/protocol_mock_objects.h"
+#include "testing/gmock/include/gmock/gmock.h"
+#include "testing/gtest/include/gtest/gtest.h"
+#include "third_party/webrtc/modules/desktop_capture/desktop_frame.h"
+#include "third_party/webrtc/modules/desktop_capture/mouse_cursor.h"
+
+using ::remoting::protocol::MockClientStub;
+
+using ::testing::_;
+using ::testing::DoAll;
+using ::testing::InSequence;
+using ::testing::InvokeWithoutArgs;
+
+namespace remoting {
+
+static const int kCursorWidth = 64;
+static const int kCursorHeight = 32;
+static const int kHotspotX = 11;
+static const int kHotspotY = 12;
+
+class ThreadCheckMouseCursorMonitor : public FakeMouseCursorMonitor {
+ public:
+  ThreadCheckMouseCursorMonitor(
+      scoped_refptr<base::SingleThreadTaskRunner> task_runner)
+      : task_runner_(task_runner) {
+  }
+  ~ThreadCheckMouseCursorMonitor() override {
+    EXPECT_TRUE(task_runner_->BelongsToCurrentThread());
+  }
+
+ private:
+  scoped_refptr<base::SingleThreadTaskRunner> task_runner_;
+
+  DISALLOW_COPY_AND_ASSIGN(ThreadCheckMouseCursorMonitor);
+};
+
+class MouseShapePumpTest : public testing::Test {
+ public:
+  MouseShapePumpTest();
+
+  void SetUp() override;
+  void TearDown() override;
+
+  void StartMouseShapePump(
+      scoped_ptr<webrtc::MouseCursorMonitor> mouse_monitor);
+  void StopMouseShapePump();
+
+  // webrtc::MouseCursorMonitor mocks.
+  void OnMouseCursorMonitorInit(
+      webrtc::MouseCursorMonitor::Callback* callback,
+      webrtc::MouseCursorMonitor::Mode mode);
+  void OnCaptureMouse();
+  void SetCursorShape(const protocol::CursorShapeInfo& cursor_shape);
+
+ protected:
+  base::MessageLoop message_loop_;
+  base::RunLoop run_loop_;
+  scoped_refptr<AutoThreadTaskRunner> capture_task_runner_;
+  scoped_refptr<AutoThreadTaskRunner> main_task_runner_;
+  scoped_ptr<MouseShapePump> pump_;
+
+  MockClientStub client_stub_;
+
+  // Points to the callback passed to webrtc::MouseCursor::Init().
+  webrtc::MouseCursorMonitor::Callback* mouse_monitor_callback_;
+
+ private:
+  DISALLOW_COPY_AND_ASSIGN(MouseShapePumpTest);
+};
+
+MouseShapePumpTest::MouseShapePumpTest()
+    : mouse_monitor_callback_(nullptr) {
+}
+
+void MouseShapePumpTest::SetUp() {
+  main_task_runner_ = new AutoThreadTaskRunner(
+      message_loop_.message_loop_proxy(), run_loop_.QuitClosure());
+  capture_task_runner_ = main_task_runner_;
+}
+
+void MouseShapePumpTest::TearDown() {
+  // Release the task runners, so that the test can quit.
+  capture_task_runner_ = nullptr;
+  main_task_runner_ = nullptr;
+
+  // Run the MessageLoop until everything has torn down.
+  run_loop_.Run();
+}
+
+void MouseShapePumpTest::StartMouseShapePump(
+    scoped_ptr<webrtc::MouseCursorMonitor> mouse_monitor) {
+  pump_.reset(new MouseShapePump(main_task_runner_, capture_task_runner_,
+                                 mouse_monitor.Pass(), &client_stub_));
+}
+
+void MouseShapePumpTest::StopMouseShapePump() {
+  pump_.reset();
+}
+
+void MouseShapePumpTest::OnCaptureMouse() {
+  EXPECT_TRUE(mouse_monitor_callback_);

[Wez, 2015/02/11 01:05:07]

nit: Does EXPECT work here, or do you need ASSERT?  (Sorry, can't remember the
restrictions off the top of my head)

[Sergey Ulanov, 2015/02/11 19:21:35]

replaced with ASSERT. the OnMouseCursor() call below will crash without it (note
that most of this code has been moved from the old VideoSchedulerTest).

+
+  scoped_ptr<webrtc::MouseCursor> mouse_cursor(
+      new webrtc::MouseCursor(
+          new webrtc::BasicDesktopFrame(
+              webrtc::DesktopSize(kCursorWidth, kCursorHeight)),
+          webrtc::DesktopVector(kHotspotX, kHotspotY)));
+
+  mouse_monitor_callback_->OnMouseCursor(mouse_cursor.release());
+}
+
+void MouseShapePumpTest::OnMouseCursorMonitorInit(
+    webrtc::MouseCursorMonitor::Callback* callback,
+    webrtc::MouseCursorMonitor::Mode mode) {
+  EXPECT_FALSE(mouse_monitor_callback_);
+  EXPECT_TRUE(callback);
+
+  mouse_monitor_callback_ = callback;
+}
+
+void MouseShapePumpTest::SetCursorShape(
+    const protocol::CursorShapeInfo& cursor_shape) {
+  EXPECT_TRUE(cursor_shape.has_width());
+  EXPECT_EQ(kCursorWidth, cursor_shape.width());
+  EXPECT_TRUE(cursor_shape.has_height());
+  EXPECT_EQ(kCursorHeight, cursor_shape.height());
+  EXPECT_TRUE(cursor_shape.has_hotspot_x());
+  EXPECT_EQ(kHotspotX, cursor_shape.hotspot_x());
+  EXPECT_TRUE(cursor_shape.has_hotspot_y());
+  EXPECT_EQ(kHotspotY, cursor_shape.hotspot_y());
+  EXPECT_TRUE(cursor_shape.has_data());
+  EXPECT_EQ(kCursorWidth * kCursorHeight * webrtc::DesktopFrame::kBytesPerPixel,
+            static_cast<int>(cursor_shape.data().size()));
+}
+
+// This test mocks MouseCursorMonitor and network layer to verify that the

[Wez, 2015/02/11 01:05:07]

What is it that it mocks from the network layer?

[Sergey Ulanov, 2015/02/11 19:21:35]

ClientStub. updated the comment.

+// MouseShapePump sends the cursor successfully.
+TEST_F(MouseShapePumpTest, FirstCursor) {
+  scoped_ptr<MockMouseCursorMonitor> cursor_monitor(
+      new MockMouseCursorMonitor());
+
+  {
+    InSequence s;
+
+    EXPECT_CALL(*cursor_monitor, Init(_, _))
+        .WillOnce(
+            Invoke(this, &MouseShapePumpTest::OnMouseCursorMonitorInit));
+
+    EXPECT_CALL(*cursor_monitor, Capture())
+        .WillRepeatedly(Invoke(this, &MouseShapePumpTest::OnCaptureMouse));
+  }
+
+  // Stop the pump once it has captured the cursor.
+  EXPECT_CALL(client_stub_, SetCursorShape(_))
+      .WillOnce(DoAll(
+          Invoke(this, &MouseShapePumpTest::SetCursorShape),
+          InvokeWithoutArgs(this, &MouseShapePumpTest::StopMouseShapePump)))
+      .RetiresOnSaturation();
+
+  // Start the pump.
+  StartMouseShapePump(cursor_monitor.Pass());
+
+  // Run until there are no more pending tasks from the MouseShapePump.
+  base::RunLoop().RunUntilIdle();
+}
+
+// Verify that the mouse monitor is torn down on the correct thread.
+TEST_F(MouseShapePumpTest, DeleteOnThreads) {
+  capture_task_runner_ = AutoThread::Create("capture", main_task_runner_);
+
+  scoped_ptr<webrtc::MouseCursorMonitor> mouse_cursor_monitor(
+      new ThreadCheckMouseCursorMonitor(capture_task_runner_));
+
+  // Start and stop the pump, so it will tear down the mouse monitor.
+  StartMouseShapePump(mouse_cursor_monitor.Pass());
+  StopMouseShapePump();
+}
+
+}  // namespace remoting