blob: 674f7d8195d7fd6fd825f741fcfd2f10f6881972 [file] [log] [blame]
// 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