chrome/browser/memory/memory_ablation_study_unittest.cc

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

#include "chrome/browser/memory/memory_ablation_study.h"
#include "base/feature_list.h"
#include "base/test/scoped_feature_list.h"
#include "base/test/task_environment.h"
#include "build/build_config.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace memory {

namespace {
BASE_FEATURE(kMemoryAblationStudy,
             "MemoryAblationStudy",
             base::FEATURE_DISABLED_BY_DEFAULT);
}  // namespace

class MemoryAblationStudyTest : public testing::Test {
 protected:
  base::test::TaskEnvironment task_environment_{
      base::test::TaskEnvironment::TimeSource::MOCK_TIME};
  base::test::ScopedFeatureList feature_list;
};

// Tests basic functionality of the MemoryAblationStudy class.
TEST_F(MemoryAblationStudyTest, Basic) {
  // Ablate 137MB.
  base::FieldTrialParams params;
  params["ablation-size-mb"] = "137";
  feature_list.InitAndEnableFeatureWithParameters(kMemoryAblationStudy, params);

  // 450s should be enough to both allocate the memory and trigger a read.
  MemoryAblationStudy study;
  task_environment_.FastForwardBy(base::Seconds(450));
  size_t total_size = 0;
  for (MemoryAblationStudy::Region& region : study.regions_) {
    total_size += region.size();
  }
  EXPECT_EQ(total_size, 137 * 1024 * 1024u);
}

}  // namespace memory