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 "client_trans_channel_callback.h"
20 #include "client_trans_session_callback.h"
21 #include "client_trans_tcp_direct_callback.h"
22 #include "client_trans_proxy_manager.h"
23 #include "client_trans_auth_manager.h"
24 #include "client_trans_udp_manager.h"
25 #include "softbus_def.h"
26 #include "softbus_errcode.h"
27
28 #define TEST_DATA_LENGTH 1024
29
30 using namespace std;
31 using namespace testing::ext;
32
33 namespace OHOS {
34 const char *g_sessionName = "ohos.distributedschedule.dms.test";
35 const char *g_networkid = "ABCDEF00ABCDEF00ABCDEF00ABCDEF00ABCDEF00ABCDEF00ABCDEF00ABCDEF00";
36
OnSessionOpened(const char * sessionName,const ChannelInfo * channel,SessionType flag)37 int32_t OnSessionOpened(const char *sessionName, const ChannelInfo *channel, SessionType flag)
38 {
39 return SOFTBUS_OK;
40 }
OnSessionClosed(int32_t channelId,int32_t channelType,ShutdownReason reason)41 int32_t OnSessionClosed(int32_t channelId, int32_t channelType, ShutdownReason reason)
42 {
43 return SOFTBUS_OK;
44 }
45
OnSessionOpenFailed(int32_t channelId,int32_t channelType,int32_t errCode)46 int32_t OnSessionOpenFailed(int32_t channelId, int32_t channelType, int32_t errCode)
47 {
48 return SOFTBUS_OK;
49 }
50
OnDataReceived(int32_t channelId,int32_t channelType,const void * data,uint32_t len,SessionPktType type)51 static int32_t OnDataReceived(int32_t channelId, int32_t channelType,
52 const void *data, uint32_t len, SessionPktType type)
53 {
54 return SOFTBUS_OK;
55 }
56
OnStreamReceived(int32_t channelId,int32_t channelType,const StreamData * data,const StreamData * ext,const StreamFrameInfo * param)57 static int32_t OnStreamReceived(int32_t channelId, int32_t channelType, const StreamData *data,
58 const StreamData *ext, const StreamFrameInfo *param)
59 {
60 return SOFTBUS_OK;
61 }
62
OnQosEvent(int32_t channelId,int32_t channelType,int32_t eventId,int32_t tvCount,const QosTv * tvList)63 static int32_t OnQosEvent(int32_t channelId, int32_t channelType, int32_t eventId,
64 int32_t tvCount, const QosTv *tvList)
65 {
66 return SOFTBUS_OK;
67 }
68
OnChannelBind(int32_t channelId,int32_t channelType)69 static int32_t OnChannelBind(int32_t channelId, int32_t channelType)
70 {
71 (void)channelId;
72 (void)channelType;
73 return SOFTBUS_OK;
74 }
75
76 static IClientSessionCallBack g_clientSessionCb = {
77 .OnSessionOpened = OnSessionOpened,
78 .OnSessionClosed = OnSessionClosed,
79 .OnSessionOpenFailed = OnSessionOpenFailed,
80 .OnDataReceived = OnDataReceived,
81 .OnStreamReceived = OnStreamReceived,
82 .OnQosEvent = OnQosEvent,
83 .OnChannelBind = OnChannelBind,
84 };
85
86 class ClientTransChannelCallbackTest : public testing::Test {
87 public:
ClientTransChannelCallbackTest()88 ClientTransChannelCallbackTest() {}
~ClientTransChannelCallbackTest()89 ~ClientTransChannelCallbackTest() {}
90 static void SetUpTestCase(void);
91 static void TearDownTestCase(void);
SetUp()92 void SetUp() override {}
TearDown()93 void TearDown() override {}
94 };
95
SetUpTestCase(void)96 void ClientTransChannelCallbackTest::SetUpTestCase(void)
97 {
98 int ret = ClientTransAuthInit(&g_clientSessionCb);
99 EXPECT_EQ(SOFTBUS_OK, ret);
100
101 ret = ClientTransProxyInit(&g_clientSessionCb);
102 EXPECT_EQ(SOFTBUS_OK, ret);
103
104 ret = ClientTransTdcSetCallBack(&g_clientSessionCb);
105 EXPECT_EQ(SOFTBUS_OK, ret);
106
107 ret = ClientTransUdpMgrInit(&g_clientSessionCb);
108 EXPECT_EQ(SOFTBUS_OK, ret);
109 }
110
TearDownTestCase(void)111 void ClientTransChannelCallbackTest::TearDownTestCase(void) {}
112
113 /**
114 * @tc.name: TransOnChannelOpenTest001
115 * @tc.desc: trans on channel open test, use the wrong or normal parameter.
116 * @tc.type: FUNC
117 * @tc.require:
118 */
119 HWTEST_F(ClientTransChannelCallbackTest, TransOnChannelOpenTest001, TestSize.Level0)
120 {
121 ChannelInfo info = {0};
122 int ret = TransOnChannelOpened(nullptr, &info);
123 EXPECT_EQ(SOFTBUS_INVALID_PARAM, ret);
124
125 ret = TransOnChannelOpened(g_sessionName, nullptr);
126 EXPECT_EQ(SOFTBUS_INVALID_PARAM, ret);
127
128 info.channelType = CHANNEL_TYPE_AUTH;
129 ret = TransOnChannelOpened(g_sessionName, &info);
130 EXPECT_EQ(SOFTBUS_OK, ret);
131
132 info.channelType = CHANNEL_TYPE_PROXY;
133 ret = TransOnChannelOpened(g_sessionName, &info);
134 EXPECT_EQ(SOFTBUS_MEM_ERR, ret);
135
136 info.channelType = CHANNEL_TYPE_TCP_DIRECT;
137 ret = TransOnChannelOpened(g_sessionName, &info);
138 EXPECT_EQ(SOFTBUS_LOCK_ERR, ret);
139
140 info.channelType = CHANNEL_TYPE_UDP;
141 info.isServer = 0;
142 ret = TransOnChannelOpened(g_sessionName, &info);
143 EXPECT_EQ(SOFTBUS_MEM_ERR, ret);
144
145 info.channelType = CHANNEL_TYPE_UDP;
146 info.isServer = 1;
147 ret = TransOnChannelOpened(g_sessionName, &info);
148 EXPECT_EQ(SOFTBUS_MEM_ERR, ret);
149
150 info.channelType = CHANNEL_TYPE_BUTT;
151 ret = TransOnChannelOpened(g_sessionName, &info);
152 EXPECT_EQ(SOFTBUS_TRANS_INVALID_CHANNEL_TYPE, ret);
153 }
154
155 /**
156 * @tc.name: TransOnChannelOpenFailedTest001
157 * @tc.desc: trans on channel open failed test, use the wrong or normal parameter.
158 * @tc.type: FUNC
159 * @tc.require:
160 */
161 HWTEST_F(ClientTransChannelCallbackTest, TransOnChannelOpenFailedTest001, TestSize.Level0)
162 {
163 int channelId = 1;
164 int ret = TransOnChannelOpenFailed(channelId, CHANNEL_TYPE_AUTH, SOFTBUS_MEM_ERR);
165 EXPECT_EQ(SOFTBUS_OK, ret);
166
167 ret = TransOnChannelOpenFailed(channelId, CHANNEL_TYPE_PROXY, SOFTBUS_MEM_ERR);
168 EXPECT_EQ(SOFTBUS_OK, ret);
169
170 ret = TransOnChannelOpenFailed(channelId, CHANNEL_TYPE_TCP_DIRECT, SOFTBUS_MEM_ERR);
171 EXPECT_EQ(SOFTBUS_OK, ret);
172
173 ret = TransOnChannelOpenFailed(channelId, CHANNEL_TYPE_UDP, SOFTBUS_MEM_ERR);
174 EXPECT_EQ(SOFTBUS_OK, ret);
175
176 ret = TransOnChannelOpenFailed(channelId, CHANNEL_TYPE_UNDEFINED, SOFTBUS_MEM_ERR);
177 EXPECT_EQ(SOFTBUS_OK, ret);
178
179 ret = TransOnChannelOpenFailed(channelId, CHANNEL_TYPE_BUTT, SOFTBUS_MEM_ERR);
180 EXPECT_EQ(SOFTBUS_TRANS_INVALID_CHANNEL_TYPE, ret);
181 }
182
183 /**
184 * @tc.name: TransOnChannelLinkDownTest001
185 * @tc.desc: trans on channel link down test, use the wrong or normal parameter.
186 * @tc.type: FUNC
187 * @tc.require:
188 */
189 HWTEST_F(ClientTransChannelCallbackTest, TransOnChannelLinkDownTest001, TestSize.Level0)
190 {
191 int ret = TransOnChannelLinkDown(nullptr, 0);
192 EXPECT_EQ(SOFTBUS_INVALID_PARAM, ret);
193
194 ret = TransOnChannelLinkDown(g_networkid, 0);
195 EXPECT_EQ(SOFTBUS_OK, ret);
196 }
197
198 /**
199 * @tc.name: TransOnChannelClosedTest001
200 * @tc.desc: trans on channel closed test, use the wrong or normal parameter.
201 * @tc.type: FUNC
202 * @tc.require:
203 */
204 HWTEST_F(ClientTransChannelCallbackTest, TransOnChannelClosedTest001, TestSize.Level0)
205 {
206 int channelId = 1;
207 int32_t messageType = MESSAGE_TYPE_NOMAL;
208 int32_t udpMessageType = MESSAGE_TYPE_CLOSE_ACK;
209 int ret = TransOnChannelClosed(channelId, CHANNEL_TYPE_AUTH, messageType, SHUTDOWN_REASON_UNKNOWN);
210 EXPECT_EQ(SOFTBUS_OK, ret);
211
212 ret = TransOnChannelClosed(channelId, CHANNEL_TYPE_AUTH, udpMessageType, SHUTDOWN_REASON_UNKNOWN);
213 EXPECT_EQ(SOFTBUS_TRANS_INVALID_CHANNEL_TYPE, ret);
214
215 ret = TransOnChannelClosed(channelId, CHANNEL_TYPE_PROXY, messageType, SHUTDOWN_REASON_UNKNOWN);
216 EXPECT_EQ(SOFTBUS_OK, ret);
217
218 ret = TransOnChannelClosed(channelId, CHANNEL_TYPE_PROXY, udpMessageType, SHUTDOWN_REASON_UNKNOWN);
219 EXPECT_EQ(SOFTBUS_TRANS_INVALID_CHANNEL_TYPE, ret);
220
221 ret = TransOnChannelClosed(channelId, CHANNEL_TYPE_UDP, udpMessageType, SHUTDOWN_REASON_UNKNOWN);
222 EXPECT_EQ(SOFTBUS_TRANS_NODE_NOT_FOUND, ret);
223
224 ret = TransOnChannelClosed(channelId, CHANNEL_TYPE_UDP, messageType, SHUTDOWN_REASON_UNKNOWN);
225 EXPECT_EQ(SOFTBUS_TRANS_UDP_GET_CHANNEL_FAILED, ret);
226
227 ret = TransOnChannelClosed(channelId, CHANNEL_TYPE_TCP_DIRECT, messageType, SHUTDOWN_REASON_UNKNOWN);
228 EXPECT_EQ(SOFTBUS_OK, ret);
229
230 ret = TransOnChannelClosed(channelId, CHANNEL_TYPE_TCP_DIRECT, udpMessageType, SHUTDOWN_REASON_UNKNOWN);
231 EXPECT_EQ(SOFTBUS_TRANS_INVALID_CHANNEL_TYPE, ret);
232
233 ret = TransOnChannelClosed(channelId, CHANNEL_TYPE_BUTT, messageType, SHUTDOWN_REASON_UNKNOWN);
234 EXPECT_EQ(SOFTBUS_TRANS_INVALID_CHANNEL_TYPE, ret);
235
236 ret = TransOnChannelClosed(channelId, CHANNEL_TYPE_TCP_DIRECT, MESSAGE_TYPE_BUTT, SHUTDOWN_REASON_UNKNOWN);
237 EXPECT_EQ(SOFTBUS_TRANS_INVALID_MESSAGE_TYPE, ret);
238 }
239
240 /**
241 * @tc.name: TransOnChannelMsgReceivedTest001
242 * @tc.desc: trans on channel msg received test, use the wrong or normal parameter.
243 * @tc.type: FUNC
244 * @tc.require:
245 */
246 HWTEST_F(ClientTransChannelCallbackTest, TransOnChannelMsgReceivedTest001, TestSize.Level0)
247 {
248 int channelId = 1;
249 const void *data = (const void *)"test";
250
251 int ret = TransOnChannelMsgReceived(channelId, CHANNEL_TYPE_AUTH, nullptr, TEST_DATA_LENGTH, TRANS_SESSION_BYTES);
252 EXPECT_EQ(SOFTBUS_INVALID_PARAM, ret);
253
254 ret = TransOnChannelMsgReceived(channelId, CHANNEL_TYPE_AUTH, data, TEST_DATA_LENGTH, TRANS_SESSION_BYTES);
255 EXPECT_EQ(SOFTBUS_OK, ret);
256
257 ret = TransOnChannelMsgReceived(channelId, CHANNEL_TYPE_PROXY, data, TEST_DATA_LENGTH, TRANS_SESSION_BYTES);
258 EXPECT_EQ(SOFTBUS_TRANS_PROXY_INVALID_CHANNEL_ID, ret);
259
260 ret = TransOnChannelMsgReceived(channelId, CHANNEL_TYPE_TCP_DIRECT, data, TEST_DATA_LENGTH, TRANS_SESSION_BYTES);
261 EXPECT_EQ(SOFTBUS_OK, ret);
262
263 ret = TransOnChannelMsgReceived(channelId, CHANNEL_TYPE_BUTT, data, TEST_DATA_LENGTH, TRANS_SESSION_BYTES);
264 EXPECT_EQ(SOFTBUS_TRANS_INVALID_CHANNEL_TYPE, ret);
265 }
266
267 /**
268 * @tc.name: TransOnChannelQosEventTest001
269 * @tc.desc: trans on channel qos event test, use the wrong or normal parameter.
270 * @tc.type: FUNC
271 * @tc.require:
272 */
273 HWTEST_F(ClientTransChannelCallbackTest, TransOnChannelQosEventTest001, TestSize.Level0)
274 {
275 int channelId = 1;
276 int32_t eventId = 0;
277 int32_t tvCount = 1;
278 const QosTv tvList = {
279 .type = WIFI_CHANNEL_QUALITY,
280 };
281
282 int ret = TransOnChannelQosEvent(channelId, CHANNEL_TYPE_UDP, eventId, tvCount, nullptr);
283 EXPECT_EQ(SOFTBUS_INVALID_PARAM, ret);
284
285 ret = TransOnChannelQosEvent(channelId, CHANNEL_TYPE_UDP, eventId, tvCount, &tvList);
286 EXPECT_EQ(SOFTBUS_TRANS_UDP_GET_CHANNEL_FAILED, ret);
287
288 ret = TransOnChannelQosEvent(channelId, CHANNEL_TYPE_BUTT, eventId, tvCount, &tvList);
289 EXPECT_EQ(SOFTBUS_TRANS_INVALID_CHANNEL_TYPE, ret);
290 }
291
292 /**
293 * @tc.name: TransOnChannelBindTest001
294 * @tc.desc: trans on channel bind test, use the wrong or normal parameter.
295 * @tc.type: FUNC
296 * @tc.require:
297 */
298 HWTEST_F(ClientTransChannelCallbackTest, TransOnChannelBindTest001, TestSize.Level0)
299 {
300 const int32_t channelId = 1;
301 int32_t ret = TransOnChannelBind(channelId, CHANNEL_TYPE_UDP);
302 EXPECT_EQ(SOFTBUS_OK, ret);
303
304 ret = TransOnChannelBind(channelId, CHANNEL_TYPE_TCP_DIRECT);
305 EXPECT_EQ(SOFTBUS_NOT_FIND, ret);
306
307 ret = TransOnChannelBind(channelId, CHANNEL_TYPE_PROXY);
308 EXPECT_EQ(SOFTBUS_OK, ret);
309
310 ret = TransOnChannelBind(channelId, CHANNEL_TYPE_AUTH);
311 EXPECT_EQ(SOFTBUS_OK, ret);
312
313 ret = TransOnChannelBind(channelId, CHANNEL_TYPE_UNDEFINED);
314 EXPECT_EQ(SOFTBUS_OK, ret);
315
316 ret = TransOnChannelBind(channelId, CHANNEL_TYPE_BUTT);
317 EXPECT_EQ(SOFTBUS_TRANS_INVALID_CHANNEL_TYPE, ret);
318 }
319 } // namespace OHOS
320