ThreadWatcher: fixes Start/StopWatchingAll.

chrome/browser/metrics/thread_watcher_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 <math.h>
 
 #include "base/basictypes.h"
 #include "base/bind.h"
 #include "base/logging.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/message_loop/message_loop.h"
 #include "base/message_loop/message_loop_proxy.h"
+#include "base/run_loop.h"
 #include "base/strings/string_number_conversions.h"
 #include "base/strings/string_split.h"
 #include "base/strings/string_tokenizer.h"
 #include "base/synchronization/condition_variable.h"
 #include "base/synchronization/lock.h"
 #include "base/threading/platform_thread.h"
 #include "base/time/time.h"
 #include "build/build_config.h"
 #include "chrome/browser/metrics/thread_watcher.h"
 #include "chrome/common/chrome_switches.h"
@@ skipping 599 matching lines @@
       base::Bind(&ThreadWatcher::DeActivateThreadWatching,
                  base::Unretained(io_watcher_)));
   WatchDogThread::PostTask(
       FROM_HERE,
       base::Bind(&ThreadWatcher::DeActivateThreadWatching,
                  base::Unretained(db_watcher_)));
 
   // Wait for the io_watcher_'s VeryLongMethod to finish.
   io_watcher_->WaitForWaitStateChange(kUnresponsiveTime * 10, ALL_DONE);
 }
+
+TEST(ThreadWatcherListTest, Restart) {
+  ThreadWatcherList::g_initialize_delay_seconds = 1;
+
+  base::MessageLoopForUI message_loop_for_ui;
+  content::TestBrowserThread ui_thread(BrowserThread::UI, &message_loop_for_ui);
+
+  scoped_ptr<WatchDogThread> watchdog_thread_(new WatchDogThread());
+  watchdog_thread_->Start();
+
+  EXPECT_FALSE(ThreadWatcherList::g_thread_watcher_list_);
+
+  // See http://crbug.com/347887.
+  // StartWatchingAll() will PostDelayedTask to create g_thread_watcher_list_,
+  // whilst StopWatchingAll() will just PostTask to destroy it.
+  // Ensure that when Stop is called, Start will NOT create
+  // g_thread_watcher_list_ later on.
+  ThreadWatcherList::StartWatchingAll(*CommandLine::ForCurrentProcess());
+  ThreadWatcherList::StopWatchingAll();
+  message_loop_for_ui.PostDelayedTask(
+      FROM_HERE,
+      message_loop_for_ui.QuitClosure(),
+      base::TimeDelta::FromSeconds(
+          ThreadWatcherList::g_initialize_delay_seconds));
+  message_loop_for_ui.Run();
+  EXPECT_FALSE(ThreadWatcherList::g_thread_watcher_list_);
+  EXPECT_TRUE(ThreadWatcherList::g_stopped_);
+
+  // Proceed with just |StartWatchingAll| and ensure it'll be started.
+  ThreadWatcherList::StartWatchingAll(*CommandLine::ForCurrentProcess());
+  message_loop_for_ui.PostDelayedTask(
+      FROM_HERE,
+      message_loop_for_ui.QuitClosure(),
+      base::TimeDelta::FromSeconds(
+          ThreadWatcherList::g_initialize_delay_seconds + 1));
+  message_loop_for_ui.Run();
+  EXPECT_TRUE(ThreadWatcherList::g_thread_watcher_list_);
+  EXPECT_FALSE(ThreadWatcherList::g_stopped_);
+
+  // Finally, StopWatchingAll() must stop.
+  ThreadWatcherList::StopWatchingAll();
+  message_loop_for_ui.PostDelayedTask(
+      FROM_HERE,
+      message_loop_for_ui.QuitClosure(),
+      base::TimeDelta::FromSeconds(
+          ThreadWatcherList::g_initialize_delay_seconds));
+  message_loop_for_ui.Run();
+  EXPECT_FALSE(ThreadWatcherList::g_thread_watcher_list_);
+  EXPECT_TRUE(ThreadWatcherList::g_stopped_);
+}