1 /*
2 * Copyright (C) 2015 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17 #include <gtest/gtest.h>
18
19 #include "event_attr.h"
20 #include "event_type.h"
21 #include "record.h"
22 #include "record_equal_test.h"
23
24 class RecordTest : public ::testing::Test {
25 protected:
SetUp()26 virtual void SetUp() {
27 const EventType* type = FindEventTypeByName("cpu-cycles");
28 ASSERT_TRUE(type != nullptr);
29 event_attr = CreateDefaultPerfEventAttr(*type);
30 event_attr.sample_id_all = 1;
31 }
32
CheckRecordMatchBinary(const Record & record)33 void CheckRecordMatchBinary(const Record& record) {
34 const char* p = record.Binary();
35 std::vector<std::unique_ptr<Record>> records =
36 ReadRecordsFromBuffer(event_attr, p, record.size());
37 ASSERT_EQ(1u, records.size());
38 CheckRecordEqual(record, *records[0]);
39 }
40
41 perf_event_attr event_attr;
42 };
43
TEST_F(RecordTest,MmapRecordMatchBinary)44 TEST_F(RecordTest, MmapRecordMatchBinary) {
45 MmapRecord record(event_attr, true, 1, 2, 0x1000, 0x2000, 0x3000,
46 "MmapRecord", 0);
47 CheckRecordMatchBinary(record);
48 }
49
TEST_F(RecordTest,CommRecordMatchBinary)50 TEST_F(RecordTest, CommRecordMatchBinary) {
51 CommRecord record(event_attr, 1, 2, "CommRecord", 0, 7);
52 CheckRecordMatchBinary(record);
53 }
54
TEST_F(RecordTest,SampleRecordMatchBinary)55 TEST_F(RecordTest, SampleRecordMatchBinary) {
56 event_attr.sample_type = PERF_SAMPLE_IP | PERF_SAMPLE_TID | PERF_SAMPLE_TIME
57 | PERF_SAMPLE_ID | PERF_SAMPLE_CPU
58 | PERF_SAMPLE_PERIOD | PERF_SAMPLE_CALLCHAIN;
59 SampleRecord record(event_attr, 1, 2, 3, 4, 5, 6, 7, {8, 9, 10});
60 CheckRecordMatchBinary(record);
61 }
62
TEST_F(RecordTest,RecordCache_smoke)63 TEST_F(RecordTest, RecordCache_smoke) {
64 event_attr.sample_id_all = 1;
65 event_attr.sample_type |= PERF_SAMPLE_TIME;
66 RecordCache cache(true, 2, 2);
67
68 // Push r1.
69 MmapRecord* r1 = new MmapRecord(event_attr, true, 1, 1, 0x100, 0x200, 0x300,
70 "mmap_record1", 0, 3);
71 cache.Push(std::unique_ptr<Record>(r1));
72 ASSERT_EQ(nullptr, cache.Pop());
73
74 // Push r2.
75 MmapRecord* r2 = new MmapRecord(event_attr, true, 1, 1, 0x100, 0x200, 0x300,
76 "mmap_record1", 0, 1);
77 cache.Push(std::unique_ptr<Record>(r2));
78 // Pop r2.
79 std::unique_ptr<Record> popped_r = cache.Pop();
80 ASSERT_TRUE(popped_r != nullptr);
81 ASSERT_EQ(r2, popped_r.get());
82 ASSERT_EQ(nullptr, cache.Pop());
83
84 // Push r3.
85 MmapRecord* r3 = new MmapRecord(event_attr, true, 1, 1, 0x100, 0x200, 0x300,
86 "mmap_record1", 0, 4);
87 cache.Push(std::unique_ptr<Record>(r3));
88 ASSERT_EQ(nullptr, cache.Pop());
89
90 // Push r4.
91 MmapRecord* r4 = new MmapRecord(event_attr, true, 1, 1, 0x100, 0x200, 0x300,
92 "mmap_record1", 0, 6);
93 cache.Push(std::unique_ptr<Record>(r4));
94 // Pop r1.
95 popped_r = cache.Pop();
96 ASSERT_TRUE(popped_r != nullptr);
97 ASSERT_EQ(r1, popped_r.get());
98 // Pop r3.
99 popped_r = cache.Pop();
100 ASSERT_TRUE(popped_r != nullptr);
101 ASSERT_EQ(r3, popped_r.get());
102 ASSERT_EQ(nullptr, cache.Pop());
103 // Pop r4.
104 std::vector<std::unique_ptr<Record>> last_records = cache.PopAll();
105 ASSERT_EQ(1u, last_records.size());
106 ASSERT_EQ(r4, last_records[0].get());
107 }
108
TEST_F(RecordTest,RecordCache_FIFO)109 TEST_F(RecordTest, RecordCache_FIFO) {
110 event_attr.sample_id_all = 1;
111 event_attr.sample_type |= PERF_SAMPLE_TIME;
112 RecordCache cache(true, 2, 2);
113 std::vector<MmapRecord*> records;
114 for (size_t i = 0; i < 10; ++i) {
115 records.push_back(new MmapRecord(event_attr, true, 1, i, 0x100, 0x200,
116 0x300, "mmap_record1", 0));
117 cache.Push(std::unique_ptr<Record>(records.back()));
118 }
119 std::vector<std::unique_ptr<Record>> out_records = cache.PopAll();
120 ASSERT_EQ(records.size(), out_records.size());
121 for (size_t i = 0; i < records.size(); ++i) {
122 ASSERT_EQ(records[i], out_records[i].get());
123 }
124 }
125
TEST_F(RecordTest,RecordCache_PushRecordVector)126 TEST_F(RecordTest, RecordCache_PushRecordVector) {
127 event_attr.sample_id_all = 1;
128 event_attr.sample_type |= PERF_SAMPLE_TIME;
129 RecordCache cache(true, 2, 2);
130 MmapRecord* r1 = new MmapRecord(event_attr, true, 1, 1, 0x100, 0x200, 0x300,
131 "mmap_record1", 0, 1);
132 MmapRecord* r2 = new MmapRecord(event_attr, true, 1, 1, 0x100, 0x200, 0x300,
133 "mmap_record1", 0, 3);
134 std::vector<std::unique_ptr<Record>> records;
135 records.push_back(std::unique_ptr<Record>(r1));
136 records.push_back(std::unique_ptr<Record>(r2));
137 cache.Push(std::move(records));
138 std::unique_ptr<Record> popped_r = cache.Pop();
139 ASSERT_TRUE(popped_r != nullptr);
140 ASSERT_EQ(r1, popped_r.get());
141 std::vector<std::unique_ptr<Record>> last_records = cache.PopAll();
142 ASSERT_EQ(1u, last_records.size());
143 ASSERT_EQ(r2, last_records[0].get());
144 }
145