1 /*
2 * Copyright (C) 2022-2024 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 <fcntl.h>
17 #include <gtest/gtest.h>
18
19 #include "hitrace_meter_test_utils.h"
20 #include "common_define.h"
21 #include "common_utils.h"
22 #include "securec.h"
23 #include "trace.h"
24
25 using namespace testing::ext;
26 using namespace OHOS::HiviewDFX::Hitrace;
27
28 namespace OHOS {
29 namespace HiviewDFX {
30 namespace HitraceTest {
31 #ifdef LOG_DOMAIN
32 #undef LOG_DOMAIN
33 #define LOG_DOMAIN 0xD002D33
34 #endif
35 #ifdef LOG_TAG
36 #undef LOG_TAG
37 #define LOG_TAG "HitraceTest"
38 #endif
39
40 constexpr uint64_t TAG = (1ULL << 62); // HITRACE_TAG_APP
41 static char g_pid[6];
42
43 class HitraceMeterNDKTest : public testing::Test {
44 public:
45 static void SetUpTestCase(void);
46 static void TearDownTestCase(void);
47 void SetUp(void);
48 void TearDown(void);
49 };
50
SetUpTestCase(void)51 void HitraceMeterNDKTest::SetUpTestCase(void)
52 {
53 std::string pidStr = std::to_string(getprocpid());
54 int ret = strcpy_s(g_pid, sizeof(g_pid), pidStr.c_str());
55 if (ret != 0) {
56 HILOG_ERROR(LOG_CORE, "pid[%{public}s] strcpy_s fail ret: %{public}d.", pidStr.c_str(), ret);
57 return;
58 }
59 ASSERT_TRUE(Init(g_pid));
60 ASSERT_TRUE(CleanFtrace());
61 }
62
TearDownTestCase(void)63 void HitraceMeterNDKTest::TearDownTestCase(void)
64 {
65 SetPropertyInner(TRACE_TAG_ENABLE_FLAGS, "0");
66 SetFtrace(TRACING_ON_NODE, false);
67 CleanTrace();
68 }
69
SetUp(void)70 void HitraceMeterNDKTest::SetUp(void)
71 {
72 ASSERT_TRUE(CleanTrace());
73 ASSERT_TRUE(SetFtrace(TRACING_ON_NODE, true)) << "SetUp: Setting tracing_on failed.";
74 std::string value = std::to_string(TAG);
75 ASSERT_TRUE(SetPropertyInner(TRACE_TAG_ENABLE_FLAGS, value)) << "SetUp: Setting enableflags failed.";
76 HILOG_INFO(LOG_CORE, "current tag is %{public}s", GetPropertyInner(TRACE_TAG_ENABLE_FLAGS, "0").c_str());
77 ASSERT_TRUE(GetPropertyInner(TRACE_TAG_ENABLE_FLAGS, "-123") == value);
78 }
79
TearDown(void)80 void HitraceMeterNDKTest::TearDown(void)
81 {
82 ASSERT_TRUE(SetPropertyInner(TRACE_TAG_ENABLE_FLAGS, "0")) << "TearDown: Setting enableflags failed.";
83 ASSERT_TRUE(SetFtrace(TRACING_ON_NODE, false)) << "TearDown: Setting tracing_on failed.";
84 ASSERT_TRUE(CleanTrace()) << "TearDown: Cleaning trace failed.";
85 }
86
87 /**
88 * @tc.name: HitraceMeterNDKInterfaceTest001
89 * @tc.desc: Testing OH_HiTrace_StartTraceEx and OH_HiTrace_FinishTraceEx
90 * @tc.type: FUNC
91 */
92 HWTEST_F(HitraceMeterNDKTest, HitraceMeterNDKInterfaceTest001, TestSize.Level1)
93 {
94 GTEST_LOG_(INFO) << "HitraceMeterNDKInterfaceTest001: start.";
95
96 const char* name = "HitraceMeterNDKInterfaceTest001";
97 const char* customArgs = "key=value";
98
99 OH_HiTrace_StartTraceEx(HITRACE_LEVEL_INFO, name, customArgs);
100 OH_HiTrace_FinishTraceEx(HITRACE_LEVEL_INFO);
101
102 std::vector<std::string> list = ReadTrace();
103 char record[RECORD_SIZE_MAX + 1] = {0};
104 TraceInfo traceInfo = {'B', HITRACE_LEVEL_INFO, TAG, 0, name, "", customArgs};
105 bool isStartSuc = GetTraceResult(traceInfo, list, record);
106 ASSERT_TRUE(isStartSuc) << "Hitrace can't find \"" << record << "\" from trace.";
107 traceInfo.type = 'E';
108 bool isFinishSuc = GetTraceResult(traceInfo, list, record);
109 ASSERT_TRUE(isFinishSuc) << "Hitrace can't find \"" << record << "\" from trace.";
110
111 GTEST_LOG_(INFO) << "HitraceMeterNDKInterfaceTest001: end.";
112 }
113
114 /**
115 * @tc.name: HitraceMeterNDKInterfaceTest002
116 * @tc.desc: Testing OH_HiTrace_StartTrace and OH_HiTrace_FinishTrace
117 * @tc.type: FUNC
118 */
119 HWTEST_F(HitraceMeterNDKTest, HitraceMeterNDKInterfaceTest002, TestSize.Level1)
120 {
121 GTEST_LOG_(INFO) << "HitraceMeterNDKInterfaceTest002: start.";
122
123 const char* name = "HitraceMeterNDKInterfaceTest002";
124
125 OH_HiTrace_StartTrace(name);
126 OH_HiTrace_FinishTrace();
127
128 std::vector<std::string> list = ReadTrace();
129 char record[RECORD_SIZE_MAX + 1] = {0};
130 TraceInfo traceInfo = {'B', HITRACE_LEVEL_COMMERCIAL, TAG, 0, name, "", ""};
131 bool isStartSuc = GetTraceResult(traceInfo, list, record);
132 ASSERT_TRUE(isStartSuc) << "Hitrace can't find \"" << record << "\" from trace.";
133 traceInfo.type = 'E';
134 bool isFinishSuc = GetTraceResult(traceInfo, list, record);
135 ASSERT_TRUE(isFinishSuc) << "Hitrace can't find \"" << record << "\" from trace.";
136
137 GTEST_LOG_(INFO) << "HitraceMeterNDKInterfaceTest002: end.";
138 }
139
140 /**
141 * @tc.name: HitraceMeterNDKInterfaceTest003
142 * @tc.desc: Testing OH_HiTrace_StartAsyncTraceEx and OH_HiTrace_FinishAsyncTraceEx
143 * @tc.type: FUNC
144 */
145 HWTEST_F(HitraceMeterNDKTest, HitraceMeterNDKInterfaceTest003, TestSize.Level1)
146 {
147 GTEST_LOG_(INFO) << "HitraceMeterNDKInterfaceTest003: start.";
148
149 const char* name = "HitraceMeterNDKInterfaceTest003";
150 const char* customArgs = "key=value";
151 const char* customCategory = "test";
152 int32_t taskId = 3;
153
154 OH_HiTrace_StartAsyncTraceEx(HITRACE_LEVEL_INFO, name, taskId, customCategory, customArgs);
155 OH_HiTrace_FinishAsyncTraceEx(HITRACE_LEVEL_INFO, name, taskId);
156
157 std::vector<std::string> list = ReadTrace();
158 char record[RECORD_SIZE_MAX + 1] = {0};
159 TraceInfo traceInfo = {'S', HITRACE_LEVEL_INFO, TAG, taskId, name, customCategory, customArgs};
160 bool isStartSuc = GetTraceResult(traceInfo, list, record);
161 ASSERT_TRUE(isStartSuc) << "Hitrace can't find \"" << record << "\" from trace.";
162 traceInfo.type = 'F';
163 bool isFinishSuc = GetTraceResult(traceInfo, list, record);
164 ASSERT_TRUE(isFinishSuc) << "Hitrace can't find \"" << record << "\" from trace.";
165
166 GTEST_LOG_(INFO) << "HitraceMeterNDKInterfaceTest003: end.";
167 }
168
169 /**
170 * @tc.name: HitraceMeterNDKInterfaceTest004
171 * @tc.desc: Testing OH_HiTrace_StartAsyncTrace and OH_HiTrace_FinishAsyncTrace
172 * @tc.type: FUNC
173 */
174 HWTEST_F(HitraceMeterNDKTest, HitraceMeterNDKInterfaceTest004, TestSize.Level1)
175 {
176 GTEST_LOG_(INFO) << "HitraceMeterNDKInterfaceTest004: start.";
177
178 const char* name = "HitraceMeterNDKInterfaceTest004";
179 int32_t taskId = 4;
180
181 OH_HiTrace_StartAsyncTrace(name, taskId);
182 OH_HiTrace_FinishAsyncTrace(name, taskId);
183
184 std::vector<std::string> list = ReadTrace();
185 char record[RECORD_SIZE_MAX + 1] = {0};
186 TraceInfo traceInfo = {'S', HITRACE_LEVEL_COMMERCIAL, TAG, taskId, name, "", ""};
187 bool isStartSuc = GetTraceResult(traceInfo, list, record);
188 ASSERT_TRUE(isStartSuc) << "Hitrace can't find \"" << record << "\" from trace.";
189 traceInfo.type = 'F';
190 bool isFinishSuc = GetTraceResult(traceInfo, list, record);
191 ASSERT_TRUE(isFinishSuc) << "Hitrace can't find \"" << record << "\" from trace.";
192
193 GTEST_LOG_(INFO) << "HitraceMeterNDKInterfaceTest004: end.";
194 }
195
196 /**
197 * @tc.name: HitraceMeterNDKInterfaceTest005
198 * @tc.desc: Testing OH_HiTrace_CountTraceEx
199 * @tc.type: FUNC
200 */
201 HWTEST_F(HitraceMeterNDKTest, HitraceMeterNDKInterfaceTest005, TestSize.Level1)
202 {
203 GTEST_LOG_(INFO) << "HitraceMeterNDKInterfaceTest005: start.";
204
205 const char* name = "HitraceMeterNDKInterfaceTest005";
206 int64_t count = 5;
207
208 OH_HiTrace_CountTraceEx(HITRACE_LEVEL_INFO, name, count);
209
210 std::vector<std::string> list = ReadTrace();
211 char record[RECORD_SIZE_MAX + 1] = {0};
212 TraceInfo traceInfo = {'C', HITRACE_LEVEL_INFO, TAG, count, name, "", ""};
213 bool isStartSuc = GetTraceResult(traceInfo, list, record);
214 ASSERT_TRUE(isStartSuc) << "Hitrace can't find \"" << record << "\" from trace.";
215
216 GTEST_LOG_(INFO) << "HitraceMeterNDKInterfaceTest005: end.";
217 }
218
219 /**
220 * @tc.name: HitraceMeterNDKInterfaceTest006
221 * @tc.desc: Testing OH_HiTrace_CountTrace
222 * @tc.type: FUNC
223 */
224 HWTEST_F(HitraceMeterNDKTest, HitraceMeterNDKInterfaceTest006, TestSize.Level1)
225 {
226 GTEST_LOG_(INFO) << "HitraceMeterNDKInterfaceTest006: start.";
227
228 const char* name = "HitraceMeterNDKInterfaceTest006";
229 int64_t count = 6;
230
231 OH_HiTrace_CountTrace(name, count);
232
233 std::vector<std::string> list = ReadTrace();
234 char record[RECORD_SIZE_MAX + 1] = {0};
235 TraceInfo traceInfo = {'C', HITRACE_LEVEL_COMMERCIAL, TAG, count, name, "", ""};
236 bool isStartSuc = GetTraceResult(traceInfo, list, record);
237 ASSERT_TRUE(isStartSuc) << "Hitrace can't find \"" << record << "\" from trace.";
238
239 GTEST_LOG_(INFO) << "HitraceMeterNDKInterfaceTest006: end.";
240 }
241
242 /**
243 * @tc.name: HitraceMeterNDKInterfaceTest007
244 * @tc.desc: Testing OH_HiTrace_IsTraceEnabled
245 * @tc.type: FUNC
246 */
247 HWTEST_F(HitraceMeterNDKTest, HitraceMeterNDKInterfaceTest007, TestSize.Level1)
248 {
249 GTEST_LOG_(INFO) << "HitraceMeterNDKInterfaceTest007: start.";
250
251 ASSERT_TRUE(OH_HiTrace_IsTraceEnabled());
252 const uint64_t tag = (1ULL << 30); // HITRACE_TAG_OHOS
253 std::string value = std::to_string(tag);
254 ASSERT_TRUE(SetPropertyInner(TRACE_TAG_ENABLE_FLAGS, value));
255 ASSERT_FALSE(OH_HiTrace_IsTraceEnabled());
256
257 GTEST_LOG_(INFO) << "HitraceMeterNDKInterfaceTest007: end.";
258 }
259 }
260 }
261 }
262