| Index: remoting/protocol/channel_multiplexer_unittest.cc
|
| diff --git a/remoting/protocol/channel_multiplexer_unittest.cc b/remoting/protocol/channel_multiplexer_unittest.cc
|
| index d8394bf03e4775417835c1685aefb3c4a02b82d8..190764ebc6edc6c5028d9d45ccdc7bc31f4d4037 100644
|
| --- a/remoting/protocol/channel_multiplexer_unittest.cc
|
| +++ b/remoting/protocol/channel_multiplexer_unittest.cc
|
| @@ -73,24 +73,20 @@
|
| void SetUp() override {
|
| // Create pair of multiplexers and connect them to each other.
|
| host_mux_.reset(new ChannelMultiplexer(
|
| - host_session_.GetTransportSession()->GetStreamChannelFactory(),
|
| - kMuxChannelName));
|
| + host_session_.GetTransportChannelFactory(), kMuxChannelName));
|
| client_mux_.reset(new ChannelMultiplexer(
|
| - client_session_.GetTransportSession()->GetStreamChannelFactory(),
|
| - kMuxChannelName));
|
| + client_session_.GetTransportChannelFactory(), 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_.GetTransportSession()
|
| - ->GetStreamChannelFactory()
|
| - ->GetFakeChannel(ChannelMultiplexer::kMuxChannelName);
|
| + host_session_.fake_channel_factory().GetFakeChannel(
|
| + ChannelMultiplexer::kMuxChannelName);
|
| FakeStreamSocket* client_socket =
|
| - client_session_.GetTransportSession()
|
| - ->GetStreamChannelFactory()
|
| - ->GetFakeChannel(ChannelMultiplexer::kMuxChannelName);
|
| + client_session_.fake_channel_factory().GetFakeChannel(
|
| + ChannelMultiplexer::kMuxChannelName);
|
| host_socket->PairWith(client_socket);
|
|
|
| // Make writes asynchronous in one direction.
|
| @@ -249,11 +245,10 @@
|
|
|
| ConnectSockets();
|
|
|
| - FakeStreamSocket* socket = host_session_.GetTransportSession()
|
| - ->GetStreamChannelFactory()
|
| - ->GetFakeChannel(kMuxChannelName);
|
| - socket->set_next_write_error(net::ERR_FAILED);
|
| - socket->set_async_write(false);
|
| + 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);
|
|
|
| scoped_refptr<net::IOBufferWithSize> buf = CreateTestBuffer(100);
|
|
|
| @@ -285,11 +280,10 @@
|
|
|
| ConnectSockets();
|
|
|
| - FakeStreamSocket* socket = host_session_.GetTransportSession()
|
| - ->GetStreamChannelFactory()
|
| - ->GetFakeChannel(kMuxChannelName);
|
| - socket->set_next_write_error(net::ERR_FAILED);
|
| - socket->set_async_write(true);
|
| + 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);
|
|
|
| scoped_refptr<net::IOBufferWithSize> buf = CreateTestBuffer(100);
|
|
|
| @@ -320,11 +314,10 @@
|
|
|
| ConnectSockets();
|
|
|
| - FakeStreamSocket* socket = host_session_.GetTransportSession()
|
| - ->GetStreamChannelFactory()
|
| - ->GetFakeChannel(kMuxChannelName);
|
| - socket->set_next_write_error(net::ERR_FAILED);
|
| - socket->set_async_write(true);
|
| + 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);
|
|
|
| scoped_refptr<net::IOBufferWithSize> buf = CreateTestBuffer(100);
|
|
|
| @@ -356,10 +349,8 @@
|
| }
|
|
|
| TEST_F(ChannelMultiplexerTest, SessionFail) {
|
| - host_session_.GetTransportSession()->GetStreamChannelFactory()
|
| - ->set_asynchronous_create(true);
|
| - host_session_.GetTransportSession()->GetStreamChannelFactory()
|
| - ->set_fail_create(true);
|
| + host_session_.fake_channel_factory().set_asynchronous_create(true);
|
| + host_session_.fake_channel_factory().set_fail_create(true);
|
|
|
| MockConnectCallback cb1;
|
| MockConnectCallback cb2;
|
|
|