[Chromoting] Retrieve process resource usage (ProcessStats) and its tests

remoting/host/host_stats_unittest.cc

+// Copyright 2017 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/host_stats.h"
+
+#include <stdint.h>
+
+#include <vector>
+
+#include "base/bind.h"
+#include "base/location.h"
+#include "base/message_loop/message_loop.h"
+#include "base/run_loop.h"
+#include "base/time/time.h"
+#include "remoting/base/process_resource_usage.h"
+#include "remoting/host/host_stats_stub.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace remoting {
+namespace host {
+
+namespace {
+
+class MockHostStatsStub : public HostStatsStub {
+ public:
+  MockHostStatsStub() = default;
+  ~MockHostStatsStub() = default;
+
+  void OnHostStats(const ProcessResourceUsage& usage) override {
+    received_.push_back(usage);
+  }
+
+  const std::vector<ProcessResourceUsage>& received() const {
+    return received_;
+  }
+
+  void Clear() {
+    received_.clear();
+  }
+
+ private:
+  std::vector<ProcessResourceUsage> received_;
+};
+
+}  // namespace
+
+TEST(HostStatsTest, ReportUsage) {
+  base::MessageLoop message_loop;
+  base::RunLoop run_loop;
+  MockHostStatsStub stub;
+  std::unique_ptr<HostStats> stats;
+  message_loop.task_runner()->PostTask(
+      FROM_HERE,
+      base::Bind([](std::unique_ptr<HostStats>* stats,
+                    MockHostStatsStub* stub) -> void {
+        stats->reset(new HostStats(stub, base::TimeDelta::FromSeconds(1)));
+      },
+      base::Unretained(&stats),
+      base::Unretained(&stub)));
+  message_loop.task_runner()->PostDelayedTask(
+      FROM_HERE,
+      base::Bind([](std::unique_ptr<HostStats>* stats) -> void {
+        stats->reset();
+      },
+      base::Unretained(&stats)),
+      base::TimeDelta::FromSecondsD(2.1));
+  message_loop.task_runner()->PostDelayedTask(
+      FROM_HERE,
+      run_loop.QuitClosure(),
+      base::TimeDelta::FromSecondsD(2.2));
+  run_loop.Run();
+
+  ASSERT_EQ(2U, stub.received().size());
+}
+
+TEST(HostStatsTest, MergeUsage) {
+  base::MessageLoop message_loop;
+  base::RunLoop run_loop;
+  MockHostStatsStub stub;
+  std::unique_ptr<HostStats> stats;
+  message_loop.task_runner()->PostTask(
+      FROM_HERE,
+      base::Bind([](std::unique_ptr<HostStats>* stats,
+                    MockHostStatsStub* stub) -> void {
+        stats->reset(new HostStats(stub, base::TimeDelta::FromSeconds(1)));
+      },
+      base::Unretained(&stats),
+      base::Unretained(&stub)));
+  message_loop.task_runner()->PostDelayedTask(
+      FROM_HERE,
+      base::Bind([](std::unique_ptr<HostStats>* stats) -> void {
+        ProcessResourceUsage fake_usage;
+        fake_usage.processor_usage = 10000000;
+        fake_usage.working_set_size = SIZE_MAX >> 1;
+        fake_usage.pagefile_size = SIZE_MAX >> 1;
+        static_cast<HostStatsStub*>(stats->get())->OnHostStats(fake_usage);
+      },
+      base::Unretained(&stats)),
+      base::TimeDelta::FromSecondsD(1.1));
+  message_loop.task_runner()->PostDelayedTask(
+      FROM_HERE,
+      base::Bind([](std::unique_ptr<HostStats>* stats) -> void {
+        stats->reset();
+      },
+      base::Unretained(&stats)),
+      base::TimeDelta::FromSecondsD(3.1));
+  message_loop.task_runner()->PostDelayedTask(
+      FROM_HERE,
+      run_loop.QuitClosure(),
+      base::TimeDelta::FromSecondsD(3.2));
+  run_loop.Run();
+
+  ASSERT_EQ(3U, stub.received().size());
+  ASSERT_LT(stub.received()[0].processor_usage, 10000000);
+  ASSERT_LT(stub.received()[0].working_set_size, SIZE_MAX >> 1);
+  ASSERT_LT(stub.received()[0].pagefile_size, SIZE_MAX >> 1);
+  ASSERT_GE(stub.received()[1].processor_usage, 10000000);
+  ASSERT_GE(stub.received()[1].working_set_size, SIZE_MAX >> 1);
+  ASSERT_GE(stub.received()[1].pagefile_size, SIZE_MAX >> 1);
+  ASSERT_GE(stub.received()[2].processor_usage, 10000000);
+  ASSERT_GE(stub.received()[2].working_set_size, SIZE_MAX >> 1);
+  ASSERT_GE(stub.received()[2].pagefile_size, SIZE_MAX >> 1);
+}
+
+}  // namespace host
+}  // namespace remoting