[Chromoting] Use ProcessStatsSender in DaemonProcess (daemon process)

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 "remoting/host/daemon_process.h"
 
 #include <stdint.h>
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
@@ skipping 22 matching lines @@
 namespace remoting {
 
 namespace {
 
 enum Messages {
   kMessageCrash = ChromotingDaemonMsg_Crash::ID,
   kMessageConfiguration = ChromotingDaemonNetworkMsg_Configuration::ID,
   kMessageConnectTerminal = ChromotingNetworkHostMsg_ConnectTerminal::ID,
   kMessageDisconnectTerminal = ChromotingNetworkHostMsg_DisconnectTerminal::ID,
   kMessageTerminalDisconnected =
-      ChromotingDaemonNetworkMsg_TerminalDisconnected::ID
+      ChromotingDaemonNetworkMsg_TerminalDisconnected::ID,
+  kMessageReportProcessStats = ChromotingAnyToNetworkMsg_ReportProcessStats::ID,
 };
 
 // Provides a public constructor allowing the test to create instances of
 // DesktopSession directly.
 class FakeDesktopSession : public DesktopSession {
  public:
   FakeDesktopSession(DaemonProcess* daemon_process, int id);
   ~FakeDesktopSession() override;
 
   void SetScreenResolution(const ScreenResolution& resolution) override {}
@@ skipping 286 matching lines @@
       ChromotingNetworkHostMsg_ConnectTerminal(id, resolution, false)));
   EXPECT_EQ(1u, desktop_sessions().size());
   EXPECT_EQ(id, desktop_sessions().front()->id());
 
   EXPECT_TRUE(daemon_process_->OnMessageReceived(
       ChromotingNetworkHostMsg_ConnectTerminal(id, resolution, false)));
   EXPECT_TRUE(desktop_sessions().empty());
   EXPECT_EQ(0, terminal_id_);
 }
 
+TEST_F(DaemonProcessTest, StartProcessStatsReport) {
+  EXPECT_CALL(*daemon_process_, Sent(Message(kMessageReportProcessStats)));
+  daemon_process_->OnMessageReceived(
+      ChromotingNetworkToAnyMsg_StartProcessStatsReport(
+          base::TimeDelta::FromMilliseconds(1)));
+  base::RunLoop run_loop;
+  ON_CALL(*daemon_process_, Sent(Message(kMessageReportProcessStats)))
+      .WillByDefault(testing::Invoke(
+          [&run_loop](const IPC::Message& message) {
+            run_loop.Quit();
+          }));
+  run_loop.Run();
+}
+
+TEST_F(DaemonProcessTest, StartProcessStatsReportWithDifferentDelta) {
+  EXPECT_CALL(*daemon_process_, Sent(Message(kMessageReportProcessStats)))
+      .Times(AnyNumber());
+  int received = 0;
+  daemon_process_->OnMessageReceived(
+      ChromotingNetworkToAnyMsg_StartProcessStatsReport(
+          base::TimeDelta::FromHours(1)));
+  daemon_process_->OnMessageReceived(
+      ChromotingNetworkToAnyMsg_StartProcessStatsReport(
+          base::TimeDelta::FromMilliseconds(1)));
+  base::RunLoop run_loop;
+  ON_CALL(*daemon_process_, Sent(Message(kMessageReportProcessStats)))
+      .WillByDefault(testing::Invoke(
+          [&run_loop, &received](const IPC::Message& message) {
+            received++;
+            if (received == 5) {
+              run_loop.Quit();
+            }
+          }));
+  run_loop.Run();
+}
+
+TEST_F(DaemonProcessTest, StopProcessStatsReportWhenTheWorkerProcessDied) {
+  daemon_process_->OnMessageReceived(
+      ChromotingNetworkToAnyMsg_StartProcessStatsReport(
+          base::TimeDelta::FromMilliseconds(1)));
+  base::RunLoop run_loop;
+  ON_CALL(*daemon_process_, Sent(Message(kMessageReportProcessStats)))
+      .WillByDefault(testing::Invoke(
+          [](const IPC::Message& message) {
+            ASSERT_TRUE(false);
+          }));
+  static_cast<WorkerProcessIpcDelegate*>(daemon_process_.get())
+      ->OnWorkerProcessStopped();
+  message_loop_.task_runner()->PostDelayedTask(
+      FROM_HERE, run_loop.QuitClosure(), base::TimeDelta::FromMilliseconds(10));
+  run_loop.Run();
+}
+
 }  // namespace remoting