Ensure that connection can be closed in response to a data message.

remoting/protocol/connection_unittest.cc

 // Copyright 2015 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 <utility>
 
 #include "base/bind.h"
 #include "base/macros.h"
 #include "base/memory/ptr_util.h"
 #include "base/message_loop/message_loop.h"
@@ skipping 330 matching lines @@
 
   Connect();
 
   std::unique_ptr<VideoStream> video_stream =
       host_connection_->StartVideoStream(
           base::WrapUnique(new TestScreenCapturer()));
 
   WaitFirstVideoFrame();
 }
 
+TEST_P(ConnectionTest, DestroyOnIncomingMessage) {
+  Connect();
+
+  KeyEvent event;
+  event.set_usb_keycode(3);
+  event.set_pressed(true);
+
+  base::RunLoop run_loop;
+
+  EXPECT_CALL(host_event_handler_,
+              OnInputEventReceived(host_connection_.get(), _));
+  EXPECT_CALL(host_input_stub_, InjectKeyEvent(EqualsKeyEvent(event)))
+      .WillOnce(/*DoAll(InvokeWithoutArgs(this, &ConnectionTest::DestroyHost),*/

[Jamie, 2016/07/23 00:03:48]

Remove the comment?

[Sergey Ulanov, 2016/07/23 06:30:03]

Duh, this line is not supposed to be commented, so the test wasn't testing what
it's supposed to test. Uncommented now and also fixed two other issues that were
affected by this bug. 

+                QuitRunLoop(&run_loop));
+
+  // Send key event from the client.
+  client_connection_->input_stub()->InjectKeyEvent(event);
+
+  run_loop.Run();
+}
+
 }  // namespace protocol
 }  // namespace remoting