1 /*
2 * Copyright (c) 2023-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 <gtest/gtest.h>
17
18 #include "gtest/gtest-message.h"
19 #include "gtest/gtest-test-part.h"
20 #include "gtest/hwext/gtest-ext.h"
21 #include "gtest/hwext/gtest-tag.h"
22 #define private public
23 #include "dev_interface_state.h"
24
25 namespace OHOS {
26 namespace NetManagerStandard {
27 namespace {
28 using namespace testing::ext;
29 constexpr const char *DEVICE_NAME = "ahaha";
30 } // namespace
31
32 class DevInterfaceStateTest : public testing::Test {
33 public:
34 static void SetUpTestCase();
35 static void TearDownTestCase();
36 void SetUp();
37 void TearDown();
38 };
39
SetUpTestCase()40 void DevInterfaceStateTest::SetUpTestCase() {}
41
TearDownTestCase()42 void DevInterfaceStateTest::TearDownTestCase() {}
43
SetUp()44 void DevInterfaceStateTest::SetUp() {}
45
TearDown()46 void DevInterfaceStateTest::TearDown() {}
47
48 HWTEST_F(DevInterfaceStateTest, DevInterfaceStateTest001, TestSize.Level1)
49 {
50 DevInterfaceState devInterfaceState;
51 devInterfaceState.SetDevName(DEVICE_NAME);
52 std::set<NetCap> netCaps;
53 netCaps.insert(NET_CAPABILITY_MMS);
54 devInterfaceState.SetNetCaps(netCaps);
55 devInterfaceState.SetLinkUp(true);
56 sptr<NetLinkInfo> linkInfo;
57 devInterfaceState.SetlinkInfo(linkInfo);
58 devInterfaceState.SetDhcpReqState(true);
59 bool getDhcpReqState = devInterfaceState.GetDhcpReqState();
60 EXPECT_TRUE(getDhcpReqState);
61
62 sptr<StaticConfiguration> config = nullptr;
63 devInterfaceState.UpdateLinkInfo(config);
64 config = new (std::nothrow) StaticConfiguration();
65 devInterfaceState.linkInfo_ = nullptr;
66 devInterfaceState.UpdateLinkInfo(config);
67 devInterfaceState.linkInfo_ = new (std::nothrow) NetLinkInfo();
68 devInterfaceState.UpdateLinkInfo(config);
69 std::string devName = devInterfaceState.GetDevName();
70 EXPECT_STREQ(devName.c_str(), "ahaha");
71 std::set<NetCap> getNetCaps = devInterfaceState.GetNetCaps();
72 EXPECT_FALSE(getNetCaps.empty());
73 bool linkUp = devInterfaceState.GetLinkUp();
74 EXPECT_TRUE(linkUp);
75 sptr<NetLinkInfo> getLinkInfo = devInterfaceState.GetLinkInfo();
76 EXPECT_NE(getLinkInfo, nullptr);
77 }
78
79 HWTEST_F(DevInterfaceStateTest, DevInterfaceStateTest002, TestSize.Level1)
80 {
81 DevInterfaceState devInterfaceState;
82 devInterfaceState.ifCfg_ = nullptr;
83 IPSetMode ipSetMod = devInterfaceState.GetIPSetMode();
84 EXPECT_EQ(ipSetMod, IPSetMode::STATIC);
85 devInterfaceState.connLinkState_ = DevInterfaceState::ConnLinkState::UNREGISTERED;
86 devInterfaceState.RemoteRegisterNetSupplier();
87 devInterfaceState.connLinkState_ = DevInterfaceState::ConnLinkState::REGISTERED;
88 devInterfaceState.RemoteUnregisterNetSupplier();
89 devInterfaceState.connLinkState_ = DevInterfaceState::ConnLinkState::LINK_UNAVAILABLE;
90 devInterfaceState.RemoteUpdateNetLinkInfo();
91 devInterfaceState.connLinkState_ = DevInterfaceState::ConnLinkState::LINK_AVAILABLE;
92 devInterfaceState.linkInfo_ = nullptr;
93 devInterfaceState.RemoteUpdateNetLinkInfo();
94 sptr<NetLinkInfo> linkInfo;
95 devInterfaceState.linkInfo_ = linkInfo;
96 devInterfaceState.RemoteUpdateNetLinkInfo();
97 devInterfaceState.connLinkState_ = DevInterfaceState::ConnLinkState::UNREGISTERED;
98 devInterfaceState.RemoteUpdateNetSupplierInfo();
99 devInterfaceState.connLinkState_ = DevInterfaceState::ConnLinkState::REGISTERED;
100 devInterfaceState.netSupplierInfo_ = nullptr;
101 devInterfaceState.RemoteUpdateNetSupplierInfo();
102 sptr<NetSupplierInfo> netSupplierInfo;
103 devInterfaceState.netSupplierInfo_ = netSupplierInfo;
104 devInterfaceState.RemoteUpdateNetSupplierInfo();
105 }
106
107 HWTEST_F(DevInterfaceStateTest, SetIfcfgTest002, TestSize.Level1)
108 {
109 DevInterfaceState devInterfaceState;
110 devInterfaceState.ifCfg_ = nullptr;
111 sptr<InterfaceConfiguration> ifCfg = new (std::nothrow) InterfaceConfiguration();
112 ifCfg->mode_ = STATIC;
113 devInterfaceState.SetIfcfg(ifCfg);
114 EXPECT_NE(devInterfaceState.GetIfcfg(), nullptr);
115 ifCfg->mode_ = DHCP;
116 devInterfaceState.SetIfcfg(ifCfg);
117 devInterfaceState.UpdateLinkInfo();
118 EXPECT_EQ(devInterfaceState.GetIfcfg()->mode_, DHCP);
119 }
120
121 HWTEST_F(DevInterfaceStateTest, DevInterfaceStateBranchTest001, TestSize.Level1)
122 {
123 DevInterfaceState devInterfaceState;
124 sptr<InterfaceConfiguration> ifCfg = new (std::nothrow) InterfaceConfiguration();
125 ifCfg->mode_ = STATIC;
126 devInterfaceState.SetLinkUp(true);
127 devInterfaceState.UpdateSupplierAvailable();
128 devInterfaceState.SetIfcfg(ifCfg);
129 EXPECT_TRUE(devInterfaceState.linkUp_);
130
131 IPSetMode mode = devInterfaceState.GetIPSetMode();
132 EXPECT_EQ(mode, STATIC);
133
134 devInterfaceState.SetLinkUp(false);
135 devInterfaceState.UpdateSupplierAvailable();
136 devInterfaceState.SetIfcfg(ifCfg);
137 EXPECT_FALSE(devInterfaceState.linkUp_);
138
139 devInterfaceState.RemoteRegisterNetSupplier();
140 devInterfaceState.RemoteUnregisterNetSupplier();
141 devInterfaceState.RemoteUpdateNetSupplierInfo();
142 HttpProxy httpProxy;
143 devInterfaceState.UpdateNetHttpProxy(httpProxy);
144
145 sptr<NetLinkInfo> linkInfo = nullptr;
146 devInterfaceState.SetlinkInfo(linkInfo);
147 devInterfaceState.UpdateNetHttpProxy(httpProxy);
148 EXPECT_TRUE(devInterfaceState.linkInfo_ == nullptr);
149
150 std::vector<INetAddr> ipAddrList;
151 std::vector<INetAddr> netMaskList;
152 devInterfaceState.CreateLocalRoute("", ipAddrList, netMaskList);
153 EXPECT_TRUE(devInterfaceState.linkInfo_ == nullptr);
154
155 std::string testString = "";
156 auto result = devInterfaceState.GetIpv4Prefix(testString, netMaskList);
157 EXPECT_TRUE(result.empty());
158
159 sptr<StaticConfiguration> config = nullptr;
160 devInterfaceState.UpdateLanLinkInfo(config);
161
162 INetAddr targetNetAddr = {};
163 devInterfaceState.GetTargetNetAddrWithSameFamily("", ipAddrList, targetNetAddr);
164 devInterfaceState.GetDumpInfo(testString);
165 EXPECT_TRUE(result.empty());
166 }
167
168 HWTEST_F(DevInterfaceStateTest, DevInterfaceStateBranchTest002, TestSize.Level1)
169 {
170 DevInterfaceState devInterfaceState;
171 devInterfaceState.GetNetCaps();
172
173 sptr<NetLinkInfo> linkInfo = nullptr;
174 devInterfaceState.SetlinkInfo(linkInfo);
175 sptr<StaticConfiguration> config = new (std::nothrow) StaticConfiguration();
176 devInterfaceState.UpdateLanLinkInfo(config);
177
178 sptr<InterfaceConfiguration> ifCfg = new (std::nothrow) InterfaceConfiguration();
179 ifCfg->mode_ = STATIC;
180 devInterfaceState.SetLancfg(ifCfg);
181 bool ret = devInterfaceState.IsLanIface();
182 EXPECT_FALSE(ret);
183
184 ifCfg->mode_ = LAN_STATIC;
185 devInterfaceState.SetLancfg(ifCfg);
186 ret = devInterfaceState.IsLanIface();
187 EXPECT_TRUE(ret);
188
189 ifCfg->mode_ = LAN_DHCP;
190 devInterfaceState.SetLancfg(ifCfg);
191 ret = devInterfaceState.IsLanIface();
192 EXPECT_TRUE(ret);
193 }
194
195 HWTEST_F(DevInterfaceStateTest, IsLanIfaceTest001, TestSize.Level1)
196 {
197 DevInterfaceState devInterfaceState;
198 devInterfaceState.ifCfg_ = nullptr;
199 bool ret = devInterfaceState.IsLanIface();
200 EXPECT_FALSE(ret);
201 }
202
203 HWTEST_F(DevInterfaceStateTest, RemoteRegisterNetSupplierTest001, TestSize.Level0)
204 {
205 DevInterfaceState devInterfaceState;
206 devInterfaceState.RemoteRegisterNetSupplier();
207 EXPECT_EQ(devInterfaceState.connLinkState_, DevInterfaceState::UNREGISTERED);
208 devInterfaceState.RemoteUnregisterNetSupplier();
209 EXPECT_EQ(devInterfaceState.connLinkState_, DevInterfaceState::UNREGISTERED);
210 }
211
212 HWTEST_F(DevInterfaceStateTest, RemoteRegisterAndUnregisterNetSupplierTest, TestSize.Level0)
213 {
214 DevInterfaceState devInterfaceState;
215 devInterfaceState.netCaps_.clear();
216 EXPECT_TRUE(devInterfaceState.netCaps_.empty());
217
218 devInterfaceState.connLinkState_ = DevInterfaceState::ConnLinkState::UNREGISTERED;
219 devInterfaceState.RemoteRegisterNetSupplier();
220
221 std::set<NetCap> getNetCaps = devInterfaceState.GetNetCaps();
222 EXPECT_FALSE(getNetCaps.empty());
223 EXPECT_TRUE(getNetCaps.find(NET_CAPABILITY_INTERNET) != getNetCaps.end());
224 devInterfaceState.RemoteUnregisterNetSupplier();
225 }
226
227 HWTEST_F(DevInterfaceStateTest, RemoteUpdateNetLinkInfoTest001, TestSize.Level0)
228 {
229 DevInterfaceState devInterfaceState;
230 sptr<NetLinkInfo> linkInfo = new NetLinkInfo();
231 devInterfaceState.SetlinkInfo(linkInfo);
232 devInterfaceState.RemoteUpdateNetLinkInfo();
233 EXPECT_EQ(devInterfaceState.connLinkState_, DevInterfaceState::UNREGISTERED);
234 sptr<NetLinkInfo> linkInfo1 = new NetLinkInfo();
235 INetAddr ipv6Addr;
236 ipv6Addr.family_ = 2;
237 linkInfo1->netAddrList_.push_back(ipv6Addr);
238 devInterfaceState.SetlinkInfo(linkInfo1);
239 devInterfaceState.RemoteUpdateNetLinkInfo();
240 EXPECT_EQ(devInterfaceState.connLinkState_, DevInterfaceState::UNREGISTERED);
241 }
242
243 HWTEST_F(DevInterfaceStateTest, UpdateNetHttpProxyTest001, TestSize.Level0)
244 {
245 DevInterfaceState devInterfaceState;
246 sptr<InterfaceConfiguration> ifCfg = new (std::nothrow) InterfaceConfiguration();
247 ifCfg->httpProxy_.SetUserId(123);
248 devInterfaceState.SetIfcfg(ifCfg);
249 HttpProxy httpProxy;
250 sptr<NetLinkInfo> linkInfo = new NetLinkInfo();
251 devInterfaceState.SetlinkInfo(linkInfo);
252 httpProxy.SetUserId(1234);
253 httpProxy.SetPort(8710);
254 devInterfaceState.UpdateNetHttpProxy(httpProxy);
255 EXPECT_EQ(devInterfaceState.connLinkState_, DevInterfaceState::UNREGISTERED);
256
257 ifCfg = new (std::nothrow) InterfaceConfiguration();
258 ifCfg->httpProxy_.SetUserId(123);
259 devInterfaceState.SetIfcfg(ifCfg);
260 devInterfaceState.connLinkState_ = DevInterfaceState::ConnLinkState::LINK_AVAILABLE;
261 devInterfaceState.UpdateNetHttpProxy(httpProxy);
262 EXPECT_NE(devInterfaceState.linkInfo_, nullptr);
263
264 ifCfg = new (std::nothrow) InterfaceConfiguration();
265 ifCfg->httpProxy_.SetUserId(123);
266 devInterfaceState.SetIfcfg(ifCfg);
267 sptr<NetLinkInfo> linkInfo1 = nullptr;
268 devInterfaceState.SetlinkInfo(linkInfo1);
269 devInterfaceState.UpdateNetHttpProxy(httpProxy);
270 EXPECT_EQ(devInterfaceState.linkInfo_, nullptr);
271 }
272
273 HWTEST_F(DevInterfaceStateTest, UpdateLinkInfoTest001, TestSize.Level0)
274 {
275 DevInterfaceState devInterfaceState;
276 sptr<InterfaceConfiguration> ifCfg = new (std::nothrow) InterfaceConfiguration();
277 ifCfg->mode_ = DHCP;
278 devInterfaceState.SetIfcfg(ifCfg);
279 devInterfaceState.UpdateLinkInfo();
280 EXPECT_NE(devInterfaceState.ifCfg_->mode_, STATIC);
281 EXPECT_EQ(devInterfaceState.connLinkState_, DevInterfaceState::UNREGISTERED);
282 }
283
284 HWTEST_F(DevInterfaceStateTest, UpdateLanLinkInfoTest001, TestSize.Level0)
285 {
286 DevInterfaceState devInterfaceState;
287 devInterfaceState.ifCfg_ = nullptr;
288 bool ret = devInterfaceState.IsLanIface();
289 EXPECT_FALSE(ret);
290 devInterfaceState.UpdateLanLinkInfo();
291
292 sptr<InterfaceConfiguration> ifCfg = new (std::nothrow) InterfaceConfiguration();
293 ifCfg->mode_ = DHCP;
294 devInterfaceState.SetIfcfg(ifCfg);
295 devInterfaceState.UpdateLanLinkInfo();
296 EXPECT_NE(devInterfaceState.ifCfg_->mode_, STATIC);
297 }
298
299 HWTEST_F(DevInterfaceStateTest, UpdateLanLinkInfoTest002, TestSize.Level0)
300 {
301 DevInterfaceState devInterfaceState;
302 sptr<StaticConfiguration> config = new StaticConfiguration();
303 devInterfaceState.linkInfo_ = nullptr;
304 devInterfaceState.UpdateLanLinkInfo(config);
305 EXPECT_EQ(devInterfaceState.connLinkState_, DevInterfaceState::UNREGISTERED);
306 }
307
308 HWTEST_F(DevInterfaceStateTest, UpdateLinkInfoTest002, TestSize.Level0)
309 {
310 DevInterfaceState devInterfaceState;
311 sptr<StaticConfiguration> config = new StaticConfiguration();
312 sptr<NetLinkInfo> linkInfo = new NetLinkInfo();
313 devInterfaceState.SetlinkInfo(linkInfo);
314 sptr<InterfaceConfiguration> ifCfg = new (std::nothrow) InterfaceConfiguration();
315 ifCfg->httpProxy_.SetUserId(123);
316 devInterfaceState.SetIfcfg(ifCfg);
317 devInterfaceState.UpdateLinkInfo(config);
318 EXPECT_EQ(devInterfaceState.connLinkState_, DevInterfaceState::UNREGISTERED);
319 }
320
321 HWTEST_F(DevInterfaceStateTest, GetRoutePrefixlenTest001, TestSize.Level0)
322 {
323 DevInterfaceState devInterfaceState;
324 std::string bySrcAddr = "123";
325 INetAddr inetaddr;
326 inetaddr.address_ = "192.168.1.1";
327 std::vector<INetAddr> fromAddrList = {inetaddr};
328 INetAddr targetNetAddr;
329 devInterfaceState.GetRoutePrefixlen(bySrcAddr, fromAddrList, targetNetAddr);
330 bySrcAddr = "192.168.1.1";
331 devInterfaceState.GetRoutePrefixlen(bySrcAddr, fromAddrList, targetNetAddr);
332 EXPECT_NE(targetNetAddr.prefixlen_, 0);
333 }
334
335 HWTEST_F(DevInterfaceStateTest, GetDumpInfoTest001, TestSize.Level0)
336 {
337 DevInterfaceState devInterfaceState;
338 sptr<NetLinkInfo> linkInfo = new NetLinkInfo();
339 devInterfaceState.SetlinkInfo(linkInfo);
340 devInterfaceState.netSupplierInfo_ = nullptr;
341 devInterfaceState.ifCfg_ = nullptr;
342 std::string info = "";
343 devInterfaceState.GetDumpInfo(info);
344 sptr<InterfaceConfiguration> ifCfg = new (std::nothrow) InterfaceConfiguration();
345 ifCfg->mode_ = DHCP;
346 devInterfaceState.SetIfcfg(ifCfg);
347 devInterfaceState.GetDumpInfo(info);
348 EXPECT_NE(info, "");
349 }
350 } // namespace NetManagerStandard
351 } // namespace OHOS