Add ability to retrieve a thread_name given a thread_id.

base/threading/thread_id_name_manager_unittest.cc

+// Copyright (c) 2012 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/threading/thread_id_name_manager.h"
+
+#include "base/threading/platform_thread.h"
+#include "base/threading/thread.h"
+#include "testing/gtest/include/gtest/gtest.h"
+#include "testing/platform_test.h"
+
+typedef PlatformTest ThreadIdNameManagerTest;
+
+namespace {
+
+static const char* kAThread = "a thread";
+static const char* kBThread = "b thread";
+
+TEST_F(ThreadIdNameManagerTest, AddThreads) {
+  base::ThreadIdNameManager* manager = base::ThreadIdNameManager::GetInstance();
+  base::Thread thread_a(kAThread);
+  base::Thread thread_b(kBThread);
+
+  thread_a.Start();
+  thread_b.Start();
+
+  EXPECT_NE(manager->GetInternedName(thread_a.thread_id()),
+      manager->GetInternedName(thread_b.thread_id()));
+
+  EXPECT_STREQ(kAThread, manager->GetNameForId(thread_a.thread_id()));
+  EXPECT_STREQ(kBThread, manager->GetNameForId(thread_b.thread_id()));
+
+  thread_b.Stop();
+  thread_a.Stop();
+}
+
+TEST_F(ThreadIdNameManagerTest, RemoveThreads) {
+  base::ThreadIdNameManager* manager = base::ThreadIdNameManager::GetInstance();
+  base::Thread thread_a(kAThread);
+
+  thread_a.Start();
+  base::PlatformThreadId b_id;
+  {
+    base::Thread thread_b(kBThread);
+    thread_b.Start();
+    b_id = thread_b.thread_id();
+    thread_b.Stop();
+  }
+  thread_a.Stop();
+
+  EXPECT_STREQ(kAThread, manager->GetNameForId(thread_a.thread_id()));
+  EXPECT_EQ(NULL, manager->GetNameForId(b_id));
+}
+
+TEST_F(ThreadIdNameManagerTest, RestartThread) {
+  base::ThreadIdNameManager* manager = base::ThreadIdNameManager::GetInstance();
+  base::Thread thread_a(kAThread);
+
+  thread_a.Start();
+  base::PlatformThreadId a_id = thread_a.thread_id();
+  thread_a.Stop();
+  EXPECT_STREQ(kAThread, manager->GetNameForId(a_id));
+
+  thread_a.Start();
+  thread_a.Stop();
+  EXPECT_STREQ(kAThread, manager->GetNameForId(a_id));
+}
+
+TEST_F(ThreadIdNameManagerTest, ThreadNameInterning) {
+  base::ThreadIdNameManager* manager = base::ThreadIdNameManager::GetInstance();
+
+  base::PlatformThreadId a_id = base::PlatformThread::CurrentId();
+  base::InternedString version = manager->GetInternedName(a_id);
+
+  base::PlatformThread::SetName("New name");
+  EXPECT_NE(version, manager->GetInternedName(a_id));
+  base::PlatformThread::SetName("");
+}
+
+TEST_F(ThreadIdNameManagerTest, ResettingNameKeepsCorrectInternedValue) {
+  base::ThreadIdNameManager* manager = base::ThreadIdNameManager::GetInstance();
+
+  base::PlatformThreadId a_id = base::PlatformThread::CurrentId();
+  base::PlatformThread::SetName("Test Name");
+  base::InternedString version = manager->GetInternedName(a_id);
+
+  base::PlatformThread::SetName("New name");
+  EXPECT_NE(version, manager->GetInternedName(a_id));
+
+  base::PlatformThread::SetName("Test Name");
+  EXPECT_EQ(version, manager->GetInternedName(a_id));
+
+  base::PlatformThread::SetName("");
+}
+
+}  // namespace