metrics: Connect leak detector to allocator

components/metrics/leak_detector/leak_detector_unittest.cc

 // Copyright 2016 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 "components/metrics/leak_detector/leak_detector.h"
 
 #include <set>
 
+#include "base/allocator/allocator_extension.h"
 #include "base/macros.h"
 #include "base/memory/scoped_ptr.h"
 #include "content/public/test/test_browser_thread_bundle.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace metrics {
 
 namespace {
 
 // Default values for LeakDetector params. See header file for the meaning of
@@ skipping 52 matching lines @@
   // test is over.
   scoped_ptr<LeakDetector> detector_;
 
  private:
   // For supporting content::BrowserThread operations.
   content::TestBrowserThreadBundle thread_bundle_;
 
   DISALLOW_COPY_AND_ASSIGN(LeakDetectorTest);
 };
 
+TEST_F(LeakDetectorTest, AllocatorHooks) {
+  EXPECT_EQ(&LeakDetector::AllocHook, base::allocator::GetSingleAllocHook());
+  EXPECT_EQ(&LeakDetector::FreeHook, base::allocator::GetSingleFreeHook());
+}
+
 TEST_F(LeakDetectorTest, NotifyObservers) {
   // Generate two sets of leak reports.
   std::vector<LeakReport> reports1(3);
   reports1[0].alloc_size_bytes = 8;
   reports1[0].call_stack = {1, 2, 3, 4};
   reports1[1].alloc_size_bytes = 16;
   reports1[1].call_stack = {5, 6, 7, 8};
   reports1[2].alloc_size_bytes = 24;
   reports1[2].call_stack = {9, 10, 11, 12};
 
@@ skipping 24 matching lines @@
   for (const TestObserver* obs : {&obs1, &obs2, &obs3}) {
     EXPECT_EQ(6U, obs->reports().size());
     for (const auto& report : {reports1[0], reports1[1], reports1[2],
                                reports2[0], reports2[1], reports2[2]}) {
       EXPECT_TRUE(obs->reports().find(report) != obs->reports().end());
     }
   }
 }
 
 }  // namespace metrics