| /* |
| * Copyright (C) 2018 The Android Open Source Project |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| #include <map> |
| #include <random> |
| #include <string> |
| |
| #include "gtest/gtest.h" |
| #include "perfetto/base/logging.h" |
| #include "perfetto/base/scoped_file.h" |
| #include "perfetto/trace_processor/trace_processor.h" |
| #include "src/base/test/utils.h" |
| #include "src/trace_processor/json_trace_parser.h" |
| |
| namespace perfetto { |
| namespace trace_processor { |
| namespace { |
| |
| class TraceProcessorIntegrationTest : public ::testing::Test { |
| public: |
| TraceProcessorIntegrationTest() |
| : processor_(TraceProcessor::CreateInstance(Config())) {} |
| |
| protected: |
| util::Status LoadTrace(const char* name, int min_chunk_size = 512) { |
| base::ScopedFstream f(fopen(base::GetTestDataPath(name).c_str(), "rb")); |
| std::minstd_rand0 rnd_engine(0); |
| std::uniform_int_distribution<> dist(min_chunk_size, 1024); |
| while (!feof(*f)) { |
| size_t chunk_size = static_cast<size_t>(dist(rnd_engine)); |
| std::unique_ptr<uint8_t[]> buf(new uint8_t[chunk_size]); |
| auto rsize = fread(reinterpret_cast<char*>(buf.get()), 1, chunk_size, *f); |
| auto status = processor_->Parse(std::move(buf), rsize); |
| if (!status.ok()) |
| return status; |
| } |
| processor_->NotifyEndOfFile(); |
| return util::OkStatus(); |
| } |
| |
| TraceProcessor::Iterator Query(const std::string& query) { |
| return processor_->ExecuteQuery(query.c_str()); |
| } |
| |
| private: |
| std::unique_ptr<TraceProcessor> processor_; |
| }; |
| |
| TEST_F(TraceProcessorIntegrationTest, AndroidSchedAndPs) { |
| ASSERT_TRUE(LoadTrace("android_sched_and_ps.pb").ok()); |
| auto it = Query( |
| "select count(*), max(ts) - min(ts) from sched " |
| "where dur != 0 and utid != 0"); |
| ASSERT_TRUE(it.Next()); |
| ASSERT_EQ(it.Get(0).type, SqlValue::kLong); |
| ASSERT_EQ(it.Get(0).long_value, 139787); |
| ASSERT_EQ(it.Get(1).type, SqlValue::kLong); |
| ASSERT_EQ(it.Get(1).long_value, 19684308497); |
| ASSERT_FALSE(it.Next()); |
| } |
| |
| TEST_F(TraceProcessorIntegrationTest, Sfgate) { |
| ASSERT_TRUE(LoadTrace("sfgate.json", strlen("{\"traceEvents\":[")).ok()); |
| auto it = |
| Query("select count(*), max(ts) - min(ts) from slices where utid != 0"); |
| ASSERT_TRUE(it.Next()); |
| ASSERT_EQ(it.Get(0).type, SqlValue::kLong); |
| ASSERT_EQ(it.Get(0).long_value, 39828); |
| ASSERT_EQ(it.Get(1).type, SqlValue::kLong); |
| ASSERT_EQ(it.Get(1).long_value, 40532506000); |
| ASSERT_FALSE(it.Next()); |
| } |
| |
| TEST_F(TraceProcessorIntegrationTest, UnsortedTrace) { |
| ASSERT_TRUE( |
| LoadTrace("unsorted_trace.json", strlen("{\"traceEvents\":[")).ok()); |
| auto it = Query("select ts, depth from slices order by ts"); |
| ASSERT_TRUE(it.Next()); |
| ASSERT_EQ(it.Get(0).type, SqlValue::kLong); |
| ASSERT_EQ(it.Get(0).long_value, 50000); |
| ASSERT_EQ(it.Get(1).type, SqlValue::kLong); |
| ASSERT_EQ(it.Get(1).long_value, 0); |
| ASSERT_TRUE(it.Next()); |
| ASSERT_EQ(it.Get(0).type, SqlValue::kLong); |
| ASSERT_EQ(it.Get(0).long_value, 100000); |
| ASSERT_EQ(it.Get(1).type, SqlValue::kLong); |
| ASSERT_EQ(it.Get(1).long_value, 1); |
| ASSERT_FALSE(it.Next()); |
| } |
| |
| TEST_F(TraceProcessorIntegrationTest, TraceBounds) { |
| ASSERT_TRUE(LoadTrace("android_sched_and_ps.pb").ok()); |
| auto it = Query("select start_ts, end_ts from trace_bounds"); |
| ASSERT_TRUE(it.Next()); |
| ASSERT_EQ(it.Get(0).type, SqlValue::kLong); |
| ASSERT_EQ(it.Get(0).long_value, 81473009948313); |
| ASSERT_EQ(it.Get(1).type, SqlValue::kLong); |
| ASSERT_EQ(it.Get(1).long_value, 81492700784311); |
| ASSERT_FALSE(it.Next()); |
| } |
| |
| // TODO(hjd): Add trace to test_data. |
| TEST_F(TraceProcessorIntegrationTest, DISABLED_AndroidBuildTrace) { |
| ASSERT_TRUE(LoadTrace("android_build_trace.json", strlen("[\n{")).ok()); |
| } |
| |
| } // namespace |
| } // namespace trace_processor |
| } // namespace perfetto |