trace_test_utils.cc 3.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112
  1. // Copyright 2021 The Chromium Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style license that can be
  3. // found in the LICENSE file.
  4. #include "base/test/trace_test_utils.h"
  5. #include "base/no_destructor.h"
  6. #include "base/task/sequenced_task_runner.h"
  7. #include "base/threading/thread_task_runner_handle.h"
  8. #include "base/tracing/perfetto_platform.h"
  9. #include "third_party/perfetto/include/perfetto/tracing.h"
  10. namespace base {
  11. namespace test {
  12. namespace {
  13. // A proxy task runner which can be dynamically pointed to route tasks into a
  14. // different task runner.
  15. class RebindableTaskRunner : public base::SequencedTaskRunner {
  16. public:
  17. RebindableTaskRunner();
  18. void set_task_runner(scoped_refptr<base::SequencedTaskRunner> task_runner) {
  19. task_runner_ = task_runner;
  20. }
  21. // base::SequencedTaskRunner implementation.
  22. bool PostDelayedTask(const base::Location& from_here,
  23. base::OnceClosure task,
  24. base::TimeDelta delay) override;
  25. bool PostNonNestableDelayedTask(const base::Location& from_here,
  26. base::OnceClosure task,
  27. base::TimeDelta delay) override;
  28. bool RunsTasksInCurrentSequence() const override;
  29. private:
  30. ~RebindableTaskRunner() override;
  31. scoped_refptr<base::SequencedTaskRunner> task_runner_;
  32. };
  33. RebindableTaskRunner::RebindableTaskRunner() = default;
  34. RebindableTaskRunner::~RebindableTaskRunner() = default;
  35. bool RebindableTaskRunner::PostDelayedTask(const base::Location& from_here,
  36. base::OnceClosure task,
  37. base::TimeDelta delay) {
  38. return task_runner_->PostDelayedTask(from_here, std::move(task), delay);
  39. }
  40. bool RebindableTaskRunner::PostNonNestableDelayedTask(
  41. const base::Location& from_here,
  42. base::OnceClosure task,
  43. base::TimeDelta delay) {
  44. return task_runner_->PostNonNestableDelayedTask(from_here, std::move(task),
  45. delay);
  46. }
  47. bool RebindableTaskRunner::RunsTasksInCurrentSequence() const {
  48. return task_runner_->RunsTasksInCurrentSequence();
  49. }
  50. RebindableTaskRunner* GetClientLibTaskRunner() {
  51. static base::NoDestructor<scoped_refptr<RebindableTaskRunner>> task_runner(
  52. MakeRefCounted<RebindableTaskRunner>());
  53. return task_runner.get()->get();
  54. }
  55. } // namespace
  56. TracingEnvironment::TracingEnvironment() {
  57. trace_event::TraceLog::GetInstance()->ResetForTesting();
  58. }
  59. TracingEnvironment::TracingEnvironment(
  60. TaskEnvironment& task_environment,
  61. scoped_refptr<SequencedTaskRunner> task_runner,
  62. tracing::PerfettoPlatform* perfetto_platform)
  63. : task_environment_(&task_environment) {
  64. // Since Perfetto's platform backend can only be initialized once in a
  65. // process, we give it a task runner that can outlive the per-test task
  66. // environment.
  67. auto* client_lib_task_runner = GetClientLibTaskRunner();
  68. if (!perfetto_platform->did_start_task_runner())
  69. perfetto_platform->StartTaskRunner(client_lib_task_runner);
  70. client_lib_task_runner->set_task_runner(std::move(task_runner));
  71. // Wait for any posted construction tasks to execute.
  72. task_environment_->RunUntilIdle();
  73. }
  74. TracingEnvironment::~TracingEnvironment() {
  75. if (task_environment_) {
  76. // Wait for any posted destruction tasks to execute.
  77. task_environment_->RunUntilIdle();
  78. }
  79. perfetto::Tracing::ResetForTesting();
  80. }
  81. // static
  82. perfetto::protos::gen::TraceConfig TracingEnvironment::GetDefaultTraceConfig() {
  83. perfetto::protos::gen::TraceConfig trace_config;
  84. auto* buffer_config = trace_config.add_buffers();
  85. buffer_config->set_size_kb(1024 * 1024);
  86. auto* data_source = trace_config.add_data_sources();
  87. auto* source_config = data_source->mutable_config();
  88. source_config->set_name("track_event");
  89. source_config->set_target_buffer(0);
  90. return trace_config;
  91. }
  92. } // namespace test
  93. } // namespace base