1
    2
    3
    4
    5
    6
    7
    8
    9
   10
   11
   12
   13
   14
   15
   16
   17
   18
   19
   20
   21
   22
   23
   24
   25
   26
   27
   28
   29
   30
   31
   32
   33
   34
   35
   36
   37
   38
   39
   40
   41
   42
   43
   44
   45
   46

base / android / pmf_utils_unittest.cc [blame]

// Copyright 2024 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "base/android/pmf_utils.h"

#include "base/files/file_util.h"
#include "build/build_config.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace base::android {

TEST(PmfUtilsTest, CalculatePrivateMemoryFootprint) {
  const char kStatusFile[] =
      "First:    1\n"
      "Second:  2 kB\n"
      "VmSwap: 10 kB\n"
      "Third:  10 kB\n"
      "VmHWM:  72 kB\n"
      "Last:     8";
  const char kStatmFile[] = "100 40 25 0 0";
  const uint64_t expected_swap_kb = 10;
  const uint64_t expected_pmf =
      (40 - 25) * getpagesize() / 1024 + expected_swap_kb;

  base::FilePath statm_path;
  EXPECT_TRUE(base::CreateTemporaryFile(&statm_path));
  EXPECT_TRUE(base::WriteFile(statm_path, kStatmFile));
  base::FilePath status_path;
  EXPECT_TRUE(base::CreateTemporaryFile(&status_path));
  EXPECT_TRUE(base::WriteFile(status_path, kStatusFile));

  base::File statm_file(statm_path,
                        base::File::FLAG_OPEN | base::File::FLAG_READ);
  base::File status_file(status_path,
                         base::File::FLAG_OPEN | base::File::FLAG_READ);

  std::optional<uint64_t> pmf =
      PmfUtils::CalculatePrivateMemoryFootprintForTesting(statm_file,
                                                          status_file);

  EXPECT_TRUE(pmf.has_value());
  EXPECT_EQ(expected_pmf, pmf.value() / 1024);
}

}  // namespace base::android