| OLD | NEW |
| 1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "chrome/browser/task_management/task_manager_tester.h" | 5 #include "chrome/browser/task_management/task_manager_tester.h" |
| 6 | 6 |
| 7 #include "base/memory/ptr_util.h" | 7 #include "base/memory/ptr_util.h" |
| 8 #include "chrome/browser/browser_process.h" | 8 #include "chrome/browser/browser_process.h" |
| 9 #include "chrome/browser/profiles/profile.h" | 9 #include "chrome/browser/profiles/profile.h" |
| 10 #include "chrome/browser/sessions/session_tab_helper.h" | 10 #include "chrome/browser/sessions/session_tab_helper.h" |
| 11 #include "chrome/browser/task_management/task_manager_interface.h" | 11 #include "chrome/browser/task_management/task_manager_interface.h" |
| 12 #include "chrome/browser/task_manager/legacy_task_manager_tester.h" | |
| 13 #include "chrome/browser/ui/browser_dialogs.h" | 12 #include "chrome/browser/ui/browser_dialogs.h" |
| 14 #include "chrome/browser/ui/task_manager/task_manager_table_model.h" | 13 #include "chrome/browser/ui/task_manager/task_manager_table_model.h" |
| 15 #include "chrome/common/chrome_switches.h" | 14 #include "chrome/common/chrome_switches.h" |
| 16 #include "chrome/grit/generated_resources.h" | 15 #include "chrome/grit/generated_resources.h" |
| 17 #include "testing/gtest/include/gtest/gtest.h" | 16 #include "testing/gtest/include/gtest/gtest.h" |
| 18 #include "ui/base/models/table_model_observer.h" | 17 #include "ui/base/models/table_model_observer.h" |
| 19 | 18 |
| 20 namespace task_management { | 19 namespace task_management { |
| 21 | 20 |
| 22 namespace { | 21 namespace { |
| 23 | 22 |
| 24 // Returns whether chrome::ShowTaskManager() will, for the current platform and | |
| 25 // command line, show a view backed by a task_management::TaskManagerTableModel. | |
| 26 bool IsNewTaskManagerViewEnabled() { | |
| 27 #if defined(OS_MACOSX) | |
| 28 if (!chrome::ToolkitViewsDialogsEnabled()) | |
| 29 return false; | |
| 30 #endif | |
| 31 return switches::NewTaskManagerEnabled(); | |
| 32 } | |
| 33 | |
| 34 // Temporarily intercepts the calls between a TableModel and its Observer, | 23 // Temporarily intercepts the calls between a TableModel and its Observer, |
| 35 // running |callback| whenever anything happens. | 24 // running |callback| whenever anything happens. |
| 36 class ScopedInterceptTableModelObserver : public ui::TableModelObserver { | 25 class ScopedInterceptTableModelObserver : public ui::TableModelObserver { |
| 37 public: | 26 public: |
| 38 ScopedInterceptTableModelObserver( | 27 ScopedInterceptTableModelObserver( |
| 39 ui::TableModel* model_to_intercept, | 28 ui::TableModel* model_to_intercept, |
| 40 ui::TableModelObserver* real_table_model_observer, | 29 ui::TableModelObserver* real_table_model_observer, |
| 41 const base::Closure& callback) | 30 const base::Closure& callback) |
| 42 : model_to_intercept_(model_to_intercept), | 31 : model_to_intercept_(model_to_intercept), |
| 43 real_table_model_observer_(real_table_model_observer), | 32 real_table_model_observer_(real_table_model_observer), |
| (...skipping 105 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 149 | 138 |
| 150 void Kill(int row) override { model_->KillTask(row); } | 139 void Kill(int row) override { model_->KillTask(row); } |
| 151 | 140 |
| 152 private: | 141 private: |
| 153 task_management::TaskManagerInterface* task_manager() { | 142 task_management::TaskManagerInterface* task_manager() { |
| 154 return model_->observed_task_manager(); | 143 return model_->observed_task_manager(); |
| 155 } | 144 } |
| 156 | 145 |
| 157 // Returns the TaskManagerTableModel for the the visible NewTaskManagerView. | 146 // Returns the TaskManagerTableModel for the the visible NewTaskManagerView. |
| 158 static task_management::TaskManagerTableModel* GetRealModel() { | 147 static task_management::TaskManagerTableModel* GetRealModel() { |
| 159 CHECK(IsNewTaskManagerViewEnabled()); | |
| 160 // This downcast is safe, as long as the new task manager is enabled. | 148 // This downcast is safe, as long as the new task manager is enabled. |
| 161 task_management::TaskManagerTableModel* result = | 149 task_management::TaskManagerTableModel* result = |
| 162 static_cast<task_management::TaskManagerTableModel*>( | 150 static_cast<task_management::TaskManagerTableModel*>( |
| 163 chrome::ShowTaskManager(nullptr)); | 151 chrome::ShowTaskManager(nullptr)); |
| 164 return result; | 152 return result; |
| 165 } | 153 } |
| 166 | 154 |
| 167 task_management::TaskManagerTableModel* model_; | 155 task_management::TaskManagerTableModel* model_; |
| 168 std::unique_ptr<ScopedInterceptTableModelObserver> interceptor_; | 156 std::unique_ptr<ScopedInterceptTableModelObserver> interceptor_; |
| 169 }; | 157 }; |
| 170 | 158 |
| 171 // static | 159 // static |
| 172 std::unique_ptr<TaskManagerTester> TaskManagerTester::Create( | 160 std::unique_ptr<TaskManagerTester> TaskManagerTester::CreateDefault( |
| 173 const base::Closure& callback) { | 161 const base::Closure& callback) { |
| 174 if (IsNewTaskManagerViewEnabled()) | 162 return base::WrapUnique(new TaskManagerTesterImpl(callback)); |
| 175 return base::WrapUnique(new TaskManagerTesterImpl(callback)); | |
| 176 else | |
| 177 return task_manager::CreateLegacyTaskManagerTester(callback); | |
| 178 } | 163 } |
| 179 | 164 |
| 180 } // namespace task_management | 165 } // namespace task_management |
| OLD | NEW |