Hook up allocator shim on mac.

base/debug/thread_heap_usage_tracker_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 "base/debug/thread_heap_usage_tracker.h"
 
 #include <map>
 
 #include "base/allocator/allocator_shim.h"
 #include "base/allocator/features.h"
@@ skipping 188 matching lines @@
     {
         &ThreadHeapUsageTrackerTest::OnAllocFn,  // alloc_function
         &ThreadHeapUsageTrackerTest::
             OnAllocZeroInitializedFn,  // alloc_zero_initialized_function
         &ThreadHeapUsageTrackerTest::
             OnAllocAlignedFn,                      // alloc_aligned_function
         &ThreadHeapUsageTrackerTest::OnReallocFn,  // realloc_function
         &ThreadHeapUsageTrackerTest::OnFreeFn,     // free_function
         &ThreadHeapUsageTrackerTest::
             OnGetSizeEstimateFn,  // get_size_estimate_function
+        nullptr,                  // batch_malloc
+        nullptr,                  // batch_free
+        nullptr,                  // free_definite_size_function
         nullptr,                  // next
 };
 
 }  // namespace
 
 TEST_F(ThreadHeapUsageTrackerTest, SimpleUsageWithExactSizeFunction) {
   set_size_function_kind(EXACT_SIZE_FUNCTION);
 
   ThreadHeapUsageTracker usage_tracker;
   usage_tracker.Start();
@@ skipping 313 matching lines @@
   ASSERT_TRUE(estimate == 0 || estimate >= kAllocSize);
 
   alloc = MockRealloc(alloc, kAllocSize);
   estimate = MockGetSizeEstimate(alloc);
   ASSERT_TRUE(estimate == 0 || estimate >= kAllocSize);
   MockFree(alloc);
 }
 
 #if BUILDFLAG(USE_EXPERIMENTAL_ALLOCATOR_SHIM)
 TEST(ThreadHeapUsageShimTest, HooksIntoMallocWhenShimAvailable) {
+#if defined(OS_MACOSX)
+  allocator::InitializeAllocatorShim();
+#endif
+
   ASSERT_FALSE(ThreadHeapUsageTracker::IsHeapTrackingEnabled());
 
   ThreadHeapUsageTracker::EnableHeapTracking();
 
   ASSERT_TRUE(ThreadHeapUsageTracker::IsHeapTrackingEnabled());
 
   const size_t kAllocSize = 9993;
   // This test verifies that the scoped heap data is affected by malloc &
   // free only when the shim is available.
   ThreadHeapUsageTracker usage_tracker;
@@ skipping 23 matching lines @@
   EXPECT_LE(u2.free_ops + 1, u3.free_ops);
 
   TestingThreadHeapUsageTracker::DisableHeapTrackingForTesting();
 
   ASSERT_FALSE(ThreadHeapUsageTracker::IsHeapTrackingEnabled());
 }
 #endif  // BUILDFLAG(USE_EXPERIMENTAL_ALLOCATOR_SHIM)
 
 }  // namespace debug
 }  // namespace base