Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(499)

Unified Diff: remoting/host/video_scheduler_unittest.cc

Issue 893353002: Rename VideoScheduler->VideoFramePipe. (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@scheduler_cleanup
Patch Set: Created 5 years, 10 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
« no previous file with comments | « remoting/host/video_scheduler.cc ('k') | remoting/remoting_host_srcs.gypi » ('j') | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: remoting/host/video_scheduler_unittest.cc
diff --git a/remoting/host/video_scheduler_unittest.cc b/remoting/host/video_scheduler_unittest.cc
deleted file mode 100644
index 5adfafa0c9ead7bcee9e69f934ab0d5bc8cfe7df..0000000000000000000000000000000000000000
--- a/remoting/host/video_scheduler_unittest.cc
+++ /dev/null
@@ -1,347 +0,0 @@
-// Copyright (c) 2012 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/video_scheduler.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/codec/video_encoder.h"
-#include "remoting/codec/video_encoder_verbatim.h"
-#include "remoting/host/fake_desktop_capturer.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"
-#include "third_party/webrtc/modules/desktop_capture/screen_capturer_mock_objects.h"
-
-using ::remoting::protocol::MockClientStub;
-using ::remoting::protocol::MockVideoStub;
-
-using ::testing::_;
-using ::testing::AtLeast;
-using ::testing::AnyNumber;
-using ::testing::DeleteArg;
-using ::testing::DoAll;
-using ::testing::Expectation;
-using ::testing::InSequence;
-using ::testing::InvokeWithoutArgs;
-using ::testing::Return;
-using ::testing::ReturnRef;
-using ::testing::SaveArg;
-
-namespace remoting {
-
-namespace {
-
-ACTION(FinishEncode) {
- scoped_ptr<VideoPacket> packet(new VideoPacket());
- return packet.release();
-}
-
-ACTION(FinishSend) {
- arg1.Run();
-}
-
-} // namespace
-
-static const int kWidth = 640;
-static const int kHeight = 480;
-static const int kCursorWidth = 64;
-static const int kCursorHeight = 32;
-static const int kHotspotX = 11;
-static const int kHotspotY = 12;
-
-class MockVideoEncoder : public VideoEncoder {
- public:
- MockVideoEncoder() {}
- virtual ~MockVideoEncoder() {}
-
- scoped_ptr<VideoPacket> Encode(const webrtc::DesktopFrame& frame) {
- return make_scoped_ptr(EncodePtr(frame));
- }
- MOCK_METHOD1(EncodePtr, VideoPacket*(const webrtc::DesktopFrame& frame));
-
- private:
- DISALLOW_COPY_AND_ASSIGN(MockVideoEncoder);
-};
-
-class ThreadCheckVideoEncoder : public VideoEncoderVerbatim {
- public:
- ThreadCheckVideoEncoder(
- scoped_refptr<base::SingleThreadTaskRunner> task_runner)
- : task_runner_(task_runner) {
- }
- ~ThreadCheckVideoEncoder() override {
- EXPECT_TRUE(task_runner_->BelongsToCurrentThread());
- }
-
- private:
- scoped_refptr<base::SingleThreadTaskRunner> task_runner_;
-
- DISALLOW_COPY_AND_ASSIGN(ThreadCheckVideoEncoder);
-};
-
-class ThreadCheckDesktopCapturer : public FakeDesktopCapturer {
- public:
- ThreadCheckDesktopCapturer(
- scoped_refptr<base::SingleThreadTaskRunner> task_runner)
- : task_runner_(task_runner) {
- }
- ~ThreadCheckDesktopCapturer() override {
- EXPECT_TRUE(task_runner_->BelongsToCurrentThread());
- }
-
- private:
- scoped_refptr<base::SingleThreadTaskRunner> task_runner_;
-
- DISALLOW_COPY_AND_ASSIGN(ThreadCheckDesktopCapturer);
-};
-
-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 VideoSchedulerTest : public testing::Test {
- public:
- VideoSchedulerTest();
-
- void SetUp() override;
- void TearDown() override;
-
- void StartVideoScheduler(
- scoped_ptr<webrtc::DesktopCapturer> capturer,
- scoped_ptr<VideoEncoder> encoder,
- scoped_ptr<webrtc::MouseCursorMonitor> mouse_monitor);
- void StopVideoScheduler();
-
- // webrtc::DesktopCapturer mocks.
- void OnCapturerStart(webrtc::DesktopCapturer::Callback* callback);
- void OnCaptureFrame(const webrtc::DesktopRegion& region);
-
- // 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> encode_task_runner_;
- scoped_refptr<AutoThreadTaskRunner> main_task_runner_;
- scoped_refptr<VideoScheduler> scheduler_;
-
- MockClientStub client_stub_;
- MockVideoStub video_stub_;
-
- // Points to the callback passed to webrtc::DesktopCapturer::Start().
- webrtc::DesktopCapturer::Callback* capturer_callback_;
-
- // Points to the callback passed to webrtc::MouseCursor::Init().
- webrtc::MouseCursorMonitor::Callback* mouse_monitor_callback_;
-
- private:
- DISALLOW_COPY_AND_ASSIGN(VideoSchedulerTest);
-};
-
-VideoSchedulerTest::VideoSchedulerTest()
- : capturer_callback_(nullptr),
- mouse_monitor_callback_(nullptr) {
-}
-
-void VideoSchedulerTest::SetUp() {
- main_task_runner_ = new AutoThreadTaskRunner(
- message_loop_.message_loop_proxy(), run_loop_.QuitClosure());
- capture_task_runner_ = main_task_runner_;
- encode_task_runner_ = main_task_runner_;
-}
-
-void VideoSchedulerTest::TearDown() {
- // Release the task runners, so that the test can quit.
- capture_task_runner_ = nullptr;
- encode_task_runner_ = nullptr;
- main_task_runner_ = nullptr;
-
- // Run the MessageLoop until everything has torn down.
- run_loop_.Run();
-}
-
-void VideoSchedulerTest::StartVideoScheduler(
- scoped_ptr<webrtc::DesktopCapturer> capturer,
- scoped_ptr<VideoEncoder> encoder,
- scoped_ptr<webrtc::MouseCursorMonitor> mouse_monitor) {
- scheduler_ = new VideoScheduler(
- capture_task_runner_,
- encode_task_runner_,
- main_task_runner_,
- capturer.Pass(),
- mouse_monitor.Pass(),
- encoder.Pass(),
- &client_stub_,
- &video_stub_);
- scheduler_->Start();
-}
-
-void VideoSchedulerTest::StopVideoScheduler() {
- scheduler_->Stop();
- scheduler_ = nullptr;
-}
-
-void VideoSchedulerTest::OnCapturerStart(
- webrtc::DesktopCapturer::Callback* callback) {
- EXPECT_FALSE(capturer_callback_);
- EXPECT_TRUE(callback);
-
- capturer_callback_ = callback;
-}
-
-void VideoSchedulerTest::OnCaptureFrame(const webrtc::DesktopRegion& region) {
- scoped_ptr<webrtc::DesktopFrame> frame(
- new webrtc::BasicDesktopFrame(webrtc::DesktopSize(kWidth, kHeight)));
- frame->mutable_updated_region()->SetRect(
- webrtc::DesktopRect::MakeXYWH(0, 0, 10, 10));
- capturer_callback_->OnCaptureCompleted(frame.release());
-}
-
-void VideoSchedulerTest::OnCaptureMouse() {
- EXPECT_TRUE(mouse_monitor_callback_);
-
- 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 VideoSchedulerTest::OnMouseCursorMonitorInit(
- webrtc::MouseCursorMonitor::Callback* callback,
- webrtc::MouseCursorMonitor::Mode mode) {
- EXPECT_FALSE(mouse_monitor_callback_);
- EXPECT_TRUE(callback);
-
- mouse_monitor_callback_ = callback;
-}
-
-void VideoSchedulerTest::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 capturer, encoder and network layer to simulate one capture
-// cycle. When the first encoded packet is submitted to the network
-// VideoScheduler is instructed to come to a complete stop. We expect the stop
-// sequence to be executed successfully.
-TEST_F(VideoSchedulerTest, StartAndStop) {
- scoped_ptr<webrtc::MockScreenCapturer> capturer(
- new webrtc::MockScreenCapturer());
- scoped_ptr<MockMouseCursorMonitor> cursor_monitor(
- new MockMouseCursorMonitor());
-
- {
- InSequence s;
-
- EXPECT_CALL(*cursor_monitor, Init(_, _))
- .WillOnce(
- Invoke(this, &VideoSchedulerTest::OnMouseCursorMonitorInit));
-
- EXPECT_CALL(*cursor_monitor, Capture())
- .WillRepeatedly(Invoke(this, &VideoSchedulerTest::OnCaptureMouse));
- }
-
- Expectation capturer_start =
- EXPECT_CALL(*capturer, Start(_))
- .WillOnce(Invoke(this, &VideoSchedulerTest::OnCapturerStart));
-
- // First the capturer is called.
- Expectation capturer_capture = EXPECT_CALL(*capturer, Capture(_))
- .After(capturer_start)
- .WillRepeatedly(Invoke(this, &VideoSchedulerTest::OnCaptureFrame));
-
- scoped_ptr<MockVideoEncoder> encoder(new MockVideoEncoder());
-
- // Expect the encoder be called.
- EXPECT_CALL(*encoder, EncodePtr(_))
- .WillRepeatedly(FinishEncode());
-
- // By default delete the arguments when ProcessVideoPacket is received.
- EXPECT_CALL(video_stub_, ProcessVideoPacketPtr(_, _))
- .WillRepeatedly(FinishSend());
-
- // When the first ProcessVideoPacket is received we stop the VideoScheduler.
- EXPECT_CALL(video_stub_, ProcessVideoPacketPtr(_, _))
- .WillOnce(DoAll(
- FinishSend(),
- InvokeWithoutArgs(this, &VideoSchedulerTest::StopVideoScheduler)))
- .RetiresOnSaturation();
-
- EXPECT_CALL(client_stub_, SetCursorShape(_))
- .WillOnce(Invoke(this, &VideoSchedulerTest::SetCursorShape));
-
- // Start video frame capture.
- scoped_ptr<webrtc::MouseCursorMonitor> mouse_cursor_monitor(
- new FakeMouseCursorMonitor());
- StartVideoScheduler(capturer.Pass(), encoder.Pass(), cursor_monitor.Pass());
-
- // Run until there are no more pending tasks from the VideoScheduler.
- // Otherwise, a lingering frame capture might attempt to trigger a capturer
- // expectation action and crash.
- base::RunLoop().RunUntilIdle();
-}
-
-// Verify that the capturer, encoder and mouse monitor are torn down on the
-// correct threads.
-TEST_F(VideoSchedulerTest, DeleteOnThreads) {
- capture_task_runner_ = AutoThread::Create("capture", main_task_runner_);
- encode_task_runner_ = AutoThread::Create("encode", main_task_runner_);
-
- scoped_ptr<webrtc::DesktopCapturer> capturer(
- new ThreadCheckDesktopCapturer(capture_task_runner_));
- scoped_ptr<VideoEncoder> encoder(
- new ThreadCheckVideoEncoder(encode_task_runner_));
- scoped_ptr<webrtc::MouseCursorMonitor> mouse_cursor_monitor(
- new ThreadCheckMouseCursorMonitor(capture_task_runner_));
-
- // Start and stop the scheduler, so it will tear down the screen capturer,
- // video encoder and mouse monitor.
- StartVideoScheduler(capturer.Pass(), encoder.Pass(),
- mouse_cursor_monitor.Pass());
- StopVideoScheduler();
-}
-
-} // namespace remoting
« no previous file with comments | « remoting/host/video_scheduler.cc ('k') | remoting/remoting_host_srcs.gypi » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698