OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "remoting/protocol/channel_multiplexer.h" | 5 #include "remoting/protocol/channel_multiplexer.h" |
6 | 6 |
7 #include "base/bind.h" | 7 #include "base/bind.h" |
8 #include "base/message_loop/message_loop.h" | 8 #include "base/message_loop/message_loop.h" |
9 #include "base/run_loop.h" | 9 #include "base/run_loop.h" |
10 #include "net/base/net_errors.h" | 10 #include "net/base/net_errors.h" |
(...skipping 54 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
65 } | 65 } |
66 | 66 |
67 void DeleteAfterSessionFail() { | 67 void DeleteAfterSessionFail() { |
68 host_mux_->CancelChannelCreation(kTestChannelName2); | 68 host_mux_->CancelChannelCreation(kTestChannelName2); |
69 DeleteAll(); | 69 DeleteAll(); |
70 } | 70 } |
71 | 71 |
72 protected: | 72 protected: |
73 virtual void SetUp() OVERRIDE { | 73 virtual void SetUp() OVERRIDE { |
74 // Create pair of multiplexers and connect them to each other. | 74 // Create pair of multiplexers and connect them to each other. |
75 host_mux_.reset(new ChannelMultiplexer(&host_session_, kMuxChannelName)); | 75 host_mux_.reset(new ChannelMultiplexer( |
76 client_mux_.reset(new ChannelMultiplexer(&client_session_, | 76 host_session_.GetTransportChannelFactory(), kMuxChannelName)); |
77 kMuxChannelName)); | 77 client_mux_.reset(new ChannelMultiplexer( |
| 78 client_session_.GetTransportChannelFactory(), kMuxChannelName)); |
78 } | 79 } |
79 | 80 |
80 // Connect sockets to each other. Must be called after we've created at least | 81 // Connect sockets to each other. Must be called after we've created at least |
81 // one channel with each multiplexer. | 82 // one channel with each multiplexer. |
82 void ConnectSockets() { | 83 void ConnectSockets() { |
83 FakeSocket* host_socket = | 84 FakeStreamSocket* host_socket = |
84 host_session_.GetStreamChannel(ChannelMultiplexer::kMuxChannelName); | 85 host_session_.fake_channel_factory().GetFakeChannel( |
85 FakeSocket* client_socket = | 86 ChannelMultiplexer::kMuxChannelName); |
86 client_session_.GetStreamChannel(ChannelMultiplexer::kMuxChannelName); | 87 FakeStreamSocket* client_socket = |
| 88 client_session_.fake_channel_factory().GetFakeChannel( |
| 89 ChannelMultiplexer::kMuxChannelName); |
87 host_socket->PairWith(client_socket); | 90 host_socket->PairWith(client_socket); |
88 | 91 |
89 // Make writes asynchronous in one direction. | 92 // Make writes asynchronous in one direction. |
90 host_socket->set_async_write(true); | 93 host_socket->set_async_write(true); |
91 } | 94 } |
92 | 95 |
93 void CreateChannel(const std::string& name, | 96 void CreateChannel(const std::string& name, |
94 scoped_ptr<net::StreamSocket>* host_socket, | 97 scoped_ptr<net::StreamSocket>* host_socket, |
95 scoped_ptr<net::StreamSocket>* client_socket) { | 98 scoped_ptr<net::StreamSocket>* client_socket) { |
96 int counter = 2; | 99 int counter = 2; |
(...skipping 138 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
235 ASSERT_NO_FATAL_FAILURE( | 238 ASSERT_NO_FATAL_FAILURE( |
236 CreateChannel(kTestChannelName, &host_socket1_, &client_socket1_)); | 239 CreateChannel(kTestChannelName, &host_socket1_, &client_socket1_)); |
237 | 240 |
238 scoped_ptr<net::StreamSocket> host_socket2_; | 241 scoped_ptr<net::StreamSocket> host_socket2_; |
239 scoped_ptr<net::StreamSocket> client_socket2_; | 242 scoped_ptr<net::StreamSocket> client_socket2_; |
240 ASSERT_NO_FATAL_FAILURE( | 243 ASSERT_NO_FATAL_FAILURE( |
241 CreateChannel(kTestChannelName2, &host_socket2_, &client_socket2_)); | 244 CreateChannel(kTestChannelName2, &host_socket2_, &client_socket2_)); |
242 | 245 |
243 ConnectSockets(); | 246 ConnectSockets(); |
244 | 247 |
245 host_session_.GetStreamChannel(kMuxChannelName)-> | 248 host_session_.fake_channel_factory().GetFakeChannel(kMuxChannelName)-> |
246 set_next_write_error(net::ERR_FAILED); | 249 set_next_write_error(net::ERR_FAILED); |
247 host_session_.GetStreamChannel(kMuxChannelName)-> | 250 host_session_.fake_channel_factory().GetFakeChannel(kMuxChannelName)-> |
248 set_async_write(false); | 251 set_async_write(false); |
249 | 252 |
250 scoped_refptr<net::IOBufferWithSize> buf = CreateTestBuffer(100); | 253 scoped_refptr<net::IOBufferWithSize> buf = CreateTestBuffer(100); |
251 | 254 |
252 MockSocketCallback cb1; | 255 MockSocketCallback cb1; |
253 MockSocketCallback cb2; | 256 MockSocketCallback cb2; |
254 | 257 |
255 EXPECT_CALL(cb1, OnDone(_)) | 258 EXPECT_CALL(cb1, OnDone(_)) |
256 .Times(0); | 259 .Times(0); |
257 EXPECT_CALL(cb2, OnDone(_)) | 260 EXPECT_CALL(cb2, OnDone(_)) |
(...skipping 15 matching lines...) Expand all Loading... |
273 | 276 |
274 TEST_F(ChannelMultiplexerTest, WriteFailAsync) { | 277 TEST_F(ChannelMultiplexerTest, WriteFailAsync) { |
275 ASSERT_NO_FATAL_FAILURE( | 278 ASSERT_NO_FATAL_FAILURE( |
276 CreateChannel(kTestChannelName, &host_socket1_, &client_socket1_)); | 279 CreateChannel(kTestChannelName, &host_socket1_, &client_socket1_)); |
277 | 280 |
278 ASSERT_NO_FATAL_FAILURE( | 281 ASSERT_NO_FATAL_FAILURE( |
279 CreateChannel(kTestChannelName2, &host_socket2_, &client_socket2_)); | 282 CreateChannel(kTestChannelName2, &host_socket2_, &client_socket2_)); |
280 | 283 |
281 ConnectSockets(); | 284 ConnectSockets(); |
282 | 285 |
283 host_session_.GetStreamChannel(kMuxChannelName)-> | 286 host_session_.fake_channel_factory().GetFakeChannel(kMuxChannelName)-> |
284 set_next_write_error(net::ERR_FAILED); | 287 set_next_write_error(net::ERR_FAILED); |
285 host_session_.GetStreamChannel(kMuxChannelName)-> | 288 host_session_.fake_channel_factory().GetFakeChannel(kMuxChannelName)-> |
286 set_async_write(true); | 289 set_async_write(true); |
287 | 290 |
288 scoped_refptr<net::IOBufferWithSize> buf = CreateTestBuffer(100); | 291 scoped_refptr<net::IOBufferWithSize> buf = CreateTestBuffer(100); |
289 | 292 |
290 MockSocketCallback cb1; | 293 MockSocketCallback cb1; |
291 MockSocketCallback cb2; | 294 MockSocketCallback cb2; |
292 EXPECT_CALL(cb1, OnDone(net::ERR_FAILED)); | 295 EXPECT_CALL(cb1, OnDone(net::ERR_FAILED)); |
293 EXPECT_CALL(cb2, OnDone(net::ERR_FAILED)); | 296 EXPECT_CALL(cb2, OnDone(net::ERR_FAILED)); |
294 | 297 |
295 EXPECT_EQ(net::ERR_IO_PENDING, | 298 EXPECT_EQ(net::ERR_IO_PENDING, |
(...skipping 11 matching lines...) Expand all Loading... |
307 } | 310 } |
308 | 311 |
309 TEST_F(ChannelMultiplexerTest, DeleteWhenFailed) { | 312 TEST_F(ChannelMultiplexerTest, DeleteWhenFailed) { |
310 ASSERT_NO_FATAL_FAILURE( | 313 ASSERT_NO_FATAL_FAILURE( |
311 CreateChannel(kTestChannelName, &host_socket1_, &client_socket1_)); | 314 CreateChannel(kTestChannelName, &host_socket1_, &client_socket1_)); |
312 ASSERT_NO_FATAL_FAILURE( | 315 ASSERT_NO_FATAL_FAILURE( |
313 CreateChannel(kTestChannelName2, &host_socket2_, &client_socket2_)); | 316 CreateChannel(kTestChannelName2, &host_socket2_, &client_socket2_)); |
314 | 317 |
315 ConnectSockets(); | 318 ConnectSockets(); |
316 | 319 |
317 host_session_.GetStreamChannel(kMuxChannelName)-> | 320 host_session_.fake_channel_factory().GetFakeChannel(kMuxChannelName)-> |
318 set_next_write_error(net::ERR_FAILED); | 321 set_next_write_error(net::ERR_FAILED); |
319 host_session_.GetStreamChannel(kMuxChannelName)-> | 322 host_session_.fake_channel_factory().GetFakeChannel(kMuxChannelName)-> |
320 set_async_write(true); | 323 set_async_write(true); |
321 | 324 |
322 scoped_refptr<net::IOBufferWithSize> buf = CreateTestBuffer(100); | 325 scoped_refptr<net::IOBufferWithSize> buf = CreateTestBuffer(100); |
323 | 326 |
324 MockSocketCallback cb1; | 327 MockSocketCallback cb1; |
325 MockSocketCallback cb2; | 328 MockSocketCallback cb2; |
326 | 329 |
327 EXPECT_CALL(cb1, OnDone(net::ERR_FAILED)) | 330 EXPECT_CALL(cb1, OnDone(net::ERR_FAILED)) |
328 .Times(AtMost(1)) | 331 .Times(AtMost(1)) |
329 .WillOnce(InvokeWithoutArgs(this, &ChannelMultiplexerTest::DeleteAll)); | 332 .WillOnce(InvokeWithoutArgs(this, &ChannelMultiplexerTest::DeleteAll)); |
(...skipping 12 matching lines...) Expand all Loading... |
342 base::Bind(&MockSocketCallback::OnDone, | 345 base::Bind(&MockSocketCallback::OnDone, |
343 base::Unretained(&cb2)))); | 346 base::Unretained(&cb2)))); |
344 | 347 |
345 base::RunLoop().RunUntilIdle(); | 348 base::RunLoop().RunUntilIdle(); |
346 | 349 |
347 // Check that the sockets were destroyed. | 350 // Check that the sockets were destroyed. |
348 EXPECT_FALSE(host_mux_.get()); | 351 EXPECT_FALSE(host_mux_.get()); |
349 } | 352 } |
350 | 353 |
351 TEST_F(ChannelMultiplexerTest, SessionFail) { | 354 TEST_F(ChannelMultiplexerTest, SessionFail) { |
352 host_session_.set_async_creation(true); | 355 host_session_.fake_channel_factory().set_asynchronous_create(true); |
353 host_session_.set_error(AUTHENTICATION_FAILED); | 356 host_session_.fake_channel_factory().set_fail_create(true); |
354 | 357 |
355 MockConnectCallback cb1; | 358 MockConnectCallback cb1; |
356 MockConnectCallback cb2; | 359 MockConnectCallback cb2; |
357 | 360 |
358 host_mux_->CreateChannel(kTestChannelName, base::Bind( | 361 host_mux_->CreateChannel(kTestChannelName, base::Bind( |
359 &MockConnectCallback::OnConnected, base::Unretained(&cb1))); | 362 &MockConnectCallback::OnConnected, base::Unretained(&cb1))); |
360 host_mux_->CreateChannel(kTestChannelName2, base::Bind( | 363 host_mux_->CreateChannel(kTestChannelName2, base::Bind( |
361 &MockConnectCallback::OnConnected, base::Unretained(&cb2))); | 364 &MockConnectCallback::OnConnected, base::Unretained(&cb2))); |
362 | 365 |
363 EXPECT_CALL(cb1, OnConnectedPtr(NULL)) | 366 EXPECT_CALL(cb1, OnConnectedPtr(NULL)) |
364 .Times(AtMost(1)) | 367 .Times(AtMost(1)) |
365 .WillOnce(InvokeWithoutArgs( | 368 .WillOnce(InvokeWithoutArgs( |
366 this, &ChannelMultiplexerTest::DeleteAfterSessionFail)); | 369 this, &ChannelMultiplexerTest::DeleteAfterSessionFail)); |
367 EXPECT_CALL(cb2, OnConnectedPtr(_)) | 370 EXPECT_CALL(cb2, OnConnectedPtr(_)) |
368 .Times(0); | 371 .Times(0); |
369 | 372 |
370 base::RunLoop().RunUntilIdle(); | 373 base::RunLoop().RunUntilIdle(); |
371 } | 374 } |
372 | 375 |
373 } // namespace protocol | 376 } // namespace protocol |
374 } // namespace remoting | 377 } // namespace remoting |
OLD | NEW |