chrome/browser/performance_monitor/metric_evaluator_helper_win_unittest.cc

// Copyright 2019 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 "chrome/browser/performance_monitor/metric_evaluator_helper_win.h"

#include "base/test/scoped_task_environment.h"
#include "base/win/scoped_com_initializer.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace performance_monitor {

class MetricEvaluatorsHelperWinTest : public testing::Test {
 public:
  MetricEvaluatorsHelperWinTest()
      : scoped_com_initializer_(base::win::ScopedCOMInitializer::kMTA) {}
  ~MetricEvaluatorsHelperWinTest() override = default;

 protected:
  base::win::ScopedCOMInitializer scoped_com_initializer_;
  base::test::ScopedTaskEnvironment scoped_task_environment_;

 private:
  DISALLOW_COPY_AND_ASSIGN(MetricEvaluatorsHelperWinTest);
};

TEST_F(MetricEvaluatorsHelperWinTest, GetFreeMemory) {
  MetricEvaluatorsHelperWin metric_evaluator_helper;
  auto value = metric_evaluator_helper.GetFreePhysicalMemoryMb();
  EXPECT_TRUE(value);
  EXPECT_GT(value.value(), 0);
}

TEST_F(MetricEvaluatorsHelperWinTest, DiskIdleTime) {
  MetricEvaluatorsHelperWin metric_evaluator_helper;
  while (!metric_evaluator_helper.wmi_refresher_initialized_for_testing())
    scoped_task_environment_.RunUntilIdle();

  // Measuring the disk idle time will always return base::nullopt for the first
  // sample on Windows.
  metric_evaluator_helper.GetDiskIdleTimePercent();
  auto refreshed_metrics = metric_evaluator_helper.GetDiskIdleTimePercent();
  EXPECT_TRUE(refreshed_metrics);
  EXPECT_LE(0.0, refreshed_metrics.value());
  EXPECT_GE(1.0, refreshed_metrics.value());
}

}  // namespace performance_monitor