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 "iptables_type.h"
19 #include "net_manager_constants.h"
20 #include "netnative_log_wrapper.h"
21 #include "netsys_controller.h"
22 #include "netsys_native_service_proxy.h"
23
24 #include "iservice_registry.h"
25 #include "system_ability_definition.h"
26
27 namespace OHOS {
28 namespace NetsysNative {
29 using namespace testing::ext;
30 using namespace NetManagerStandard;
31 constexpr uint32_t TEST_UID = 11256;
32 class BandwidthManagerTest : public testing::Test {
33 public:
34 static void SetUpTestCase();
35 static void TearDownTestCase();
36 void SetUp();
37 void TearDown();
38 };
39
SetUpTestCase()40 void BandwidthManagerTest::SetUpTestCase()
41 {
42 NetsysController::GetInstance().BandwidthEnableDataSaver(false);
43 }
44
TearDownTestCase()45 void BandwidthManagerTest::TearDownTestCase() {}
46
SetUp()47 void BandwidthManagerTest::SetUp() {}
48
TearDown()49 void BandwidthManagerTest::TearDown() {}
50
51 /**
52 * @tc.name: BandwidthEnableDataSaverTest001
53 * @tc.desc: Test BandwidthManager BandwidthEnableDataSaver.
54 * @tc.type: FUNC
55 */
56 HWTEST_F(BandwidthManagerTest, BandwidthEnableDataSaverTest001, TestSize.Level1)
57 {
58 int32_t ret = NetsysController::GetInstance().BandwidthEnableDataSaver(true);
59 EXPECT_EQ(ret, NETMANAGER_SUCCESS);
60 }
61
62 /**
63 * @tc.name: BandwidthEnableDataSaverTest002
64 * @tc.desc: Test BandwidthManager BandwidthEnableDataSaver.
65 * @tc.type: FUNC
66 */
67 HWTEST_F(BandwidthManagerTest, BandwidthEnableDataSaverTest002, TestSize.Level1)
68 {
69 int32_t ret = NetsysController::GetInstance().BandwidthEnableDataSaver(false);
70 EXPECT_EQ(ret, NETMANAGER_SUCCESS);
71 }
72
73 /**
74 * @tc.name: BandwidthEnableDataSaverTest003
75 * @tc.desc: Test BandwidthManager BandwidthEnableDataSaver.
76 * @tc.type: FUNC
77 */
78 HWTEST_F(BandwidthManagerTest, BandwidthEnableDataSaverTest003, TestSize.Level1)
79 {
80 int32_t ret = NetsysController::GetInstance().BandwidthEnableDataSaver(true);
81 EXPECT_EQ(ret, NETMANAGER_SUCCESS);
82 ret = NetsysController::GetInstance().BandwidthEnableDataSaver(true);
83 EXPECT_EQ(ret, NETMANAGER_ERROR);
84 }
85
86 /**
87 * @tc.name: BandwidthEnableDataSaverTest004
88 * @tc.desc: Test BandwidthManager BandwidthEnableDataSaver.
89 * @tc.type: FUNC
90 */
91 HWTEST_F(BandwidthManagerTest, BandwidthEnableDataSaverTest004, TestSize.Level1)
92 {
93 int32_t ret = NetsysController::GetInstance().BandwidthEnableDataSaver(false);
94 EXPECT_EQ(ret, NETMANAGER_SUCCESS);
95 ret = NetsysController::GetInstance().BandwidthEnableDataSaver(false);
96 EXPECT_EQ(ret, NETMANAGER_ERROR);
97 }
98
99 /**
100 * @tc.name: BandwidthSetIfaceQuotaTest001
101 * @tc.desc: Test BandwidthManager BandwidthSetIfaceQuota.
102 * @tc.type: FUNC
103 */
104 HWTEST_F(BandwidthManagerTest, BandwidthSetIfaceQuotaTest001, TestSize.Level1)
105 {
106 int32_t ret = NetsysController::GetInstance().BandwidthSetIfaceQuota("iface0", 2097152);
107 EXPECT_EQ(ret, NETMANAGER_SUCCESS);
108 }
109
110 /**
111 * @tc.name: BandwidthSetIfaceQuotaTest002
112 * @tc.desc: Test BandwidthManager BandwidthSetIfaceQuota.
113 * @tc.type: FUNC
114 */
115 HWTEST_F(BandwidthManagerTest, BandwidthSetIfaceQuotaTest002, TestSize.Level1)
116 {
117 int32_t ret = NetsysController::GetInstance().BandwidthSetIfaceQuota("wlan0", 2097152);
118 EXPECT_EQ(ret, NETMANAGER_SUCCESS);
119 }
120
121 /**
122 * @tc.name: BandwidthSetIfaceQuotaTest003
123 * @tc.desc: Test BandwidthManager BandwidthSetIfaceQuota.
124 * @tc.type: FUNC
125 */
126 HWTEST_F(BandwidthManagerTest, BandwidthSetIfaceQuotaTest003, TestSize.Level1)
127 {
128 int32_t ret = NetsysController::GetInstance().BandwidthSetIfaceQuota("wlan0", 5000000000000000);
129 EXPECT_EQ(ret, NETMANAGER_SUCCESS);
130 }
131
132 /**
133 * @tc.name: BandwidthSetIfaceQuotaTest004
134 * @tc.desc: Test BandwidthManager BandwidthSetIfaceQuota.
135 * @tc.type: FUNC
136 */
137 HWTEST_F(BandwidthManagerTest, BandwidthSetIfaceQuotaTest004, TestSize.Level1)
138 {
139 int32_t ret = NetsysController::GetInstance().BandwidthSetIfaceQuota("wlan0", 5000000000000);
140 EXPECT_EQ(ret, NETMANAGER_SUCCESS);
141 ret = NetsysController::GetInstance().BandwidthSetIfaceQuota("wlan0", 2097152);
142 EXPECT_EQ(ret, NETMANAGER_SUCCESS);
143 }
144
145 /**
146 * @tc.name: BandwidthRemoveIfaceQuotaTest001
147 * @tc.desc: Test BandwidthManager BandwidthRemoveIfaceQuota.
148 * @tc.type: FUNC
149 */
150 HWTEST_F(BandwidthManagerTest, BandwidthRemoveIfaceQuotaTest001, TestSize.Level1)
151 {
152 int32_t ret = NetsysController::GetInstance().BandwidthRemoveIfaceQuota("iface0");
153 EXPECT_EQ(ret, NETMANAGER_SUCCESS);
154 }
155
156 /**
157 * @tc.name: BandwidthRemoveIfaceQuotaTest002
158 * @tc.desc: Test BandwidthManager BandwidthRemoveIfaceQuota.
159 * @tc.type: FUNC
160 */
161 HWTEST_F(BandwidthManagerTest, BandwidthRemoveIfaceQuotaTest002, TestSize.Level1)
162 {
163 int32_t ret = NetsysController::GetInstance().BandwidthRemoveIfaceQuota("wlan0");
164 EXPECT_EQ(ret, NETMANAGER_SUCCESS);
165 }
166
167 /**
168 * @tc.name: BandwidthAddDeniedListTest001
169 * @tc.desc: Test BandwidthManager BandwidthAddDeniedList.
170 * @tc.type: FUNC
171 */
172 HWTEST_F(BandwidthManagerTest, BandwidthAddDeniedListTest001, TestSize.Level1)
173 {
174 int32_t ret = NetsysController::GetInstance().BandwidthAddDeniedList(TEST_UID);
175 EXPECT_EQ(ret, NETMANAGER_SUCCESS);
176 ret = NetsysController::GetInstance().BandwidthRemoveDeniedList(TEST_UID);
177 ASSERT_EQ(ret, NETMANAGER_SUCCESS);
178 }
179
180 /**
181 * @tc.name: BandwidthAddDeniedListTest002
182 * @tc.desc: Test BandwidthManager BandwidthAddDeniedList.
183 * @tc.type: FUNC
184 */
185 HWTEST_F(BandwidthManagerTest, BandwidthAddDeniedListTest002, TestSize.Level1)
186 {
187 int32_t ret = NetsysController::GetInstance().BandwidthAddDeniedList(TEST_UID);
188 EXPECT_EQ(ret, NETMANAGER_SUCCESS);
189 ret = NetsysController::GetInstance().BandwidthAddDeniedList(TEST_UID);
190 EXPECT_EQ(ret, NETMANAGER_ERROR);
191 }
192
193 /**
194 * @tc.name: BandwidthRemoveDeniedListTest001
195 * @tc.desc: Test BandwidthManager BandwidthRemoveDeniedList.
196 * @tc.type: FUNC
197 */
198 HWTEST_F(BandwidthManagerTest, BandwidthRemoveDeniedListTest001, TestSize.Level1)
199 {
200 int32_t ret = NetsysController::GetInstance().BandwidthRemoveDeniedList(TEST_UID);
201 EXPECT_EQ(ret, NETMANAGER_SUCCESS);
202 }
203
204 /**
205 * @tc.name: BandwidthRemoveDeniedListTest002
206 * @tc.desc: Test BandwidthManager BandwidthRemoveDeniedList.
207 * @tc.type: FUNC
208 */
209 HWTEST_F(BandwidthManagerTest, BandwidthRemoveDeniedListTest002, TestSize.Level1)
210 {
211 int32_t ret = NetsysController::GetInstance().BandwidthAddDeniedList(TEST_UID);
212 ASSERT_EQ(ret, NETMANAGER_SUCCESS);
213 ret = NetsysController::GetInstance().BandwidthRemoveDeniedList(TEST_UID);
214 EXPECT_EQ(ret, NETMANAGER_SUCCESS);
215 ret = NetsysController::GetInstance().BandwidthRemoveDeniedList(TEST_UID);
216 EXPECT_EQ(ret, NETMANAGER_ERROR);
217 }
218
219 /**
220 * @tc.name: BandwidthAddAllowedListTest001
221 * @tc.desc: Test BandwidthManager BandwidthAddAllowedList.
222 * @tc.type: FUNC
223 */
224 HWTEST_F(BandwidthManagerTest, BandwidthAddAllowedListTest001, TestSize.Level1)
225 {
226 int32_t ret = NetsysController::GetInstance().BandwidthAddAllowedList(TEST_UID);
227 EXPECT_EQ(ret, NETMANAGER_SUCCESS);
228 }
229
230 /**
231 * @tc.name: BandwidthAddAllowedListTest002
232 * @tc.desc: Test BandwidthManager BandwidthAddAllowedList.
233 * @tc.type: FUNC
234 */
235 HWTEST_F(BandwidthManagerTest, BandwidthAddAllowedListTest002, TestSize.Level1)
236 {
237 int32_t ret = NetsysController::GetInstance().BandwidthAddAllowedList(TEST_UID);
238 EXPECT_EQ(ret, NETMANAGER_ERROR);
239 }
240
241 /**
242 * @tc.name: BandwidthRemoveAllowedListTest001
243 * @tc.desc: Test BandwidthManager BandwidthRemoveAllowedList.
244 * @tc.type: FUNC
245 */
246 HWTEST_F(BandwidthManagerTest, BandwidthRemoveAllowedListTest001, TestSize.Level1)
247 {
248 int32_t ret = NetsysController::GetInstance().BandwidthRemoveAllowedList(TEST_UID);
249 EXPECT_EQ(ret, NETMANAGER_SUCCESS);
250 }
251
252 /**
253 * @tc.name: BandwidthRemoveAllowedListTest002
254 * @tc.desc: Test BandwidthManager BandwidthRemoveAllowedList.
255 * @tc.type: FUNC
256 */
257 HWTEST_F(BandwidthManagerTest, BandwidthRemoveAllowedListTest002, TestSize.Level1)
258 {
259 int32_t ret = NetsysController::GetInstance().BandwidthAddAllowedList(TEST_UID);
260 ASSERT_EQ(ret, NETMANAGER_SUCCESS);
261 ret = NetsysController::GetInstance().BandwidthRemoveAllowedList(TEST_UID);
262 EXPECT_EQ(ret, NETMANAGER_SUCCESS);
263 ret = NetsysController::GetInstance().BandwidthRemoveAllowedList(TEST_UID);
264 EXPECT_EQ(ret, NETMANAGER_ERROR);
265 }
266 } // namespace NetsysNative
267 } // namespace OHOS
268