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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
|
// Copyright 2020 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "components/performance_manager/test_support/test_harness_helper.h"
#include <utility>
#include "base/check.h"
#include "base/functional/callback.h"
#include "components/performance_manager/embedder/performance_manager_registry.h"
#include "components/performance_manager/performance_manager_impl.h"
#include "content/public/browser/browser_context.h"
namespace performance_manager {
PerformanceManagerTestHarnessHelper::PerformanceManagerTestHarnessHelper() =
default;
PerformanceManagerTestHarnessHelper::~PerformanceManagerTestHarnessHelper() =
default;
void PerformanceManagerTestHarnessHelper::SetUp() {
// Allow this to be called multiple times.
if (perf_man_.get())
return;
perf_man_ = PerformanceManagerImpl::Create();
GraphImpl* graph = PerformanceManagerImpl::GetGraphImpl();
graph_features_.ConfigureGraph(graph);
if (graph_impl_callback_) {
std::move(graph_impl_callback_).Run(graph);
}
registry_ = PerformanceManagerRegistry::Create();
}
void PerformanceManagerTestHarnessHelper::TearDown() {
// Have the performance manager destroy itself.
registry_->TearDown();
registry_.reset();
PerformanceManagerImpl::Destroy(std::move(perf_man_));
}
void PerformanceManagerTestHarnessHelper::OnWebContentsCreated(
content::WebContents* contents) {
registry_->CreatePageNodeForWebContents(contents);
}
void PerformanceManagerTestHarnessHelper::OnBrowserContextAdded(
content::BrowserContext* browser_context) {
registry_->NotifyBrowserContextAdded(browser_context);
}
void PerformanceManagerTestHarnessHelper::OnBrowserContextRemoved(
content::BrowserContext* browser_context) {
registry_->NotifyBrowserContextRemoved(browser_context);
}
GraphFeatures& PerformanceManagerTestHarnessHelper::GetGraphFeatures() {
// Calling GetGraphFeatures() after the performance manager is initialized
// would have no effect, so it is restricted here.
CHECK(!perf_man_);
return graph_features_;
}
} // namespace performance_manager
|
