// Copyright 2021 The Chromium Authors. All rights reserved. // 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 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 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> task_runner( MakeRefCounted()); return task_runner.get()->get(); } } // namespace TracingEnvironment::TracingEnvironment() { trace_event::TraceLog::GetInstance()->ResetForTesting(); } TracingEnvironment::TracingEnvironment( TaskEnvironment& task_environment, scoped_refptr 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