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 <algorithm>
17 #include <gtest/gtest.h>
18
19 #include "net_manager_constants.h"
20 #include "netnative_log_wrapper.h"
21 #include "netsys_controller.h"
22 #define private public
23 #define protected public
24 #include "physical_network.h"
25 #undef private
26 #undef protected
27
28 namespace OHOS {
29 namespace NetsysNative {
30 namespace {
31 using namespace testing::ext;
32 using namespace OHOS::nmd;
33 using namespace OHOS::NetManagerStandard;
34 constexpr uint16_t TEST_NETID = 2000;
35 } // namespace
36 class PhysicalNetworkTest : public testing::Test {
37 public:
38 static void SetUpTestCase();
39 static void TearDownTestCase();
40 void SetUp();
41 void TearDown();
42 static inline std::shared_ptr<PhysicalNetwork> instance_ = nullptr;
43 };
44
SetUpTestCase()45 void PhysicalNetworkTest::SetUpTestCase()
46 {
47 instance_ = std::make_shared<PhysicalNetwork>(TEST_NETID, NetworkPermission::PERMISSION_NETWORK);
48 }
49
TearDownTestCase()50 void PhysicalNetworkTest::TearDownTestCase() {}
51
SetUp()52 void PhysicalNetworkTest::SetUp() {}
53
TearDown()54 void PhysicalNetworkTest::TearDown() {}
55
56 HWTEST_F(PhysicalNetworkTest, AddInterfaceTest001, TestSize.Level1)
57 {
58 NETNATIVE_LOGI("AddInterfaceTest001 enter");
59 PhysicalNetwork physicNetwork(2, NetworkPermission::PERMISSION_NETWORK);
60 std::string interfaceName1 = "wlan0";
61 auto ifaceList = NetManagerStandard::NetsysController::GetInstance().InterfaceGetList();
62 bool wlan0NotExist = std::find(ifaceList.begin(), ifaceList.end(), interfaceName1) == ifaceList.end();
63 if (wlan0NotExist) {
64 return;
65 }
66 int32_t ret = physicNetwork.AddInterface(interfaceName1);
67 EXPECT_EQ(ret, 0);
68 physicNetwork.AddDefault();
69 physicNetwork.interfaces_.insert(interfaceName1);
70 ret = physicNetwork.AddInterface(interfaceName1);
71 EXPECT_EQ(ret, 0);
72 std::string interfaceName2 = "eth1";
73 ifaceList = NetManagerStandard::NetsysController::GetInstance().InterfaceGetList();
74 bool eth1Exist = std::find(ifaceList.begin(), ifaceList.end(), interfaceName2) != ifaceList.end();
75 if (eth1Exist) {
76 ret = physicNetwork.AddInterface(interfaceName2);
77 EXPECT_EQ(ret, 0);
78 }
79
80 ret = physicNetwork.RemoveInterface(interfaceName1);
81 physicNetwork.RemoveDefault();
82 physicNetwork.RemoveInterface(interfaceName2);
83 std::string interfaceName3 = "eth2";
84 physicNetwork.RemoveInterface(interfaceName3);
85 physicNetwork.IsPhysical();
86 physicNetwork.GetPermission();
87 physicNetwork.GetNetworkType();
88 }
89
90 HWTEST_F(PhysicalNetworkTest, AddInterfaceTest002, TestSize.Level1)
91 {
92 std::string interfaceName = "test";
93 auto ret = instance_->AddInterface(interfaceName);
94 EXPECT_EQ(ret, NETMANAGER_ERROR);
95 interfaceName = "wlan0";
96 ret = instance_->AddInterface(interfaceName);
97 EXPECT_EQ(ret, NETMANAGER_SUCCESS);
98 interfaceName = "wlan1";
99 auto ifaceList = NetManagerStandard::NetsysController::GetInstance().InterfaceGetList();
100 bool wlan1Exist = std::find(ifaceList.begin(), ifaceList.end(), interfaceName) != ifaceList.end();
101 if (wlan1Exist) {
102 return;
103 }
104 ret = instance_->AddInterface(interfaceName);
105 EXPECT_EQ(ret, NETMANAGER_ERROR);
106 }
107
108 HWTEST_F(PhysicalNetworkTest, AddInterfaceTest003, TestSize.Level1)
109 {
110 instance_->interfaces_.clear();
111 std::string interfaceName = "wlan0";
112 instance_->isDefault_ = true;
113 auto ret = instance_->AddInterface(interfaceName);
114 EXPECT_EQ(ret, NETMANAGER_SUCCESS);
115 }
116
117 HWTEST_F(PhysicalNetworkTest, AddInterfaceTest004, TestSize.Level1)
118 {
119 instance_->interfaces_.clear();
120 std::string interfaceName = "wlan0";
121 instance_->isDefault_ = false;
122 auto ret = instance_->AddInterface(interfaceName);
123 EXPECT_EQ(ret, NETMANAGER_SUCCESS);
124 }
125
126 HWTEST_F(PhysicalNetworkTest, RemoveInterfaceTest001, TestSize.Level1)
127 {
128 std::string interfaceName = "test";
129 auto ret = instance_->RemoveInterface(interfaceName);
130 EXPECT_EQ(ret, NETMANAGER_SUCCESS);
131 interfaceName = "wlan0";
132 ret = instance_->RemoveInterface(interfaceName);
133 EXPECT_EQ(ret, NETMANAGER_ERROR);
134 interfaceName = "wlan1";
135 ret = instance_->RemoveInterface(interfaceName);
136 EXPECT_EQ(ret, NETMANAGER_SUCCESS);
137 }
138
139 HWTEST_F(PhysicalNetworkTest, RemoveInterfaceTest002, TestSize.Level1)
140 {
141 instance_->interfaces_.clear();
142 std::string interfaceName = "wlan0";
143 instance_->isDefault_ = true;
144 auto ret = instance_->AddInterface(interfaceName);
145 ret = instance_->RemoveInterface(interfaceName);
146 EXPECT_EQ(ret, NETMANAGER_ERROR);
147 }
148
149 HWTEST_F(PhysicalNetworkTest, RemoveInterfaceTest003, TestSize.Level1)
150 {
151 instance_->interfaces_.clear();
152 std::string interfaceName = "wlan0";
153 instance_->isDefault_ = false;
154 auto ret = instance_->AddInterface(interfaceName);
155 ret = instance_->RemoveInterface(interfaceName);
156 EXPECT_EQ(ret, NETMANAGER_ERROR);
157 }
158
159 HWTEST_F(PhysicalNetworkTest, RemoveInterfaceTest004, TestSize.Level1)
160 {
161 instance_->interfaces_.clear();
162 std::string interfaceName = "wlan0";
163 instance_->isDefault_ = false;
164 auto ret = instance_->RemoveInterface(interfaceName);
165 EXPECT_EQ(ret, NETMANAGER_SUCCESS);
166 }
167
168 HWTEST_F(PhysicalNetworkTest, AddDefaultTest001, TestSize.Level1)
169 {
170 instance_->AddDefault();
171 EXPECT_TRUE(instance_->isDefault_);
172 instance_->RemoveDefault();
173 EXPECT_FALSE(instance_->isDefault_);
174 }
175
176 HWTEST_F(PhysicalNetworkTest, RemoveDefaultTest001, TestSize.Level1)
177 {
178 instance_->RemoveDefault();
179 EXPECT_FALSE(instance_->isDefault_);
180 }
181 } // namespace NetsysNative
182 } // namespace OHOS
183