1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73
|
// Copyright 2024 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/task_environment.h"
#include "base/test/test_trace_processor.h"
#include "base/test/trace_test_utils.h"
#include "base/trace_event/trace_event.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "third_party/perfetto/include/perfetto/tracing/tracing.h"
namespace base::test {
class TestTraceProcessorExample : public ::testing::Test {
private:
base::test::TracingEnvironment tracing_environment_;
base::test::TaskEnvironment task_environment_;
};
TEST_F(TestTraceProcessorExample, Basic) {
TestTraceProcessor test_trace_processor;
test_trace_processor.StartTrace("");
{
// A simple trace event inside of a scope so it gets flushed properly.
TRACE_EVENT("test_category", "test_event");
}
auto status = test_trace_processor.StopAndParseTrace();
ASSERT_TRUE(status.ok()) << status.message();
auto result = test_trace_processor.RunQuery(R"(
SELECT
name
FROM slice
WHERE category = 'test_category'
)");
ASSERT_TRUE(result.has_value()) << result.error();
EXPECT_THAT(result.value(),
::testing::ElementsAre(std::vector<std::string>{"name"},
std::vector<std::string>{"test_event"}));
}
TEST_F(TestTraceProcessorExample, BasicTraceConfig) {
TestTraceProcessor test_trace_processor;
// Start tracing with a category filter string set in the trace config.
test_trace_processor.StartTrace(base::test::DefaultTraceConfig(
"test_category", /*privacy_filtering=*/false));
{
// A simple trace event inside of a scope so it gets flushed properly.
TRACE_EVENT("test_category", "test_event");
}
auto status = test_trace_processor.StopAndParseTrace();
ASSERT_TRUE(status.ok()) << status.message();
auto result = test_trace_processor.RunQuery(R"(
SELECT
name
FROM slice
)");
ASSERT_TRUE(result.has_value()) << result.error();
EXPECT_THAT(result.value(),
::testing::ElementsAre(std::vector<std::string>{"name"},
std::vector<std::string>{"test_event"}));
}
} // namespace base::test
|