OLD | NEW |
| (Empty) |
1 // Copyright 2013 The Chromium Authors. All rights reserved. | |
2 // Use of this source code is governed by a BSD-style license that can be | |
3 // found in the LICENSE file. | |
4 | |
5 #include "content/browser/renderer_host/media/desktop_capture_device_ash.h" | |
6 | |
7 #include "base/synchronization/waitable_event.h" | |
8 #include "content/browser/browser_thread_impl.h" | |
9 #include "media/video/capture/video_capture_types.h" | |
10 #include "testing/gmock/include/gmock/gmock.h" | |
11 #include "testing/gtest/include/gtest/gtest.h" | |
12 #include "ui/aura/client/window_tree_client.h" | |
13 #include "ui/aura/test/aura_test_helper.h" | |
14 #include "ui/aura/test/test_window_delegate.h" | |
15 #include "ui/aura/window.h" | |
16 | |
17 using ::testing::_; | |
18 using ::testing::AnyNumber; | |
19 using ::testing::DoAll; | |
20 using ::testing::Expectation; | |
21 using ::testing::InvokeWithoutArgs; | |
22 using ::testing::SaveArg; | |
23 | |
24 namespace content { | |
25 namespace { | |
26 | |
27 const int kFrameRate = 30; | |
28 | |
29 class MockDeviceClient : public media::VideoCaptureDevice::Client { | |
30 public: | |
31 MOCK_METHOD2(ReserveOutputBuffer, | |
32 scoped_refptr<Buffer>(media::VideoFrame::Format format, | |
33 const gfx::Size& dimensions)); | |
34 MOCK_METHOD0(OnError, void()); | |
35 MOCK_METHOD7(OnIncomingCapturedFrame, | |
36 void(const uint8* data, | |
37 int length, | |
38 base::Time timestamp, | |
39 int rotation, | |
40 bool flip_vert, | |
41 bool flip_horiz, | |
42 const media::VideoCaptureFormat& frame_format)); | |
43 MOCK_METHOD5(OnIncomingCapturedBuffer, | |
44 void(const scoped_refptr<Buffer>& buffer, | |
45 media::VideoFrame::Format format, | |
46 const gfx::Size& dimensions, | |
47 base::Time timestamp, | |
48 int frame_rate)); | |
49 }; | |
50 | |
51 // Test harness that sets up a minimal environment with necessary stubs. | |
52 class DesktopCaptureDeviceAshTest : public testing::Test { | |
53 public: | |
54 DesktopCaptureDeviceAshTest() | |
55 : browser_thread_for_ui_(BrowserThread::UI, &message_loop_) {} | |
56 virtual ~DesktopCaptureDeviceAshTest() {} | |
57 | |
58 protected: | |
59 virtual void SetUp() OVERRIDE { | |
60 helper_.reset(new aura::test::AuraTestHelper(&message_loop_)); | |
61 helper_->SetUp(); | |
62 | |
63 // We need a window to cover desktop area so that DesktopCaptureDeviceAsh | |
64 // can use gfx::NativeWindow::GetWindowAtScreenPoint() to locate the | |
65 // root window associated with the primary display. | |
66 gfx::Rect desktop_bounds = root_window()->bounds(); | |
67 window_delegate_.reset(new aura::test::TestWindowDelegate()); | |
68 desktop_window_.reset(new aura::Window(window_delegate_.get())); | |
69 desktop_window_->Init(ui::LAYER_TEXTURED); | |
70 desktop_window_->SetBounds(desktop_bounds); | |
71 aura::client::ParentWindowWithContext( | |
72 desktop_window_.get(), root_window(), desktop_bounds); | |
73 desktop_window_->Show(); | |
74 } | |
75 | |
76 virtual void TearDown() OVERRIDE { | |
77 helper_->RunAllPendingInMessageLoop(); | |
78 root_window()->RemoveChild(desktop_window_.get()); | |
79 desktop_window_.reset(); | |
80 window_delegate_.reset(); | |
81 helper_->TearDown(); | |
82 } | |
83 | |
84 aura::Window* root_window() { return helper_->root_window(); } | |
85 | |
86 private: | |
87 base::MessageLoopForUI message_loop_; | |
88 BrowserThreadImpl browser_thread_for_ui_; | |
89 scoped_ptr<aura::test::AuraTestHelper> helper_; | |
90 scoped_ptr<aura::Window> desktop_window_; | |
91 scoped_ptr<aura::test::TestWindowDelegate> window_delegate_; | |
92 | |
93 DISALLOW_COPY_AND_ASSIGN(DesktopCaptureDeviceAshTest); | |
94 }; | |
95 | |
96 TEST_F(DesktopCaptureDeviceAshTest, StartAndStop) { | |
97 DesktopMediaID source(DesktopMediaID::TYPE_SCREEN, 0); | |
98 scoped_ptr<media::VideoCaptureDevice> capture_device( | |
99 DesktopCaptureDeviceAsh::Create(source)); | |
100 | |
101 scoped_ptr<MockDeviceClient> client(new MockDeviceClient()); | |
102 EXPECT_CALL(*client, OnError()).Times(0); | |
103 | |
104 media::VideoCaptureParams capture_params; | |
105 capture_params.requested_format.frame_size.SetSize(640, 480); | |
106 capture_params.requested_format.frame_rate = kFrameRate; | |
107 capture_params.requested_format.pixel_format = media::PIXEL_FORMAT_I420; | |
108 capture_params.allow_resolution_change = false; | |
109 capture_device->AllocateAndStart( | |
110 capture_params, client.PassAs<media::VideoCaptureDevice::Client>()); | |
111 capture_device->StopAndDeAllocate(); | |
112 } | |
113 | |
114 } // namespace | |
115 } // namespace content | |
OLD | NEW |