Add gnubby authentication to remoting host

remoting/host/host_mock_objects.h

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #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/gnubby_connection.h"
+#include "remoting/host/gnubby_connection_factory.h"
 #include "remoting/host/host_status_observer.h"
 #include "remoting/host/input_injector.h"
 #include "remoting/host/screen_controls.h"
 #include "remoting/host/screen_resolution.h"
 #include "remoting/proto/control.pb.h"
 #include "testing/gmock/include/gmock/gmock.h"
 
 namespace base {
 class SingleThreadTaskRunner;
 }  // namespace base
@@ skipping 96 matching lines @@
   MOCK_METHOD1(OnClientConnected, void(const std::string& jid));
   MOCK_METHOD1(OnClientDisconnected, void(const std::string& jid));
   MOCK_METHOD3(OnClientRouteChange,
                void(const std::string& jid,
                     const std::string& channel_name,
                     const protocol::TransportRoute& route));
   MOCK_METHOD1(OnStart, void(const std::string& xmpp_login));
   MOCK_METHOD0(OnShutdown, void());
 };
 
+class MockGnubbyAuthHandler : public GnubbyAuthHandler {
+ public:
+  MockGnubbyAuthHandler(
+      scoped_refptr<base::SingleThreadTaskRunner> network_task_runner,
+      GnubbyConnectionFactory* gnubby_connection_factory,
+      protocol::ClientStub* client_stub);
+  virtual ~MockGnubbyAuthHandler();
+
+  MOCK_METHOD1(DeliverClientMessage, void(const std::string message));
+  MOCK_METHOD1(ConnectionClosed, void(int connection_id));
+  MOCK_METHOD2(ConnectionError, void(int connection_id, int result));
+  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);
+};
+
+class MockGnubbyConnection : public GnubbyConnection {
+ public:
+  MockGnubbyConnection(
+      scoped_refptr<base::SingleThreadTaskRunner> network_task_runner,
+      GnubbyAuthHandler* auth_handler,
+      int connection_id,
+      net::StreamSocket* socket);
+  virtual ~MockGnubbyConnection();
+
+  MOCK_METHOD0(Read, void());
+  MOCK_METHOD1(Write, void(const std::string& data));
+
+ private:
+  DISALLOW_COPY_AND_ASSIGN(MockGnubbyConnection);
+};
+
+class MockGnubbyConnectionFactory : public GnubbyConnectionFactory {
+ public:
+  MockGnubbyConnectionFactory();
+  virtual ~MockGnubbyConnectionFactory();
+
+  MOCK_METHOD4(Create,
+               GnubbyConnection*(scoped_refptr<base::SingleThreadTaskRunner>
+                                     network_task_runner,
+                                 GnubbyAuthHandler* auth_handler,
+                                 int connection_id,
+                                 net::StreamSocket* socket));
+
+ private:
+  DISALLOW_COPY_AND_ASSIGN(MockGnubbyConnectionFactory);
+};
+
 }  // namespace remoting
 
 #endif  // REMOTING_HOST_HOST_MOCK_OBJECTS_H_