Hide WorkerProcessHost from chrome.

chrome/browser/debugger/devtools_sanity_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/bind.h"
 #include "base/cancelable_callback.h"
 #include "base/command_line.h"
 #include "base/compiler_specific.h"
 #include "base/memory/ref_counted.h"
 #include "base/path_service.h"
 #include "base/stringprintf.h"
 #include "base/test/test_timeouts.h"
 #include "base/utf_string_conversions.h"
 #include "chrome/browser/debugger/devtools_window.h"
 #include "chrome/browser/extensions/extension_host.h"
 #include "chrome/browser/extensions/extension_service.h"
 #include "chrome/browser/extensions/unpacked_installer.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/tab_contents/tab_contents_wrapper.h"
 #include "chrome/common/chrome_notification_types.h"
 #include "chrome/common/chrome_paths.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/test/base/in_process_browser_test.h"
 #include "chrome/test/base/ui_test_utils.h"
 #include "content/browser/renderer_host/render_view_host.h"
-#include "content/browser/worker_host/worker_process_host.h"
-#include "content/public/browser/browser_child_process_host_iterator.h"
 #include "content/public/browser/child_process_data.h"
 #include "content/public/browser/content_browser_client.h"
 #include "content/public/browser/devtools_agent_host_registry.h"
 #include "content/public/browser/devtools_client_host.h"
 #include "content/public/browser/devtools_manager.h"
 #include "content/public/browser/notification_registrar.h"
 #include "content/public/browser/notification_service.h"
 #include "content/public/browser/render_view_host_delegate.h"
 #include "content/public/browser/web_contents.h"
 #include "content/public/browser/worker_service.h"
@@ skipping 246 matching lines @@
   class WorkerCreationObserver : public WorkerServiceObserver {
    public:
     explicit WorkerCreationObserver(WorkerData* worker_data)
         : worker_data_(worker_data) {
     }
 
    private:
     virtual ~WorkerCreationObserver() {}
 
     virtual void WorkerCreated (
-        WorkerProcessHost* process,
-        const WorkerProcessHost::WorkerInstance& instance) OVERRIDE {
-      worker_data_->worker_process_id = process->GetData().id;
-      worker_data_->worker_route_id = instance.worker_route_id();
+        const GURL& url,
+        const string16& name,
+        int process_id,
+        int route_id) OVERRIDE {
+      worker_data_->worker_process_id = process_id;
+      worker_data_->worker_route_id = route_id;
       WorkerService::GetInstance()->RemoveObserver(this);
       BrowserThread::PostTask(BrowserThread::UI, FROM_HERE,
           MessageLoop::QuitClosure());
       delete this;
     }
-    virtual void WorkerDestroyed(
-        WorkerProcessHost*,
-        int worker_route_id) OVERRIDE {}
-    virtual void WorkerContextStarted(
-        WorkerProcessHost*,
-        int worker_route_id) OVERRIDE {}
     scoped_refptr<WorkerData> worker_data_;
   };
 
   class WorkerTerminationObserver : public WorkerServiceObserver {
    public:
     explicit WorkerTerminationObserver(WorkerData* worker_data)
         : worker_data_(worker_data) {
     }
 
    private:
     virtual ~WorkerTerminationObserver() {}
 
-    virtual void WorkerCreated (
-        WorkerProcessHost* process,
-        const WorkerProcessHost::WorkerInstance& instance) OVERRIDE {}
-    virtual void WorkerDestroyed(
-        WorkerProcessHost* process,
-        int worker_route_id) OVERRIDE {
-      ASSERT_EQ(worker_data_->worker_process_id, process->GetData().id);
-      ASSERT_EQ(worker_data_->worker_route_id, worker_route_id);
+    virtual void WorkerDestroyed(int process_id, int route_id) OVERRIDE {
+      ASSERT_EQ(worker_data_->worker_process_id, process_id);
+      ASSERT_EQ(worker_data_->worker_route_id, route_id);
       WorkerService::GetInstance()->RemoveObserver(this);
       BrowserThread::PostTask(BrowserThread::UI, FROM_HERE,
           MessageLoop::QuitClosure());
       delete this;
     }
-    virtual void WorkerContextStarted(
-        WorkerProcessHost*,
-        int worker_route_id) OVERRIDE {}
     scoped_refptr<WorkerData> worker_data_;
   };
 
   void RunTest(const char* test_name, const char* test_page) {
     ASSERT_TRUE(test_server()->Start());
     GURL url = test_server()->GetURL(test_page);
     ui_test_utils::NavigateToURL(browser(), url);
 
     scoped_refptr<WorkerData> worker_data = WaitForFirstSharedWorker();
     OpenDevToolsWindowForSharedWorker(worker_data.get());
     RunTestFunction(window_, test_name);
     CloseDevToolsWindow();
   }
 
-  static void TerminateWorkerOnIOThread(
-      scoped_refptr<WorkerData> worker_data) {
-    for (WorkerProcessHostIterator iter; !iter.Done(); ++iter) {
-      if (iter.GetData().id == worker_data->worker_process_id) {
-        iter->TerminateWorker(worker_data->worker_route_id);
-        WorkerService::GetInstance()->AddObserver(
-            new WorkerTerminationObserver(worker_data));
-        return;
-      }
+  static void TerminateWorkerOnIOThread(scoped_refptr<WorkerData> worker_data) {
+    if (WorkerService::GetInstance()->TerminateWorker(
+            worker_data->worker_process_id, worker_data->worker_route_id)) {
+      WorkerService::GetInstance()->AddObserver(
+          new WorkerTerminationObserver(worker_data));
+      return;
     }
     FAIL() << "Failed to terminate worker.\n";
   }
 
   static void TerminateWorker(scoped_refptr<WorkerData> worker_data) {
     BrowserThread::PostTask(
         BrowserThread::IO, FROM_HERE,
         base::Bind(&TerminateWorkerOnIOThread, worker_data));
     ui_test_utils::RunMessageLoop();
   }
 
   static void WaitForFirstSharedWorkerOnIOThread(
       scoped_refptr<WorkerData> worker_data) {
-    for (WorkerProcessHostIterator iter; !iter.Done(); ++iter) {
-      const WorkerProcessHost::Instances& instances = iter->instances();
-      for (WorkerProcessHost::Instances::const_iterator i = instances.begin();
-           i != instances.end(); ++i) {
-
-        worker_data->worker_process_id = iter.GetData().id;
-        worker_data->worker_route_id = i->worker_route_id();
-        BrowserThread::PostTask(BrowserThread::UI, FROM_HERE,
-            MessageLoop::QuitClosure());
-        return;
-      }
+    std::vector<WorkerService::WorkerInfo> worker_info =
+        WorkerService::GetInstance()->GetWorkers();
+    if (!worker_info.empty()) {
+      worker_data->worker_process_id = worker_info[0].process_id;
+      worker_data->worker_route_id = worker_info[0].route_id;
+      BrowserThread::PostTask(BrowserThread::UI, FROM_HERE,
+          MessageLoop::QuitClosure());
+      return;
     }
 
     WorkerService::GetInstance()->AddObserver(
         new WorkerCreationObserver(worker_data.get()));
   }
 
   static scoped_refptr<WorkerData> WaitForFirstSharedWorker() {
     scoped_refptr<WorkerData> worker_data(new WorkerData());
     BrowserThread::PostTask(
         BrowserThread::IO, FROM_HERE,
@@ skipping 192 matching lines @@
 
     // Reload page to restart the worker.
     ui_test_utils::NavigateToURL(browser(), url);
 
     // Wait until worker script is paused on the debugger statement.
     RunTestFunction(window_, "testPauseInSharedWorkerInitialization");
     CloseDevToolsWindow();
 }
 
 }  // namespace