| OLD | NEW |
| 1 /* | 1 /* |
| 2 * Copyright (c) 2013 The WebRTC project authors. All Rights Reserved. | 2 * Copyright (c) 2013 The WebRTC project authors. All Rights Reserved. |
| 3 * | 3 * |
| 4 * Use of this source code is governed by a BSD-style license | 4 * Use of this source code is governed by a BSD-style license |
| 5 * that can be found in the LICENSE file in the root of the source | 5 * that can be found in the LICENSE file in the root of the source |
| 6 * tree. An additional intellectual property rights grant can be found | 6 * tree. An additional intellectual property rights grant can be found |
| 7 * in the file PATENTS. All contributing project authors may | 7 * in the file PATENTS. All contributing project authors may |
| 8 * be found in the AUTHORS file in the root of the source tree. | 8 * be found in the AUTHORS file in the root of the source tree. |
| 9 */ | 9 */ |
| 10 | 10 |
| (...skipping 14 matching lines...) Expand all Loading... |
| 25 | 25 |
| 26 using ::testing::_; | 26 using ::testing::_; |
| 27 using ::testing::AnyNumber; | 27 using ::testing::AnyNumber; |
| 28 using ::testing::Return; | 28 using ::testing::Return; |
| 29 | 29 |
| 30 namespace webrtc { | 30 namespace webrtc { |
| 31 | 31 |
| 32 class ScreenCapturerMacTest : public testing::Test { | 32 class ScreenCapturerMacTest : public testing::Test { |
| 33 public: | 33 public: |
| 34 // Verifies that the whole screen is initially dirty. | 34 // Verifies that the whole screen is initially dirty. |
| 35 void CaptureDoneCallback1(DesktopCapturer::Result result, | 35 void CaptureDoneCallback1(DesktopFrame* frame); |
| 36 std::unique_ptr<DesktopFrame>* frame); | |
| 37 | 36 |
| 38 // Verifies that a rectangle explicitly marked as dirty is propagated | 37 // Verifies that a rectangle explicitly marked as dirty is propagated |
| 39 // correctly. | 38 // correctly. |
| 40 void CaptureDoneCallback2(DesktopCapturer::Result result, | 39 void CaptureDoneCallback2(DesktopFrame* frame); |
| 41 std::unique_ptr<DesktopFrame>* frame); | |
| 42 | 40 |
| 43 protected: | 41 protected: |
| 44 void SetUp() override { | 42 void SetUp() override { |
| 45 capturer_.reset( | 43 capturer_.reset( |
| 46 ScreenCapturer::Create(DesktopCaptureOptions::CreateDefault())); | 44 ScreenCapturer::Create(DesktopCaptureOptions::CreateDefault())); |
| 47 } | 45 } |
| 48 | 46 |
| 49 std::unique_ptr<ScreenCapturer> capturer_; | 47 std::unique_ptr<ScreenCapturer> capturer_; |
| 50 MockScreenCapturerCallback callback_; | 48 MockScreenCapturerCallback callback_; |
| 51 }; | 49 }; |
| 52 | 50 |
| 53 void ScreenCapturerMacTest::CaptureDoneCallback1( | 51 void ScreenCapturerMacTest::CaptureDoneCallback1( |
| 54 DesktopCapturer::Result result, | 52 DesktopFrame* frame) { |
| 55 std::unique_ptr<DesktopFrame>* frame) { | 53 std::unique_ptr<DesktopFrame> owned_frame(frame); |
| 56 EXPECT_EQ(result, DesktopCapturer::Result::SUCCESS); | |
| 57 | 54 |
| 58 MacDesktopConfiguration config = MacDesktopConfiguration::GetCurrent( | 55 MacDesktopConfiguration config = MacDesktopConfiguration::GetCurrent( |
| 59 MacDesktopConfiguration::BottomLeftOrigin); | 56 MacDesktopConfiguration::BottomLeftOrigin); |
| 60 | 57 |
| 61 // Verify that the region contains full frame. | 58 // Verify that the region contains full frame. |
| 62 DesktopRegion::Iterator it((*frame)->updated_region()); | 59 DesktopRegion::Iterator it(frame->updated_region()); |
| 63 EXPECT_TRUE(!it.IsAtEnd() && it.rect().equals(config.pixel_bounds)); | 60 EXPECT_TRUE(!it.IsAtEnd() && it.rect().equals(config.pixel_bounds)); |
| 64 } | 61 } |
| 65 | 62 |
| 66 void ScreenCapturerMacTest::CaptureDoneCallback2( | 63 void ScreenCapturerMacTest::CaptureDoneCallback2( |
| 67 DesktopCapturer::Result result, | 64 DesktopFrame* frame) { |
| 68 std::unique_ptr<DesktopFrame>* frame) { | 65 std::unique_ptr<DesktopFrame> owned_frame(frame); |
| 69 EXPECT_EQ(result, DesktopCapturer::Result::SUCCESS); | |
| 70 | 66 |
| 71 MacDesktopConfiguration config = MacDesktopConfiguration::GetCurrent( | 67 MacDesktopConfiguration config = MacDesktopConfiguration::GetCurrent( |
| 72 MacDesktopConfiguration::BottomLeftOrigin); | 68 MacDesktopConfiguration::BottomLeftOrigin); |
| 73 int width = config.pixel_bounds.width(); | 69 int width = config.pixel_bounds.width(); |
| 74 int height = config.pixel_bounds.height(); | 70 int height = config.pixel_bounds.height(); |
| 75 | 71 |
| 76 EXPECT_EQ(width, (*frame)->size().width()); | 72 EXPECT_EQ(width, frame->size().width()); |
| 77 EXPECT_EQ(height, (*frame)->size().height()); | 73 EXPECT_EQ(height, frame->size().height()); |
| 78 EXPECT_TRUE((*frame)->data() != NULL); | 74 EXPECT_TRUE(frame->data() != NULL); |
| 79 // Depending on the capture method, the screen may be flipped or not, so | 75 // Depending on the capture method, the screen may be flipped or not, so |
| 80 // the stride may be positive or negative. | 76 // the stride may be positive or negative. |
| 81 EXPECT_EQ(static_cast<int>(sizeof(uint32_t) * width), | 77 EXPECT_EQ(static_cast<int>(sizeof(uint32_t) * width), |
| 82 abs((*frame)->stride())); | 78 abs(frame->stride())); |
| 83 } | 79 } |
| 84 | 80 |
| 85 TEST_F(ScreenCapturerMacTest, Capture) { | 81 TEST_F(ScreenCapturerMacTest, Capture) { |
| 86 EXPECT_CALL(callback_, | 82 EXPECT_CALL(callback_, OnCaptureCompleted(_)) |
| 87 OnCaptureResultPtr(DesktopCapturer::Result::SUCCESS, _)) | |
| 88 .Times(2) | 83 .Times(2) |
| 89 .WillOnce(Invoke(this, &ScreenCapturerMacTest::CaptureDoneCallback1)) | 84 .WillOnce(Invoke(this, &ScreenCapturerMacTest::CaptureDoneCallback1)) |
| 90 .WillOnce(Invoke(this, &ScreenCapturerMacTest::CaptureDoneCallback2)); | 85 .WillOnce(Invoke(this, &ScreenCapturerMacTest::CaptureDoneCallback2)); |
| 91 | 86 |
| 92 SCOPED_TRACE(""); | 87 SCOPED_TRACE(""); |
| 93 capturer_->Start(&callback_); | 88 capturer_->Start(&callback_); |
| 94 | 89 |
| 95 // Check that we get an initial full-screen updated. | 90 // Check that we get an initial full-screen updated. |
| 96 capturer_->Capture(DesktopRegion()); | 91 capturer_->Capture(DesktopRegion()); |
| 97 | 92 |
| 98 // Check that subsequent dirty rects are propagated correctly. | 93 // Check that subsequent dirty rects are propagated correctly. |
| 99 capturer_->Capture(DesktopRegion()); | 94 capturer_->Capture(DesktopRegion()); |
| 100 } | 95 } |
| 101 | 96 |
| 102 } // namespace webrtc | 97 } // namespace webrtc |
| OLD | NEW |