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),
+                      QuitRunLoop(&run_loop)));
+
+  // Send key event from the client.
+  client_connection_->input_stub()->InjectKeyEvent(event);
+
+  run_loop.Run();
+}
+
 }  // namespace protocol
 }  // namespace remoting