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