Index: remoting/host/host_mock_objects.h |
diff --git a/remoting/host/host_mock_objects.h b/remoting/host/host_mock_objects.h |
index b5f1db9b810ff79a231adecc0d8b232710253531..fa630ce04fe3b9c30df8279207d75de5c4dfc9c2 100644 |
--- a/remoting/host/host_mock_objects.h |
+++ b/remoting/host/host_mock_objects.h |
@@ -5,11 +5,14 @@ |
#ifndef REMOTING_HOST_HOST_MOCK_OBJECTS_H_ |
#define REMOTING_HOST_HOST_MOCK_OBJECTS_H_ |
+#include <string> |
+ |
#include "net/base/ip_endpoint.h" |
#include "remoting/host/chromoting_host_context.h" |
#include "remoting/host/client_session.h" |
#include "remoting/host/client_session_control.h" |
#include "remoting/host/desktop_environment.h" |
+#include "remoting/host/gnubby_auth_handler.h" |
#include "remoting/host/host_status_observer.h" |
#include "remoting/host/input_injector.h" |
#include "remoting/host/screen_controls.h" |
@@ -126,6 +129,22 @@ class MockHostStatusObserver : public HostStatusObserver { |
MOCK_METHOD0(OnShutdown, void()); |
}; |
+class MockGnubbyAuthHandler : public GnubbyAuthHandler { |
+ public: |
+ MockGnubbyAuthHandler(protocol::ClientStub* client_stub); |
+ virtual ~MockGnubbyAuthHandler(); |
+ |
+ MOCK_METHOD1(DeliverClientMessage, void(const std::string message)); |
+ MOCK_CONST_METHOD2(DeliverHostControlMessage, |
+ void(const std::string control_type, |
+ const std::string data)); |
+ MOCK_CONST_METHOD2(DeliverHostDataMessage, |
+ void(int connection_id, const std::string data)); |
+ |
+ private: |
+ DISALLOW_COPY_AND_ASSIGN(MockGnubbyAuthHandler); |
+}; |
+ |
} // namespace remoting |
#endif // REMOTING_HOST_HOST_MOCK_OBJECTS_H_ |