Index: remoting/protocol/channel_multiplexer_unittest.cc |
diff --git a/remoting/protocol/channel_multiplexer_unittest.cc b/remoting/protocol/channel_multiplexer_unittest.cc |
index 190764ebc6edc6c5028d9d45ccdc7bc31f4d4037..d8394bf03e4775417835c1685aefb3c4a02b82d8 100644 |
--- a/remoting/protocol/channel_multiplexer_unittest.cc |
+++ b/remoting/protocol/channel_multiplexer_unittest.cc |
@@ -73,20 +73,24 @@ class ChannelMultiplexerTest : public testing::Test { |
void SetUp() override { |
// Create pair of multiplexers and connect them to each other. |
host_mux_.reset(new ChannelMultiplexer( |
- host_session_.GetTransportChannelFactory(), kMuxChannelName)); |
+ host_session_.GetTransportSession()->GetStreamChannelFactory(), |
+ kMuxChannelName)); |
client_mux_.reset(new ChannelMultiplexer( |
- client_session_.GetTransportChannelFactory(), kMuxChannelName)); |
+ client_session_.GetTransportSession()->GetStreamChannelFactory(), |
+ kMuxChannelName)); |
} |
// Connect sockets to each other. Must be called after we've created at least |
// one channel with each multiplexer. |
void ConnectSockets() { |
FakeStreamSocket* host_socket = |
- host_session_.fake_channel_factory().GetFakeChannel( |
- ChannelMultiplexer::kMuxChannelName); |
+ host_session_.GetTransportSession() |
+ ->GetStreamChannelFactory() |
+ ->GetFakeChannel(ChannelMultiplexer::kMuxChannelName); |
FakeStreamSocket* client_socket = |
- client_session_.fake_channel_factory().GetFakeChannel( |
- ChannelMultiplexer::kMuxChannelName); |
+ client_session_.GetTransportSession() |
+ ->GetStreamChannelFactory() |
+ ->GetFakeChannel(ChannelMultiplexer::kMuxChannelName); |
host_socket->PairWith(client_socket); |
// Make writes asynchronous in one direction. |
@@ -245,10 +249,11 @@ TEST_F(ChannelMultiplexerTest, WriteFailSync) { |
ConnectSockets(); |
- host_session_.fake_channel_factory().GetFakeChannel(kMuxChannelName)-> |
- set_next_write_error(net::ERR_FAILED); |
- host_session_.fake_channel_factory().GetFakeChannel(kMuxChannelName)-> |
- set_async_write(false); |
+ FakeStreamSocket* socket = host_session_.GetTransportSession() |
+ ->GetStreamChannelFactory() |
+ ->GetFakeChannel(kMuxChannelName); |
+ socket->set_next_write_error(net::ERR_FAILED); |
+ socket->set_async_write(false); |
scoped_refptr<net::IOBufferWithSize> buf = CreateTestBuffer(100); |
@@ -280,10 +285,11 @@ TEST_F(ChannelMultiplexerTest, WriteFailAsync) { |
ConnectSockets(); |
- host_session_.fake_channel_factory().GetFakeChannel(kMuxChannelName)-> |
- set_next_write_error(net::ERR_FAILED); |
- host_session_.fake_channel_factory().GetFakeChannel(kMuxChannelName)-> |
- set_async_write(true); |
+ FakeStreamSocket* socket = host_session_.GetTransportSession() |
+ ->GetStreamChannelFactory() |
+ ->GetFakeChannel(kMuxChannelName); |
+ socket->set_next_write_error(net::ERR_FAILED); |
+ socket->set_async_write(true); |
scoped_refptr<net::IOBufferWithSize> buf = CreateTestBuffer(100); |
@@ -314,10 +320,11 @@ TEST_F(ChannelMultiplexerTest, DeleteWhenFailed) { |
ConnectSockets(); |
- host_session_.fake_channel_factory().GetFakeChannel(kMuxChannelName)-> |
- set_next_write_error(net::ERR_FAILED); |
- host_session_.fake_channel_factory().GetFakeChannel(kMuxChannelName)-> |
- set_async_write(true); |
+ FakeStreamSocket* socket = host_session_.GetTransportSession() |
+ ->GetStreamChannelFactory() |
+ ->GetFakeChannel(kMuxChannelName); |
+ socket->set_next_write_error(net::ERR_FAILED); |
+ socket->set_async_write(true); |
scoped_refptr<net::IOBufferWithSize> buf = CreateTestBuffer(100); |
@@ -349,8 +356,10 @@ TEST_F(ChannelMultiplexerTest, DeleteWhenFailed) { |
} |
TEST_F(ChannelMultiplexerTest, SessionFail) { |
- host_session_.fake_channel_factory().set_asynchronous_create(true); |
- host_session_.fake_channel_factory().set_fail_create(true); |
+ host_session_.GetTransportSession()->GetStreamChannelFactory() |
+ ->set_asynchronous_create(true); |
+ host_session_.GetTransportSession()->GetStreamChannelFactory() |
+ ->set_fail_create(true); |
MockConnectCallback cb1; |
MockConnectCallback cb2; |