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 "gtest/gtest.h"
17
18 #include "utils.h"
19
20 #define private public
21 #define protected public
22 #include "mem_mgr_client.h"
23 #include "bundle_priority_list.h"
24 #include "mem_mgr_constant.h"
25 #include "app_state_subscriber.h"
26 #undef private
27 #undef protected
28
29 namespace OHOS {
30 namespace Memory {
31 using namespace testing;
32 using namespace testing::ext;
33
34 class AppStateSubscriberTest : public AppStateSubscriber {
35 public:
36 void OnConnected() override;
37 void OnDisconnected() override;
38 void OnAppStateChanged(int32_t pid, int32_t uid, int32_t state) override;
39 void ForceReclaim(int32_t pid, int32_t uid) override;
40 void OnTrim(SystemMemoryLevel level) override;
41 void OnRemoteDied(const wptr<IRemoteObject> &object) override;
42 };
43
OnConnected()44 void AppStateSubscriberTest::OnConnected()
45 {
46 }
47
OnDisconnected()48 void AppStateSubscriberTest::OnDisconnected()
49 {
50 }
51
OnAppStateChanged(int32_t pid,int32_t uid,int32_t state)52 void AppStateSubscriberTest::OnAppStateChanged(int32_t pid, int32_t uid, int32_t state)
53 {
54 }
55
ForceReclaim(int32_t pid,int32_t uid)56 void AppStateSubscriberTest::ForceReclaim(int32_t pid, int32_t uid)
57 {
58 }
59
OnTrim(SystemMemoryLevel level)60 void AppStateSubscriberTest::OnTrim(SystemMemoryLevel level)
61 {
62 }
63
OnRemoteDied(const wptr<IRemoteObject> & object)64 void AppStateSubscriberTest::OnRemoteDied(const wptr<IRemoteObject> &object)
65 {
66 }
67
68 class InnerkitsTest : public testing::Test {
69 public:
70 static void SetUpTestCase();
71 static void TearDownTestCase();
72 void SetUp();
73 void TearDown();
74 static std::shared_ptr<MemMgrClient> memMgrClient_;
75 static std::vector<std::shared_ptr<AppStateSubscriberTest>> appStateSubscriberTests;
76 };
77 std::vector<std::shared_ptr<AppStateSubscriberTest>> InnerkitsTest::appStateSubscriberTests =
78 std::vector<std::shared_ptr<AppStateSubscriberTest>>();
79
SetUpTestCase()80 void InnerkitsTest::SetUpTestCase()
81 {
82 }
83
TearDownTestCase()84 void InnerkitsTest::TearDownTestCase()
85 {
86 }
87
SetUp()88 void InnerkitsTest::SetUp()
89 {
90 }
91
TearDown()92 void InnerkitsTest::TearDown()
93 {
94 for (auto appStateSubscriberTest_ : appStateSubscriberTests) {
95 MemMgrClient::GetInstance().UnsubscribeAppState(*appStateSubscriberTest_);
96 }
97 appStateSubscriberTests.clear();
98 }
99
100 HWTEST_F(InnerkitsTest, GetBundlePriorityList_Test, TestSize.Level1)
101 {
102 BundlePriorityList bundlePrioList;
103 int32_t ret = MemMgrClient::GetInstance().GetBundlePriorityList(bundlePrioList);
104 EXPECT_EQ(ret, 0);
105 bundlePrioList.Show();
106 }
107
108 HWTEST_F(InnerkitsTest, GetPriorityDescTest, TestSize.Level1)
109 {
110 auto ptr = ReclaimPriorityMapping.find(RECLAIM_PRIORITY_SYSTEM);
111 EXPECT_EQ(ptr->second, "System");
112 }
113
114 HWTEST_F(InnerkitsTest, NotifyDistDevStatus_Test, TestSize.Level1)
115 {
116 int32_t ret = MemMgrClient::GetInstance().NotifyDistDevStatus(123, 456, "dist_dev_test", true);
117 EXPECT_EQ(ret, 0);
118 }
119
120 HWTEST_F(InnerkitsTest, GetKillLevelOfLmkd_Test, TestSize.Level1)
121 {
122 int32_t killLevel;
123 int32_t ret = MemMgrClient::GetInstance().GetKillLevelOfLmkd(killLevel);
124 EXPECT_EQ(ret, 0);
125 printf("lmkd kill level: %d\n", killLevel);
126 }
127
128 #ifdef USE_PURGEABLE_MEMORY
129
130 HWTEST_F(InnerkitsTest, RegisterActiveApps_Test, TestSize.Level1)
131 {
132 int32_t pid = 1234;
133 int32_t uid = 20012001;
134 int32_t ret = MemMgrClient::GetInstance().RegisterActiveApps(pid, uid);
135 EXPECT_EQ(ret, 0);
136 }
137
138 HWTEST_F(InnerkitsTest, DeregisterActiveApps_Test, TestSize.Level1)
139 {
140 int32_t pid = 1234;
141 int32_t uid = 20012001;
142 int32_t ret = MemMgrClient::GetInstance().DeregisterActiveApps(pid, uid);
143 EXPECT_EQ(ret, 0);
144 }
145
146 HWTEST_F(InnerkitsTest, SubscribeAppState_Test, TestSize.Level1)
147 {
148 std::shared_ptr<AppStateSubscriberTest> appStateSubscriberTest_1 = std::make_shared<AppStateSubscriberTest>();
149 EXPECT_NE(appStateSubscriberTest_1, nullptr);
150 int32_t ret = MemMgrClient::GetInstance().SubscribeAppState(*appStateSubscriberTest_1);
151 appStateSubscriberTests.emplace_back(appStateSubscriberTest_1);
152 EXPECT_EQ(ret, 0);
153 }
154
155 HWTEST_F(InnerkitsTest, UnsubscribeAppState_Test, TestSize.Level1)
156 {
157 std::shared_ptr<AppStateSubscriberTest> appStateSubscriberTest_1 = std::make_shared<AppStateSubscriberTest>();
158 EXPECT_NE(appStateSubscriberTest_1, nullptr);
159 MemMgrClient::GetInstance().SubscribeAppState(*appStateSubscriberTest_1);
160 appStateSubscriberTests.emplace_back(appStateSubscriberTest_1);
161 int32_t ret = MemMgrClient::GetInstance().UnsubscribeAppState(*appStateSubscriberTest_1);
162 EXPECT_EQ(ret, 0);
163 }
164
165 HWTEST_F(InnerkitsTest, GetAvailableMemory_Test, TestSize.Level1)
166 {
167 int32_t ret = MemMgrClient::GetInstance().GetAvailableMemory();
168 EXPECT_NE(ret, 0);
169 }
170 #else
171 HWTEST_F(InnerkitsTest, RegisterActiveApps_Test, TestSize.Level1)
172 {
173 int32_t pid = 1234;
174 int32_t uid = 20012001;
175 int32_t ret = MemMgrClient::GetInstance().RegisterActiveApps(pid, uid);
176 EXPECT_EQ(ret, -1);
177 }
178
179 HWTEST_F(InnerkitsTest, DeregisterActiveApps_Test, TestSize.Level1)
180 {
181 int32_t pid = 1234;
182 int32_t uid = 20012001;
183 int32_t ret = MemMgrClient::GetInstance().DeregisterActiveApps(pid, uid);
184 EXPECT_EQ(ret, -1);
185 }
186
187 HWTEST_F(InnerkitsTest, SubscribeAppState_Test, TestSize.Level1)
188 {
189 std::shared_ptr<AppStateSubscriberTest> appStateSubscriberTest_1 = std::make_shared<AppStateSubscriberTest>();
190 EXPECT_NE(appStateSubscriberTest_1, nullptr);
191 int32_t ret = MemMgrClient::GetInstance().SubscribeAppState(*appStateSubscriberTest_1);
192 appStateSubscriberTests.emplace_back(appStateSubscriberTest_1);
193 EXPECT_EQ(ret, -1);
194 }
195
196 HWTEST_F(InnerkitsTest, UnsubscribeAppState_Test, TestSize.Level1)
197 {
198 std::shared_ptr<AppStateSubscriberTest> appStateSubscriberTest_1 = std::make_shared<AppStateSubscriberTest>();
199 EXPECT_NE(appStateSubscriberTest_1, nullptr);
200 MemMgrClient::GetInstance().SubscribeAppState(*appStateSubscriberTest_1);
201 appStateSubscriberTests.emplace_back(appStateSubscriberTest_1);
202 int32_t ret = MemMgrClient::GetInstance().UnsubscribeAppState(*appStateSubscriberTest_1);
203 EXPECT_EQ(ret, -1);
204 }
205
206 HWTEST_F(InnerkitsTest, GetAvailableMemory_Test, TestSize.Level1)
207 {
208 int32_t ret = MemMgrClient::GetInstance().GetAvailableMemory();
209 EXPECT_EQ(ret, -1);
210 }
211 #endif // USE_PURGEABLE_MEMORY
212
213 HWTEST_F(InnerkitsTest, GetReclaimPriorityByPid_Test, TestSize.Level1)
214 {
215 int32_t pid = 1;
216 int32_t priority = 1001;
217 int32_t ret = MemMgrClient::GetInstance().GetReclaimPriorityByPid(pid, priority);
218 int32_t error_num = 300;
219 EXPECT_EQ(ret, error_num);
220 EXPECT_EQ(priority, 1001);
221 }
222
223 }
224 }
225