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 "base/basictypes.h" | 5 #include "base/basictypes.h" |
6 #include "base/bind.h" | 6 #include "base/bind.h" |
7 #include "base/memory/ref_counted.h" | 7 #include "base/memory/ref_counted.h" |
8 #include "base/message_loop.h" | 8 #include "base/message_loop.h" |
9 #include "base/win/scoped_handle.h" | 9 #include "base/win/scoped_handle.h" |
10 #include "base/win/scoped_process_information.h" | 10 #include "base/win/scoped_process_information.h" |
(...skipping 44 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
55 }; | 55 }; |
56 | 56 |
57 class MockIpcDelegate : public WorkerProcessIpcDelegate { | 57 class MockIpcDelegate : public WorkerProcessIpcDelegate { |
58 public: | 58 public: |
59 MockIpcDelegate() {} | 59 MockIpcDelegate() {} |
60 virtual ~MockIpcDelegate() {} | 60 virtual ~MockIpcDelegate() {} |
61 | 61 |
62 // WorkerProcessIpcDelegate interface. | 62 // WorkerProcessIpcDelegate interface. |
63 MOCK_METHOD1(OnChannelConnected, void(int32)); | 63 MOCK_METHOD1(OnChannelConnected, void(int32)); |
64 MOCK_METHOD1(OnMessageReceived, bool(const IPC::Message&)); | 64 MOCK_METHOD1(OnMessageReceived, bool(const IPC::Message&)); |
65 MOCK_METHOD0(OnPermanentError, void()); | 65 MOCK_METHOD1(OnPermanentError, void(int)); |
66 | 66 |
67 private: | 67 private: |
68 DISALLOW_COPY_AND_ASSIGN(MockIpcDelegate); | 68 DISALLOW_COPY_AND_ASSIGN(MockIpcDelegate); |
69 }; | 69 }; |
70 | 70 |
71 class MockWorkerListener : public IPC::Listener { | 71 class MockWorkerListener : public IPC::Listener { |
72 public: | 72 public: |
73 MockWorkerListener() {} | 73 MockWorkerListener() {} |
74 virtual ~MockWorkerListener() {} | 74 virtual ~MockWorkerListener() {} |
75 | 75 |
(...skipping 304 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
380 event_handler_->OnProcessLaunched(copy.Pass()); | 380 event_handler_->OnProcessLaunched(copy.Pass()); |
381 } | 381 } |
382 | 382 |
383 TEST_F(WorkerProcessLauncherTest, Start) { | 383 TEST_F(WorkerProcessLauncherTest, Start) { |
384 EXPECT_CALL(*launcher_delegate_, LaunchProcess(_)) | 384 EXPECT_CALL(*launcher_delegate_, LaunchProcess(_)) |
385 .Times(1) | 385 .Times(1) |
386 .WillRepeatedly(Invoke(this, &WorkerProcessLauncherTest::LaunchProcess)); | 386 .WillRepeatedly(Invoke(this, &WorkerProcessLauncherTest::LaunchProcess)); |
387 | 387 |
388 EXPECT_CALL(server_listener_, OnChannelConnected(_)) | 388 EXPECT_CALL(server_listener_, OnChannelConnected(_)) |
389 .Times(0); | 389 .Times(0); |
390 EXPECT_CALL(server_listener_, OnPermanentError()) | 390 EXPECT_CALL(server_listener_, OnPermanentError(_)) |
391 .Times(0); | 391 .Times(0); |
392 | 392 |
393 StartWorker(); | 393 StartWorker(); |
394 StopWorker(); | 394 StopWorker(); |
395 message_loop_.Run(); | 395 message_loop_.Run(); |
396 } | 396 } |
397 | 397 |
398 // Starts and connects to the worker process. Expect OnChannelConnected to be | 398 // Starts and connects to the worker process. Expect OnChannelConnected to be |
399 // called. | 399 // called. |
400 TEST_F(WorkerProcessLauncherTest, StartAndConnect) { | 400 TEST_F(WorkerProcessLauncherTest, StartAndConnect) { |
401 EXPECT_CALL(*launcher_delegate_, LaunchProcess(_)) | 401 EXPECT_CALL(*launcher_delegate_, LaunchProcess(_)) |
402 .Times(1) | 402 .Times(1) |
403 .WillRepeatedly(Invoke( | 403 .WillRepeatedly(Invoke( |
404 this, &WorkerProcessLauncherTest::LaunchProcessAndConnect)); | 404 this, &WorkerProcessLauncherTest::LaunchProcessAndConnect)); |
405 | 405 |
406 EXPECT_CALL(server_listener_, OnChannelConnected(_)) | 406 EXPECT_CALL(server_listener_, OnChannelConnected(_)) |
407 .Times(1) | 407 .Times(1) |
408 .WillOnce(InvokeWithoutArgs(this, | 408 .WillOnce(InvokeWithoutArgs(this, |
409 &WorkerProcessLauncherTest::StopWorker)); | 409 &WorkerProcessLauncherTest::StopWorker)); |
410 EXPECT_CALL(server_listener_, OnPermanentError()) | 410 EXPECT_CALL(server_listener_, OnPermanentError(_)) |
411 .Times(0); | 411 .Times(0); |
412 | 412 |
413 StartWorker(); | 413 StartWorker(); |
414 message_loop_.Run(); | 414 message_loop_.Run(); |
415 } | 415 } |
416 | 416 |
417 // Kills the worker process after the 1st connect and expects it to be | 417 // Kills the worker process after the 1st connect and expects it to be |
418 // restarted. | 418 // restarted. |
419 TEST_F(WorkerProcessLauncherTest, Restart) { | 419 TEST_F(WorkerProcessLauncherTest, Restart) { |
420 EXPECT_CALL(*launcher_delegate_, LaunchProcess(_)) | 420 EXPECT_CALL(*launcher_delegate_, LaunchProcess(_)) |
421 .Times(2) | 421 .Times(2) |
422 .WillRepeatedly(Invoke( | 422 .WillRepeatedly(Invoke( |
423 this, &WorkerProcessLauncherTest::LaunchProcessAndConnect)); | 423 this, &WorkerProcessLauncherTest::LaunchProcessAndConnect)); |
424 Expectation first_connect = | 424 Expectation first_connect = |
425 EXPECT_CALL(server_listener_, OnChannelConnected(_)) | 425 EXPECT_CALL(server_listener_, OnChannelConnected(_)) |
426 .Times(2) | 426 .Times(2) |
427 .WillOnce(InvokeWithoutArgs(CreateFunctor( | 427 .WillOnce(InvokeWithoutArgs(CreateFunctor( |
428 this, &WorkerProcessLauncherTest::TerminateWorker, | 428 this, &WorkerProcessLauncherTest::TerminateWorker, |
429 CONTROL_C_EXIT))) | 429 CONTROL_C_EXIT))) |
430 .WillOnce(InvokeWithoutArgs(this, | 430 .WillOnce(InvokeWithoutArgs(this, |
431 &WorkerProcessLauncherTest::StopWorker)); | 431 &WorkerProcessLauncherTest::StopWorker)); |
432 | 432 |
433 EXPECT_CALL(server_listener_, OnPermanentError()) | 433 EXPECT_CALL(server_listener_, OnPermanentError(_)) |
434 .Times(0); | 434 .Times(0); |
435 | 435 |
436 StartWorker(); | 436 StartWorker(); |
437 message_loop_.Run(); | 437 message_loop_.Run(); |
438 } | 438 } |
439 | 439 |
440 // Drops the IPC channel to the worker process after the 1st connect and expects | 440 // Drops the IPC channel to the worker process after the 1st connect and expects |
441 // the worker process to be restarted. | 441 // the worker process to be restarted. |
442 TEST_F(WorkerProcessLauncherTest, DropIpcChannel) { | 442 TEST_F(WorkerProcessLauncherTest, DropIpcChannel) { |
443 EXPECT_CALL(*launcher_delegate_, LaunchProcess(_)) | 443 EXPECT_CALL(*launcher_delegate_, LaunchProcess(_)) |
444 .Times(2) | 444 .Times(2) |
445 .WillRepeatedly(Invoke( | 445 .WillRepeatedly(Invoke( |
446 this, &WorkerProcessLauncherTest::LaunchProcessAndConnect)); | 446 this, &WorkerProcessLauncherTest::LaunchProcessAndConnect)); |
447 | 447 |
448 Expectation first_connect = | 448 Expectation first_connect = |
449 EXPECT_CALL(server_listener_, OnChannelConnected(_)) | 449 EXPECT_CALL(server_listener_, OnChannelConnected(_)) |
450 .Times(2) | 450 .Times(2) |
451 .WillOnce(InvokeWithoutArgs( | 451 .WillOnce(InvokeWithoutArgs( |
452 this, &WorkerProcessLauncherTest::DisconnectClient)) | 452 this, &WorkerProcessLauncherTest::DisconnectClient)) |
453 .WillOnce(InvokeWithoutArgs( | 453 .WillOnce(InvokeWithoutArgs( |
454 this, &WorkerProcessLauncherTest::StopWorker)); | 454 this, &WorkerProcessLauncherTest::StopWorker)); |
455 | 455 |
456 EXPECT_CALL(server_listener_, OnPermanentError()) | 456 EXPECT_CALL(server_listener_, OnPermanentError(_)) |
457 .Times(0); | 457 .Times(0); |
458 | 458 |
459 StartWorker(); | 459 StartWorker(); |
460 message_loop_.Run(); | 460 message_loop_.Run(); |
461 } | 461 } |
462 | 462 |
463 // Returns a permanent error exit code and expects OnPermanentError() to be | 463 // Returns a permanent error exit code and expects OnPermanentError() to be |
464 // invoked. | 464 // invoked. |
465 TEST_F(WorkerProcessLauncherTest, PermanentError) { | 465 TEST_F(WorkerProcessLauncherTest, PermanentError) { |
466 EXPECT_CALL(*launcher_delegate_, LaunchProcess(_)) | 466 EXPECT_CALL(*launcher_delegate_, LaunchProcess(_)) |
467 .Times(1) | 467 .Times(1) |
468 .WillRepeatedly(Invoke( | 468 .WillRepeatedly(Invoke( |
469 this, &WorkerProcessLauncherTest::LaunchProcessAndConnect)); | 469 this, &WorkerProcessLauncherTest::LaunchProcessAndConnect)); |
470 | 470 |
471 EXPECT_CALL(server_listener_, OnChannelConnected(_)) | 471 EXPECT_CALL(server_listener_, OnChannelConnected(_)) |
472 .Times(1) | 472 .Times(1) |
473 .WillOnce(InvokeWithoutArgs(CreateFunctor( | 473 .WillOnce(InvokeWithoutArgs(CreateFunctor( |
474 this, &WorkerProcessLauncherTest::TerminateWorker, | 474 this, &WorkerProcessLauncherTest::TerminateWorker, |
475 kMinPermanentErrorExitCode))); | 475 kMinPermanentErrorExitCode))); |
476 EXPECT_CALL(server_listener_, OnPermanentError()) | 476 EXPECT_CALL(server_listener_, OnPermanentError(_)) |
477 .Times(1) | 477 .Times(1) |
478 .WillOnce(InvokeWithoutArgs(this, | 478 .WillOnce(InvokeWithoutArgs(this, |
479 &WorkerProcessLauncherTest::StopWorker)); | 479 &WorkerProcessLauncherTest::StopWorker)); |
480 | 480 |
481 StartWorker(); | 481 StartWorker(); |
482 message_loop_.Run(); | 482 message_loop_.Run(); |
483 } | 483 } |
484 | 484 |
485 // Requests the worker to crash and expects it to honor the request. | 485 // Requests the worker to crash and expects it to honor the request. |
486 TEST_F(WorkerProcessLauncherTest, Crash) { | 486 TEST_F(WorkerProcessLauncherTest, Crash) { |
(...skipping 38 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
525 EXPECT_CALL(client_listener_, OnCrash(_, _, _)) | 525 EXPECT_CALL(client_listener_, OnCrash(_, _, _)) |
526 .Times(1) | 526 .Times(1) |
527 .WillOnce(InvokeWithoutArgs( | 527 .WillOnce(InvokeWithoutArgs( |
528 this, &WorkerProcessLauncherTest::SendFakeMessageToLauncher)); | 528 this, &WorkerProcessLauncherTest::SendFakeMessageToLauncher)); |
529 | 529 |
530 StartWorker(); | 530 StartWorker(); |
531 message_loop_.Run(); | 531 message_loop_.Run(); |
532 } | 532 } |
533 | 533 |
534 } // namespace remoting | 534 } // namespace remoting |
OLD | NEW |