1 /*
2 * Copyright (c) 2022 Huawei Device Co., Ltd.
3 * Licensed under the Apache License, Version 2.0 (the "License");
4 * you may not use this file except in compliance with the License.
5 * You may obtain a copy of the License at
6 *
7 * http://www.apache.org/licenses/LICENSE-2.0
8 *
9 * Unless required by applicable law or agreed to in writing, software
10 * distributed under the License is distributed on an "AS IS" BASIS,
11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 * See the License for the specific language governing permissions and
13 * limitations under the License.
14 */
15
16 #include "agent/tracing_impl.h"
17 #include "ecmascript/tests/test_helper.h"
18 #include "protocol_handler.h"
19
20 using namespace panda::ecmascript;
21 using namespace panda::ecmascript::tooling;
22
23 namespace panda::test {
24 class TracingImplTest : public testing::Test {
25 public:
SetUpTestCase()26 static void SetUpTestCase()
27 {
28 GTEST_LOG_(INFO) << "SetUpTestCase";
29 }
30
TearDownTestCase()31 static void TearDownTestCase()
32 {
33 GTEST_LOG_(INFO) << "TearDownCase";
34 }
35
SetUp()36 void SetUp() override
37 {
38 TestHelper::CreateEcmaVMWithScope(ecmaVm, thread, scope);
39 }
40
TearDown()41 void TearDown() override
42 {
43 TestHelper::DestroyEcmaVMWithScope(ecmaVm, scope);
44 }
45
46 protected:
47 EcmaVM *ecmaVm {nullptr};
48 EcmaHandleScope *scope {nullptr};
49 JSThread *thread {nullptr};
50 };
51
HWTEST_F_L0(TracingImplTest,StartAndEndTest)52 HWTEST_F_L0(TracingImplTest, StartAndEndTest)
53 {
54 ProtocolChannel *channel = nullptr;
55 auto tracing = std::make_unique<TracingImpl>(ecmaVm, channel);
56 auto params = std::make_unique<StartParams>();
57 params->SetCategories("cpu_profiler");
58 DispatchResponse response = tracing->Start(std::move(params));
59 ASSERT_TRUE(response.IsOk());
60
61 auto profileInfo = tracing->End();
62 ASSERT_TRUE(profileInfo != nullptr);
63 }
64
HWTEST_F_L0(TracingImplTest,GetCategoriesTest)65 HWTEST_F_L0(TracingImplTest, GetCategoriesTest)
66 {
67 ProtocolChannel *channel = nullptr;
68 auto tracing = std::make_unique<TracingImpl>(ecmaVm, channel);
69 std::vector<std::string> categories = {};
70 DispatchResponse response = tracing->GetCategories(categories);
71 ASSERT_TRUE(response.GetMessage() == "GetCategories not support now.");
72 }
73
HWTEST_F_L0(TracingImplTest,RecordClockSyncMarkerTest)74 HWTEST_F_L0(TracingImplTest, RecordClockSyncMarkerTest)
75 {
76 ProtocolChannel *channel = nullptr;
77 auto tracing = std::make_unique<TracingImpl>(ecmaVm, channel);
78 std::string syncId;
79 DispatchResponse response = tracing->RecordClockSyncMarker(syncId);
80 ASSERT_TRUE(response.GetMessage() == "RecordClockSyncMarker not support now.");
81 }
82
HWTEST_F_L0(TracingImplTest,RequestMemoryDumpTest)83 HWTEST_F_L0(TracingImplTest, RequestMemoryDumpTest)
84 {
85 ProtocolChannel *channel = nullptr;
86 auto tracing = std::make_unique<TracingImpl>(ecmaVm, channel);
87 auto params = std::make_unique<RequestMemoryDumpParams>();
88 std::string dumpGuid;
89 bool success = false;
90 DispatchResponse response = tracing->RequestMemoryDump(std::move(params), dumpGuid, success);
91 ASSERT_TRUE(response.GetMessage() == "RequestMemoryDump not support now.");
92 }
93
HWTEST_F_L0(TracingImplTest,DispatcherImplDispatchTest)94 HWTEST_F_L0(TracingImplTest, DispatcherImplDispatchTest)
95 {
96 std::string result = "";
97 std::function<void(const void*, const std::string &)> callback =
98 [&result]([[maybe_unused]] const void *ptr, const std::string &temp) { result = temp; };
99 ProtocolChannel *channel = new ProtocolHandler(callback, ecmaVm);
100 auto tracing = std::make_unique<TracingImpl>(ecmaVm, channel);
101 auto dispatcherImpl = std::make_unique<TracingImpl::DispatcherImpl>(channel, std::move(tracing));
102 std::string msg = std::string() + R"({"id":0,"method":"Debugger.Test","params":{}})";
103 DispatchRequest request(msg);
104 dispatcherImpl->Dispatch(request);
105 ASSERT_TRUE(result.find("Unknown method: Test") != std::string::npos);
106 msg = std::string() + R"({"id":0,"method":"Debugger.end","params":{}})";
107 DispatchRequest request1 = DispatchRequest(msg);
108 dispatcherImpl->Dispatch(request1);
109 if (channel) {
110 delete channel;
111 channel = nullptr;
112 }
113 ASSERT_TRUE(result.find("\"id\":0,") != std::string::npos);
114 }
115
HWTEST_F_L0(TracingImplTest,DispatcherImplEndTest)116 HWTEST_F_L0(TracingImplTest, DispatcherImplEndTest)
117 {
118 std::string result = "";
119 std::function<void(const void*, const std::string &)> callback =
120 [&result]([[maybe_unused]] const void *ptr, const std::string &temp) { result = temp; };
121 ProtocolChannel *channel = new ProtocolHandler(callback, ecmaVm);
122 auto tracing = std::make_unique<TracingImpl>(ecmaVm, channel);
123 auto dispatcherImpl = std::make_unique<TracingImpl::DispatcherImpl>(channel, std::move(tracing));
124 std::string msg = std::string() + R"({"id":0,"method":"Debugger.end","params":{}})";
125 DispatchRequest request = DispatchRequest(msg);
126 dispatcherImpl->Dispatch(request);
127 if (channel) {
128 delete channel;
129 channel = nullptr;
130 }
131 ASSERT_TRUE(result.find("\"id\":0,") != std::string::npos);
132 }
133
HWTEST_F_L0(TracingImplTest,DispatcherImplRepeateStartTest)134 HWTEST_F_L0(TracingImplTest, DispatcherImplRepeateStartTest)
135 {
136 std::string result = "";
137 std::function<void(const void*, const std::string &)> callback =
138 [&result]([[maybe_unused]] const void *ptr, const std::string &temp) { result = temp; };
139 ProtocolChannel *channel = new ProtocolHandler(callback, ecmaVm);
140 auto tracing = std::make_unique<TracingImpl>(ecmaVm, channel);
141 auto dispatcherImpl = std::make_unique<TracingImpl::DispatcherImpl>(channel, std::move(tracing));
142 ecmaVm->SetLoop(uv_default_loop());
143 std::string msg = std::string() + R"({"id":0,"method":"Debugger.start","params":{"categories":"cpu_profiler"}})";
144 DispatchRequest request = DispatchRequest(msg);
145 dispatcherImpl->Dispatch(request);
146 ASSERT_TRUE(result.find("\"id\":0,") != std::string::npos);
147 std::string msg1 = std::string() + R"({"id":0,"method":"Debugger.start","params":{"categories":"cpu_profiler"}})";
148 DispatchRequest request1 = DispatchRequest(msg1);
149 dispatcherImpl->Dispatch(request1);
150 ASSERT_TRUE(result.find("\"id\":0,") != std::string::npos);
151 if (channel) {
152 delete channel;
153 channel = nullptr;
154 }
155 }
156
HWTEST_F_L0(TracingImplTest,DispatcherImplStartAndEndTest)157 HWTEST_F_L0(TracingImplTest, DispatcherImplStartAndEndTest)
158 {
159 std::string result = "";
160 std::function<void(const void*, const std::string &)> callback =
161 [&result]([[maybe_unused]] const void *ptr, const std::string &temp) { result = temp; };
162 ProtocolChannel *channel = new ProtocolHandler(callback, ecmaVm);
163 auto tracing = std::make_unique<TracingImpl>(ecmaVm, channel);
164 auto dispatcherImpl = std::make_unique<TracingImpl::DispatcherImpl>(channel, std::move(tracing));
165 ecmaVm->SetLoop(uv_default_loop());
166 std::string msg = std::string() + R"({"id":0,"method":"Debugger.start","params":{"categories":"cpu_profiler"}})";
167 DispatchRequest request = DispatchRequest(msg);
168 dispatcherImpl->Dispatch(request);
169 ASSERT_TRUE(result.find("\"id\":0,") != std::string::npos);
170 std::string msg1 = std::string() + R"({"id":0,"method":"Debugger.end","params":{}})";
171 DispatchRequest request1 = DispatchRequest(msg1);
172 dispatcherImpl->Dispatch(request1);
173 ASSERT_TRUE(result.find("Tracing.tracingComplete") != std::string::npos);
174 if (channel) {
175 delete channel;
176 channel = nullptr;
177 }
178 }
179
HWTEST_F_L0(TracingImplTest,DispatcherImplGetCategoriesTest)180 HWTEST_F_L0(TracingImplTest, DispatcherImplGetCategoriesTest)
181 {
182 std::string result = "";
183 std::function<void(const void*, const std::string &)> callback =
184 [&result]([[maybe_unused]] const void *ptr, const std::string &temp) { result = temp; };
185 ProtocolChannel *channel = new ProtocolHandler(callback, ecmaVm);
186 auto tracing = std::make_unique<TracingImpl>(ecmaVm, channel);
187 auto dispatcherImpl = std::make_unique<TracingImpl::DispatcherImpl>(channel, std::move(tracing));
188 std::string msg = std::string() + R"({"id":0,"method":"Debugger.getCategories","params":{}})";
189 DispatchRequest request = DispatchRequest(msg);
190 dispatcherImpl->Dispatch(request);
191 if (channel) {
192 delete channel;
193 channel = nullptr;
194 }
195 ASSERT_TRUE(result.find("GetCategories not support now") != std::string::npos);
196 }
197
HWTEST_F_L0(TracingImplTest,DispatcherImplRecordClockSyncMarkerTest)198 HWTEST_F_L0(TracingImplTest, DispatcherImplRecordClockSyncMarkerTest)
199 {
200 std::string result = "";
201 std::function<void(const void*, const std::string &)> callback =
202 [&result]([[maybe_unused]] const void *ptr, const std::string &temp) { result = temp; };
203 ProtocolChannel *channel = new ProtocolHandler(callback, ecmaVm);
204 auto tracing = std::make_unique<TracingImpl>(ecmaVm, channel);
205 auto dispatcherImpl = std::make_unique<TracingImpl::DispatcherImpl>(channel, std::move(tracing));
206 std::string msg = std::string() + R"({"id":0,"method":"Debugger.recordClockSyncMarker","params":{}})";
207 DispatchRequest request = DispatchRequest(msg);
208 dispatcherImpl->Dispatch(request);
209 if (channel) {
210 delete channel;
211 channel = nullptr;
212 }
213 ASSERT_TRUE(result.find("RecordClockSyncMarker not support now.") != std::string::npos);
214 }
215
HWTEST_F_L0(TracingImplTest,DispatcherImplRequestMemoryDumpTest)216 HWTEST_F_L0(TracingImplTest, DispatcherImplRequestMemoryDumpTest)
217 {
218 std::string result = "";
219 std::function<void(const void*, const std::string &)> callback =
220 [&result]([[maybe_unused]] const void *ptr, const std::string &temp) { result = temp; };
221 ProtocolChannel *channel = new ProtocolHandler(callback, ecmaVm);
222 auto tracing = std::make_unique<TracingImpl>(ecmaVm, channel);
223 auto dispatcherImpl = std::make_unique<TracingImpl::DispatcherImpl>(channel, std::move(tracing));
224 std::string msg = std::string() + R"({"id":0,"method":"Debugger.requestMemoryDump","params":{}})";
225 DispatchRequest request = DispatchRequest(msg);
226 dispatcherImpl->Dispatch(request);
227 if (channel) {
228 delete channel;
229 channel = nullptr;
230 }
231 ASSERT_TRUE(result.find("RequestMemoryDump not support now.") != std::string::npos);
232 }
233
HWTEST_F_L0(TracingImplTest,DispatcherImplStartDumpTest)234 HWTEST_F_L0(TracingImplTest, DispatcherImplStartDumpTest)
235 {
236 std::string result = "";
237 std::function<void(const void*, const std::string &)> callback =
238 [&result]([[maybe_unused]] const void *ptr, const std::string &temp) { result = temp; };
239 ProtocolChannel *channel = new ProtocolHandler(callback, ecmaVm);
240 auto tracing = std::make_unique<TracingImpl>(ecmaVm, channel);
241 auto dispatcherImpl = std::make_unique<TracingImpl::DispatcherImpl>(channel, std::move(tracing));
242 std::string msg = std::string() + R"({"id":0,"method":"Debugger.start","params":{"categories":"cpu_profiler"}})";
243 DispatchRequest request = DispatchRequest(msg);
244 dispatcherImpl->Dispatch(request);
245 if (channel) {
246 delete channel;
247 channel = nullptr;
248 }
249 ASSERT_TRUE(result.find("\"id\":0,") != std::string::npos);
250 }
251
HWTEST_F_L0(TracingImplTest,FrontendBufferUsageTest)252 HWTEST_F_L0(TracingImplTest, FrontendBufferUsageTest)
253 {
254 std::string result = "";
255 std::function<void(const void*, const std::string &)> callback =
256 [&result]([[maybe_unused]] const void *ptr, const std::string &temp) { result = temp; };
257 ProtocolChannel *channel = nullptr;
258 auto frontend = std::make_unique<TracingImpl::Frontend>(channel);
259 frontend->BufferUsage(0, 0, 0);
260 ASSERT_TRUE(result == "");
261 if (!channel) {
262 channel = new ProtocolHandler(callback, ecmaVm);
263 }
264 auto frontend1 = std::make_unique<TracingImpl::Frontend>(channel);
265 frontend1->BufferUsage(0, 0, 0);
266 if (channel) {
267 delete channel;
268 channel = nullptr;
269 }
270 ASSERT_TRUE(result.find("Tracing.bufferUsage") != std::string::npos);
271 }
272
HWTEST_F_L0(TracingImplTest,FrontendDataCollectedTest)273 HWTEST_F_L0(TracingImplTest, FrontendDataCollectedTest)
274 {
275 int64_t ts = 604898475815;
276 TraceEvent event("timeline", "UpdateCounters", "I", getpid(), 1415);
277 event.SetTs(ts);
278 event.SetTts(ts);
279 event.SetS("t");
280 std::string args = "{\"data\":{\"jsHeapSizeUsed\":" + std::to_string(1024) + "}}";
281 event.SetArgs(args);
282 std::unique_ptr<std::vector<TraceEvent>> traceEvents = std::make_unique<std::vector<TraceEvent>>();
283 traceEvents->emplace_back(event);
284
285 std::string result = "";
286 std::function<void(const void*, const std::string &)> callback =
287 [&result]([[maybe_unused]] const void *ptr, const std::string &temp) { result = temp; };
288 ProtocolChannel *channel = nullptr;
289 auto frontend = std::make_unique<TracingImpl::Frontend>(channel);
290 frontend->DataCollected(std::move(traceEvents));
291 ASSERT_TRUE(result == "");
292 if (!channel) {
293 channel = new ProtocolHandler(callback, ecmaVm);
294 }
295 auto frontend1 = std::make_unique<TracingImpl::Frontend>(channel);
296 std::unique_ptr<std::vector<TraceEvent>> traceEvents1 = std::make_unique<std::vector<TraceEvent>>();
297 traceEvents1->emplace_back(event);
298 frontend1->DataCollected(std::move(traceEvents1));
299 if (channel) {
300 delete channel;
301 channel = nullptr;
302 }
303 ASSERT_TRUE(result.find("Tracing.dataCollected") != std::string::npos);
304 }
305
HWTEST_F_L0(TracingImplTest,FrontendTracingCompleteTest)306 HWTEST_F_L0(TracingImplTest, FrontendTracingCompleteTest)
307 {
308 std::string result = "";
309 std::function<void(const void*, const std::string &)> callback =
310 [&result]([[maybe_unused]] const void *ptr, const std::string &temp) { result = temp; };
311 ProtocolChannel *channel = nullptr;
312 auto frontend = std::make_unique<TracingImpl::Frontend>(channel);
313 frontend->TracingComplete();
314 ASSERT_TRUE(result == "");
315 if (!channel) {
316 channel = new ProtocolHandler(callback, ecmaVm);
317 }
318 auto frontend1 = std::make_unique<TracingImpl::Frontend>(channel);
319 frontend1->TracingComplete();
320 if (channel) {
321 delete channel;
322 channel = nullptr;
323 }
324 ASSERT_TRUE(result.find("Tracing.tracingComplete") != std::string::npos);
325 }
326 } // namespace panda::test