1 files changed, 91 insertions, 0 deletions
diff --git a/chromium/chrome/common/heap_profiler_controller_unittest.cc b/chromium/chrome/common/heap_profiler_controller_unittest.cc
new file mode 100644
index 00000000000..c0ff97dd2f6
--- /dev/null
+++ b/chromium/chrome/common/heap_profiler_controller_unittest.cc
@@ -0,0 +1,91 @@
+// Copyright 2018 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/common/heap_profiler_controller.h"
+
+#include "base/sampling_heap_profiler/sampling_heap_profiler.h"
+#include "base/test/bind_test_util.h"
+#include "base/test/scoped_feature_list.h"
+#include "base/test/task_environment.h"
+#include "build/build_config.h"
+#include "components/metrics/call_stack_profile_builder.h"
+#include "components/metrics/call_stack_profile_metrics_provider.h"
+#include "testing/gtest/include/gtest/gtest.h"
+#include "third_party/metrics_proto/sampled_profile.pb.h"
+
+// TODO(crbug.com/961073): Fix memory leaks in tests and re-enable on LSAN.
+#ifdef LEAK_SANITIZER
+#define MAYBE_EmptyProfileIsNotEmitted DISABLED_EmptyProfileIsNotEmitted
+#else
+#define MAYBE_EmptyProfileIsNotEmitted EmptyProfileIsNotEmitted
+#endif
+
+class HeapProfilerControllerTest : public testing::Test {
+ protected:
+  base::test::TaskEnvironment task_environment{
+      base::test::TaskEnvironment::TimeSource::MOCK_TIME};
+};
+
+TEST_F(HeapProfilerControllerTest, MAYBE_EmptyProfileIsNotEmitted) {
+  HeapProfilerController controller;
+  metrics::CallStackProfileBuilder::SetBrowserProcessReceiverCallback(
+      base::BindLambdaForTesting(
+          [](base::TimeTicks time, metrics::SampledProfile profile) {
+            ADD_FAILURE();
+          }));
+  controller.Start();
+
+  task_environment.FastForwardBy(base::TimeDelta::FromDays(365));
+}
+
+// Sampling profiler is not capable of unwinding stack on Android under tests.
+#if !defined(OS_ANDROID)
+TEST_F(HeapProfilerControllerTest, ProfileCollectionsScheduler) {
+  constexpr size_t kAllocationSize = 42 * 1024;
+  constexpr int kSnapshotsToCollect = 3;
+
+  auto controller = std::make_unique<HeapProfilerController>();
+  int profile_count = 0;
+
+  auto check_profile = [&](base::TimeTicks time,
+                           metrics::SampledProfile profile) {
+    EXPECT_EQ(metrics::SampledProfile::PERIODIC_HEAP_COLLECTION,
+              profile.trigger_event());
+    EXPECT_LT(0, profile.call_stack_profile().stack_sample_size());
+
+    bool found = false;
+    for (const metrics::CallStackProfile::StackSample& sample :
+         profile.call_stack_profile().stack_sample()) {
+      if (sample.has_weight() &&
+          static_cast<size_t>(sample.weight()) >= kAllocationSize) {
+        found = true;
+        break;
+      }
+    }
+    EXPECT_TRUE(found);
+
+    if (++profile_count == kSnapshotsToCollect)
+      controller.reset();
+  };
+
+  base::test::ScopedFeatureList feature_list;
+  feature_list.InitAndEnableFeature(
+      metrics::CallStackProfileMetricsProvider::kHeapProfilerReporting);
+  metrics::CallStackProfileBuilder::SetBrowserProcessReceiverCallback(
+      base::BindLambdaForTesting(check_profile));
+  base::SamplingHeapProfiler::Get()->SetSamplingInterval(1024);
+
+  controller->Start();
+
+  auto* sampler = base::PoissonAllocationSampler::Get();
+  sampler->SuppressRandomnessForTest(true);
+  sampler->RecordAlloc(reinterpret_cast<void*>(0x1337), kAllocationSize,
+                       base::PoissonAllocationSampler::kMalloc, nullptr);
+  sampler->RecordAlloc(reinterpret_cast<void*>(0x7331), kAllocationSize,
+                       base::PoissonAllocationSampler::kMalloc, nullptr);
+
+  task_environment.FastForwardUntilNoTasksRemain();
+  EXPECT_LE(kSnapshotsToCollect, profile_count);
+}
+#endif