1 /*
2 * Copyright (c) 2022-2023 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 "common_list.h"
20 #include "bus_center_decision_center.h"
21 #include "bus_center_decision_center_deps_mock.h"
22 #include "message_handler.h"
23 #include "softbus_conn_interface.h"
24 #include "bus_center_manager.h"
25 #include "lnn_distributed_net_ledger.h"
26 #include "lnn_net_builder.h"
27 #include "softbus_adapter_mem.h"
28 #include "softbus_adapter_thread.h"
29 #include "softbus_error_code.h"
30 #include "softbus_utils.h"
31 #include "softbus_bus_center.h"
32 #include "bus_center_event.h"
33
34 using namespace testing;
35 using namespace testing::ext;
36 #define NETWORK_ID_BUF_LEN 65
37 #define NODE_NETWORK_ID "ABCDEF00ABCDEF00ABCDEF00ABCDEF00ABCDEF00ABCDEF00ABCDEF00ABCDEF0"
38
39 namespace OHOS {
40 LnnEnhanceFuncList g_lnnEnhanceFuncListTest = { nullptr };
41
MockLnnVirtualLinkInit(void)42 static int32_t MockLnnVirtualLinkInit(void)
43 {
44 return SOFTBUS_OK;
45 }
46
MockLnnVirtualLinkDeinit(void)47 static void MockLnnVirtualLinkDeinit(void)
48 {
49 return;
50 }
51
52 class BusCenterDecisionTest : public testing::Test {
53 protected:
54 static void SetUpTestCase(void);
55 static void TearDownTestCase(void);
56 void SetUp();
57 void TearDown();
58 };
59
SetUpTestCase(void)60 void BusCenterDecisionTest::SetUpTestCase(void)
61 {
62 }
63
TearDownTestCase(void)64 void BusCenterDecisionTest::TearDownTestCase(void)
65 {
66 }
67
SetUp(void)68 void BusCenterDecisionTest::SetUp(void)
69 {
70 }
71
TearDown(void)72 void BusCenterDecisionTest::TearDown(void)
73 {
74 }
75
76 /*
77 * @tc.name: BusCenterDecisionTest001
78 * @tc.desc:
79 * @tc.type: FUNC
80 * @tc.require: 1
81 */
82 HWTEST_F(BusCenterDecisionTest, BusCenterDecisionTest001, TestSize.Level1)
83 {
84 ConnectOption option;
85 memset_s(&option, sizeof(ConnectOption), 0, sizeof(ConnectOption));
86 option.type = CONNECT_BR;
87 EXPECT_NO_FATAL_FAILURE(LnnDCReportConnectException(&option, 1));
88 option.type = CONNECT_P2P;
89 EXPECT_NO_FATAL_FAILURE(LnnDCReportConnectException(&option, 1));
90 }
91
92 /*
93 * @tc.name: BusCenterDecisionTest002
94 * @tc.desc:bus center decision test
95 * @tc.type: FUNC
96 * @tc.require: 1
97 */
98 HWTEST_F(BusCenterDecisionTest, BusCenterDecisionTest002, TestSize.Level1)
99 {
100 ConnectOption option;
101 memset_s(&option, sizeof(ConnectOption), 0, sizeof(ConnectOption));
102 option.type = CONNECT_BR;
103 EXPECT_NO_FATAL_FAILURE(LnnDCClearConnectException(&option));
104 option.type = CONNECT_P2P;
105 EXPECT_NO_FATAL_FAILURE(LnnDCClearConnectException(&option));
106 }
107
108 /*
109 * @tc.name: BusCenterDecisionTest003
110 * @tc.desc:bus center decision test
111 * @tc.type: FUNC
112 * @tc.require: 1
113 */
114 HWTEST_F(BusCenterDecisionTest, BusCenterDecisionTest003, TestSize.Level1)
115 {
116 NodeBasicInfo info;
117 (void)memset_s(&info, sizeof(info), 0, sizeof(info));
118 bool isOnline = true;
119 EXPECT_NO_FATAL_FAILURE(LnnDCProcessOnlineState(isOnline, &info));
120 isOnline = false;
121 EXPECT_NO_FATAL_FAILURE(LnnDCProcessOnlineState(isOnline, &info));
122 (void)strncpy_s(info.networkId, NETWORK_ID_BUF_LEN, NODE_NETWORK_ID, strlen(NODE_NETWORK_ID));
123 EXPECT_NO_FATAL_FAILURE(LnnDCProcessOnlineState(isOnline, &info));
124 }
125
126 /*
127 * @tc.name: BusCenterDecisionTest004
128 * @tc.desc:bus center decision test
129 * @tc.type: FUNC
130 * @tc.require: 1
131 */
132 HWTEST_F(BusCenterDecisionTest, BusCenterDecisionTest004, TestSize.Level1)
133 {
134 NiceMock<BusCenterDecisionCenterDepsInterfaceMock> BusCenterDecisionMock;
135 EXPECT_CALL(BusCenterDecisionMock, CreateSoftBusList).WillOnce(Return(nullptr));
136 int32_t ret = InitDecisionCenter();
137 EXPECT_NE(ret, SOFTBUS_OK);
138 }
139
140 /*
141 * @tc.name: BusCenterDecisionTest005
142 * @tc.desc:InitDecisionCenter test
143 * @tc.type: FUNC
144 * @tc.require: 1
145 */
146 HWTEST_F(BusCenterDecisionTest, BusCenterDecisionTest005, TestSize.Level1)
147 {
148 // list will free when go to TransSrvDataListDeinit
149 SoftBusList *list = (SoftBusList *)SoftBusCalloc(sizeof(SoftBusList));
150 ASSERT_TRUE(list != nullptr);
151 SoftBusMutexAttr mutexAttr;
152 mutexAttr.type = SOFTBUS_MUTEX_RECURSIVE;
153 SoftBusMutexInit(&list->lock, &mutexAttr);
154 ListInit(&list->list);
155 NiceMock<BusCenterDecisionCenterDepsInterfaceMock> BusCenterDecisionMock;
156 EXPECT_CALL(BusCenterDecisionMock, CreateSoftBusList).WillOnce(Return(list));
157 EXPECT_CALL(BusCenterDecisionMock, LnnEnhanceFuncListGet).WillRepeatedly(Return(&g_lnnEnhanceFuncListTest));
158 g_lnnEnhanceFuncListTest.lnnVirtualLinkInit = MockLnnVirtualLinkInit;
159 g_lnnEnhanceFuncListTest.lnnVirtualLinkDeinit = MockLnnVirtualLinkDeinit;
160 int32_t ret = InitDecisionCenter();
161 EXPECT_EQ(SOFTBUS_OK, ret);
162 EXPECT_NO_FATAL_FAILURE(DeinitDecisionCenter());
163 }
164
165 /*
166 * @tc.name: BusCenterDecisionTest006
167 * @tc.desc: InitDecisionCenter test
168 * @tc.type: FUNC
169 * @tc.require: 1
170 */
171 HWTEST_F(BusCenterDecisionTest, BusCenterDecisionTest006, TestSize.Level1)
172 {
173 // list will free when go to TransSrvDataListDeinit
174 SoftBusList *list = (SoftBusList *)SoftBusCalloc(sizeof(SoftBusList));
175 ASSERT_TRUE(list != nullptr);
176 SoftBusMutexAttr mutexAttr;
177 mutexAttr.type = SOFTBUS_MUTEX_RECURSIVE;
178 SoftBusMutexInit(&list->lock, &mutexAttr);
179 ListInit(&list->list);
180 NiceMock<BusCenterDecisionCenterDepsInterfaceMock> BusCenterDecisionMock;
181 EXPECT_CALL(BusCenterDecisionMock, CreateSoftBusList).WillOnce(Return(list));
182 EXPECT_CALL(BusCenterDecisionMock, LnnEnhanceFuncListGet).WillRepeatedly(Return(&g_lnnEnhanceFuncListTest));
183 g_lnnEnhanceFuncListTest.lnnVirtualLinkInit = nullptr;
184 g_lnnEnhanceFuncListTest.lnnVirtualLinkDeinit = nullptr;
185 int32_t ret = InitDecisionCenter();
186 EXPECT_EQ(SOFTBUS_OK, ret);
187 EXPECT_NO_FATAL_FAILURE(DeinitDecisionCenter());
188 }
189
190 /*
191 * @tc.name: BusCenterDecisionTest007
192 * @tc.desc: InitDecisionCenter test
193 * @tc.type: FUNC
194 * @tc.require: 1
195 */
196 HWTEST_F(BusCenterDecisionTest, BusCenterDecisionTest007, TestSize.Level1)
197 {
198 // list will free when go to TransSrvDataListDeinit
199 SoftBusList *list = (SoftBusList *)SoftBusCalloc(sizeof(SoftBusList));
200 ASSERT_TRUE(list != nullptr);
201 SoftBusMutexAttr mutexAttr;
202 mutexAttr.type = SOFTBUS_MUTEX_RECURSIVE;
203 SoftBusMutexInit(&list->lock, &mutexAttr);
204 ListInit(&list->list);
205 NiceMock<BusCenterDecisionCenterDepsInterfaceMock> BusCenterDecisionMock;
206 EXPECT_CALL(BusCenterDecisionMock, CreateSoftBusList).WillOnce(Return(list));
207 EXPECT_CALL(BusCenterDecisionMock, LnnEnhanceFuncListGet).WillRepeatedly(Return(&g_lnnEnhanceFuncListTest));
208 g_lnnEnhanceFuncListTest.registAuthTransListener = nullptr;
209 int32_t ret = InitDecisionCenter();
210 EXPECT_EQ(SOFTBUS_OK, ret);
211 EXPECT_NO_FATAL_FAILURE(DeinitDecisionCenter());
212 }
213 }
214
215