Add TransportSession interface to prepare for WebRTC-based transport.

remoting/protocol/connection_to_client_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/connection_to_client.h"
 
 #include "base/bind.h"
 #include "base/message_loop/message_loop.h"
 #include "base/run_loop.h"
 #include "remoting/base/constants.h"
@@ skipping 56 matching lines @@
 };
 
 TEST_F(ConnectionToClientTest, SendUpdateStream) {
   scoped_ptr<VideoPacket> packet(new VideoPacket());
   viewer_->video_stub()->ProcessVideoPacket(packet.Pass(), base::Closure());
 
   base::RunLoop().RunUntilIdle();
 
   // Verify that something has been written.
   // TODO(sergeyu): Verify that the correct data has been written.
-  ASSERT_TRUE(
-      session_->fake_channel_factory().GetFakeChannel(kVideoChannelName));
-  EXPECT_FALSE(session_->fake_channel_factory()
-                   .GetFakeChannel(kVideoChannelName)->written_data().empty());
+  FakeStreamSocket* channel = session_->GetTransportSession()
+                                  ->GetStreamChannelFactory()
+                                  ->GetFakeChannel(kVideoChannelName);
+  ASSERT_TRUE(channel);
+  EXPECT_FALSE(channel->written_data().empty());
 
   // And then close the connection to ConnectionToClient.
   viewer_->Disconnect();
 
   base::RunLoop().RunUntilIdle();
 }
 
 TEST_F(ConnectionToClientTest, NoWriteAfterDisconnect) {
   scoped_ptr<VideoPacket> packet(new VideoPacket());
   viewer_->video_stub()->ProcessVideoPacket(packet.Pass(), base::Closure());
@@ skipping 13 matching lines @@
   base::RunLoop().RunUntilIdle();
 
   EXPECT_CALL(handler_, OnConnectionClosed(viewer_.get(), SESSION_REJECTED));
   session_->set_error(SESSION_REJECTED);
   session_->event_handler()->OnSessionStateChange(Session::FAILED);
   base::RunLoop().RunUntilIdle();
 }
 
 }  // namespace protocol
 }  // namespace remoting