src/third_party/webrtc/modules/desktop_capture/screen_capturer_mock_objects.h

   1 /*
   2  *  Copyright (c) 2013 The WebRTC project authors. All Rights Reserved.
   3  *
   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
   6  *  tree. An additional intellectual property rights grant can be found
   7  *  in the file PATENTS.  All contributing project authors may
   8  *  be found in the AUTHORS file in the root of the source tree.
   9  */
  10
  11 #ifndef WEBRTC_MODULES_DESKTOP_CAPTURE_SCREEN_CAPTURER_MOCK_OBJECTS_H_
  12 #define WEBRTC_MODULES_DESKTOP_CAPTURE_SCREEN_CAPTURER_MOCK_OBJECTS_H_
  13
  14 #include "testing/gmock/include/gmock/gmock.h"
  15 #include "webrtc/modules/desktop_capture/screen_capturer.h"
  16
  17 namespace webrtc {
  18
  19 class MockScreenCapturer : public ScreenCapturer {
  20  public:
  21   MockScreenCapturer() {}
  22   virtual ~MockScreenCapturer() {}
  23
  24   MOCK_METHOD1(Start, void(Callback* callback));
  25   MOCK_METHOD1(Capture, void(const DesktopRegion& region));
  26   MOCK_METHOD1(GetScreenList, bool(ScreenList* screens));
  27   MOCK_METHOD1(SelectScreen, bool(ScreenId id));
  28
  29  private:
  30   DISALLOW_COPY_AND_ASSIGN(MockScreenCapturer);
  31 };
  32
  33 class MockScreenCapturerCallback : public ScreenCapturer::Callback {
  34  public:
  35   MockScreenCapturerCallback() {}
  36   virtual ~MockScreenCapturerCallback() {}
  37
  38   MOCK_METHOD1(CreateSharedMemory, SharedMemory*(size_t));
  39   MOCK_METHOD1(OnCaptureCompleted, void(DesktopFrame*));
  40
  41  private:
  42   DISALLOW_COPY_AND_ASSIGN(MockScreenCapturerCallback);
  43 };
  44
  45 }  // namespace webrtc
  46
  47 #endif  // WEBRTC_MODULES_DESKTOP_CAPTURE_SCREEN_CAPTURER_MOCK_OBJECTS_H_