Index: remoting/host/host_mock_objects.h |
diff --git a/remoting/host/host_mock_objects.h b/remoting/host/host_mock_objects.h |
index 0dd4020321aa0e4d3184727df90b73ad3d0da672..c325af48ac1edcda53d28f8bf06d7337a090afef 100644 |
--- a/remoting/host/host_mock_objects.h |
+++ b/remoting/host/host_mock_objects.h |
@@ -6,6 +6,7 @@ |
#define REMOTING_HOST_HOST_MOCK_OBJECTS_H_ |
#include "net/base/ip_endpoint.h" |
+#include "remoting/base/capabilities.h" |
#include "remoting/host/chromoting_host_context.h" |
#include "remoting/host/client_session.h" |
#include "remoting/host/client_session_control.h" |
@@ -34,6 +35,8 @@ class MockDesktopEnvironment : public DesktopEnvironment { |
MOCK_METHOD0(CreateInputInjectorPtr, InputInjector*()); |
MOCK_METHOD0(CreateScreenControlsPtr, ScreenControls*()); |
MOCK_METHOD0(CreateVideoCapturerPtr, media::ScreenCapturer*()); |
+ MOCK_CONST_METHOD0(GetCapabilities, Capabilities()); |
+ MOCK_METHOD1(SetCapabilities, void(const Capabilities&)); |
// DesktopEnvironment implementation. |
virtual scoped_ptr<AudioCapturer> CreateAudioCapturer() OVERRIDE; |