[Chromoting] Use ProcessStatsSender in host process

remoting/host/client_session_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/client_session.h"
 
 #include <stdint.h>
 
 #include <algorithm>
 #include <string>
 #include <utility>
 #include <vector>
 
+#include "base/bind.h"
+#include "base/location.h"
 #include "base/memory/ptr_util.h"
 #include "base/message_loop/message_loop.h"
 #include "base/run_loop.h"
 #include "base/strings/string_split.h"
 #include "base/strings/string_util.h"
 #include "build/build_config.h"
 #include "remoting/base/auto_thread_task_runner.h"
 #include "remoting/base/constants.h"
 #include "remoting/codec/video_encoder_verbatim.h"
 #include "remoting/host/desktop_environment.h"
 #include "remoting/host/fake_desktop_environment.h"
 #include "remoting/host/fake_host_extension.h"
 #include "remoting/host/fake_mouse_cursor_monitor.h"
 #include "remoting/host/host_extension.h"
 #include "remoting/host/host_extension_session.h"
 #include "remoting/host/host_mock_objects.h"
+#include "remoting/protocol/errors.h"
 #include "remoting/protocol/fake_connection_to_client.h"
 #include "remoting/protocol/fake_desktop_capturer.h"
+#include "remoting/protocol/fake_process_stats_stub.h"
 #include "remoting/protocol/fake_session.h"
 #include "remoting/protocol/protocol_mock_objects.h"
 #include "remoting/protocol/test_event_matchers.h"
 #include "testing/gmock/include/gmock/gmock-matchers.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "third_party/webrtc/modules/desktop_capture/desktop_geometry.h"
 
 namespace remoting {
 
 using protocol::FakeSession;
@@ skipping 438 matching lines @@
   session->SetAttachment(0, std::move(configuration));
   CreateClientSession(std::move(session));
   ConnectClientSession();
   ASSERT_FALSE(desktop_environment_factory_->last_desktop_environment()
                    ->options()
                    .desktop_capture_options()
                    ->allow_directx_capturer());
 }
 #endif
 
+TEST_F(ClientSessionTest,
+       ProcessStatsSenderShouldBeCorrectlyStartedAndStopped) {
+  base::RunLoop run_loop;
+  protocol::FakeProcessStatsStub stats_stub;
+  auto session = base::MakeUnique<protocol::FakeSession>();
+  auto configuration = base::MakeUnique<buzz::XmlElement>(
+      buzz::QName(kChromotingXmlNamespace, "host-configuration"));
+  configuration->SetBodyText("Track-Process-Stats-Interval-Ms:1");
+  session->SetAttachment(0, std::move(configuration));
+  CreateClientSession(std::move(session));
+  connection_->set_stats_stub(&stats_stub);
+  stats_stub.set_expected_usage_count(5);
+  stats_stub.set_quit_closure(base::Bind([](ClientSession* client_session) {
+    client_session->OnConnectionClosed(protocol::MAX_SESSION_LENGTH);
+  },
+  base::Unretained(client_session_.get())));
+  ConnectClientSession();
+  task_runner_->PostDelayedTask(FROM_HERE,
+      base::Bind([](base::RunLoop* run_loop) {
+        run_loop->Quit();
+      },
+      base::Unretained(&run_loop)),
+      base::TimeDelta::FromMilliseconds(100));
+  run_loop.Run();
+  ASSERT_EQ(stats_stub.received().size(), 5U);
+}
+
 }  // namespace remoting