| OLD | NEW |
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | 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 | 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 "remoting/host/shaped_screen_capturer.h" | 5 #include "remoting/host/shaped_desktop_capturer.h" |
| 6 | 6 |
| 7 #include "remoting/host/desktop_shape_tracker.h" | 7 #include "remoting/host/desktop_shape_tracker.h" |
| 8 #include "remoting/host/fake_screen_capturer.h" | 8 #include "remoting/host/fake_desktop_capturer.h" |
| 9 #include "testing/gtest/include/gtest/gtest.h" | 9 #include "testing/gtest/include/gtest/gtest.h" |
| 10 #include "third_party/webrtc/modules/desktop_capture/desktop_frame.h" | 10 #include "third_party/webrtc/modules/desktop_capture/desktop_frame.h" |
| 11 #include "third_party/webrtc/modules/desktop_capture/desktop_geometry.h" | 11 #include "third_party/webrtc/modules/desktop_capture/desktop_geometry.h" |
| 12 #include "third_party/webrtc/modules/desktop_capture/desktop_region.h" | 12 #include "third_party/webrtc/modules/desktop_capture/desktop_region.h" |
| 13 | 13 |
| 14 namespace remoting { | 14 namespace remoting { |
| 15 | 15 |
| 16 class FakeDesktopShapeTracker : public DesktopShapeTracker { | 16 class FakeDesktopShapeTracker : public DesktopShapeTracker { |
| 17 public: | 17 public: |
| 18 FakeDesktopShapeTracker() {} | 18 FakeDesktopShapeTracker() {} |
| (...skipping 13 matching lines...) Expand all Loading... |
| 32 virtual const webrtc::DesktopRegion& desktop_shape() OVERRIDE { | 32 virtual const webrtc::DesktopRegion& desktop_shape() OVERRIDE { |
| 33 // desktop_shape() can't be called before RefreshDesktopShape(). | 33 // desktop_shape() can't be called before RefreshDesktopShape(). |
| 34 EXPECT_FALSE(shape_.is_empty()); | 34 EXPECT_FALSE(shape_.is_empty()); |
| 35 return shape_; | 35 return shape_; |
| 36 } | 36 } |
| 37 | 37 |
| 38 private: | 38 private: |
| 39 webrtc::DesktopRegion shape_; | 39 webrtc::DesktopRegion shape_; |
| 40 }; | 40 }; |
| 41 | 41 |
| 42 class ShapedScreenCapturerTest : public testing::Test, | 42 class ShapedDesktopCapturerTest : public testing::Test, |
| 43 public webrtc::DesktopCapturer::Callback { | 43 public webrtc::DesktopCapturer::Callback { |
| 44 public: | 44 public: |
| 45 // webrtc::DesktopCapturer::Callback interface | 45 // webrtc::DesktopCapturer::Callback interface |
| 46 virtual webrtc::SharedMemory* CreateSharedMemory(size_t size) OVERRIDE { | 46 virtual webrtc::SharedMemory* CreateSharedMemory(size_t size) OVERRIDE { |
| 47 return NULL; | 47 return NULL; |
| 48 } | 48 } |
| 49 | 49 |
| 50 virtual void OnCaptureCompleted(webrtc::DesktopFrame* frame) OVERRIDE { | 50 virtual void OnCaptureCompleted(webrtc::DesktopFrame* frame) OVERRIDE { |
| 51 last_frame_.reset(frame); | 51 last_frame_.reset(frame); |
| 52 } | 52 } |
| 53 | 53 |
| 54 scoped_ptr<webrtc::DesktopFrame> last_frame_; | 54 scoped_ptr<webrtc::DesktopFrame> last_frame_; |
| 55 }; | 55 }; |
| 56 | 56 |
| 57 // Verify that captured frame have shape. | 57 // Verify that captured frame have shape. |
| 58 TEST_F(ShapedScreenCapturerTest, Basic) { | 58 TEST_F(ShapedDesktopCapturerTest, Basic) { |
| 59 ShapedScreenCapturer capturer( | 59 ShapedDesktopCapturer capturer( |
| 60 scoped_ptr<webrtc::ScreenCapturer>(new FakeScreenCapturer()), | 60 scoped_ptr<webrtc::DesktopCapturer>(new FakeDesktopCapturer()), |
| 61 scoped_ptr<DesktopShapeTracker>(new FakeDesktopShapeTracker())); | 61 scoped_ptr<DesktopShapeTracker>(new FakeDesktopShapeTracker())); |
| 62 capturer.Start(this); | 62 capturer.Start(this); |
| 63 capturer.Capture(webrtc::DesktopRegion()); | 63 capturer.Capture(webrtc::DesktopRegion()); |
| 64 ASSERT_TRUE(last_frame_.get()); | 64 ASSERT_TRUE(last_frame_.get()); |
| 65 ASSERT_TRUE(last_frame_->shape()); | 65 ASSERT_TRUE(last_frame_->shape()); |
| 66 EXPECT_TRUE( | 66 EXPECT_TRUE( |
| 67 FakeDesktopShapeTracker::CreateShape().Equals(*last_frame_->shape())); | 67 FakeDesktopShapeTracker::CreateShape().Equals(*last_frame_->shape())); |
| 68 } | 68 } |
| 69 | 69 |
| 70 } // namespace remoting | 70 } // namespace remoting |
| OLD | NEW |