Fix ConnectionToClient to connect stubs only after authentication.

remoting/protocol/protocol_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_PROTOCOL_PROTOCOL_MOCK_OBJECTS_H_
 #define REMOTING_PROTOCOL_PROTOCOL_MOCK_OBJECTS_H_
 
 #include <map>
 #include <string>
 
@@ skipping 24 matching lines @@
   MockConnectionToClient(Session* session,
                          HostStub* host_stub);
   virtual ~MockConnectionToClient();
 
   MOCK_METHOD1(Init, void(Session* session));
   MOCK_METHOD0(video_stub, VideoStub*());
   MOCK_METHOD0(client_stub, ClientStub*());
   MOCK_METHOD0(session, Session*());
   MOCK_METHOD0(Disconnect, void());
 
+  void set_clipboard_stub(ClipboardStub* clipboard_stub) override {
+    clipboard_stub_ = clipboard_stub;
+  }
+  void set_host_stub(HostStub* host_stub) override { host_stub_ = host_stub; }
+  void set_input_stub(InputStub* input_stub) override {
+    input_stub_ = input_stub;
+  }
+
+  ClipboardStub* clipboard_stub() { return clipboard_stub_; }
+  HostStub* host_stub() { return host_stub_; }
+  InputStub* input_stub() { return input_stub_; }
+
  private:
+  ClipboardStub* clipboard_stub_;
+  HostStub* host_stub_;
+  InputStub* input_stub_;
+
   DISALLOW_COPY_AND_ASSIGN(MockConnectionToClient);
 };
 
 class MockConnectionToClientEventHandler :
       public ConnectionToClient::EventHandler {
  public:
   MockConnectionToClientEventHandler();
   virtual ~MockConnectionToClientEventHandler();
 
   MOCK_METHOD1(OnConnectionAuthenticating,
@@ skipping 198 matching lines @@
   // Runs tasks synchronously instead of posting them to |task_runner|.
   void PostTask(const scoped_refptr<base::SingleThreadTaskRunner>& task_runner,
                 const tracked_objects::Location& from_here,
                 const base::Closure& task) override;
 };
 
 }  // namespace protocol
 }  // namespace remoting
 
 #endif  // REMOTING_PROTOCOL_PROTOCOL_MOCK_OBJECTS_H_