1 /*
2 * Copyright (C) 2020 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 "src/trace_processor/importers/common/async_track_set_tracker.h"
18
19 #include "src/trace_processor/importers/common/args_tracker.h"
20 #include "src/trace_processor/importers/common/global_args_tracker.h"
21 #include "src/trace_processor/importers/common/track_tracker.h"
22 #include "src/trace_processor/types/trace_processor_context.h"
23 #include "test/gtest_and_gmock.h"
24
25 namespace perfetto {
26 namespace trace_processor {
27
28 class AsyncTrackSetTrackerUnittest : public testing::Test {
29 public:
AsyncTrackSetTrackerUnittest()30 AsyncTrackSetTrackerUnittest() {
31 context_.storage.reset(new TraceStorage());
32 context_.global_args_tracker.reset(
33 new GlobalArgsTracker(context_.storage.get()));
34 context_.args_tracker.reset(new ArgsTracker(&context_));
35 context_.track_tracker.reset(new TrackTracker(&context_));
36 context_.async_track_set_tracker.reset(new AsyncTrackSetTracker(&context_));
37
38 storage_ = context_.storage.get();
39 tracker_ = context_.async_track_set_tracker.get();
40
41 nestable_id_ =
42 tracker_->InternProcessTrackSet(1, storage_->InternString("test"));
43 legacy_unnestable_id_ = tracker_->InternAndroidLegacyUnnestableTrackSet(
44 2, storage_->InternString("test"));
45 }
46
47 protected:
48 TraceStorage* storage_ = nullptr;
49 AsyncTrackSetTracker* tracker_ = nullptr;
50
51 AsyncTrackSetTracker::TrackSetId nestable_id_;
52 AsyncTrackSetTracker::TrackSetId legacy_unnestable_id_;
53
54 private:
55 TraceProcessorContext context_;
56 };
57
58 namespace {
59
TEST_F(AsyncTrackSetTrackerUnittest,Smoke)60 TEST_F(AsyncTrackSetTrackerUnittest, Smoke) {
61 auto set_id = tracker_->InternAndroidLegacyUnnestableTrackSet(
62 1, storage_->InternString("test"));
63
64 auto begin = tracker_->Begin(set_id, 1);
65 auto end = tracker_->End(set_id, 1);
66
67 ASSERT_EQ(begin, end);
68
69 uint32_t row = *storage_->process_track_table().id().IndexOf(begin);
70 ASSERT_EQ(storage_->process_track_table().upid()[row], 1u);
71 ASSERT_EQ(storage_->process_track_table().name()[row],
72 storage_->InternString("test"));
73 }
74
TEST_F(AsyncTrackSetTrackerUnittest,EndFirst)75 TEST_F(AsyncTrackSetTrackerUnittest, EndFirst) {
76 auto end = tracker_->End(nestable_id_, 1);
77
78 uint32_t row = *storage_->process_track_table().id().IndexOf(end);
79 ASSERT_EQ(storage_->process_track_table().upid()[row], 1u);
80 ASSERT_EQ(storage_->process_track_table().name()[row],
81 storage_->InternString("test"));
82 }
83
TEST_F(AsyncTrackSetTrackerUnittest,LegacySaturating)84 TEST_F(AsyncTrackSetTrackerUnittest, LegacySaturating) {
85 auto begin = tracker_->Begin(legacy_unnestable_id_, 1);
86 auto begin_2 = tracker_->Begin(legacy_unnestable_id_, 1);
87
88 ASSERT_EQ(begin, begin_2);
89 }
90
TEST_F(AsyncTrackSetTrackerUnittest,DoubleBegin)91 TEST_F(AsyncTrackSetTrackerUnittest, DoubleBegin) {
92 auto begin = tracker_->Begin(nestable_id_, 1);
93 auto end = tracker_->End(nestable_id_, 1);
94 auto begin_2 = tracker_->Begin(nestable_id_, 1);
95
96 ASSERT_EQ(begin, end);
97 ASSERT_EQ(begin, begin_2);
98 }
99
TEST_F(AsyncTrackSetTrackerUnittest,Nesting)100 TEST_F(AsyncTrackSetTrackerUnittest, Nesting) {
101 auto begin = tracker_->Begin(nestable_id_, 1);
102 auto begin_nested = tracker_->Begin(nestable_id_, 1);
103 auto begin_other = tracker_->Begin(nestable_id_, 2);
104 auto end_nested = tracker_->End(nestable_id_, 1);
105 auto end = tracker_->End(nestable_id_, 1);
106 auto end_other = tracker_->Begin(nestable_id_, 2);
107
108 ASSERT_EQ(begin, begin_nested);
109 ASSERT_NE(begin, begin_other);
110 ASSERT_EQ(begin_nested, end_nested);
111 ASSERT_EQ(begin, end);
112 ASSERT_EQ(begin_other, end_other);
113 }
114
TEST_F(AsyncTrackSetTrackerUnittest,NestableMultipleEndAfterBegin)115 TEST_F(AsyncTrackSetTrackerUnittest, NestableMultipleEndAfterBegin) {
116 auto begin = tracker_->Begin(nestable_id_, 1);
117 auto end = tracker_->End(nestable_id_, 1);
118 auto end_2 = tracker_->End(nestable_id_, 1);
119
120 ASSERT_EQ(begin, end);
121 ASSERT_EQ(end, end_2);
122 }
123
TEST_F(AsyncTrackSetTrackerUnittest,OnlyScoped)124 TEST_F(AsyncTrackSetTrackerUnittest, OnlyScoped) {
125 TrackId a = tracker_->Scoped(nestable_id_, 100, 10);
126 TrackId b = tracker_->Scoped(nestable_id_, 105, 2);
127 TrackId c = tracker_->Scoped(nestable_id_, 107, 3);
128 TrackId d = tracker_->Scoped(nestable_id_, 110, 5);
129
130 ASSERT_NE(a, b);
131 ASSERT_EQ(b, c);
132 ASSERT_EQ(a, d);
133 }
134
TEST_F(AsyncTrackSetTrackerUnittest,MixScopedAndBeginEnd)135 TEST_F(AsyncTrackSetTrackerUnittest, MixScopedAndBeginEnd) {
136 TrackId a = tracker_->Scoped(nestable_id_, 100, 10);
137
138 TrackId begin = tracker_->Begin(nestable_id_, 777);
139 TrackId end = tracker_->End(nestable_id_, 777);
140
141 TrackId b = tracker_->Scoped(nestable_id_, 105, 2);
142
143 ASSERT_NE(a, begin);
144 ASSERT_NE(b, begin);
145 ASSERT_EQ(begin, end);
146 }
147
TEST_F(AsyncTrackSetTrackerUnittest,DifferentTracksInterleave)148 TEST_F(AsyncTrackSetTrackerUnittest, DifferentTracksInterleave) {
149 TrackId b1 = tracker_->Begin(nestable_id_, 666);
150 TrackId b2 = tracker_->Begin(legacy_unnestable_id_, 777);
151 TrackId e1 = tracker_->End(nestable_id_, 666);
152 TrackId e2 = tracker_->End(legacy_unnestable_id_, 777);
153
154 ASSERT_EQ(b1, e1);
155 ASSERT_EQ(b2, e2);
156 ASSERT_NE(b1, b2);
157 }
158
TEST_F(AsyncTrackSetTrackerUnittest,DifferentCookieInterleave)159 TEST_F(AsyncTrackSetTrackerUnittest, DifferentCookieInterleave) {
160 TrackId b1 = tracker_->Begin(legacy_unnestable_id_, 666);
161 TrackId b2 = tracker_->Begin(legacy_unnestable_id_, 777);
162 TrackId e1 = tracker_->End(legacy_unnestable_id_, 666);
163 TrackId e2 = tracker_->End(legacy_unnestable_id_, 777);
164
165 ASSERT_EQ(b1, e1);
166 ASSERT_EQ(b2, e2);
167 ASSERT_NE(b1, b2);
168 }
169
TEST_F(AsyncTrackSetTrackerUnittest,DifferentCookieSequential)170 TEST_F(AsyncTrackSetTrackerUnittest, DifferentCookieSequential) {
171 TrackId b1 = tracker_->Begin(legacy_unnestable_id_, 666);
172 TrackId e1 = tracker_->End(legacy_unnestable_id_, 666);
173 TrackId b2 = tracker_->Begin(legacy_unnestable_id_, 777);
174 TrackId e2 = tracker_->End(legacy_unnestable_id_, 777);
175
176 ASSERT_EQ(b1, e1);
177 ASSERT_EQ(b1, b2);
178 ASSERT_EQ(b2, e2);
179 }
180
181 } // namespace
182 } // namespace trace_processor
183 } // namespace perfetto
184