OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "media/video/capture/screen/screen_capturer.h" | 5 #include "media/video/capture/screen/screen_capturer.h" |
6 | 6 |
7 #include "base/bind.h" | 7 #include "base/bind.h" |
8 #if defined(OS_MACOSX) | 8 #if defined(OS_MACOSX) |
9 #include "base/mac/mac_util.h" | 9 #include "base/mac/mac_util.h" |
10 #endif // defined(OS_MACOSX) | 10 #endif // defined(OS_MACOSX) |
11 #include "media/video/capture/screen/screen_capture_data.h" | |
12 #include "media/video/capture/screen/screen_capturer_mock_objects.h" | 11 #include "media/video/capture/screen/screen_capturer_mock_objects.h" |
13 #include "testing/gmock/include/gmock/gmock.h" | 12 #include "testing/gmock/include/gmock/gmock.h" |
14 #include "testing/gtest/include/gtest/gtest.h" | 13 #include "testing/gtest/include/gtest/gtest.h" |
| 14 #include "third_party/webrtc/modules/desktop_capture/desktop_frame.h" |
| 15 #include "third_party/webrtc/modules/desktop_capture/desktop_region.h" |
15 | 16 |
16 using ::testing::_; | 17 using ::testing::_; |
17 using ::testing::AnyNumber; | 18 using ::testing::AnyNumber; |
18 using ::testing::Return; | 19 using ::testing::Return; |
| 20 using ::testing::SaveArg; |
| 21 |
| 22 const int kTestSharedMemoryId = 123; |
19 | 23 |
20 namespace media { | 24 namespace media { |
21 | 25 |
22 MATCHER(DirtyRegionIsNonEmptyRect, "") { | |
23 const SkRegion& dirty_region = arg->dirty_region(); | |
24 const SkIRect& dirty_region_bounds = dirty_region.getBounds(); | |
25 if (dirty_region_bounds.isEmpty()) { | |
26 return false; | |
27 } | |
28 return dirty_region == SkRegion(dirty_region_bounds); | |
29 } | |
30 | |
31 class ScreenCapturerTest : public testing::Test { | 26 class ScreenCapturerTest : public testing::Test { |
32 public: | 27 public: |
33 scoped_refptr<SharedBuffer> CreateSharedBuffer(uint32 size); | 28 webrtc::SharedMemory* CreateSharedMemory(size_t size); |
34 | 29 |
35 protected: | 30 protected: |
36 scoped_ptr<ScreenCapturer> capturer_; | 31 scoped_ptr<ScreenCapturer> capturer_; |
37 MockScreenCapturerDelegate delegate_; | 32 MockMouseShapeObserver mouse_observer_; |
| 33 MockScreenCapturerCallback callback_; |
38 }; | 34 }; |
39 | 35 |
40 scoped_refptr<SharedBuffer> ScreenCapturerTest::CreateSharedBuffer( | 36 class FakeSharedMemory : public webrtc::SharedMemory { |
41 uint32 size) { | 37 public: |
42 return scoped_refptr<SharedBuffer>(new SharedBuffer(size)); | 38 FakeSharedMemory(char* buffer, size_t size) |
| 39 : SharedMemory(buffer, size, 0, kTestSharedMemoryId), |
| 40 buffer_(buffer) { |
| 41 } |
| 42 virtual ~FakeSharedMemory() { |
| 43 delete[] buffer_; |
| 44 } |
| 45 private: |
| 46 char* buffer_; |
| 47 DISALLOW_COPY_AND_ASSIGN(FakeSharedMemory); |
| 48 }; |
| 49 |
| 50 webrtc::SharedMemory* ScreenCapturerTest::CreateSharedMemory(size_t size) { |
| 51 return new FakeSharedMemory(new char[size], size); |
43 } | 52 } |
44 | 53 |
45 TEST_F(ScreenCapturerTest, StartCapturer) { | 54 TEST_F(ScreenCapturerTest, StartCapturer) { |
46 capturer_ = ScreenCapturer::Create(); | 55 capturer_ = ScreenCapturer::Create(); |
47 capturer_->Start(&delegate_); | 56 capturer_->SetMouseShapeObserver(&mouse_observer_); |
| 57 capturer_->Start(&callback_); |
48 } | 58 } |
49 | 59 |
50 TEST_F(ScreenCapturerTest, Capture) { | 60 TEST_F(ScreenCapturerTest, Capture) { |
51 // Assume that Start() treats the screen as invalid initially. | 61 // Assume that Start() treats the screen as invalid initially. |
52 EXPECT_CALL(delegate_, | 62 webrtc::DesktopFrame* frame = NULL; |
53 OnCaptureCompleted(DirtyRegionIsNonEmptyRect())); | 63 EXPECT_CALL(callback_, OnCaptureCompleted(_)) |
54 EXPECT_CALL(delegate_, OnCursorShapeChangedPtr(_)) | 64 .WillOnce(SaveArg<0>(&frame)); |
| 65 EXPECT_CALL(mouse_observer_, OnCursorShapeChangedPtr(_)) |
55 .Times(AnyNumber()); | 66 .Times(AnyNumber()); |
56 | 67 |
57 EXPECT_CALL(delegate_, CreateSharedBuffer(_)) | 68 EXPECT_CALL(callback_, CreateSharedMemory(_)) |
58 .Times(AnyNumber()) | 69 .Times(AnyNumber()) |
59 .WillRepeatedly(Return(scoped_refptr<SharedBuffer>())); | 70 .WillRepeatedly(Return(static_cast<webrtc::SharedMemory*>(NULL))); |
60 | 71 |
61 capturer_ = ScreenCapturer::Create(); | 72 capturer_ = ScreenCapturer::Create(); |
62 capturer_->Start(&delegate_); | 73 capturer_->Start(&callback_); |
63 capturer_->CaptureFrame(); | 74 capturer_->Capture(webrtc::DesktopRegion()); |
| 75 |
| 76 ASSERT_TRUE(frame); |
| 77 EXPECT_GT(frame->size().width(), 0); |
| 78 EXPECT_GT(frame->size().height(), 0); |
| 79 EXPECT_GE(frame->stride(), |
| 80 frame->size().width() * webrtc::DesktopFrame::kBytesPerPixel); |
| 81 EXPECT_TRUE(frame->shared_memory() == NULL); |
| 82 |
| 83 // Verify that the region contains whole screen. |
| 84 EXPECT_FALSE(frame->updated_region().is_empty()); |
| 85 webrtc::DesktopRegion::Iterator it(frame->updated_region()); |
| 86 ASSERT_TRUE(!it.IsAtEnd()); |
| 87 EXPECT_TRUE(it.rect().equals(webrtc::DesktopRect::MakeSize(frame->size()))); |
| 88 it.Advance(); |
| 89 EXPECT_TRUE(it.IsAtEnd()); |
| 90 |
| 91 delete frame; |
64 } | 92 } |
65 | 93 |
66 #if defined(OS_WIN) | 94 #if defined(OS_WIN) |
67 | 95 |
68 TEST_F(ScreenCapturerTest, UseSharedBuffers) { | 96 TEST_F(ScreenCapturerTest, UseSharedBuffers) { |
69 EXPECT_CALL(delegate_, | 97 webrtc::DesktopFrame* frame = NULL; |
70 OnCaptureCompleted(DirtyRegionIsNonEmptyRect())); | 98 EXPECT_CALL(callback_, OnCaptureCompleted(_)) |
71 EXPECT_CALL(delegate_, OnCursorShapeChangedPtr(_)) | 99 .WillOnce(SaveArg<0>(&frame)); |
| 100 EXPECT_CALL(mouse_observer_, OnCursorShapeChangedPtr(_)) |
72 .Times(AnyNumber()); | 101 .Times(AnyNumber()); |
73 | 102 |
74 EXPECT_CALL(delegate_, CreateSharedBuffer(_)) | 103 EXPECT_CALL(callback_, CreateSharedMemory(_)) |
75 .Times(AnyNumber()) | 104 .Times(AnyNumber()) |
76 .WillRepeatedly(Invoke(this, &ScreenCapturerTest::CreateSharedBuffer)); | 105 .WillRepeatedly(Invoke(this, &ScreenCapturerTest::CreateSharedMemory)); |
77 EXPECT_CALL(delegate_, ReleaseSharedBuffer(_)) | |
78 .Times(AnyNumber()); | |
79 | 106 |
80 capturer_ = ScreenCapturer::Create(); | 107 capturer_ = ScreenCapturer::Create(); |
81 capturer_->Start(&delegate_); | 108 capturer_->Start(&callback_); |
82 capturer_->CaptureFrame(); | 109 capturer_->Capture(webrtc::DesktopRegion()); |
83 capturer_.reset(); | 110 |
| 111 ASSERT_TRUE(frame); |
| 112 ASSERT_TRUE(frame->shared_memory()); |
| 113 EXPECT_EQ(frame->shared_memory()->id(), kTestSharedMemoryId); |
| 114 |
| 115 delete frame; |
84 } | 116 } |
85 | 117 |
86 #endif // defined(OS_WIN) | 118 #endif // defined(OS_WIN) |
87 | 119 |
88 } // namespace media | 120 } // namespace media |
OLD | NEW |