The worker process launcher can now ask the worker to crash.

remoting/host/daemon_process_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 "base/bind.h"
 #include "base/bind_helpers.h"
+#include "base/location.h"
 #include "base/memory/ref_counted.h"
 #include "base/process.h"
 #include "ipc/ipc_message.h"
 #include "ipc/ipc_message_macros.h"
 #include "ipc/ipc_platform_file.h"
 #include "remoting/base/auto_thread_task_runner.h"
 #include "remoting/host/chromoting_messages.h"
 #include "remoting/host/daemon_process.h"
 #include "remoting/host/desktop_session.h"
 #include "testing/gmock_mutant.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 using testing::_;
 using testing::AnyNumber;
 using testing::InSequence;
 
 namespace remoting {
 
 namespace {
 
 enum Messages {
-  kMessageCrash = ChromotingDaemonNetworkMsg_Crash::ID,
+  kMessageCrash = ChromotingDaemonMsg_Crash::ID,
   kMessageConfiguration = ChromotingDaemonNetworkMsg_Configuration::ID,
   kMessageConnectTerminal = ChromotingNetworkHostMsg_ConnectTerminal::ID,
   kMessageDisconnectTerminal = ChromotingNetworkHostMsg_DisconnectTerminal::ID,
   kMessageTerminalDisconnected =
       ChromotingDaemonNetworkMsg_TerminalDisconnected::ID
 };
 
 // Provides a public constructor allowing the test to create instances of
 // DesktopSession directly.
 class FakeDesktopSession : public DesktopSession {
@@ skipping 21 matching lines @@
   virtual bool OnMessageReceived(const IPC::Message& message) OVERRIDE;
   virtual void SendToNetwork(IPC::Message* message) OVERRIDE;
 
   MOCK_METHOD1(Received, void(const IPC::Message&));
   MOCK_METHOD1(Sent, void(const IPC::Message&));
 
   MOCK_METHOD3(OnDesktopSessionAgentAttached,
                bool(int, base::ProcessHandle, IPC::PlatformFileForTransit));
 
   MOCK_METHOD1(DoCreateDesktopSessionPtr, DesktopSession*(int));
+  MOCK_METHOD1(DoCrashNetworkProcess, void(const tracked_objects::Location&));
   MOCK_METHOD0(LaunchNetworkProcess, void());
 
  private:
   DISALLOW_COPY_AND_ASSIGN(MockDaemonProcess);
 };
 
 FakeDesktopSession::FakeDesktopSession(DaemonProcess* daemon_process, int id)
     : DesktopSession(daemon_process, id) {
 }
 
@@ skipping 36 matching lines @@
 class DaemonProcessTest : public testing::Test {
  public:
   DaemonProcessTest();
   virtual ~DaemonProcessTest();
 
   virtual void SetUp() OVERRIDE;
   virtual void TearDown() OVERRIDE;
 
   // DaemonProcess mocks
   DesktopSession* DoCreateDesktopSession(int terminal_id);
+  void DoCrashNetworkProcess(const tracked_objects::Location& location);
   void LaunchNetworkProcess();
 
   // Deletes |daemon_process_|.
   void DeleteDaemonProcess();
 
   // Quits |message_loop_|.
   void QuitMessageLoop();
 
   void StartDaemonProcess();
 
@@ skipping 24 matching lines @@
                  base::Unretained(this)));
   daemon_process_.reset(
       new MockDaemonProcess(task_runner, task_runner,
                             base::Bind(&DaemonProcessTest::DeleteDaemonProcess,
                                        base::Unretained(this))));
 
   // Set up daemon process mocks.
   EXPECT_CALL(*daemon_process_, DoCreateDesktopSessionPtr(_))
       .Times(AnyNumber())
       .WillRepeatedly(Invoke(this, &DaemonProcessTest::DoCreateDesktopSession));
+  EXPECT_CALL(*daemon_process_, DoCrashNetworkProcess(_))
+      .Times(AnyNumber())
+      .WillRepeatedly(Invoke(this, &DaemonProcessTest::DoCrashNetworkProcess));
   EXPECT_CALL(*daemon_process_, LaunchNetworkProcess())
       .Times(AnyNumber())
       .WillRepeatedly(Invoke(this, &DaemonProcessTest::LaunchNetworkProcess));
 }
 
 void DaemonProcessTest::TearDown() {
   daemon_process_->Stop();
   message_loop_.Run();
 }
 
 DesktopSession* DaemonProcessTest::DoCreateDesktopSession(int terminal_id) {
   return new FakeDesktopSession(daemon_process_.get(), terminal_id);
 }
 
+void DaemonProcessTest::DoCrashNetworkProcess(
+    const tracked_objects::Location& location) {
+  daemon_process_->SendToNetwork(
+      new ChromotingDaemonMsg_Crash(location.function_name(),
+                                    location.file_name(),
+                                    location.line_number()));
+}
+
 void DaemonProcessTest::LaunchNetworkProcess() {
   terminal_id_ = 0;
   daemon_process_->OnChannelConnected(0);
 }
 
 void DaemonProcessTest::DeleteDaemonProcess() {
   daemon_process_.reset();
 }
 
 void DaemonProcessTest::QuitMessageLoop() {
@@ skipping 124 matching lines @@
   EXPECT_EQ(1u, desktop_sessions().size());
   EXPECT_EQ(id, desktop_sessions().front()->id());
 
   EXPECT_TRUE(daemon_process_->OnMessageReceived(
       ChromotingNetworkHostMsg_ConnectTerminal(id, params, false)));
   EXPECT_TRUE(desktop_sessions().empty());
   EXPECT_EQ(0, terminal_id_);
 }
 
 }  // namespace remoting