1 files changed, 24 insertions, 0 deletions
diff --git a/remoting/protocol/mock_objects.h b/remoting/protocol/mock_objects.h
index 5acb131..b10c139 100644
--- a/remoting/protocol/mock_objects.h
+++ b/remoting/protocol/mock_objects.h
@@ -7,6 +7,8 @@
 
 #include "remoting/proto/internal.pb.h"
 #include "remoting/protocol/connection_to_client.h"
+#include "remoting/protocol/host_stub.h"
+#include "remoting/protocol/input_stub.h"
 #include "testing/gmock/include/gmock/gmock.h"
 
 namespace remoting {
@@ -44,6 +46,28 @@ class MockConnectionToClientEventHandler :
   DISALLOW_COPY_AND_ASSIGN(MockConnectionToClientEventHandler);
 };
 
+class MockInputStub : public InputStub {
+ public:
+  MockInputStub() {}
+
+  MOCK_METHOD2(InjectKeyEvent, void(const KeyEvent* event, Task* done));
+  MOCK_METHOD2(InjectMouseEvent, void(const MouseEvent* event, Task* done));
+
+ private:
+  DISALLOW_COPY_AND_ASSIGN(MockInputStub);
+};
+
+class MockHostStub : public HostStub {
+ public:
+  MockHostStub() {}
+
+  MOCK_METHOD2(SuggestResolution, void(const SuggestResolutionRequest* msg,
+                                       Task* done));
+
+ private:
+  DISALLOW_COPY_AND_ASSIGN(MockHostStub);
+};
+
 }  // namespace protocol
 }  // namespace remoting