Fix subtle bug with the in-process utility thread usage in unit tests.

content/public/test/test_utils.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 "content/public/test/test_utils.h"
 
 #include "base/bind.h"
 #include "base/message_loop/message_loop.h"
 #include "base/run_loop.h"
 #include "base/strings/utf_string_conversions.h"
 #include "base/values.h"
+#include "content/public/browser/browser_child_process_host_iterator.h"
 #include "content/public/browser/notification_service.h"
+#include "content/public/browser/render_process_host.h"
 #include "content/public/browser/render_view_host.h"
+#include "content/public/common/process_type.h"
 #include "content/public/test/test_launcher.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace content {
 
 namespace {
 
 // Number of times to repost a Quit task so that the MessageLoop finishes up
 // pending tasks and tasks posted by those pending tasks without risking the
 // potential hang behavior of MessageLoop::QuitWhenIdle.
@@ skipping 211 matching lines @@
     return;
 
   seen_ = true;
   if (!running_)
     return;
 
   message_loop_runner_->Quit();
   running_ = false;
 }
 
+InProcessUtilityThreadHelper::InProcessUtilityThreadHelper()
+    : child_thread_count_(0) {
+  RenderProcessHost::SetRunRendererInProcess(true);
+  BrowserChildProcessObserver::Add(this);
+}
+
+InProcessUtilityThreadHelper::~InProcessUtilityThreadHelper() {
+  if (child_thread_count_) {
+    DCHECK(BrowserThread::IsMessageLoopValid(BrowserThread::UI));
+    DCHECK(BrowserThread::IsMessageLoopValid(BrowserThread::IO));
+    runner_ = new MessageLoopRunner;
+    runner_->Run();
+  }
+  BrowserChildProcessObserver::Remove(this);
+  RenderProcessHost::SetRunRendererInProcess(false);
+}
+
+void InProcessUtilityThreadHelper::BrowserChildProcessHostConnected(
+    const ChildProcessData& data) {
+  child_thread_count_++;
+}
+
+void InProcessUtilityThreadHelper::BrowserChildProcessHostDisconnected(
+    const ChildProcessData& data) {
+  if (--child_thread_count_)
+    return;
+
+  if (runner_.get())
+    runner_->Quit();
+}
+
 }  // namespace content