Chromium -> Mojo roll.

base/trace_event/memory_allocator_dump_unittest.cc

+// Copyright 2015 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/trace_event/memory_allocator_dump.h"
+
+#include "base/trace_event/memory_dump_provider.h"
+#include "base/trace_event/process_memory_dump.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace base {
+namespace trace_event {
+
+namespace {
+class FakeMemoryAllocatorDumpProvider : public MemoryDumpProvider {
+ public:
+  FakeMemoryAllocatorDumpProvider() {
+    DeclareAllocatorAttribute({"attr1", "count"});
+    DeclareAllocatorAttribute({"attr2", "bytes"});
+  }
+
+  bool DumpInto(ProcessMemoryDump* pmd) override {
+    MemoryAllocatorDump* mad_foo = pmd->CreateAllocatorDump("foo");
+    mad_foo->set_physical_size_in_bytes(4096);
+    mad_foo->set_allocated_objects_count(42);
+    mad_foo->set_allocated_objects_size_in_bytes(1000);
+    mad_foo->SetExtraAttribute("attr1", 1234);
+    mad_foo->SetExtraAttribute("attr2", 99);
+
+    MemoryAllocatorDump* mad_bar = pmd->CreateAllocatorDump("foo/bar", mad_foo);
+    mad_bar->set_physical_size_in_bytes(1);
+    mad_bar->set_allocated_objects_count(2);
+    mad_bar->set_allocated_objects_size_in_bytes(3);
+
+    pmd->CreateAllocatorDump("baz");
+    // Leave the rest of |baz| deliberately uninitialized, to check that
+    // CreateAllocatorDump returns a properly zero-initialized object.
+
+    return true;
+  }
+
+  const char* GetFriendlyName() const override { return "mock_allocator"; }
+};
+}  // namespace
+
+TEST(MemoryAllocatorDumpTest, DumpIntoProcessMemoryDump) {
+  FakeMemoryAllocatorDumpProvider fmadp;
+  ProcessMemoryDump pmd;
+
+  fmadp.DumpInto(&pmd);
+
+  ASSERT_EQ(3u, pmd.allocator_dumps().size());
+
+  const MemoryAllocatorDump* mad_foo = pmd.GetAllocatorDump("foo");
+  ASSERT_NE(nullptr, mad_foo);
+  EXPECT_EQ("foo", mad_foo->name());
+  ASSERT_EQ(nullptr, mad_foo->parent());
+  EXPECT_EQ(4096u, mad_foo->physical_size_in_bytes());
+  EXPECT_EQ(42u, mad_foo->allocated_objects_count());
+  EXPECT_EQ(1000u, mad_foo->allocated_objects_size_in_bytes());
+
+  // Check the extra attributes of |mad_foo|.
+  EXPECT_EQ(1234, mad_foo->GetExtraIntegerAttribute("attr1"));
+  EXPECT_EQ(99, mad_foo->GetExtraIntegerAttribute("attr2"));
+
+  const MemoryAllocatorDump* mad_bar = pmd.GetAllocatorDump("foo/bar");
+  ASSERT_NE(nullptr, mad_bar);
+  EXPECT_EQ("foo/bar", mad_bar->name());
+  ASSERT_EQ(mad_foo, mad_bar->parent());
+  EXPECT_EQ(1u, mad_bar->physical_size_in_bytes());
+  EXPECT_EQ(2u, mad_bar->allocated_objects_count());
+  EXPECT_EQ(3u, mad_bar->allocated_objects_size_in_bytes());
+
+  const MemoryAllocatorDump* mad_baz = pmd.GetAllocatorDump("baz");
+  ASSERT_NE(nullptr, mad_baz);
+  EXPECT_EQ("baz", mad_baz->name());
+  ASSERT_EQ(nullptr, mad_baz->parent());
+  EXPECT_EQ(0u, mad_baz->physical_size_in_bytes());
+  EXPECT_EQ(0u, mad_baz->allocated_objects_count());
+  EXPECT_EQ(0u, mad_baz->allocated_objects_size_in_bytes());
+}
+
+// DEATH tests are not supported in Android / iOS.
+#if !defined(NDEBUG) && !defined(OS_ANDROID) && !defined(OS_IOS)
+TEST(MemoryAllocatorDumpTest, ForbidDuplicatesDeathTest) {
+  FakeMemoryAllocatorDumpProvider fmadp;
+  ProcessMemoryDump pmd;
+  pmd.CreateAllocatorDump("dump_1");
+  pmd.CreateAllocatorDump("dump_2");
+  ASSERT_DEATH(pmd.CreateAllocatorDump("dump_1"), "");
+}
+#endif
+
+}  // namespace trace_event
+}  // namespace base