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 ret = physicNetwork.AddInterface(interfaceName2);
74 EXPECT_EQ(ret, 0);
75
76 ret = physicNetwork.RemoveInterface(interfaceName1);
77 physicNetwork.RemoveDefault();
78 physicNetwork.RemoveInterface(interfaceName2);
79 std::string interfaceName3 = "eth2";
80 physicNetwork.RemoveInterface(interfaceName3);
81 physicNetwork.IsPhysical();
82 physicNetwork.GetPermission();
83 physicNetwork.GetNetworkType();
84 }
85
86 HWTEST_F(PhysicalNetworkTest, AddInterfaceTest002, TestSize.Level1)
87 {
88 std::string interfaceName = "test";
89 auto ret = instance_->AddInterface(interfaceName);
90 EXPECT_EQ(ret, NETMANAGER_ERROR);
91 interfaceName = "wlan0";
92 ret = instance_->AddInterface(interfaceName);
93 EXPECT_EQ(ret, NETMANAGER_SUCCESS);
94 interfaceName = "wlan1";
95 ret = instance_->AddInterface(interfaceName);
96 EXPECT_EQ(ret, NETMANAGER_ERROR);
97 }
98
99 HWTEST_F(PhysicalNetworkTest, AddInterfaceTest003, TestSize.Level1)
100 {
101 instance_->interfaces_.clear();
102 std::string interfaceName = "wlan0";
103 instance_->isDefault_ = true;
104 auto ret = instance_->AddInterface(interfaceName);
105 EXPECT_EQ(ret, NETMANAGER_SUCCESS);
106 }
107
108 HWTEST_F(PhysicalNetworkTest, AddInterfaceTest004, TestSize.Level1)
109 {
110 instance_->interfaces_.clear();
111 std::string interfaceName = "wlan0";
112 instance_->isDefault_ = false;
113 auto ret = instance_->AddInterface(interfaceName);
114 EXPECT_EQ(ret, NETMANAGER_SUCCESS);
115 }
116
117 HWTEST_F(PhysicalNetworkTest, RemoveInterfaceTest001, TestSize.Level1)
118 {
119 std::string interfaceName = "test";
120 auto ret = instance_->RemoveInterface(interfaceName);
121 EXPECT_EQ(ret, NETMANAGER_SUCCESS);
122 interfaceName = "wlan0";
123 ret = instance_->RemoveInterface(interfaceName);
124 EXPECT_EQ(ret, NETMANAGER_ERROR);
125 interfaceName = "wlan1";
126 ret = instance_->RemoveInterface(interfaceName);
127 EXPECT_EQ(ret, NETMANAGER_SUCCESS);
128 }
129
130 HWTEST_F(PhysicalNetworkTest, RemoveInterfaceTest002, TestSize.Level1)
131 {
132 instance_->interfaces_.clear();
133 std::string interfaceName = "wlan0";
134 instance_->isDefault_ = true;
135 auto ret = instance_->AddInterface(interfaceName);
136 ret = instance_->RemoveInterface(interfaceName);
137 EXPECT_EQ(ret, NETMANAGER_ERROR);
138 }
139
140 HWTEST_F(PhysicalNetworkTest, RemoveInterfaceTest003, TestSize.Level1)
141 {
142 instance_->interfaces_.clear();
143 std::string interfaceName = "wlan0";
144 instance_->isDefault_ = false;
145 auto ret = instance_->AddInterface(interfaceName);
146 ret = instance_->RemoveInterface(interfaceName);
147 EXPECT_EQ(ret, NETMANAGER_ERROR);
148 }
149
150 HWTEST_F(PhysicalNetworkTest, RemoveInterfaceTest004, TestSize.Level1)
151 {
152 instance_->interfaces_.clear();
153 std::string interfaceName = "wlan0";
154 instance_->isDefault_ = false;
155 auto ret = instance_->RemoveInterface(interfaceName);
156 EXPECT_EQ(ret, NETMANAGER_SUCCESS);
157 }
158
159 HWTEST_F(PhysicalNetworkTest, AddDefaultTest001, TestSize.Level1)
160 {
161 instance_->AddDefault();
162 EXPECT_TRUE(instance_->isDefault_);
163 instance_->RemoveDefault();
164 EXPECT_FALSE(instance_->isDefault_);
165 }
166
167 HWTEST_F(PhysicalNetworkTest, RemoveDefaultTest001, TestSize.Level1)
168 {
169 instance_->RemoveDefault();
170 EXPECT_FALSE(instance_->isDefault_);
171 }
172 } // namespace NetsysNative
173 } // namespace OHOS
174