Close all writers before JingleSession is destroyed.

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/message_loop.h"
 #include "remoting/base/base_mock_objects.h"
 #include "remoting/protocol/fake_session.h"
 #include "remoting/protocol/connection_to_client.h"
 #include "remoting/protocol/protocol_mock_objects.h"
 #include "testing/gmock/include/gmock/gmock.h"
@@ skipping 37 matching lines @@
  private:
   DISALLOW_COPY_AND_ASSIGN(ConnectionToClientTest);
 };
 
 TEST_F(ConnectionToClientTest, SendUpdateStream) {
   // Then send the actual data.
   VideoPacket* packet = new VideoPacket();
   viewer_->video_stub()->ProcessVideoPacket(
       packet, new DeleteTask<VideoPacket>(packet));
 
+  message_loop_.RunAllPending();
+
   // And then close the connection to ConnectionToClient.
   viewer_->Disconnect();
 
   message_loop_.RunAllPending();
 
   // Verify that something has been written.
   // TODO(sergeyu): Verify that the correct data has been written.
   EXPECT_GT(session_->video_channel()->written_data().size(), 0u);
 }
 
+TEST_F(ConnectionToClientTest, NoWriteAfterDisconnect) {
+  // Then send the actual data.
+  VideoPacket* packet = new VideoPacket();
+  viewer_->video_stub()->ProcessVideoPacket(
+      packet, new DeleteTask<VideoPacket>(packet));
+
+  // And then close the connection to ConnectionToClient.
+  viewer_->Disconnect();
+
+  message_loop_.RunAllPending();
+
+  // Nothing should be written because connection has been closed.
+  EXPECT_EQ(session_->video_channel()->written_data().size(), 0u);
+}
+
 TEST_F(ConnectionToClientTest, StateChange) {
   EXPECT_CALL(handler_, OnConnectionClosed(viewer_.get()));
   session_->state_change_callback()->Run(protocol::Session::CLOSED);
   message_loop_.RunAllPending();
 
   EXPECT_CALL(handler_, OnConnectionFailed(viewer_.get()));
   session_->state_change_callback()->Run(protocol::Session::FAILED);
   message_loop_.RunAllPending();
 }
 
 // Test that we can close client connection more than once.
 TEST_F(ConnectionToClientTest, Close) {
   viewer_->Disconnect();
   message_loop_.RunAllPending();
   EXPECT_TRUE(session_->is_closed());
 
   viewer_->Disconnect();
   message_loop_.RunAllPending();
 }
 
 }  // namespace protocol
 }  // namespace remoting