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 "memcg_mgr.h"
23 #undef private
24 #undef protected
25
26 namespace OHOS {
27 namespace Memory {
28 using namespace testing;
29 using namespace testing::ext;
30
31 class MemcgMgrTest : public testing::Test {
32 public:
33 static void SetUpTestCase();
34 static void TearDownTestCase();
35 void SetUp();
36 void TearDown();
37 };
38
SetUpTestCase()39 void MemcgMgrTest::SetUpTestCase()
40 {
41 }
42
TearDownTestCase()43 void MemcgMgrTest::TearDownTestCase()
44 {
45 }
46
SetUp()47 void MemcgMgrTest::SetUp()
48 {
49 }
50
TearDown()51 void MemcgMgrTest::TearDown()
52 {
53 }
54
55 HWTEST_F(MemcgMgrTest, GetRootMemcgTest, TestSize.Level1)
56 {
57 Memcg* memcg = MemcgMgr::GetInstance().GetRootMemcg();
58 EXPECT_EQ(memcg != nullptr, true);
59 EXPECT_STREQ(memcg->GetMemcgPath_().c_str(), "/dev/memcg");
60 }
61
62 HWTEST_F(MemcgMgrTest, SetRootMemcgParaTest, TestSize.Level1)
63 {
64 EXPECT_EQ(MemcgMgr::GetInstance().SetRootMemcgPara(), true);
65 }
66
67 HWTEST_F(MemcgMgrTest, GetUserMemcgTest, TestSize.Level1)
68 {
69 unsigned int memcgId = 123456u; // ensure it is a new ID
70 EXPECT_EQ(MemcgMgr::GetInstance().GetUserMemcg(memcgId) == nullptr, true);
71 }
72
73 HWTEST_F(MemcgMgrTest, AddUserMemcgTest, TestSize.Level1)
74 {
75 unsigned int memcgId = 123456u; // ensure it is a new ID
76 EXPECT_EQ(MemcgMgr::GetInstance().GetUserMemcg(memcgId) == nullptr, true);
77 EXPECT_EQ(MemcgMgr::GetInstance().AddUserMemcg(memcgId) != nullptr, true);
78 EXPECT_EQ(MemcgMgr::GetInstance().GetUserMemcg(memcgId) != nullptr, true);
79 EXPECT_EQ(MemcgMgr::GetInstance().RemoveUserMemcg(memcgId), true);
80 EXPECT_EQ(MemcgMgr::GetInstance().GetUserMemcg(memcgId) == nullptr, true);
81 }
82
83 HWTEST_F(MemcgMgrTest, RemoveUserMemcgTest, TestSize.Level1)
84 {
85 unsigned int memcgId = 123456u; // ensure it is a new ID
86 EXPECT_EQ(MemcgMgr::GetInstance().AddUserMemcg(memcgId) != nullptr, true);
87 EXPECT_EQ(MemcgMgr::GetInstance().RemoveUserMemcg(memcgId), true);
88 EXPECT_EQ(MemcgMgr::GetInstance().GetUserMemcg(memcgId) == nullptr, true);
89 EXPECT_EQ(MemcgMgr::GetInstance().RemoveUserMemcg(memcgId), false);
90 }
91
92 HWTEST_F(MemcgMgrTest, UpdateMemcgScoreAndReclaimRatiosTest, TestSize.Level1)
93 {
94 unsigned int memcgId = 123456u; // ensure it is a new ID
95 EXPECT_EQ(MemcgMgr::GetInstance().AddUserMemcg(memcgId) != nullptr, true);
96 ReclaimRatios ratios;
97 EXPECT_EQ(MemcgMgr::GetInstance().UpdateMemcgScoreAndReclaimRatios(memcgId, 200, ratios), true);
98 MemcgMgr::GetInstance().RemoveUserMemcg(memcgId);
99 }
100
101 HWTEST_F(MemcgMgrTest, AddProcToMemcgTest, TestSize.Level1)
102 {
103 unsigned int memcgId = 123456u; // ensure it is a new ID
104 EXPECT_EQ(MemcgMgr::GetInstance().AddProcToMemcg(1234567, memcgId), true);
105 EXPECT_EQ(MemcgMgr::GetInstance().GetUserMemcg(memcgId) != nullptr, true);
106 EXPECT_EQ(MemcgMgr::GetInstance().RemoveUserMemcg(memcgId), true);
107 }
108
109 HWTEST_F(MemcgMgrTest, SwapInMemcgTest, TestSize.Level1)
110 {
111 unsigned int memcgId = 123456u; // ensure it is a new ID
112 EXPECT_EQ(MemcgMgr::GetInstance().SwapInMemcg(memcgId), false);
113 EXPECT_EQ(MemcgMgr::GetInstance().AddUserMemcg(memcgId) != nullptr, true);
114 EXPECT_EQ(MemcgMgr::GetInstance().SwapInMemcg(memcgId), true);
115 EXPECT_EQ(MemcgMgr::GetInstance().RemoveUserMemcg(memcgId), true);
116 }
117
118 HWTEST_F(MemcgMgrTest, GetMemcgSwapInfoTest, TestSize.Level1)
119 {
120 unsigned int memcgId = 123456u; // ensure it is a new ID
121 EXPECT_EQ(MemcgMgr::GetInstance().GetMemcgSwapInfo(memcgId) == nullptr, true);
122 EXPECT_EQ(MemcgMgr::GetInstance().AddUserMemcg(memcgId) != nullptr, true);
123 EXPECT_EQ(MemcgMgr::GetInstance().GetMemcgSwapInfo(memcgId) != nullptr, true);
124 EXPECT_EQ(MemcgMgr::GetInstance().RemoveUserMemcg(memcgId), true);
125 EXPECT_EQ(MemcgMgr::GetInstance().GetMemcgSwapInfo(memcgId) == nullptr, true);
126 }
127
128 HWTEST_F(MemcgMgrTest, GetMemcgMemInfoTest, TestSize.Level1)
129 {
130 unsigned int memcgId = 123456u; // ensure it is a new ID
131 EXPECT_EQ(MemcgMgr::GetInstance().GetMemcgMemInfo(memcgId) == nullptr, true);
132 EXPECT_EQ(MemcgMgr::GetInstance().AddUserMemcg(memcgId) != nullptr, true);
133 EXPECT_EQ(MemcgMgr::GetInstance().GetMemcgMemInfo(memcgId) != nullptr, true);
134 EXPECT_EQ(MemcgMgr::GetInstance().RemoveUserMemcg(memcgId), true);
135 EXPECT_EQ(MemcgMgr::GetInstance().GetMemcgMemInfo(memcgId) == nullptr, true);
136 }
137 }
138 }
139