Chromium Code Reviews| 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; |
| 19 | 20 |
| 20 namespace media { | 21 namespace media { |
| 21 | 22 |
| 22 MATCHER(DirtyRegionIsNonEmptyRect, "") { | 23 MATCHER(DirtyRegionIsNonEmptyRect, "") { |
| 23 const SkRegion& dirty_region = arg->dirty_region(); | 24 const webrtc::DesktopRegion& updated_region = arg->updated_region(); |
| 24 const SkIRect& dirty_region_bounds = dirty_region.getBounds(); | 25 webrtc::DesktopRegion::Iterator it(updated_region); |
| 25 if (dirty_region_bounds.isEmpty()) { | 26 return !it.IsAtEnd() && (it.Advance(), it.IsAtEnd()); |
|
alexeypa (please no reviews)
2013/05/08 22:24:59
The name of the matcher does not reflect what it d
Sergey Ulanov
2013/05/09 18:49:02
I think it's correct. This matcher verifies that t
| |
| 26 return false; | |
| 27 } | |
| 28 return dirty_region == SkRegion(dirty_region_bounds); | |
| 29 } | 27 } |
| 30 | 28 |
| 31 class ScreenCapturerTest : public testing::Test { | 29 class ScreenCapturerTest : public testing::Test { |
| 32 public: | 30 public: |
| 33 scoped_refptr<SharedBuffer> CreateSharedBuffer(uint32 size); | |
| 34 | 31 |
| 35 protected: | 32 protected: |
| 36 scoped_ptr<ScreenCapturer> capturer_; | 33 scoped_ptr<ScreenCapturer> capturer_; |
| 37 MockScreenCapturerDelegate delegate_; | 34 MockMouseShapeObserver mouse_observer_; |
| 35 MockScreenCapturerCallback callback_; | |
| 38 }; | 36 }; |
| 39 | 37 |
| 40 scoped_refptr<SharedBuffer> ScreenCapturerTest::CreateSharedBuffer( | |
| 41 uint32 size) { | |
| 42 return scoped_refptr<SharedBuffer>(new SharedBuffer(size)); | |
| 43 } | |
| 44 | |
| 45 TEST_F(ScreenCapturerTest, StartCapturer) { | 38 TEST_F(ScreenCapturerTest, StartCapturer) { |
| 46 capturer_ = ScreenCapturer::Create(); | 39 capturer_ = ScreenCapturer::Create(); |
| 47 capturer_->Start(&delegate_); | 40 capturer_->SetMouseShapeObserver(&mouse_observer_); |
| 41 capturer_->Start(&callback_); | |
| 48 } | 42 } |
| 49 | 43 |
| 50 TEST_F(ScreenCapturerTest, Capture) { | 44 TEST_F(ScreenCapturerTest, Capture) { |
| 51 // Assume that Start() treats the screen as invalid initially. | 45 // Assume that Start() treats the screen as invalid initially. |
| 52 EXPECT_CALL(delegate_, | 46 EXPECT_CALL(callback_, |
| 53 OnCaptureCompleted(DirtyRegionIsNonEmptyRect())); | 47 OnCaptureCompleted(DirtyRegionIsNonEmptyRect())); |
| 54 EXPECT_CALL(delegate_, OnCursorShapeChangedPtr(_)) | 48 EXPECT_CALL(mouse_observer_, OnCursorShapeChangedPtr(_)) |
| 55 .Times(AnyNumber()); | 49 .Times(AnyNumber()); |
| 56 | 50 |
| 57 EXPECT_CALL(delegate_, CreateSharedBuffer(_)) | 51 EXPECT_CALL(callback_, CreateSharedMemory(_)) |
| 58 .Times(AnyNumber()) | 52 .Times(AnyNumber()) |
| 59 .WillRepeatedly(Return(scoped_refptr<SharedBuffer>())); | 53 .WillRepeatedly(Return(static_cast<webrtc::SharedMemory*>(NULL))); |
| 60 | 54 |
| 61 capturer_ = ScreenCapturer::Create(); | 55 capturer_ = ScreenCapturer::Create(); |
| 62 capturer_->Start(&delegate_); | 56 capturer_->Start(&callback_); |
| 63 capturer_->CaptureFrame(); | 57 capturer_->Capture(webrtc::DesktopRegion()); |
| 64 } | 58 } |
| 65 | 59 |
| 66 #if defined(OS_WIN) | 60 #if defined(OS_WIN) |
| 67 | 61 |
| 68 TEST_F(ScreenCapturerTest, UseSharedBuffers) { | 62 TEST_F(ScreenCapturerTest, UseSharedBuffers) { |
| 69 EXPECT_CALL(delegate_, | 63 EXPECT_CALL(callback_, |
| 70 OnCaptureCompleted(DirtyRegionIsNonEmptyRect())); | 64 OnCaptureCompleted(DirtyRegionIsNonEmptyRect())); |
| 71 EXPECT_CALL(delegate_, OnCursorShapeChangedPtr(_)) | 65 EXPECT_CALL(mouse_observer_, OnCursorShapeChangedPtr(_)) |
| 72 .Times(AnyNumber()); | 66 .Times(AnyNumber()); |
| 73 | 67 |
| 74 EXPECT_CALL(delegate_, CreateSharedBuffer(_)) | 68 EXPECT_CALL(callback_, CreateSharedBuffer(_)) |
| 75 .Times(AnyNumber()) | 69 .Times(AnyNumber()) |
| 76 .WillRepeatedly(Invoke(this, &ScreenCapturerTest::CreateSharedBuffer)); | 70 .WillRepeatedly(Invoke(this, &ScreenCapturerTest::CreateSharedBuffer)); |
| 77 EXPECT_CALL(delegate_, ReleaseSharedBuffer(_)) | 71 EXPECT_CALL(callback_, ReleaseSharedBuffer(_)) |
| 78 .Times(AnyNumber()); | 72 .Times(AnyNumber()); |
| 79 | 73 |
| 80 capturer_ = ScreenCapturer::Create(); | 74 capturer_ = ScreenCapturer::Create(); |
| 81 capturer_->Start(&delegate_); | 75 capturer_->Start(&callback_); |
| 82 capturer_->CaptureFrame(); | 76 capturer_->CaptureFrame(); |
| 83 capturer_.reset(); | 77 capturer_.reset(); |
| 84 } | 78 } |
| 85 | 79 |
| 86 #endif // defined(OS_WIN) | 80 #endif // defined(OS_WIN) |
| 87 | 81 |
| 88 } // namespace media | 82 } // namespace media |
| OLD | NEW |