| OLD | NEW |
| (Empty) |
| 1 // Copyright 2015 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #ifndef CHROME_BROWSER_TASK_MANAGEMENT_TASK_MANAGEMENT_BROWSERTEST_UTIL_H_ | |
| 6 #define CHROME_BROWSER_TASK_MANAGEMENT_TASK_MANAGEMENT_BROWSERTEST_UTIL_H_ | |
| 7 | |
| 8 #include "base/macros.h" | |
| 9 #include "chrome/browser/task_management/providers/task_provider_observer.h" | |
| 10 #include "chrome/browser/task_management/providers/web_contents/web_contents_tag
s_manager.h" | |
| 11 #include "chrome/browser/task_management/providers/web_contents/web_contents_tas
k_provider.h" | |
| 12 | |
| 13 namespace task_management { | |
| 14 | |
| 15 // Defines a test class that will act as a task manager that is designed to | |
| 16 // only observe the WebContents-based tasks. | |
| 17 class MockWebContentsTaskManager : public TaskProviderObserver { | |
| 18 public: | |
| 19 MockWebContentsTaskManager(); | |
| 20 ~MockWebContentsTaskManager() override; | |
| 21 | |
| 22 // task_management::TaskProviderObserver: | |
| 23 void TaskAdded(Task* task) override; | |
| 24 void TaskRemoved(Task* task) override; | |
| 25 | |
| 26 // Start / Stop observing the WebContentsTaskProvider. | |
| 27 void StartObserving(); | |
| 28 void StopObserving(); | |
| 29 | |
| 30 const std::vector<Task*>& tasks() const { return tasks_; } | |
| 31 | |
| 32 private: | |
| 33 std::vector<Task*> tasks_; | |
| 34 WebContentsTaskProvider provider_; | |
| 35 | |
| 36 DISALLOW_COPY_AND_ASSIGN(MockWebContentsTaskManager); | |
| 37 }; | |
| 38 | |
| 39 } // namespace task_management | |
| 40 | |
| 41 #endif // CHROME_BROWSER_TASK_MANAGEMENT_TASK_MANAGEMENT_BROWSERTEST_UTIL_H_ | |
| OLD | NEW |