Improving the process model extension API

chrome/browser/task_manager/task_manager_browsertest_util.cc

 // Copyright (c) 2011 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/message_loop.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/task_manager/task_manager.h"
 #include "chrome/browser/task_manager/task_manager_browsertest_util.h"
 #include "chrome/browser/ui/browser.h"
@@ skipping 21 matching lines @@
   }
 
   virtual void OnItemsAdded(int start, int length) OVERRIDE {
     OnResourceChange();
   }
 
   virtual void OnItemsRemoved(int start, int length) OVERRIDE {
     OnResourceChange();
   }
 
+  virtual void OnItemsToBeRemoved(int start, int length) OVERRIDE {
+    OnResourceChange();
+  }
+
  private:
   void OnResourceChange() {
     if (model_->ResourceCount() == target_resource_count_)
       MessageLoopForUI::current()->Quit();
   }
 
   const TaskManagerModel* model_;
   const int target_resource_count_;
 };
 
@@ skipping 25 matching lines @@
   ui_test_utils::WindowedNotificationObserver observer(
       chrome::NOTIFICATION_TASK_MANAGER_WINDOW_READY,
       content::Source<TaskManagerModel>(TaskManager::GetInstance()->model()));
   browser->window()->ShowTaskManager();
   observer.Wait();
 #else
   browser->window()->ShowTaskManager();
 #endif  // defined(WEBUI_TASK_MANAGER)
 }