Crash the network process when a fatal daemon-to-network protocol error encountered.

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/memory/ref_counted.h"
 #include "base/single_thread_task_runner.h"
 #include "ipc/ipc_message.h"
 #include "ipc/ipc_message_macros.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,
   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 16 matching lines @@
   virtual scoped_ptr<DesktopSession> DoCreateDesktopSession(
       int terminal_id) OVERRIDE;
   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_METHOD1(DoCreateDesktopSessionPtr, DesktopSession*(int));
   MOCK_METHOD0(LaunchNetworkProcess, void());
-  MOCK_METHOD0(RestartNetworkProcess, void());
 
  private:
   DISALLOW_COPY_AND_ASSIGN(MockDaemonProcess);
 };
 
 FakeDesktopSession::FakeDesktopSession(DaemonProcess* daemon_process, int id)
     : DesktopSession(daemon_process, id) {
 }
 
 FakeDesktopSession::~FakeDesktopSession() {
@@ skipping 188 matching lines @@
       ChromotingNetworkHostMsg_DisconnectTerminal(id)));
   EXPECT_TRUE(desktop_sessions().empty());
 }
 
 // Tries to close an invalid terminal ID and expects the network process to be
 // restarted.
 TEST_F(DaemonProcessTest, InvalidDisconnectTerminal) {
   InSequence s;
   EXPECT_CALL(*daemon_process_, Sent(Message(kMessageConfiguration)));
   EXPECT_CALL(*daemon_process_, Received(Message(kMessageDisconnectTerminal)));
-  EXPECT_CALL(*daemon_process_, RestartNetworkProcess())
-      .WillRepeatedly(Invoke(this, &DaemonProcessTest::LaunchNetworkProcess));
+  EXPECT_CALL(*daemon_process_, Sent(Message(kMessageCrash)))
+      .WillOnce(InvokeWithoutArgs(this,
+                                  &DaemonProcessTest::LaunchNetworkProcess));
   EXPECT_CALL(*daemon_process_, Sent(Message(kMessageConfiguration)));
 
   StartDaemonProcess();
 
   int id = terminal_id_++;
 
   EXPECT_TRUE(daemon_process_->OnMessageReceived(
       ChromotingNetworkHostMsg_DisconnectTerminal(id)));
   EXPECT_TRUE(desktop_sessions().empty());
   EXPECT_EQ(0, terminal_id_);
 }
 
 // Tries to open an invalid terminal ID and expects the network process to be
 // restarted.
 TEST_F(DaemonProcessTest, InvalidConnectTerminal) {
   InSequence s;
   EXPECT_CALL(*daemon_process_, Sent(Message(kMessageConfiguration)));
   EXPECT_CALL(*daemon_process_, Received(Message(kMessageConnectTerminal)));
   EXPECT_CALL(*daemon_process_, Received(Message(kMessageConnectTerminal)));
-  EXPECT_CALL(*daemon_process_, RestartNetworkProcess())
-      .WillRepeatedly(Invoke(this, &DaemonProcessTest::LaunchNetworkProcess));
+  EXPECT_CALL(*daemon_process_, Sent(Message(kMessageCrash)))
+      .WillOnce(InvokeWithoutArgs(this,
+                                  &DaemonProcessTest::LaunchNetworkProcess));
   EXPECT_CALL(*daemon_process_, Sent(Message(kMessageConfiguration)));
 
   StartDaemonProcess();
 
   int id = terminal_id_++;
 
   EXPECT_TRUE(daemon_process_->OnMessageReceived(
       ChromotingNetworkHostMsg_ConnectTerminal(id)));
   EXPECT_EQ(1u, desktop_sessions().size());
   EXPECT_EQ(id, desktop_sessions().front()->id());
 
   EXPECT_TRUE(daemon_process_->OnMessageReceived(
       ChromotingNetworkHostMsg_ConnectTerminal(id)));
   EXPECT_TRUE(desktop_sessions().empty());
   EXPECT_EQ(0, terminal_id_);
 }
 
 }  // namespace remoting