| OLD | NEW |
| 1 // Copyright 2015 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_management_browsertest_util.h" | 5 #include "chrome/browser/task_management/mock_web_contents_task_manager.h" |
| 6 | 6 |
| 7 #include "base/stl_util.h" | 7 #include "base/stl_util.h" |
| 8 #include "build/build_config.h" | 8 #include "build/build_config.h" |
| 9 | 9 |
| 10 #if defined(OS_CHROMEOS) | 10 #if defined(OS_CHROMEOS) |
| 11 #include "chrome/browser/chromeos/resource_reporter/resource_reporter.h" | 11 #include "chrome/browser/chromeos/resource_reporter/resource_reporter.h" |
| 12 #endif // defined(OS_CHROMEOS) | 12 #endif // defined(OS_CHROMEOS) |
| 13 | 13 |
| 14 namespace task_management { | 14 namespace task_management { |
| 15 | 15 |
| 16 MockWebContentsTaskManager::MockWebContentsTaskManager() { | 16 MockWebContentsTaskManager::MockWebContentsTaskManager() {} |
| 17 } | |
| 18 | 17 |
| 19 MockWebContentsTaskManager::~MockWebContentsTaskManager() { | 18 MockWebContentsTaskManager::~MockWebContentsTaskManager() {} |
| 20 } | |
| 21 | 19 |
| 22 void MockWebContentsTaskManager::TaskAdded(Task* task) { | 20 void MockWebContentsTaskManager::TaskAdded(Task* task) { |
| 23 DCHECK(task); | 21 DCHECK(task); |
| 24 DCHECK(!ContainsValue(tasks_, task)); | 22 DCHECK(!ContainsValue(tasks_, task)); |
| 25 tasks_.push_back(task); | 23 tasks_.push_back(task); |
| 26 } | 24 } |
| 27 | 25 |
| 28 void MockWebContentsTaskManager::TaskRemoved(Task* task) { | 26 void MockWebContentsTaskManager::TaskRemoved(Task* task) { |
| 29 DCHECK(task); | 27 DCHECK(task); |
| 30 DCHECK(ContainsValue(tasks_, task)); | 28 DCHECK(ContainsValue(tasks_, task)); |
| 31 tasks_.erase(std::find(tasks_.begin(), tasks_.end(), task)); | 29 tasks_.erase(std::find(tasks_.begin(), tasks_.end(), task)); |
| 32 } | 30 } |
| 33 | 31 |
| 34 void MockWebContentsTaskManager::StartObserving() { | 32 void MockWebContentsTaskManager::StartObserving() { |
| 35 #if defined(OS_CHROMEOS) | 33 #if defined(OS_CHROMEOS) |
| 36 // On ChromeOS, the ResourceReporter needs to be turned off so as not to | 34 // On ChromeOS, the ResourceReporter needs to be turned off so as not to |
| 37 // interfere with the tests. | 35 // interfere with the tests. |
| 38 chromeos::ResourceReporter::GetInstance()->StopMonitoring(); | 36 chromeos::ResourceReporter::GetInstance()->StopMonitoring(); |
| 39 #endif // defined(OS_CHROMEOS) | 37 #endif // defined(OS_CHROMEOS) |
| 40 | 38 |
| 41 provider_.SetObserver(this); | 39 provider_.SetObserver(this); |
| 42 } | 40 } |
| 43 | 41 |
| 44 void MockWebContentsTaskManager::StopObserving() { | 42 void MockWebContentsTaskManager::StopObserving() { |
| 45 provider_.ClearObserver(); | 43 provider_.ClearObserver(); |
| 46 } | 44 } |
| 47 | 45 |
| 48 } // namespace task_management | 46 } // namespace task_management |
| OLD | NEW |