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