blob: 344e17b0498336fd77e39afb469e37c6000ad1ee [file] [log] [blame]
/*
* Copyright (C) 2019 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 <cstdint>
#include <utility>
#include <vector>
#include "src/trace_processor/containers/string_pool.h"
#include "src/trace_processor/db/column.h"
#include "src/trace_processor/db/column/types.h"
#include "src/trace_processor/db/column_storage.h"
#include "src/trace_processor/tables/py_tables_unittest_py.h"
#include "test/gtest_and_gmock.h"
namespace perfetto::trace_processor::tables {
TestEventTable::~TestEventTable() = default;
TestEventChildTable::~TestEventChildTable() = default;
TestSliceTable::~TestSliceTable() = default;
TestArgsTable::~TestArgsTable() = default;
namespace {
class PyTablesUnittest : public ::testing::Test {
protected:
StringPool pool_;
TestEventTable event_{&pool_};
TestEventChildTable event_child_{&pool_, &event_};
TestSliceTable slice_{&pool_, &event_};
TestArgsTable args_{&pool_};
};
TEST_F(PyTablesUnittest, EventTableProprties) {
ASSERT_STREQ(TestEventTable::Name(), "event");
ASSERT_EQ(TestEventTable::ColumnIndex::id, 0u);
ASSERT_EQ(TestEventTable::ColumnIndex::type, 1u);
ASSERT_EQ(TestEventTable::ColumnIndex::ts, 2u);
ASSERT_EQ(TestEventTable::ColumnIndex::arg_set_id, 3u);
ASSERT_EQ(TestEventTable::ColumnFlag::ts,
ColumnLegacy::Flag::kSorted | ColumnLegacy::Flag::kNonNull);
ASSERT_EQ(TestEventTable::ColumnFlag::arg_set_id,
ColumnLegacy::Flag::kNonNull);
}
TEST_F(PyTablesUnittest, ArgsTableProprties) {
ASSERT_STREQ(TestArgsTable::Name(), "args");
ASSERT_EQ(TestArgsTable::ColumnIndex::id, 0u);
ASSERT_EQ(TestArgsTable::ColumnIndex::type, 1u);
ASSERT_EQ(TestArgsTable::ColumnIndex::arg_set_id, 2u);
ASSERT_EQ(TestArgsTable::ColumnFlag::arg_set_id,
ColumnLegacy::Flag::kSorted | ColumnLegacy::Flag::kSetId |
ColumnLegacy::Flag::kNonNull);
}
TEST_F(PyTablesUnittest, InsertEvent) {
event_.Insert(TestEventTable::Row(100, 0));
ASSERT_EQ(pool_.Get(event_[0].type()).ToStdString(), "event");
ASSERT_EQ(event_[0].ts(), 100);
ASSERT_EQ(event_[0].arg_set_id(), 0u);
}
TEST_F(PyTablesUnittest, InsertEventSpecifyCols) {
TestEventTable::Row row;
row.ts = 100;
row.arg_set_id = 0;
event_.Insert(row);
ASSERT_EQ(pool_.Get(event_[0].type()).ToStdString(), "event");
ASSERT_EQ(event_[0].ts(), 100);
ASSERT_EQ(event_[0].arg_set_id(), 0u);
}
TEST_F(PyTablesUnittest, MutableColumn) {
event_.Insert(TestEventTable::Row(100, 0));
ASSERT_EQ(event_[0].ts(), 100);
ASSERT_EQ(event_[0].arg_set_id(), 0u);
}
TEST_F(PyTablesUnittest, ShrinkToFit) {
event_.Insert(TestEventTable::Row(100, 0));
event_.ShrinkToFit();
// Unfortunately given the loose restrictions on shrink_to_fit provided by the
// standard library, we cannot really assert anything. Just call the method to
// ensure it doesn't cause crashes.
}
TEST_F(PyTablesUnittest, FindById) {
auto id_and_row = event_.Insert(TestEventTable::Row(100, 0));
auto row_ref = event_.FindById(id_and_row.id);
ASSERT_EQ(row_ref->ToRowNumber().row_number(), id_and_row.row);
ASSERT_EQ(row_ref->id(), id_and_row.id);
ASSERT_EQ(row_ref->ts(), 100);
ASSERT_EQ(row_ref->arg_set_id(), 0u);
}
TEST_F(PyTablesUnittest, ChildFindById) {
event_.Insert(TestEventTable::Row(50, 0));
auto id_and_row = slice_.Insert(TestSliceTable::Row(100, 0, 10));
auto row_ref = slice_.FindById(id_and_row.id);
ASSERT_EQ(row_ref->ToRowNumber().row_number(), id_and_row.row);
ASSERT_EQ(row_ref->id(), id_and_row.id);
ASSERT_EQ(row_ref->ts(), 100);
ASSERT_EQ(row_ref->arg_set_id(), 0u);
ASSERT_EQ(row_ref->dur(), 10u);
}
TEST_F(PyTablesUnittest, ChildTableStatics) {
ASSERT_EQ(TestSliceTable::ColumnFlag::dur, ColumnLegacy::Flag::kNonNull);
ASSERT_EQ(TestSliceTable::ColumnIndex::id, 0u);
ASSERT_EQ(TestSliceTable::ColumnIndex::type, 1u);
ASSERT_EQ(TestSliceTable::ColumnIndex::ts, 2u);
ASSERT_EQ(TestSliceTable::ColumnIndex::arg_set_id, 3u);
ASSERT_EQ(TestSliceTable::ColumnIndex::dur, 4u);
}
TEST_F(PyTablesUnittest, ParentAndChildInsert) {
event_.Insert(TestEventTable::Row(50, 0));
slice_.Insert(TestSliceTable::Row(100, 1, 10));
event_.Insert(TestEventTable::Row(150, 2));
slice_.Insert(TestSliceTable::Row(200, 3, 20));
ASSERT_EQ(event_.row_count(), 4u);
ASSERT_EQ(event_[0].id(), TestEventTable::Id{0});
ASSERT_EQ(pool_.Get(event_[0].type()), "event");
ASSERT_EQ(event_[0].ts(), 50);
ASSERT_EQ(event_[1].id(), TestEventTable::Id{1});
ASSERT_EQ(pool_.Get(event_[1].type()), "slice");
ASSERT_EQ(event_[1].ts(), 100);
ASSERT_EQ(event_[2].id(), TestEventTable::Id{2});
ASSERT_EQ(pool_.Get(event_[2].type()), "event");
ASSERT_EQ(event_[2].ts(), 150);
ASSERT_EQ(event_[3].id(), TestEventTable::Id{3});
ASSERT_EQ(pool_.Get(event_[3].type()), "slice");
ASSERT_EQ(event_[3].ts(), 200);
ASSERT_EQ(slice_.row_count(), 2u);
ASSERT_EQ(slice_[0].id(), TestEventTable::Id{1});
ASSERT_EQ(pool_.Get(slice_[0].type()), "slice");
ASSERT_EQ(slice_[0].ts(), 100);
ASSERT_EQ(slice_[0].dur(), 10);
ASSERT_EQ(slice_[1].id(), TestEventTable::Id{3});
ASSERT_EQ(pool_.Get(slice_[1].type()), "slice");
ASSERT_EQ(slice_[1].ts(), 200);
ASSERT_EQ(slice_[1].dur(), 20);
}
TEST_F(PyTablesUnittest, Extend) {
event_.Insert(TestEventTable::Row(50, 0));
event_.Insert(TestEventTable::Row(100, 1));
event_.Insert(TestEventTable::Row(150, 2));
ColumnStorage<int64_t> dur;
dur.Append(512);
dur.Append(1024);
dur.Append(2048);
auto slice_ext = TestSliceTable::ExtendParent(event_, std::move(dur));
ASSERT_EQ(slice_ext->row_count(), 3u);
ASSERT_EQ((*slice_ext)[0].ts(), 50);
ASSERT_EQ((*slice_ext)[0].dur(), 512);
ASSERT_EQ((*slice_ext)[1].ts(), 100);
ASSERT_EQ((*slice_ext)[1].dur(), 1024);
ASSERT_EQ((*slice_ext)[2].ts(), 150);
ASSERT_EQ((*slice_ext)[2].dur(), 2048);
}
TEST_F(PyTablesUnittest, SelectAndExtend) {
event_.Insert(TestEventTable::Row(50, 0));
event_.Insert(TestEventTable::Row(100, 1));
event_.Insert(TestEventTable::Row(150, 2));
std::vector<TestEventTable::RowNumber> rows;
rows.emplace_back(1);
ColumnStorage<int64_t> dur;
dur.Append(1024);
auto slice_ext = TestSliceTable::SelectAndExtendParent(
event_, std::move(rows), std::move(dur));
ASSERT_EQ(slice_ext->row_count(), 1u);
ASSERT_EQ((*slice_ext)[0].ts(), 100);
ASSERT_EQ((*slice_ext)[0].dur(), 1024);
}
TEST_F(PyTablesUnittest, SetIdColumns) {
StringPool pool;
TestArgsTable table{&pool};
table.Insert(TestArgsTable::Row(0, 100));
table.Insert(TestArgsTable::Row(0, 200));
table.Insert(TestArgsTable::Row(2, 200));
table.Insert(TestArgsTable::Row(3, 300));
table.Insert(TestArgsTable::Row(4, 200));
table.Insert(TestArgsTable::Row(4, 500));
table.Insert(TestArgsTable::Row(4, 900));
table.Insert(TestArgsTable::Row(4, 200));
table.Insert(TestArgsTable::Row(8, 400));
ASSERT_EQ(table.row_count(), 9u);
ASSERT_TRUE(table.arg_set_id().IsSetId());
// Verify that not-present ids are not returned.
{
static constexpr uint32_t kFilterArgSetId = 1;
Query q;
q.constraints = {table.arg_set_id().eq(kFilterArgSetId)};
auto res = table.FilterToIterator(q);
ASSERT_TRUE(!res);
}
{
static constexpr uint32_t kFilterArgSetId = 9;
Query q;
q.constraints = {table.arg_set_id().eq(kFilterArgSetId)};
auto it = table.FilterToIterator(q);
ASSERT_TRUE(!it);
}
// Verify that filtering equality for real arg set ids works as expected.
{
static constexpr uint32_t kFilterArgSetId = 4;
Query q;
q.constraints = {table.arg_set_id().eq(kFilterArgSetId)};
uint32_t cnt = 0;
for (auto it = table.FilterToIterator(q); it; ++it, ++cnt) {
ASSERT_EQ(it.arg_set_id(), kFilterArgSetId);
}
ASSERT_EQ(cnt, 4u);
}
{
static constexpr uint32_t kFilterArgSetId = 0;
Query q;
q.constraints = {table.arg_set_id().eq(kFilterArgSetId)};
uint32_t cnt = 0;
for (auto it = table.FilterToIterator(q); it; ++it, ++cnt) {
ASSERT_EQ(it.arg_set_id(), kFilterArgSetId);
}
ASSERT_EQ(cnt, 2u);
}
{
static constexpr uint32_t kFilterArgSetId = 8;
Query q;
q.constraints = {table.arg_set_id().eq(kFilterArgSetId)};
uint32_t cnt = 0;
for (auto it = table.FilterToIterator(q); it; ++it, ++cnt) {
ASSERT_EQ(it.arg_set_id(), kFilterArgSetId);
}
ASSERT_EQ(cnt, 1u);
}
// Verify that filtering equality for arg set ids after filtering another
// column works.
{
static constexpr uint32_t kFilterArgSetId = 4;
Query q;
q.constraints = {table.int_value().eq(200),
table.arg_set_id().eq(kFilterArgSetId)};
uint32_t cnt = 0;
for (auto it = table.FilterToIterator(q); it; ++it, ++cnt) {
ASSERT_EQ(it.arg_set_id(), kFilterArgSetId);
}
ASSERT_EQ(cnt, 2u);
}
}
} // namespace
} // namespace perfetto::trace_processor::tables