112 lines
3.8 KiB
C++
112 lines
3.8 KiB
C++
// 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/tracing/perfetto_platform.h"
|
|
#include "third_party/perfetto/include/perfetto/tracing/core/data_source_config.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
|