Index: remoting/protocol/connection_to_client_unittest.cc |
diff --git a/remoting/protocol/connection_to_client_unittest.cc b/remoting/protocol/connection_to_client_unittest.cc |
index 7a21096c79c92f93426d716b7d4ea49f67df88a0..bb73a8f3d1cb36ffee2a2f72329c9af936cd83ca 100644 |
--- a/remoting/protocol/connection_to_client_unittest.cc |
+++ b/remoting/protocol/connection_to_client_unittest.cc |
@@ -27,8 +27,9 @@ class ConnectionToClientTest : public testing::Test { |
session_->set_message_loop(&message_loop_); |
// Allocate a ClientConnection object with the mock objects. |
- viewer_ = new ConnectionToClient(&message_loop_, &handler_, &input_stub_); |
+ viewer_ = new ConnectionToClient(&message_loop_, &handler_); |
viewer_->set_host_stub(&host_stub_); |
+ viewer_->set_input_stub(&input_stub_); |
viewer_->Init(session_); |
EXPECT_CALL(handler_, OnConnectionOpened(viewer_.get())); |
session_->state_change_callback()->Run( |