1// Copyright 2014 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 "remoting/host/shaped_desktop_capturer.h"
6
7#include "remoting/host/desktop_shape_tracker.h"
8#include "remoting/host/fake_desktop_capturer.h"
9#include "testing/gtest/include/gtest/gtest.h"
10#include "third_party/webrtc/modules/desktop_capture/desktop_frame.h"
11#include "third_party/webrtc/modules/desktop_capture/desktop_geometry.h"
12#include "third_party/webrtc/modules/desktop_capture/desktop_region.h"
13
14namespace remoting {
15
16class FakeDesktopShapeTracker : public DesktopShapeTracker {
17 public:
18  FakeDesktopShapeTracker() {}
19  virtual ~FakeDesktopShapeTracker() {}
20
21  static webrtc::DesktopRegion CreateShape() {
22    webrtc::DesktopRegion result;
23    result.AddRect(webrtc::DesktopRect::MakeXYWH(0, 0, 5, 5));
24    result.AddRect(webrtc::DesktopRect::MakeXYWH(5, 5, 5, 5));
25    return result;
26  }
27
28  virtual void RefreshDesktopShape() OVERRIDE {
29    shape_ = CreateShape();
30  }
31
32  virtual const webrtc::DesktopRegion& desktop_shape() OVERRIDE {
33    // desktop_shape() can't be called before RefreshDesktopShape().
34    EXPECT_FALSE(shape_.is_empty());
35    return shape_;
36  }
37
38 private:
39  webrtc::DesktopRegion shape_;
40};
41
42class ShapedDesktopCapturerTest : public testing::Test,
43                                 public webrtc::DesktopCapturer::Callback {
44 public:
45  // webrtc::DesktopCapturer::Callback interface
46  virtual webrtc::SharedMemory* CreateSharedMemory(size_t size) OVERRIDE {
47    return NULL;
48  }
49
50  virtual void OnCaptureCompleted(webrtc::DesktopFrame* frame) OVERRIDE {
51    last_frame_.reset(frame);
52  }
53
54  scoped_ptr<webrtc::DesktopFrame> last_frame_;
55};
56
57// Verify that captured frame have shape.
58TEST_F(ShapedDesktopCapturerTest, Basic) {
59  ShapedDesktopCapturer capturer(
60      scoped_ptr<webrtc::DesktopCapturer>(new FakeDesktopCapturer()),
61      scoped_ptr<DesktopShapeTracker>(new FakeDesktopShapeTracker()));
62  capturer.Start(this);
63  capturer.Capture(webrtc::DesktopRegion());
64  ASSERT_TRUE(last_frame_.get());
65  ASSERT_TRUE(last_frame_->shape());
66  EXPECT_TRUE(
67      FakeDesktopShapeTracker::CreateShape().Equals(*last_frame_->shape()));
68}
69
70}  // namespace remoting
71