chrome/browser/memory/memory_ablation_study_unittest.cc

// Copyright 2021 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/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 {
const 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.
#if defined(OS_WIN)
// TODO(https://crbug.com/1245173) Flaky on Win7
#define MAYBE_Basic DISABLED_Basic
#else
#define MAYBE_Basic Basic
#endif
TEST_F(MemoryAblationStudyTest, MAYBE_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