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
18 #include "bus_center_manager.h"
19 #include "bus_center_mock.h"
20 #include "disc_manager_mock.h"
21 #include "lnn_coap_discovery_impl.h"
22 #include "lnn_coap_discovery_impl.c"
23 #include "lnn_discovery_manager.h"
24 #include "lnn_discovery_manager.c"
25 #include "lnn_hichain_mock.h"
26 #include "softbus_errcode.h"
27
28 namespace OHOS {
29 using namespace testing;
30 using namespace testing::ext;
31
32 class LNNDiscoveryInterfaceTest : public testing::Test {
33 public:
34 static void SetUpTestCase();
35 static void TearDownTestCase();
36 void SetUp();
37 void TearDown();
38 };
39
SetUpTestCase()40 void LNNDiscoveryInterfaceTest::SetUpTestCase()
41 {
42 }
43
TearDownTestCase()44 void LNNDiscoveryInterfaceTest::TearDownTestCase()
45 {
46 }
47
SetUp()48 void LNNDiscoveryInterfaceTest::SetUp()
49 {
50 }
51
TearDown()52 void LNNDiscoveryInterfaceTest::TearDown()
53 {
54 }
55
OnRefreshDeviceFoundTest(const char * pkgName,const DeviceInfo * device,const InnerDeviceInfoAddtions * additions)56 static int32_t OnRefreshDeviceFoundTest(const char *pkgName, const DeviceInfo *device,
57 const InnerDeviceInfoAddtions *additions)
58 {
59 return SOFTBUS_OK;
60 }
61
62 static IServerDiscInnerCallback g_discInnerCb = {
63 .OnServerDeviceFound = OnRefreshDeviceFoundTest,
64 };
65
LnnCoapTest(void)66 static int32_t LnnCoapTest(void)
67 {
68 return SOFTBUS_ERR;
69 }
70
LnnCoapFuncTest(void)71 static int32_t LnnCoapFuncTest(void)
72 {
73 return SOFTBUS_OK;
74 }
75
76 /*
77 * @tc.name: LNN_PUBLISH_SERVICE_TEST_001
78 * @tc.desc: device found test
79 * @tc.type: FUNC
80 * @tc.require:
81 */
82 HWTEST_F(LNNDiscoveryInterfaceTest, LNN_PUBLISH_SERVICE_TEST_001, TestSize.Level1)
83 {
84 DeviceInfo device;
85 InnerDeviceInfoAddtions additions;
86
87 (void)memset_s(&device, sizeof(DeviceInfo), 0, sizeof(DeviceInfo));
88 (void)memset_s(&additions, sizeof(InnerDeviceInfoAddtions), 0, sizeof(InnerDeviceInfoAddtions));
89 BusCenterMock busCenterMock;
90 busCenterMock.SetupSuccessStub();
91 DeviceFound(nullptr, &additions);
92 DeviceFound(&device, &additions);
93 device.addr[0].type = CONNECTION_ADDR_WLAN;
94 DeviceFound(&device, &additions);
95 device.addr[0].type = CONNECTION_ADDR_ETH;
96 DeviceFound(&device, &additions);
97 device.addr[0].type = CONNECTION_ADDR_BR;
98 DeviceFound(&device, &additions);
99 device.addr[0].info.ip.port = 22;
100 DeviceFound(&device, &additions);
101
102 const char *pkgName = "testpkgName";
103 PublishInfo info;
104 bool isInnerRequest = false;
105 DiscManagerInterfaceMock discMock;
106 (void)memset_s(&info, sizeof(PublishInfo), 0, sizeof(PublishInfo));
107 EXPECT_CALL(discMock, DiscPublishService).WillRepeatedly(Return(SOFTBUS_ERR));
108 int32_t ret = LnnPublishService(pkgName, &info, isInnerRequest);
109 EXPECT_TRUE(ret == SOFTBUS_ERR);
110 }
111
112 /*
113 * @tc.name: LNN_PUBLISH_SERVICE_TEST_002
114 * @tc.desc: lnn publish service test
115 * @tc.type: FUNC
116 * @tc.require:
117 */
118 HWTEST_F(LNNDiscoveryInterfaceTest, LNN_PUBLISH_SERVICE_TEST_002, TestSize.Level1)
119 {
120 const char *pkgName = "testpkgName";
121 PublishInfo info;
122 bool isInnerRequest = false;
123 DiscManagerInterfaceMock discMock;
124 int32_t ret = 0;
125
126 (void)memset_s(&info, sizeof(PublishInfo), 0, sizeof(PublishInfo));
127 EXPECT_CALL(discMock, DiscPublishService).WillRepeatedly(Return(SOFTBUS_ERR));
128 ret = LnnPublishService(pkgName, &info, isInnerRequest);
129 EXPECT_TRUE(ret == SOFTBUS_ERR);
130 EXPECT_CALL(discMock, DiscPublishService).WillRepeatedly(Return(SOFTBUS_OK));
131 ret = LnnPublishService(pkgName, &info, isInnerRequest);
132 EXPECT_TRUE(ret == SOFTBUS_OK);
133 isInnerRequest = true;
134 EXPECT_CALL(discMock, DiscStartScan).WillRepeatedly(Return(SOFTBUS_ERR));
135 ret = LnnPublishService(pkgName, &info, isInnerRequest);
136 EXPECT_TRUE(ret == SOFTBUS_ERR);
137 EXPECT_CALL(discMock, DiscStartScan).WillRepeatedly(Return(SOFTBUS_OK));
138 ret = LnnPublishService(pkgName, &info, isInnerRequest);
139 EXPECT_TRUE(ret == SOFTBUS_OK);
140 }
141
142 /*
143 * @tc.name: LNN_UNPUBLISH_SERVICE_TEST_001
144 * @tc.desc: lnn unpublish service test
145 * @tc.type: FUNC
146 * @tc.require:
147 */
148 HWTEST_F(LNNDiscoveryInterfaceTest, LNN_UNPUBLISH_SERVICE_TEST_001, TestSize.Level1)
149 {
150 const char *pkgName = "testpkgName";
151 int32_t publishId = 0;
152 bool isInnerRequest = false;
153 DiscManagerInterfaceMock discMock;
154 int32_t ret = 0;
155
156 EXPECT_CALL(discMock, DiscUnPublishService).WillRepeatedly(Return(SOFTBUS_ERR));
157 ret = LnnUnPublishService(pkgName, publishId, isInnerRequest);
158 EXPECT_TRUE(ret == SOFTBUS_ERR);
159 EXPECT_CALL(discMock, DiscUnPublishService).WillRepeatedly(Return(SOFTBUS_OK));
160 ret = LnnUnPublishService(pkgName, publishId, isInnerRequest);
161 EXPECT_TRUE(ret == SOFTBUS_OK);
162 isInnerRequest = true;
163 EXPECT_CALL(discMock, DiscUnpublish).WillRepeatedly(Return(SOFTBUS_ERR));
164 ret = LnnUnPublishService(pkgName, publishId, isInnerRequest);
165 EXPECT_TRUE(ret == SOFTBUS_ERR);
166 EXPECT_CALL(discMock, DiscUnpublish).WillRepeatedly(Return(SOFTBUS_OK));
167 ret = LnnUnPublishService(pkgName, publishId, isInnerRequest);
168 EXPECT_TRUE(ret == SOFTBUS_OK);
169 }
170
171 /*
172 * @tc.name: LNN_START_DISC_DEVICE_TEST_001
173 * @tc.desc: lnn start disc device test
174 * @tc.type: FUNC
175 * @tc.require:
176 */
177 HWTEST_F(LNNDiscoveryInterfaceTest, LNN_START_DISC_DEVICE_TEST_001, TestSize.Level1)
178 {
179 const char *pkgName = "testpkgName";
180 SubscribeInfo info;
181 bool isInnerRequest = false;
182 DiscManagerInterfaceMock discMock;
183 int32_t ret = 0;
184
185 InnerCallback cb = {
186 .serverCb = g_discInnerCb,
187 };
188 (void)memset_s(&info, sizeof(SubscribeInfo), 0, sizeof(SubscribeInfo));
189 EXPECT_CALL(discMock, DiscStartDiscovery).WillRepeatedly(Return(SOFTBUS_ERR));
190 ret = LnnStartDiscDevice(pkgName, &info, &cb, isInnerRequest);
191 EXPECT_TRUE(ret == SOFTBUS_ERR);
192 EXPECT_CALL(discMock, DiscStartDiscovery).WillRepeatedly(Return(SOFTBUS_OK));
193 ret = LnnStartDiscDevice(pkgName, &info, &cb, isInnerRequest);
194 EXPECT_TRUE(ret == SOFTBUS_OK);
195 isInnerRequest = true;
196 EXPECT_CALL(discMock, DiscSetDiscoverCallback).WillRepeatedly(Return(SOFTBUS_ERR));
197 ret = LnnStartDiscDevice(pkgName, &info, &cb, isInnerRequest);
198 EXPECT_TRUE(ret == SOFTBUS_ERR);
199 EXPECT_CALL(discMock, DiscSetDiscoverCallback).WillRepeatedly(Return(SOFTBUS_OK));
200 EXPECT_CALL(discMock, DiscStartAdvertise).WillRepeatedly(Return(SOFTBUS_ERR));
201 ret = LnnStartDiscDevice(pkgName, &info, &cb, isInnerRequest);
202 EXPECT_TRUE(ret == SOFTBUS_ERR);
203 EXPECT_CALL(discMock, DiscSetDiscoverCallback).WillRepeatedly(Return(SOFTBUS_OK));
204 EXPECT_CALL(discMock, DiscStartAdvertise).WillRepeatedly(Return(SOFTBUS_OK));
205 ret = LnnStartDiscDevice(pkgName, &info, &cb, isInnerRequest);
206 EXPECT_TRUE(ret == SOFTBUS_OK);
207 }
208
209 /*
210 * @tc.name: LNN_STOP_DISC_DEVICE_TEST_001
211 * @tc.desc: lnn stop disc device test
212 * @tc.type: FUNC
213 * @tc.require:
214 */
215 HWTEST_F(LNNDiscoveryInterfaceTest, LNN_STOP_DISC_DEVICE_TEST_001, TestSize.Level1)
216 {
217 const char *pkgName = "testpkgName";
218 int32_t subscribeId = 0;
219 bool isInnerRequest = false;
220 DiscManagerInterfaceMock discMock;
221 int32_t ret = 0;
222
223 EXPECT_CALL(discMock, DiscStopDiscovery).WillRepeatedly(Return(SOFTBUS_ERR));
224 ret = LnnStopDiscDevice(pkgName, subscribeId, isInnerRequest);
225 EXPECT_TRUE(ret == SOFTBUS_ERR);
226 EXPECT_CALL(discMock, DiscStopDiscovery).WillRepeatedly(Return(SOFTBUS_OK));
227 ret = LnnStopDiscDevice(pkgName, subscribeId, isInnerRequest);
228 EXPECT_TRUE(ret == SOFTBUS_OK);
229 isInnerRequest = true;
230 EXPECT_CALL(discMock, DiscStopAdvertise).WillRepeatedly(Return(SOFTBUS_ERR));
231 ret = LnnStopDiscDevice(pkgName, subscribeId, isInnerRequest);
232 EXPECT_TRUE(ret == SOFTBUS_ERR);
233 EXPECT_CALL(discMock, DiscStopAdvertise).WillRepeatedly(Return(SOFTBUS_OK));
234 ret = LnnStopDiscDevice(pkgName, subscribeId, isInnerRequest);
235 EXPECT_TRUE(ret == SOFTBUS_OK);
236 }
237
238 /*
239 * @tc.name: LNN_START_PUBLISH_TEST_001
240 * @tc.desc: lnn start publish test
241 * @tc.type: FUNC
242 * @tc.require:
243 */
244 HWTEST_F(LNNDiscoveryInterfaceTest, LNN_START_PUBLISH_TEST_001, TestSize.Level1)
245 {
246 g_discoveryImpl[0].StopPublishImpl = nullptr;
247 LnnStopPublish();
248 g_discoveryImpl[0].StopPublishImpl = LnnCoapTest;
249 LnnStopPublish();
250 g_discoveryImpl[0].StopPublishImpl = LnnCoapFuncTest;
251 LnnStopPublish();
252
253 g_discoveryImpl[0].StartPublishImpl = nullptr;
254 int32_t ret = LnnStartPublish();
255 EXPECT_TRUE(ret == SOFTBUS_OK);
256 g_discoveryImpl[0].StartPublishImpl = LnnCoapTest;
257 ret = LnnStartPublish();
258 EXPECT_TRUE(ret == SOFTBUS_ERR);
259 g_discoveryImpl[0].StartPublishImpl = LnnCoapFuncTest;
260 ret = LnnStartPublish();
261 EXPECT_TRUE(ret == SOFTBUS_OK);
262 }
263
264 /*
265 * @tc.name: LNN_START_DISCOVERY_TEST_001
266 * @tc.desc: lnn start discovery test
267 * @tc.type: FUNC
268 * @tc.require:
269 */
270 HWTEST_F(LNNDiscoveryInterfaceTest, LNN_START_DISCOVERY_TEST_001, TestSize.Level1)
271 {
272 g_discoveryImpl[0].StopDiscoveryImpl = nullptr;
273 LnnStopDiscovery();
274 g_discoveryImpl[0].StopDiscoveryImpl = LnnCoapTest;
275 LnnStopDiscovery();
276 g_discoveryImpl[0].StopDiscoveryImpl = LnnCoapFuncTest;
277 LnnStopDiscovery();
278
279 g_discoveryImpl[0].StartDiscoveryImpl = nullptr;
280 int32_t ret = LnnStartDiscovery();
281 EXPECT_TRUE(ret == SOFTBUS_OK);
282 g_discoveryImpl[0].StartDiscoveryImpl = LnnCoapTest;
283 ret = LnnStartDiscovery();
284 EXPECT_TRUE(ret == SOFTBUS_ERR);
285 g_discoveryImpl[0].StartDiscoveryImpl = LnnCoapFuncTest;
286 ret = LnnStartDiscovery();
287 EXPECT_TRUE(ret == SOFTBUS_OK);
288 }
289 } // namespace OHOS
290