[heap] Improve size profiling for ArrayBuffer tracking

test/cctest/heap/test-array-buffer-tracker.cc

 // Copyright 2016 the V8 project 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 "src/heap/array-buffer-tracker.h"
 #include "test/cctest/cctest.h"
 #include "test/cctest/heap/heap-utils.h"
 
 namespace {
 
@@ skipping 296 matching lines @@
     Handle<JSArrayBuffer> buf1 = v8::Utils::OpenHandle(*ab1);
     heap::FillCurrentPage(heap->new_space());
     Local<v8::ArrayBuffer> ab2 = v8::ArrayBuffer::New(isolate, 100);
     Handle<JSArrayBuffer> buf2 = v8::Utils::OpenHandle(*ab2);
     CHECK_NE(Page::FromAddress(buf1->address()),
              Page::FromAddress(buf2->address()));
     heap::GcAndSweep(heap, OLD_SPACE);
   }
 }
 
+TEST(ArrayBuffer_RetainedCounterPromotion) {
+  // The test checks that retained counters on ArrayBufferTracker are consistent
+  // with where the buffers are on the heap.
+  CcTest::InitializeVM();
+  LocalContext env;
+  v8::Isolate* isolate = env->GetIsolate();
+  Heap* heap = reinterpret_cast<Isolate*>(isolate)->heap();
+  ArrayBufferTracker* tracker = heap->array_buffer_tracker();
+  {
+    v8::HandleScope handle_scope(isolate);
+    Local<v8::ArrayBuffer> ab = v8::ArrayBuffer::New(isolate, 100);
+    Handle<JSArrayBuffer> buf = v8::Utils::OpenHandle(*ab);
+    CHECK(heap->InNewSpace(*buf));
+    CHECK_EQ(tracker->retained_from_new_space(), 100u);
+    CHECK_EQ(tracker->retained_from_old_space(), 0u);
+    heap::GcAndSweep(heap, OLD_SPACE);
+    CHECK(heap->InNewSpace(*buf));
+    CHECK_EQ(tracker->retained_from_new_space(), 100u);
+    CHECK_EQ(tracker->retained_from_old_space(), 0u);
+    heap::GcAndSweep(heap, OLD_SPACE);
+    CHECK(heap->InOldSpace(*buf));
+    CHECK_EQ(tracker->retained_from_new_space(), 0u);
+    CHECK_EQ(tracker->retained_from_old_space(), 100u);
+  }
+}
+
+TEST(ArrayBuffer_RetainedCounterNewSpace) {
+  // The test checks that retained counters on ArrayBufferTracker are consistent
+  // with where the buffers are on the heap.
+  CcTest::InitializeVM();
+  LocalContext env;
+  v8::Isolate* isolate = env->GetIsolate();
+  Heap* heap = reinterpret_cast<Isolate*>(isolate)->heap();
+  ArrayBufferTracker* tracker = heap->array_buffer_tracker();
+  {
+    v8::HandleScope handle_scope(isolate);
+    {
+      v8::HandleScope tmp_scope(isolate);
+      Local<v8::ArrayBuffer> ab = v8::ArrayBuffer::New(isolate, 100);
+      Handle<JSArrayBuffer> buf = v8::Utils::OpenHandle(*ab);
+      CHECK(heap->InNewSpace(*buf));
+    }
+    {
+      v8::HandleScope tmp_scope(isolate);
+      Local<v8::ArrayBuffer> ab = v8::ArrayBuffer::New(isolate, 50);
+      Handle<JSArrayBuffer> buf = v8::Utils::OpenHandle(*ab);
+      CHECK(heap->InNewSpace(*buf));
+      CHECK_EQ(tracker->retained_from_new_space(), 150u);
+      CHECK_EQ(tracker->retained_from_old_space(), 0u);
+      heap::GcAndSweep(heap, NEW_SPACE);
+      CHECK(heap->InNewSpace(*buf));
+      CHECK_EQ(tracker->retained_from_new_space(), 50u);
+      CHECK_EQ(tracker->retained_from_old_space(), 0u);
+    }
+    heap::GcAndSweep(heap, NEW_SPACE);
+    CHECK_EQ(tracker->retained_from_new_space(), 0u);
+    CHECK_EQ(tracker->retained_from_old_space(), 0u);
+  }
+}
+
 }  // namespace internal
 }  // namespace v8