Add support for multiplexed channels in remoting::protocol::Session interface.

remoting/protocol/channel_multiplexer_unittest.cc

 // Copyright (c) 2012 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 "remoting/protocol/channel_multiplexer.h"
 
 #include "base/bind.h"
 #include "base/message_loop.h"
 #include "net/base/net_errors.h"
 #include "net/socket/socket.h"
@@ skipping 38 matching lines @@
     host_mux_.reset();
     client_mux_.reset();
   }
 
  protected:
   virtual void SetUp() OVERRIDE {
     // Create pair of multiplexers and connect them to each other.
     host_mux_.reset(new ChannelMultiplexer(&host_session_, kMuxChannelName));
     client_mux_.reset(new ChannelMultiplexer(&client_session_,
                                              kMuxChannelName));
+  }
+
+  // Connect sockets to each other. Must be called after we've created at least
+  // one channel with each multiplexer.
+  void ConnectSockets() {
     FakeSocket* host_socket =
         host_session_.GetStreamChannel(ChannelMultiplexer::kMuxChannelName);
     FakeSocket* client_socket =
         client_session_.GetStreamChannel(ChannelMultiplexer::kMuxChannelName);
     host_socket->PairWith(client_socket);
 
     // Make writes asynchronous in one direction.
     host_socket->set_async_write(true);
   }
 
@@ skipping 47 matching lines @@
   scoped_ptr<net::StreamSocket> host_socket2_;
   scoped_ptr<net::StreamSocket> client_socket2_;
 };
 
 
 TEST_F(ChannelMultiplexerTest, OneChannel) {
   scoped_ptr<net::StreamSocket> host_socket;
   scoped_ptr<net::StreamSocket> client_socket;
   ASSERT_NO_FATAL_FAILURE(CreateChannel("test", &host_socket, &client_socket));
 
+  ConnectSockets();
+
   StreamConnectionTester tester(host_socket.get(), client_socket.get(),
                                 kMessageSize, kMessages);
   tester.Start();
   message_loop_.Run();
   tester.CheckResults();
 }
 
 TEST_F(ChannelMultiplexerTest, TwoChannels) {
   scoped_ptr<net::StreamSocket> host_socket1_;
   scoped_ptr<net::StreamSocket> client_socket1_;
   ASSERT_NO_FATAL_FAILURE(
       CreateChannel("test", &host_socket1_, &client_socket1_));
 
   scoped_ptr<net::StreamSocket> host_socket2_;
   scoped_ptr<net::StreamSocket> client_socket2_;
   ASSERT_NO_FATAL_FAILURE(
       CreateChannel("ch2", &host_socket2_, &client_socket2_));
 
+  ConnectSockets();
+
   StreamConnectionTester tester1(host_socket1_.get(), client_socket1_.get(),
                                 kMessageSize, kMessages);
   StreamConnectionTester tester2(host_socket2_.get(), client_socket2_.get(),
                                  kMessageSize, kMessages);
   tester1.Start();
   tester2.Start();
   while (!tester1.done() || !tester2.done()) {
     message_loop_.Run();
   }
   tester1.CheckResults();
@@ skipping 15 matching lines @@
   scoped_ptr<net::StreamSocket> host_socket3;
   scoped_ptr<net::StreamSocket> client_socket3;
   ASSERT_NO_FATAL_FAILURE(
       CreateChannel("test3", &host_socket3, &client_socket3));
 
   scoped_ptr<net::StreamSocket> host_socket4;
   scoped_ptr<net::StreamSocket> client_socket4;
   ASSERT_NO_FATAL_FAILURE(
       CreateChannel("ch4", &host_socket4, &client_socket4));
 
+  ConnectSockets();
+
   StreamConnectionTester tester1(host_socket1_.get(), client_socket1_.get(),
                                 kMessageSize, kMessages);
   StreamConnectionTester tester2(host_socket2_.get(), client_socket2_.get(),
                                  kMessageSize, kMessages);
   StreamConnectionTester tester3(client_socket3.get(), host_socket3.get(),
                                  kMessageSize, kMessages);
   StreamConnectionTester tester4(client_socket4.get(), host_socket4.get(),
                                  kMessageSize, kMessages);
   tester1.Start();
   tester2.Start();
@@ skipping 13 matching lines @@
   scoped_ptr<net::StreamSocket> host_socket1_;
   scoped_ptr<net::StreamSocket> client_socket1_;
   ASSERT_NO_FATAL_FAILURE(
       CreateChannel("test", &host_socket1_, &client_socket1_));
 
   scoped_ptr<net::StreamSocket> host_socket2_;
   scoped_ptr<net::StreamSocket> client_socket2_;
   ASSERT_NO_FATAL_FAILURE(
       CreateChannel("ch2", &host_socket2_, &client_socket2_));
 
+  ConnectSockets();
+
   host_session_.GetStreamChannel(kMuxChannelName)->
       set_next_write_error(net::ERR_FAILED);
   host_session_.GetStreamChannel(kMuxChannelName)->
       set_async_write(false);
 
   scoped_refptr<net::IOBufferWithSize> buf = CreateTestBuffer(100);
 
   MockSocketCallback cb1;
   MockSocketCallback cb2;
 
@@ skipping 10 matching lines @@
   message_loop_.RunAllPending();
 }
 
 TEST_F(ChannelMultiplexerTest, AsyncFail) {
   ASSERT_NO_FATAL_FAILURE(
       CreateChannel("test", &host_socket1_, &client_socket1_));
 
   ASSERT_NO_FATAL_FAILURE(
       CreateChannel("ch2", &host_socket2_, &client_socket2_));
 
+  ConnectSockets();
+
   host_session_.GetStreamChannel(kMuxChannelName)->
       set_next_write_error(net::ERR_FAILED);
   host_session_.GetStreamChannel(kMuxChannelName)->
       set_async_write(true);
 
   scoped_refptr<net::IOBufferWithSize> buf = CreateTestBuffer(100);
 
   MockSocketCallback cb1;
   MockSocketCallback cb2;
   EXPECT_CALL(cb1, OnDone(net::ERR_FAILED));
   EXPECT_CALL(cb2, OnDone(net::ERR_FAILED));
 
   EXPECT_EQ(net::ERR_IO_PENDING,
             host_socket1_->Write(buf, buf->size(), base::Bind(
                 &MockSocketCallback::OnDone, base::Unretained(&cb1))));
   EXPECT_EQ(net::ERR_IO_PENDING,
             host_socket2_->Write(buf, buf->size(), base::Bind(
                 &MockSocketCallback::OnDone, base::Unretained(&cb2))));
 
   message_loop_.RunAllPending();
 }
 
 TEST_F(ChannelMultiplexerTest, DeleteWhenFailed) {
   ASSERT_NO_FATAL_FAILURE(
       CreateChannel("test", &host_socket1_, &client_socket1_));
   ASSERT_NO_FATAL_FAILURE(
       CreateChannel("ch2", &host_socket2_, &client_socket2_));
 
+  ConnectSockets();
+
   host_session_.GetStreamChannel(kMuxChannelName)->
       set_next_write_error(net::ERR_FAILED);
   host_session_.GetStreamChannel(kMuxChannelName)->
       set_async_write(true);
 
   scoped_refptr<net::IOBufferWithSize> buf = CreateTestBuffer(100);
 
   MockSocketCallback cb1;
   MockSocketCallback cb2;
 
@@ skipping 12 matching lines @@
                 &MockSocketCallback::OnDone, base::Unretained(&cb2))));
 
   message_loop_.RunAllPending();
 
   // Check that the sockets were destroyed.
   EXPECT_FALSE(host_mux_.get());
 }
 
 }  // namespace protocol
 }  // namespace remoting