Move ConnectionToClient::EventHandler from ChromotingHost to ClientSession

remoting/protocol/connection_to_client_unittest.cc

 // Copyright (c) 2011 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.
 
 #include "base/bind.h"
 #include "base/message_loop.h"
 #include "base/message_loop_proxy.h"
 #include "remoting/base/base_mock_objects.h"
 #include "remoting/base/constants.h"
 #include "remoting/protocol/fake_session.h"
@@ skipping 13 matching lines @@
   ConnectionToClientTest() {
   }
 
  protected:
   virtual void SetUp() {
     session_ = new protocol::FakeSession();
     session_->set_message_loop(&message_loop_);
 
     // Allocate a ClientConnection object with the mock objects.
     viewer_ = new ConnectionToClient(
-        base::MessageLoopProxy::current(), &handler_);
+        base::MessageLoopProxy::current(), session_);
     viewer_->set_host_stub(&host_stub_);
     viewer_->set_input_stub(&input_stub_);
-    viewer_->Init(session_);
+    viewer_->SetEventHandler(&handler_);
     EXPECT_CALL(handler_, OnConnectionOpened(viewer_.get()));
     session_->state_change_callback().Run(
         protocol::Session::CONNECTED);
     session_->state_change_callback().Run(
         protocol::Session::CONNECTED_CHANNELS);
     message_loop_.RunAllPending();
   }
 
   MessageLoop message_loop_;
   MockConnectionToClientEventHandler handler_;
@@ skipping 56 matching lines @@
 // Test that we can close client connection more than once.
 TEST_F(ConnectionToClientTest, Close) {
   viewer_->Disconnect();
   message_loop_.RunAllPending();
   viewer_->Disconnect();
   message_loop_.RunAllPending();
 }
 
 }  // namespace protocol
 }  // namespace remoting