123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112 |
- // 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<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
|