psi_memory_parser_linux_unittest.cc 4.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163
  1. // Copyright 2021 The Chromium Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style license that can be
  3. // found in the LICENSE file.
  4. #include "components/metrics/psi_memory_parser.h"
  5. #include <memory>
  6. #include "base/files/file_util.h"
  7. #include "base/metrics/statistics_recorder.h"
  8. #include "base/test/metrics/histogram_tester.h"
  9. #include "testing/gtest/include/gtest/gtest.h"
  10. namespace metrics {
  11. namespace {
  12. // Just as the kernel outputs.
  13. const char kFileContents1[] =
  14. "some avg10=23.10 avg60=5.06 avg300=15.10 total=417963\n"
  15. "full avg10=9.00 avg60=19.20 avg300=3.23 total=205933\n";
  16. // Number of decimals not consistent, slightly malformed - but acceptable.
  17. const char kFileContents2[] =
  18. "some avg10=24 avg60=5.06 avg300=15.10 total=417963\n"
  19. "full avg10=9.2 avg60=19.20 avg300=3.23 total=205933\n";
  20. } // namespace
  21. class PSIMemoryParserTest : public testing::Test {
  22. public:
  23. PSIMemoryParserTest() = default;
  24. ~PSIMemoryParserTest() override = default;
  25. void Init(uint32_t period) {
  26. cit_ = std::make_unique<PSIMemoryParser>(period);
  27. }
  28. uint32_t GetPeriod() { return cit_->GetPeriod(); }
  29. base::HistogramTester& Histograms() { return histogram_tester_; }
  30. std::unique_ptr<PSIMemoryParser>& Cit() { return cit_; }
  31. const std::string& GetMetricPrefix() { return cit_->metric_prefix_; }
  32. void KillCit() { cit_.reset(); }
  33. private:
  34. std::unique_ptr<PSIMemoryParser> cit_;
  35. base::HistogramTester histogram_tester_;
  36. };
  37. TEST_F(PSIMemoryParserTest, CustomInterval) {
  38. Init(60u);
  39. EXPECT_EQ(60u, GetPeriod());
  40. }
  41. TEST_F(PSIMemoryParserTest, InvalidInterval) {
  42. Init(15u);
  43. EXPECT_EQ(10u, GetPeriod());
  44. }
  45. TEST_F(PSIMemoryParserTest, InternalsA) {
  46. Init(10u);
  47. std::string testContent1 = "prefix" + GetMetricPrefix() + "9.37 suffix";
  48. EXPECT_EQ(10u, GetPeriod());
  49. size_t s = 0;
  50. size_t e = 0;
  51. EXPECT_EQ(false, internal::FindMiddleString(testContent1, 0, "nothere",
  52. "suffix", &s, &e));
  53. EXPECT_EQ(false, internal::FindMiddleString(testContent1, 0, "prefix",
  54. "notthere", &s, &e));
  55. EXPECT_EQ(true, internal::FindMiddleString(testContent1, 0, "prefix",
  56. "suffix", &s, &e));
  57. EXPECT_EQ(6u, s);
  58. EXPECT_EQ(17u, e);
  59. EXPECT_EQ(937, Cit()->GetMetricValue(testContent1, s, e));
  60. std::string testContent2 = "extra " + testContent1;
  61. EXPECT_EQ(true, internal::FindMiddleString(testContent2, 0, "prefix",
  62. "suffix", &s, &e));
  63. EXPECT_EQ(12u, s);
  64. EXPECT_EQ(23u, e);
  65. EXPECT_EQ(937, Cit()->GetMetricValue(testContent2, s, e));
  66. }
  67. TEST_F(PSIMemoryParserTest, InternalsB) {
  68. Init(300);
  69. int msome;
  70. int mfull;
  71. ParsePSIMemStatus stat;
  72. stat = Cit()->ParseMetrics(kFileContents1, &msome, &mfull);
  73. EXPECT_EQ(ParsePSIMemStatus::kSuccess, stat);
  74. EXPECT_EQ(1510, msome);
  75. EXPECT_EQ(323, mfull);
  76. }
  77. TEST_F(PSIMemoryParserTest, InternalsC) {
  78. Init(60);
  79. int msome;
  80. int mfull;
  81. ParsePSIMemStatus stat;
  82. stat = Cit()->ParseMetrics(kFileContents1, &msome, &mfull);
  83. EXPECT_EQ(ParsePSIMemStatus::kSuccess, stat);
  84. EXPECT_EQ(506, msome);
  85. EXPECT_EQ(1920, mfull);
  86. }
  87. TEST_F(PSIMemoryParserTest, InternalsD) {
  88. Init(10);
  89. int msome;
  90. int mfull;
  91. ParsePSIMemStatus stat;
  92. stat = Cit()->ParseMetrics(kFileContents1, &msome, &mfull);
  93. EXPECT_EQ(ParsePSIMemStatus::kSuccess, stat);
  94. EXPECT_EQ(2310, msome);
  95. EXPECT_EQ(900, mfull);
  96. }
  97. TEST_F(PSIMemoryParserTest, InternalsE) {
  98. Init(10);
  99. int msome;
  100. int mfull;
  101. ParsePSIMemStatus stat;
  102. stat = Cit()->ParseMetrics(kFileContents2, &msome, &mfull);
  103. EXPECT_EQ(ParsePSIMemStatus::kSuccess, stat);
  104. EXPECT_EQ(2400, msome);
  105. EXPECT_EQ(920, mfull);
  106. }
  107. TEST_F(PSIMemoryParserTest, ParseResultCounter) {
  108. Init(10);
  109. Cit()->LogParseStatus(ParsePSIMemStatus::kSuccess);
  110. Cit()->LogParseStatus(ParsePSIMemStatus::kInvalidMetricFormat);
  111. Cit()->LogParseStatus(ParsePSIMemStatus::kInvalidMetricFormat);
  112. Histograms().ExpectBucketCount("ChromeOS.CWP.ParsePSIMemory",
  113. ParsePSIMemStatus::kSuccess, 1);
  114. Histograms().ExpectBucketCount("ChromeOS.CWP.ParsePSIMemory",
  115. ParsePSIMemStatus::kInvalidMetricFormat, 2);
  116. }
  117. } // namespace metrics