Track thread activities in order to diagnose hangs.

base/metrics/activity_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/metrics/activity_tracker.h"
+
+#include <memory>
+
+#include "base/bind.h"
+#include "base/files/file.h"
+#include "base/files/file_util.h"
+#include "base/files/memory_mapped_file.h"
+#include "base/files/scoped_temp_dir.h"
+#include "base/memory/ptr_util.h"
+#include "base/pending_task.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace base {
+
+namespace {
+
+class TestActivityTracker : public ThreadActivityTracker {
+ public:
+  TestActivityTracker(std::unique_ptr<char[]> memory, size_t mem_size)
+      : ThreadActivityTracker(memset(memory.get(), 0, mem_size), mem_size),
+        mem_segment_(std::move(memory)) {}
+
+  ~TestActivityTracker() override {
+    delete GlobalActivityTracker::Get();
+  }
+
+ private:
+  std::unique_ptr<char[]> mem_segment_;
+};
+
+}  // namespace
+
+
+class ActivityTrackerTest : public testing::Test {
+ public:
+  const int kMemorySize = 1 << 10;  // 1KiB
+  const int kStackSize  = 256;
+
+  ActivityTrackerTest() {}
+
+  ~ActivityTrackerTest() {}
+
+  std::unique_ptr<ThreadActivityTracker> CreateActivityTracker() {
+    std::unique_ptr<char[]> memory(new char[kStackSize]);
+    return WrapUnique(new TestActivityTracker(std::move(memory), kStackSize));
+  }
+
+  static void DoNothing() {}
+};
+
+TEST_F(ActivityTrackerTest, PushPopTest) {
+  std::unique_ptr<ThreadActivityTracker> tracker = CreateActivityTracker();
+  std::vector<ThreadActivityTracker::StackEntry> stack;
+
+  ASSERT_EQ(0U, tracker->CopyStack(&stack));
+  ASSERT_EQ(0U, stack.size());
+
+  char source1;
+  tracker->RecordStart(&source1, ThreadActivityTracker::ACT_TASK, 2, 3);
+  ASSERT_EQ(1U, tracker->CopyStack(&stack));
+  ASSERT_EQ(1U, stack.size());
+  EXPECT_NE(0, stack[0].time_ticks);
+  EXPECT_EQ(ThreadActivityTracker::ACT_TASK, stack[0].activity_type);
+  EXPECT_EQ(reinterpret_cast<intptr_t>(&source1), stack[0].source_address);
+  EXPECT_EQ(2, stack[0].method_address);
+  EXPECT_EQ(3U, stack[0].sequence_id);
+
+  char source2;
+  tracker->RecordStart(&source2, ThreadActivityTracker::ACT_LOCK, 3, 4);
+  ASSERT_EQ(2U, tracker->CopyStack(&stack));
+  ASSERT_EQ(2U, stack.size());
+  EXPECT_LE(stack[0].time_ticks, stack[1].time_ticks);
+  EXPECT_EQ(ThreadActivityTracker::ACT_LOCK, stack[1].activity_type);
+  EXPECT_EQ(reinterpret_cast<intptr_t>(&source2), stack[1].source_address);
+  EXPECT_EQ(3, stack[1].method_address);
+  EXPECT_EQ(4U, stack[1].sequence_id);
+
+  tracker->RecordFinish(&source2);
+  ASSERT_EQ(1U, tracker->CopyStack(&stack));
+  ASSERT_EQ(1U, stack.size());
+  EXPECT_EQ(ThreadActivityTracker::ACT_TASK, stack[0].activity_type);
+  EXPECT_EQ(reinterpret_cast<intptr_t>(&source1), stack[0].source_address);
+  EXPECT_EQ(2, stack[0].method_address);
+  EXPECT_EQ(3U, stack[0].sequence_id);
+
+  tracker->RecordFinish(&source1);
+  ASSERT_EQ(0U, tracker->CopyStack(&stack));
+  ASSERT_EQ(0U, stack.size());
+}
+
+TEST_F(ActivityTrackerTest, ScopedTaskTest) {
+  GlobalActivityTracker::CreateWithLocalMemory(kMemorySize, 0, "", kStackSize);
+
+  ThreadActivityTracker* tracker =
+      GlobalActivityTracker::Get()->GetOrCreateTrackerForCurrentThread();
+  std::vector<ThreadActivityTracker::StackEntry> stack;
+
+  ASSERT_EQ(0U, tracker->CopyStack(&stack));
+  ASSERT_EQ(0U, stack.size());
+
+  {
+    PendingTask task1(FROM_HERE, base::Bind(&DoNothing));
+    ScopedTaskActivity activity1(task1);
+
+    ASSERT_EQ(1U, tracker->CopyStack(&stack));
+    ASSERT_EQ(1U, stack.size());
+    EXPECT_EQ(ThreadActivityTracker::ACT_TASK, stack[0].activity_type);
+
+    {
+      PendingTask task2(FROM_HERE, base::Bind(&DoNothing));
+      ScopedTaskActivity activity2(task2);
+
+      ASSERT_EQ(2U, tracker->CopyStack(&stack));
+      ASSERT_EQ(2U, stack.size());
+      EXPECT_EQ(ThreadActivityTracker::ACT_TASK, stack[1].activity_type);
+    }
+
+    ASSERT_EQ(1U, tracker->CopyStack(&stack));
+    ASSERT_EQ(1U, stack.size());
+    EXPECT_EQ(ThreadActivityTracker::ACT_TASK, stack[0].activity_type);
+  }
+
+  ASSERT_EQ(0U, tracker->CopyStack(&stack));
+  ASSERT_EQ(0U, stack.size());
+}
+
+}  // namespace base