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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
|
// Copyright 2021 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "base/test/trace_test_utils.h"
#include "base/no_destructor.h"
#include "base/task/sequenced_task_runner.h"
#include "base/threading/thread_task_runner_handle.h"
#include "base/tracing/perfetto_platform.h"
#include "third_party/perfetto/include/perfetto/tracing.h"
namespace base {
namespace test {
namespace {
// A proxy task runner which can be dynamically pointed to route tasks into a
// different task runner.
class RebindableTaskRunner : public base::SequencedTaskRunner {
public:
RebindableTaskRunner();
void set_task_runner(scoped_refptr<base::SequencedTaskRunner> task_runner) {
task_runner_ = task_runner;
}
// base::SequencedTaskRunner implementation.
bool PostDelayedTask(const base::Location& from_here,
base::OnceClosure task,
base::TimeDelta delay) override;
bool PostNonNestableDelayedTask(const base::Location& from_here,
base::OnceClosure task,
base::TimeDelta delay) override;
bool RunsTasksInCurrentSequence() const override;
private:
~RebindableTaskRunner() override;
scoped_refptr<base::SequencedTaskRunner> task_runner_;
};
RebindableTaskRunner::RebindableTaskRunner() = default;
RebindableTaskRunner::~RebindableTaskRunner() = default;
bool RebindableTaskRunner::PostDelayedTask(const base::Location& from_here,
base::OnceClosure task,
base::TimeDelta delay) {
return task_runner_->PostDelayedTask(from_here, std::move(task), delay);
}
bool RebindableTaskRunner::PostNonNestableDelayedTask(
const base::Location& from_here,
base::OnceClosure task,
base::TimeDelta delay) {
return task_runner_->PostNonNestableDelayedTask(from_here, std::move(task),
delay);
}
bool RebindableTaskRunner::RunsTasksInCurrentSequence() const {
return task_runner_->RunsTasksInCurrentSequence();
}
RebindableTaskRunner* GetClientLibTaskRunner() {
static base::NoDestructor<scoped_refptr<RebindableTaskRunner>> task_runner(
MakeRefCounted<RebindableTaskRunner>());
return task_runner.get()->get();
}
} // namespace
TracingEnvironment::TracingEnvironment() {
trace_event::TraceLog::GetInstance()->ResetForTesting();
}
TracingEnvironment::TracingEnvironment(
TaskEnvironment& task_environment,
scoped_refptr<SequencedTaskRunner> task_runner,
tracing::PerfettoPlatform* perfetto_platform)
: task_environment_(&task_environment) {
// Since Perfetto's platform backend can only be initialized once in a
// process, we give it a task runner that can outlive the per-test task
// environment.
auto* client_lib_task_runner = GetClientLibTaskRunner();
if (!perfetto_platform->did_start_task_runner())
perfetto_platform->StartTaskRunner(client_lib_task_runner);
client_lib_task_runner->set_task_runner(std::move(task_runner));
// Wait for any posted construction tasks to execute.
task_environment_->RunUntilIdle();
}
TracingEnvironment::~TracingEnvironment() {
if (task_environment_) {
// Wait for any posted destruction tasks to execute.
task_environment_->RunUntilIdle();
}
perfetto::Tracing::ResetForTesting();
}
// static
perfetto::protos::gen::TraceConfig TracingEnvironment::GetDefaultTraceConfig() {
perfetto::protos::gen::TraceConfig trace_config;
auto* buffer_config = trace_config.add_buffers();
buffer_config->set_size_kb(1024 * 1024);
auto* data_source = trace_config.add_data_sources();
auto* source_config = data_source->mutable_config();
source_config->set_name("track_event");
source_config->set_target_buffer(0);
return trace_config;
}
} // namespace test
} // namespace base
|
