Add parser for new cpu freq file

chrome/browser/chromeos/power/cpu_data_collector_unittest.cc

Index: chrome/browser/chromeos/power/cpu_data_collector_unittest.cc
diff --git a/chrome/browser/chromeos/power/cpu_data_collector_unittest.cc b/chrome/browser/chromeos/power/cpu_data_collector_unittest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..4dd834d5e39a06ee57ce6b73b4d3a2862f9bead7
--- /dev/null
+++ b/chrome/browser/chromeos/power/cpu_data_collector_unittest.cc
@@ -0,0 +1,139 @@
+// 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 <string>
+#include <vector>
+
+#include "base/files/file.h"
+#include "base/files/file_util.h"
+#include "base/files/scoped_temp_dir.h"
+#include "chrome/browser/chromeos/power/cpu_data_collector.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+using base::TimeDelta;
+
+namespace chromeos {
+namespace {
+
+// The suffix path of fake cpu frequency file for cpu0.
+constexpr char kTimeInStateSuffixPathCpu0[] = "cpu0/time_in_state";
+
+// The suffix path of fake cpu frequency file for cpu1.
+constexpr char kTimeInStateSuffixPathCpu1[] = "cpu1/time_in_state";
+
+// The suffix path of fake cpu frequency file for all cpus.
+constexpr char kAllTimeInStateSuffixPath[] = "all_time_in_state";
+
+// The string content of the fake cpu frequency file for cpu0.
+constexpr char kTimeInStateContentCpu0[] =
+    "20000 30000000\n"
+    "60000 90000\n"
+    "100000 50000\n";
+
+// The string content of the fake cpu frequency file for cpu1.
+constexpr char kTimeInStateContentCpu1[] =
+    "20000 31000000\n"
+    "60000 91000\n"
+    "100000 51000\n";
+
+// The string content of the fake cpu frequency file for all cpus.
+constexpr char kAllTimeInStateContent[] =
+    "freq\t\tcpu0\t\tcpu1\t\t\n"
+    "20000\t\t30000000\t\t31000000\t\t\n"
+    "60000\t\t90000\t\t91000\t\t\n"
+    "100000\t\t50000\t\t51000\t\t\n";
+
+}  // namespace
+
+class CpuDataCollectorTest : public testing::Test {
+ public:
+  CpuDataCollectorTest()
+      : kExpectedCpuFreqStateNames({"20", "60", "100"}),
+        kExpectedTimeInStateCpu0({TimeDelta::FromMilliseconds(300000000),
+                                  TimeDelta::FromMilliseconds(900000),
+                                  TimeDelta::FromMilliseconds(500000)}),
+        kExpectedTimeInStateCpu1({TimeDelta::FromMilliseconds(310000000),
+                                  TimeDelta::FromMilliseconds(910000),
+                                  TimeDelta::FromMilliseconds(510000)}) {}
+
+  void SetUp() override {

[xiyuan, 2017/05/04 21:25:35]

nit: // testing::Test:

[weidongg, 2017/05/04 21:38:50]

Done.

+    ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
+
+    time_in_state_path_cpu0_ =
+        temp_dir_.GetPath().AppendASCII(kTimeInStateSuffixPathCpu0);
+    time_in_state_path_cpu1_ =
+        temp_dir_.GetPath().AppendASCII(kTimeInStateSuffixPathCpu1);
+    all_time_in_state_path_ =
+        temp_dir_.GetPath().AppendASCII(kAllTimeInStateSuffixPath);
+
+    ASSERT_TRUE(base::CreateTemporaryFile(&time_in_state_path_cpu0_));
+    ASSERT_TRUE(base::CreateTemporaryFile(&time_in_state_path_cpu1_));
+    ASSERT_TRUE(base::CreateTemporaryFile(&all_time_in_state_path_));
+    ASSERT_TRUE(base::WriteFile(
+                    time_in_state_path_cpu0_, kTimeInStateContentCpu0,
+                    static_cast<int>(strlen(kTimeInStateContentCpu0))) != -1);
+    ASSERT_TRUE(base::WriteFile(
+                    time_in_state_path_cpu1_, kTimeInStateContentCpu1,
+                    static_cast<int>(strlen(kTimeInStateContentCpu1))) != -1);
+    ASSERT_TRUE(base::WriteFile(
+                    all_time_in_state_path_, kAllTimeInStateContent,
+                    static_cast<int>(strlen(kAllTimeInStateContent))) != -1);
+  }
+
+ protected:
+  // Expected cpu frequency state names after calling |ReadCpuFreqTimeInState|
+  // or |ReadCpuFreqAllTimeInState|
+  const std::vector<std::string> kExpectedCpuFreqStateNames;
+
+  // Expected time_in_state of sample for cpu0.
+  const std::vector<TimeDelta> kExpectedTimeInStateCpu0;
+
+  // Expected time_in_state of sample for cpu1.
+  const std::vector<TimeDelta> kExpectedTimeInStateCpu1;
+
+  base::ScopedTempDir temp_dir_;
+  base::FilePath time_in_state_path_cpu0_;
+  base::FilePath time_in_state_path_cpu1_;
+  base::FilePath all_time_in_state_path_;
+
+ private:
+  DISALLOW_COPY_AND_ASSIGN(CpuDataCollectorTest);
+};
+
+TEST_F(CpuDataCollectorTest, ReadCpuFreqTimeInState) {
+  std::vector<std::string> cpu_freq_state_names;
+  CpuDataCollector::StateOccupancySample freq_sample_cpu0;
+  CpuDataCollector::StateOccupancySample freq_sample_cpu1;
+
+  CpuDataCollector::ReadCpuFreqTimeInState(
+      time_in_state_path_cpu0_, &cpu_freq_state_names, &freq_sample_cpu0);
+  EXPECT_EQ(kExpectedCpuFreqStateNames, cpu_freq_state_names);
+  EXPECT_EQ(kExpectedTimeInStateCpu0, freq_sample_cpu0.time_in_state);
+
+  cpu_freq_state_names.clear();
+  CpuDataCollector::ReadCpuFreqTimeInState(
+      time_in_state_path_cpu1_, &cpu_freq_state_names, &freq_sample_cpu1);
+  EXPECT_EQ(kExpectedCpuFreqStateNames, cpu_freq_state_names);
+  EXPECT_EQ(kExpectedTimeInStateCpu1, freq_sample_cpu1.time_in_state);
+}
+
+TEST_F(CpuDataCollectorTest, ReadCpuFreqAllTimeInState) {
+  std::vector<std::string> cpu_freq_state_names;
+  std::vector<CpuDataCollector::StateOccupancySample> freq_samples;
+  CpuDataCollector::StateOccupancySample freq_sample_cpu0;
+  CpuDataCollector::StateOccupancySample freq_sample_cpu1;
+  // |ReadCpuFreqAllTimeInState| only completes sample for cpu that is online.
+  freq_sample_cpu0.cpu_online = true;
+  freq_sample_cpu1.cpu_online = true;
+  freq_samples.push_back(freq_sample_cpu0);
+  freq_samples.push_back(freq_sample_cpu1);
+
+  CpuDataCollector::ReadCpuFreqAllTimeInState(
+      2, all_time_in_state_path_, &cpu_freq_state_names, &freq_samples);
+  EXPECT_EQ(kExpectedCpuFreqStateNames, cpu_freq_state_names);
+  EXPECT_EQ(kExpectedTimeInStateCpu0, freq_samples[0].time_in_state);
+  EXPECT_EQ(kExpectedTimeInStateCpu1, freq_samples[1].time_in_state);
+}
+
+}  // namespace chromeos