| Index: chrome/browser/task_manager/task_manager_browsertest_util.cc
|
| ===================================================================
|
| --- chrome/browser/task_manager/task_manager_browsertest_util.cc (revision 95017)
|
| +++ chrome/browser/task_manager/task_manager_browsertest_util.cc (working copy)
|
| @@ -1,126 +0,0 @@
|
| -// 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"
|
| -#include "chrome/browser/ui/browser_window.h"
|
| -#include "chrome/common/chrome_notification_types.h"
|
| -#include "chrome/test/ui_test_utils.h"
|
| -#include "content/common/notification_source.h"
|
| -
|
| -namespace {
|
| -
|
| -class ResourceChangeObserver : public TaskManagerModelObserver {
|
| - public:
|
| - ResourceChangeObserver(const TaskManagerModel* model,
|
| - int target_resource_count)
|
| - : model_(model),
|
| - target_resource_count_(target_resource_count) {
|
| - }
|
| -
|
| - virtual void OnModelChanged() OVERRIDE {
|
| - OnResourceChange();
|
| - }
|
| -
|
| - virtual void OnItemsChanged(int start, int length) OVERRIDE {
|
| - OnResourceChange();
|
| - }
|
| -
|
| - virtual void OnItemsAdded(int start, int length) OVERRIDE {
|
| - OnResourceChange();
|
| - }
|
| -
|
| - virtual void OnItemsRemoved(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_;
|
| -};
|
| -
|
| -// Helper class used to wait for a TaskManager to be ready.
|
| -class TaskManagerReadyListener : public NotificationObserver {
|
| - public:
|
| - explicit TaskManagerReadyListener(TaskManagerModel* model) {
|
| - registrar_.Add(this, chrome::NOTIFICATION_TASK_MANAGER_WINDOW_READY,
|
| - Source<TaskManagerModel>(model));
|
| - }
|
| -
|
| - virtual void Observe(int type,
|
| - const NotificationSource& source,
|
| - const NotificationDetails& details) {
|
| - // Quit once the BackgroundContents has been loaded.
|
| - if (type == chrome::NOTIFICATION_TASK_MANAGER_WINDOW_READY)
|
| - MessageLoopForUI::current()->Quit();
|
| - }
|
| - private:
|
| - NotificationRegistrar registrar_;
|
| -};
|
| -
|
| -// Helper class used to wait for a BackgroundContents to finish loading.
|
| -class BackgroundContentsListener : public NotificationObserver {
|
| - public:
|
| - explicit BackgroundContentsListener(Profile* profile) {
|
| - registrar_.Add(this, chrome::NOTIFICATION_BACKGROUND_CONTENTS_NAVIGATED,
|
| - Source<Profile>(profile));
|
| - }
|
| -
|
| - virtual void Observe(int type,
|
| - const NotificationSource& source,
|
| - const NotificationDetails& details) {
|
| - // Quit once the BackgroundContents has been loaded.
|
| - if (type == chrome::NOTIFICATION_BACKGROUND_CONTENTS_NAVIGATED)
|
| - MessageLoopForUI::current()->Quit();
|
| - }
|
| -
|
| - void ShowTaskManagerAndWait() {
|
| - ui_test_utils::RunMessageLoop();
|
| - }
|
| -
|
| - private:
|
| - NotificationRegistrar registrar_;
|
| -};
|
| -
|
| -} // namespace
|
| -
|
| -// static
|
| -void TaskManagerBrowserTestUtil::WaitForResourceChange(int target_count) {
|
| - TaskManagerModel* model = TaskManager::GetInstance()->model();
|
| -
|
| - if (model->ResourceCount() == target_count)
|
| - return;
|
| - ResourceChangeObserver observer(model, target_count);
|
| - model->AddObserver(&observer);
|
| - ui_test_utils::RunMessageLoop();
|
| - model->RemoveObserver(&observer);
|
| -}
|
| -
|
| -// static
|
| -void TaskManagerBrowserTestUtil::ShowTaskManagerAndWaitForReady(
|
| - Browser* browser) {
|
| -#if defined(WEBUI_TASK_MANAGER)
|
| - TaskManagerReadyListener listener(TaskManager::GetInstance()->model());
|
| - browser->window()->ShowTaskManager();
|
| - ui_test_utils::RunMessageLoop();
|
| -#else
|
| - browser->window()->ShowTaskManager();
|
| -#endif // defined(WEBUI_TASK_MANAGER)
|
| -}
|
| -
|
| -// static
|
| -void TaskManagerBrowserTestUtil::WaitForBackgroundContents(Browser* browser) {
|
| - BackgroundContentsListener listener(browser->profile());
|
| - ui_test_utils::RunMessageLoop();
|
| -}
|
| -
|
|
|