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 #include <securec.h>
18
19 #include "bus_center_event.h"
20 #include "lnn_net_ledger_mock.h"
21 #include "lnn_sync_info_manager.h"
22 #include "lnn_service_mock.h"
23 #include "lnn_topo_manager.h"
24 #include "lnn_trans_mock.h"
25 #include "message_handler.h"
26 #include "softbus_adapter_mem.h"
27 #include "softbus_common.h"
28 #include "softbus_errcode.h"
29 #include "softbus_log.h"
30
31 #define LNN_RELATION_JOIN_THREAD 1
32 #define LNN_RELATION_ERROR 0
33 namespace OHOS {
34 using namespace testing::ext;
35 using namespace testing;
36
37 constexpr char UDID[] = "123456789";
38 constexpr char PEER1_UDID[] = "123456789";
39 constexpr uint32_t LEN = CONNECTION_ADDR_MAX + 1;
40 constexpr uint32_t LEN2 = CONNECTION_ADDR_MAX;
41 constexpr uint32_t LEN3 = 8;
42 constexpr int32_t CHANNELID = 0;
43
44 class LnnTopoManagerTest : public testing::Test {
45 public:
46 static void SetUpTestCase();
47 static void TearDownTestCase();
48 void SetUp();
49 void TearDown();
50 };
51
SetUpTestCase()52 void LnnTopoManagerTest::SetUpTestCase()
53 {
54 LooperInit();
55 }
56
TearDownTestCase()57 void LnnTopoManagerTest::TearDownTestCase()
58 {
59 LooperDeinit();
60 }
61
SetUp()62 void LnnTopoManagerTest::SetUp()
63 {
64 }
65
TearDown()66 void LnnTopoManagerTest::TearDown()
67 {
68 }
69
GetEventHandler(LnnEventType event,LnnEventHandler & handler)70 static bool GetEventHandler(LnnEventType event, LnnEventHandler &handler)
71 {
72 if (LnnServicetInterfaceMock::g_lnnEventHandlers.find(event) !=
73 LnnServicetInterfaceMock::g_lnnEventHandlers.end()) {
74 handler = LnnServicetInterfaceMock::g_lnnEventHandlers[event];
75 return true;
76 }
77 return false;
78 }
79
InitMock(LnnTransInterfaceMock & transMock,LnnServicetInterfaceMock & serviceMock)80 void InitMock(LnnTransInterfaceMock &transMock, LnnServicetInterfaceMock &serviceMock)
81 {
82 EXPECT_CALL(transMock, TransRegisterNetworkingChannelListener).WillRepeatedly(
83 LnnTransInterfaceMock::ActionOfTransRegister);
84 LnnInitSyncInfoManager();
85 ON_CALL(serviceMock, LnnRegisterEventHandler).WillByDefault(
86 LnnServicetInterfaceMock::ActionOfLnnRegisterEventHandler);
87 int ret = LnnInitTopoManager();
88 EXPECT_TRUE(ret == SOFTBUS_OK);
89 }
90
DeinitMock(LnnTransInterfaceMock & transMock,LnnServicetInterfaceMock & serviceMock)91 void DeinitMock(LnnTransInterfaceMock &transMock, LnnServicetInterfaceMock &serviceMock)
92 {
93 ON_CALL(serviceMock, LnnUnregisterEventHandler).WillByDefault(Return());
94 LnnDeinitTopoManager();
95 LnnDeinitSyncInfoManager();
96 }
97
98 /*
99 * @tc.name: LNN_GET_RELATION_TEST_001
100 * @tc.desc: len is not CONNECTION_ADDR_MAX return SOFTBUS_INVALID_PARAM
101 * @tc.type: FUNC
102 * @tc.require: I5OMIK
103 */
104 HWTEST_F(LnnTopoManagerTest, LNN_GET_RELATION_TEST_001, TestSize.Level1)
105 {
106 uint8_t num = 0;
107 int ret = LnnGetRelation(UDID, PEER1_UDID, &num, LEN);
108 EXPECT_TRUE(ret == SOFTBUS_INVALID_PARAM);
109 }
110
111 /*
112 * @tc.name: LNN_GET_RELATION_TEST_002
113 * @tc.desc: Udid and PeerUdid not find return SOFTBUS_NOT_FIND
114 * @tc.type: FUNC
115 * @tc.require: I5OMIK
116 */
117 HWTEST_F(LnnTopoManagerTest, LNN_GET_RELATION_TEST_002, TestSize.Level1)
118 {
119 NiceMock<LnnServicetInterfaceMock> serviceMock;
120 NiceMock<LnnNetLedgertInterfaceMock> ledgerMock;
121 NiceMock<LnnTransInterfaceMock> transMock;
122 InitMock(transMock, serviceMock);
123 uint8_t num = 0;
124 int ret = LnnGetRelation(UDID, PEER1_UDID, &num, LEN2);
125 EXPECT_TRUE(ret == SOFTBUS_NOT_FIND);
126 DeinitMock(transMock, serviceMock);
127 }
128
129 /*
130 * @tc.name: LNN_GET_ALL_RELATION_TEST_001
131 * @tc.desc: relationNum is NULL return SOFTBUS_INVALID_PARAM
132 * @tc.type: FUNC
133 * @tc.require: I5OMIK
134 */
135 HWTEST_F(LnnTopoManagerTest, LNN_GET_ALL_RELATION_TEST_001, TestSize.Level1)
136 {
137 uint32_t *relationNum = nullptr;
138 LnnRelation *relation = nullptr;
139 int ret = LnnGetAllRelation(&relation, relationNum);
140 EXPECT_TRUE(ret == SOFTBUS_INVALID_PARAM);
141 }
142
143 /*
144 * @tc.name: LNN_GET_RELATION_TEST_002
145 * @tc.desc: *invalid parameter
146 * @tc.type: FUNC
147 * @tc.require: I5OMIK
148 */
149 HWTEST_F(LnnTopoManagerTest, LNN_GET_ALL_RELATION_TEST_002, TestSize.Level1)
150 {
151 NiceMock<LnnServicetInterfaceMock> serviceMock;
152 NiceMock<LnnNetLedgertInterfaceMock> ledgerMock;
153 NiceMock<LnnTransInterfaceMock> transMock;
154 InitMock(transMock, serviceMock);
155 uint32_t num = 0;
156 LnnRelation *relation = nullptr;
157 int ret = LnnGetAllRelation(&relation, &num);
158 EXPECT_TRUE(ret == SOFTBUS_OK);
159 num = 1;
160 ret = LnnGetAllRelation(&relation, &num);
161 SoftBusFree(relation);
162 EXPECT_TRUE(ret == SOFTBUS_OK);
163 DeinitMock(transMock, serviceMock);
164 }
165
166 /*
167 * @tc.name: LNN_INIT_TOPO_MANAGER_TEST_001
168 * @tc.desc: test LnnInitTopoManage
169 * @tc.type: FUNC
170 * @tc.require: I5OMIK
171 */
172 HWTEST_F(LnnTopoManagerTest, LNN_INIT_TOPO_MANAGER_TEST_001, TestSize.Level1)
173 {
174 NiceMock<LnnServicetInterfaceMock> serviceMock;
175 EXPECT_CALL(serviceMock, LnnRegisterEventHandler(_, _)).WillRepeatedly(Return(SOFTBUS_ERR));
176 int ret = LnnInitTopoManager();
177 EXPECT_TRUE(ret = SOFTBUS_ERR);
178 }
179
180 /*
181 * @tc.name: LNN_INIT_TOPO_MANAGER_TEST_002
182 * @tc.desc: test LnnInitTopoManage
183 * @tc.type: FUNC
184 * @tc.require: I5OMIK
185 */
186 HWTEST_F(LnnTopoManagerTest, LNN_INIT_TOPO_MANAGER_TEST_002, TestSize.Level1)
187 {
188 NiceMock<LnnServicetInterfaceMock> serviceMock;
189 NiceMock<LnnNetLedgertInterfaceMock> ledgerMock;
190 NiceMock<LnnTransInterfaceMock> transMock;
191 InitMock(transMock, serviceMock);
192
193 LnnEventHandler handler;
194 bool isGet = GetEventHandler(LNN_EVENT_RELATION_CHANGED, handler);
195 ASSERT_TRUE(isGet == true);
196
197 LnnRelationChanedEventInfo eventInfo = {
198 .basic.event = LNN_EVENT_RELATION_CHANGED,
199 .type = CONNECTION_ADDR_BR,
200 .relation = LNN_RELATION_JOIN_THREAD,
201 .isJoin = true,
202 .udid = nullptr,
203 };
204 handler(nullptr);
205 handler((const LnnEventBasicInfo *)&eventInfo);
206 eventInfo.udid = UDID;
207
208 EXPECT_CALL(ledgerMock, LnnGetLnnRelation).WillOnce(Return(SOFTBUS_INVALID_PARAM));
209 handler((const LnnEventBasicInfo *)&eventInfo);
210 SoftBusSleepMs(5500);
211
212 EXPECT_CALL(ledgerMock, LnnGetLnnRelation).WillRepeatedly(LnnNetLedgertInterfaceMock::ActionOfLnnGetLnnRelation);
213 eventInfo.relation = LNN_RELATION_ERROR;
214 handler((const LnnEventBasicInfo *)&eventInfo);
215 SoftBusSleepMs(5500);
216 }
217
218 /*
219 * @tc.name: LNN_INIT_TOPO_MANAGER_TEST_003
220 * @tc.desc: test LnnInitTopoManage
221 * @tc.type: FUNC
222 * @tc.require: I5OMIK
223 */
224 HWTEST_F(LnnTopoManagerTest, LNN_INIT_TOPO_MANAGER_TEST_003, TestSize.Level1)
225 {
226 NiceMock<LnnServicetInterfaceMock> serviceMock;
227 NiceMock<LnnNetLedgertInterfaceMock> ledgerMock;
228 NiceMock<LnnTransInterfaceMock> transMock;
229 InitMock(transMock, serviceMock);
230 LnnEventHandler handler;
231 bool isGet = GetEventHandler(LNN_EVENT_RELATION_CHANGED, handler);
232 ASSERT_TRUE(isGet == true);
233 LnnRelationChanedEventInfo eventInfo = {
234 .basic.event = LNN_EVENT_RELATION_CHANGED,
235 .type = CONNECTION_ADDR_BR,
236 .relation = LNN_RELATION_JOIN_THREAD,
237 .isJoin = true,
238 .udid = UDID,
239 };
240 EXPECT_CALL(ledgerMock, LnnGetLnnRelation).WillRepeatedly(LnnNetLedgertInterfaceMock::ActionOfLnnGetLnnRelation);
241 EXPECT_CALL(ledgerMock, LnnGetLocalStrInfo).WillOnce(Return(SOFTBUS_ERR)).WillRepeatedly(
242 LnnNetLedgertInterfaceMock::ActionOfLnnGetLocalStrInfo);
243 handler((const LnnEventBasicInfo *)&eventInfo);
244 SoftBusSleepMs(5500);
245
246 EXPECT_CALL(ledgerMock, LnnConvertDlId).WillOnce(LnnNetLedgertInterfaceMock::ActionOfLnnConvertDlId).
247 WillRepeatedly(LnnNetLedgertInterfaceMock::ActionOfLnnConvertDlId1);
248 EXPECT_CALL(ledgerMock, LnnGetAllOnlineNodeInfo).WillOnce(Return(SOFTBUS_OK));
249 handler((const LnnEventBasicInfo *)&eventInfo);
250 SoftBusSleepMs(5500);
251 }
252
253 /*
254 * @tc.name: LNN_INIT_TOPO_MANAGER_TEST_004
255 * @tc.desc: test LnnInitTopoManage
256 * @tc.type: FUNC
257 * @tc.require: I5OMIK
258 */
259 HWTEST_F(LnnTopoManagerTest, LNN_INIT_TOPO_MANAGER_TEST_004, TestSize.Level1)
260 {
261 NiceMock<LnnServicetInterfaceMock> serviceMock;
262 NiceMock<LnnNetLedgertInterfaceMock> ledgerMock;
263 NiceMock<LnnTransInterfaceMock> transMock;
264 InitMock(transMock, serviceMock);
265 char msg[LEN3] = {0};
266 *(int32_t *)msg = LNN_INFO_TYPE_TOPO_UPDATE;
267 if (memcpy_s(msg + sizeof(int32_t), LEN3 - sizeof(int32_t), "abc", strlen("abc") + 1) != EOK) {
268 SoftBusLog(SOFTBUS_LOG_LNN, SOFTBUS_LOG_ERROR, "copy sync info msg fail");
269 }
270 LnnTransInterfaceMock::g_networkListener->onChannelOpened(CHANNELID, UDID, true);
271 LnnTransInterfaceMock::g_networkListener->onMessageReceived(CHANNELID, msg, LEN3);
272 DeinitMock(transMock, serviceMock);
273 SoftBusSleepMs(20);
274 }
275 } // namespace OHOS
276