1 /*
2  * Copyright (c) 2023 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 #include <gtest/gtest.h>
16 #include <iservice_registry.h>
17 #include "dump_manager_cpu_client.h"
18 #include "dump_manager_cpu_service.h"
19 #include "dump_common_utils.h"
20 #include "inner/dump_service_id.h"
21 using namespace std;
22 using namespace testing::ext;
23 using namespace OHOS;
24 using namespace OHOS::HiviewDFX;
25 namespace OHOS {
26 namespace HiviewDFX {
27 class HidumperCpuServiceTest : public testing::Test {
28 public:
29     static void SetUpTestCase(void);
30     static void TearDownTestCase(void);
31     void SetUp();
32     void TearDown();
33 protected:
34     static const uint32_t TEST_PID = 1;
35 };
36 
SetUpTestCase(void)37 void HidumperCpuServiceTest::SetUpTestCase(void)
38 {
39 }
40 
TearDownTestCase(void)41 void HidumperCpuServiceTest::TearDownTestCase(void)
42 {
43 }
44 
SetUp(void)45 void HidumperCpuServiceTest::SetUp(void)
46 {
47 }
48 
TearDown(void)49 void HidumperCpuServiceTest::TearDown(void)
50 {
51 }
52 
53 /**
54  * @tc.name: HidumperCpuServiceTest001
55  * @tc.desc: Test DumpManagerCpuService service ready.
56  * @tc.type: FUNC
57  */
58 HWTEST_F(HidumperCpuServiceTest, HidumperCpuServiceTest001, TestSize.Level3)
59 {
60     sptr<ISystemAbilityManager> sam = SystemAbilityManagerClient::GetInstance().GetSystemAbilityManager();
61     ASSERT_TRUE(sam != nullptr) << "HidumperCpuServiceTest001 fail to get GetSystemAbilityManager";
62     sptr<IRemoteObject> remoteObject = sam->CheckSystemAbility(DFX_SYS_HIDUMPER_CPU_ABILITY_ID);
63     ASSERT_TRUE(remoteObject != nullptr) << "GetCpuSystemAbility failed.";
64 }
65 
66 /**
67  * @tc.name: HidumperCpuServiceTest002
68  * @tc.desc: Test of send request to dump spuusage.
69  * @tc.type: FUNC
70  */
71 HWTEST_F(HidumperCpuServiceTest, HidumperCpuServiceTest002, TestSize.Level3)
72 {
73     auto dumpManagerCpuService = std::make_shared<DumpManagerCpuService>();
74     dumpManagerCpuService->StartService();
75     dumpManagerCpuService->OnStart();
76     auto dumpCpuData = std::make_shared<DumpCpuData>();
77     int ret = dumpManagerCpuService->Request(*dumpCpuData);
78     ASSERT_TRUE(ret == 0) << "dump cpuusage failed.";
79 }
80 
81 /**
82  * @tc.name: HidumperCpuServiceTest003
83  * @tc.desc: Test of send request to dump spuusage.
84  * @tc.type: FUNC
85  */
86 HWTEST_F(HidumperCpuServiceTest, HidumperCpuServiceTest003, TestSize.Level3)
87 {
88     auto dumpManagerCpuService = std::make_shared<DumpManagerCpuService>();
89     dumpManagerCpuService->StartService();
90     double cpuUsage;
91     int ret = dumpManagerCpuService->GetCpuUsageByPid(TEST_PID, cpuUsage);
92     ASSERT_TRUE(ret == 0) << "dump cpuusage failed.";
93 }
94 
95 /**
96  * @tc.name: HidumperCpuServiceTest004
97  * @tc.desc: Test error cpuservice.
98  * @tc.type: FUNC
99  */
100 HWTEST_F(HidumperCpuServiceTest, HidumperCpuServiceTest004, TestSize.Level3)
101 {
102     auto dumpManagerCpuService = std::make_shared<DumpManagerCpuService>();
103     dumpManagerCpuService->OnStart();
104     dumpManagerCpuService->OnStart();
105     dumpManagerCpuService->curCPUInfo_ = nullptr;
106     int32_t ret = dumpManagerCpuService->DumpCpuUsageData();
107     ASSERT_TRUE(ret != 0) << "dump cpuusage failed.";
108 }
109 } // namespace HiviewDFX
110 } // namespace OHOS